added dataset attribute and check for evaluations

Agenta-AI · Feb 15, 2024 · 2dc4b22 · 2dc4b22
1 parent 7960a48
commit 2dc4b22
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/agenta-web/cypress/support/commands/evaluations.ts b/agenta-web/cypress/support/commands/evaluations.ts
@@ -100,7 +100,16 @@ Cypress.Commands.add("removeLlmProviderKey", () => {
 })
 
 Cypress.Commands.add("createNewEvaluation", () => {
-    cy.get('[data-cy="new-evaluation-button"]').click()
+    cy.request({
+        url: `${Cypress.env().baseApiURL}/evaluations/?app_id=${app_id}`,
+        method: "GET",
+    }).then((resp) => {
+        if (resp.body.length) {
+            cy.get('[data-cy="new-evaluation-button"]').click()
+        } else {
+            cy.get('[data-cy="new-evaluation-button__no_variants"]').click()
+        }
+    })
     cy.get(".ant-modal-content").should("exist")
 
     cy.get('[data-cy="select-testset-group"]').click()

diff --git a/agenta-web/src/components/pages/evaluations/evaluationResults/EmptyEvaluations.tsx b/agenta-web/src/components/pages/evaluations/evaluationResults/EmptyEvaluations.tsx
@@ -75,6 +75,7 @@ const EmptyEvaluations: React.FC<Props> = ({onConfigureEvaluators, onBeginEvalua
                             icon={<PlusCircleOutlined />}
                             type="default"
                             onClick={onBeginEvaluation}
+                            data-cy={"new-evaluation-button__no_variants"}
                         >
                             Begin Evaluation Now
                         </Button>