From 3964366fc3ada6b8d14e5390e456e965ffe03e6b Mon Sep 17 00:00:00 2001
From: Mahmoud Mabrouk <mahmoud.mabrouk@tu-berlin.de>
Date: Wed, 24 Jan 2024 12:53:56 +0100
Subject: [PATCH] Refactor test set creation buttons

---
 .../pages/apps/[app_id]/testsets/index.tsx    | 22 ++++++++-----------
 1 file changed, 9 insertions(+), 13 deletions(-)

diff --git a/agenta-web/src/pages/apps/[app_id]/testsets/index.tsx b/agenta-web/src/pages/apps/[app_id]/testsets/index.tsx
index b9221f5726..8e53259f89 100644
--- a/agenta-web/src/pages/apps/[app_id]/testsets/index.tsx
+++ b/agenta-web/src/pages/apps/[app_id]/testsets/index.tsx
@@ -94,21 +94,17 @@ export default function Testsets() {
                         >
                             <Button>Create a test set with UI</Button>
                         </Link>
-                        {isDemo() ? (
-                            <Tooltip title="API test set creation is unavailable in the demo version. Check out the self-hosted open-source version at https://github.com/agenta-ai/agenta">
-                                <Button disabled>Create a test set with API</Button>
-                            </Tooltip>
-                        ) : (
-                            <Link
-                                data-cy="testset-new-api-link"
-                                href={`/apps/${appId}/testsets/new/api`}
-                            >
-                                <Button>Create a test set with API</Button>
+                        <Link
+                            data-cy="testset-new-api-link"
+                            href={`/apps/${appId}/testsets/new/api`}
+                        >
+                            <Button>Create a test set with API</Button>
+                        </Link>
+                        {!isDemo() && (
+                            <Link href={`/apps/${appId}/testsets/new/endpoint`}>
+                                <Button>Import from Endpoint</Button>
                             </Link>
                         )}
-                        <Link href={`/apps/${appId}/testsets/new/endpoint`}>
-                            <Button>Import from Endpoint</Button>
-                        </Link>
                     </div>
 
                     <Link href={`/apps/${appId}/evaluations`} className={classes.startLink}>