Refactor test set creation buttons

Agenta-AI · Jan 24, 2024 · 3964366 · 3964366
1 parent 80441cd
commit 3964366
Showing 1 changed file with 9 additions and 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
@@ -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}>