diff --git a/apps/api/src/__tests__/e2e_full_withAuth/index.test.ts b/apps/api/src/__tests__/e2e_full_withAuth/index.test.ts index 43e9ea1d..af7fe4a3 100644 --- a/apps/api/src/__tests__/e2e_full_withAuth/index.test.ts +++ b/apps/api/src/__tests__/e2e_full_withAuth/index.test.ts @@ -149,7 +149,7 @@ describe("E2E Tests for API Routes", () => { expect(response.body.data).toHaveProperty("metadata"); expect(response.body.data.content).toContain("_Roast_"); expect(response.body.data.markdown).toContain("_Roast_"); - expect(response.body.data.html).toContain(" { extractorOptions: { mode: "llm-extraction-from-raw-html", extractionPrompt: - "Based on the information on the page, find what the company's mission is and whether it supports SSO, and whether it is open source", + "Based on the information on the page, what are the primary and secondary CTA buttons?", extractionSchema: { type: "object", properties: { - company_mission: { + primary_cta: { type: "string", }, - supports_sso: { - type: "boolean", - }, - is_open_source: { - type: "boolean", + secondary_cta: { + type: "string", }, }, - required: ["company_mission", "supports_sso", "is_open_source"], + required: ["primary_cta", "secondary_cta"], }, }, }); @@ -1237,14 +1234,11 @@ describe("E2E Tests for API Routes", () => { let llmExtraction = response.body.data.llm_extraction; // Check if the llm_extraction object has the required properties with correct types and values - expect(llmExtraction).toHaveProperty("company_mission"); - expect(typeof llmExtraction.company_mission).toBe("string"); - expect(llmExtraction).toHaveProperty("supports_sso"); - expect(llmExtraction.supports_sso).toBe(true); - expect(typeof llmExtraction.supports_sso).toBe("boolean"); - expect(llmExtraction).toHaveProperty("is_open_source"); - expect(llmExtraction.is_open_source).toBe(false); - expect(typeof llmExtraction.is_open_source).toBe("boolean"); + expect(llmExtraction).toHaveProperty("primary_cta"); + expect(typeof llmExtraction.primary_cta).toBe("string"); + expect(llmExtraction).toHaveProperty("secondary_cta"); + expect(typeof llmExtraction.secondary_cta).toBe("string"); + }, 60000); // 60 secs });