From c9b0a80522a03642d51b79a7183bc06ba4d3fc82 Mon Sep 17 00:00:00 2001 From: Nicolas Date: Wed, 20 Nov 2024 10:23:44 -0800 Subject: [PATCH] Nick: --- .../src/__tests__/e2e_extract/index.test.ts | 25 +++++++++++++++++++ .../scrapeURL/transformers/llmExtract.ts | 1 + 2 files changed, 26 insertions(+) diff --git a/apps/api/src/__tests__/e2e_extract/index.test.ts b/apps/api/src/__tests__/e2e_extract/index.test.ts index f97b111e3..31f14f4a8 100644 --- a/apps/api/src/__tests__/e2e_extract/index.test.ts +++ b/apps/api/src/__tests__/e2e_extract/index.test.ts @@ -150,4 +150,29 @@ describe("E2E Tests for Extract API Routes", () => { // expect(response.body).toHaveProperty("data"); // expect(response.body.data?.pciDssCompliance).toBe(true); }, 60000); + + it.concurrent("should return Greenhouse Applicant Tracking System for Abnormal Security", async () => { + const response = await request(TEST_URL) + .post("/v1/extract") + .set("Authorization", `Bearer ${process.env.TEST_API_KEY}`) + .set("Content-Type", "application/json") + .send({ + urls: ["https://careers.abnormalsecurity.com/jobs/6119456003?gh_jid=6119456003"], + prompt: "what applicant tracking system is this company using?", + schema: { + type: "object", + properties: { + isGreenhouseATS: { type: "boolean" }, + answer: { type: "string" } + } + }, + allowExternalLinks: true + }) + + console.log(response.body); + expect(response.statusCode).toBe(200); + expect(response.body).toHaveProperty("data"); + expect(response.body.data?.isGreenhouseATS).toBe(true); + }, 60000); + }); diff --git a/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts b/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts index ab6194233..7c2a66968 100644 --- a/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts +++ b/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts @@ -123,6 +123,7 @@ export async function generateOpenAICompletions(logger: Logger, options: Extract const jsonCompletion = await openai.beta.chat.completions.parse({ model, + temperature: 0, messages: [ { role: "system",