From 2245650bc3db47ad8c5ef12e40fe24bccc475375 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gerg=C5=91=20M=C3=B3ricz?= <mo.geryy@gmail.com>
Date: Tue, 15 Apr 2025 23:54:56 -0700
Subject: [PATCH] fix

---
 .../src/lib/extract/completions/analyzeSchemaAndPrompt.ts    | 5 +++--
 apps/api/src/lib/extract/extraction-service.ts               | 4 ++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/apps/api/src/lib/extract/completions/analyzeSchemaAndPrompt.ts b/apps/api/src/lib/extract/completions/analyzeSchemaAndPrompt.ts
index b02588be..98e3ccc0 100644
--- a/apps/api/src/lib/extract/completions/analyzeSchemaAndPrompt.ts
+++ b/apps/api/src/lib/extract/completions/analyzeSchemaAndPrompt.ts
@@ -8,14 +8,15 @@ import {
   buildAnalyzeSchemaPrompt,
   buildAnalyzeSchemaUserPrompt,
 } from "../build-prompts";
-import { logger } from "../../../lib/logger";
 import { jsonSchema } from "ai";
 import { getModel } from "../../../lib/generic-ai";
+import { Logger } from "winston";
 
 export async function analyzeSchemaAndPrompt(
   urls: string[],
   schema: any,
   prompt: string,
+  logger: Logger,
 ): Promise<{
   isMultiEntity: boolean;
   multiEntityKeys: string[];
@@ -26,7 +27,7 @@ export async function analyzeSchemaAndPrompt(
 }> {
   let cost = 0;
   if (!schema) {
-    const genRes = await generateSchemaFromPrompt(prompt);
+    const genRes = await generateSchemaFromPrompt(prompt, logger);
     schema = genRes.extract;
     cost = genRes.cost;
   }
diff --git a/apps/api/src/lib/extract/extraction-service.ts b/apps/api/src/lib/extract/extraction-service.ts
index 1a6a4262..87f4f76a 100644
--- a/apps/api/src/lib/extract/extraction-service.ts
+++ b/apps/api/src/lib/extract/extraction-service.ts
@@ -178,7 +178,7 @@ export async function performExtraction(
 
   let reqSchema = request.schema;
   if (!reqSchema && request.prompt) {
-    const schemaGenRes = await generateSchemaFromPrompt(request.prompt);
+    const schemaGenRes = await generateSchemaFromPrompt(request.prompt, logger);
     reqSchema = schemaGenRes.extract;
     costTracking.otherCallCount++;
     costTracking.otherCost += schemaGenRes.cost;
@@ -214,7 +214,7 @@ export async function performExtraction(
     keyIndicators,
     tokenUsage: schemaAnalysisTokenUsage,
     cost: schemaAnalysisCost,
-  } = await analyzeSchemaAndPrompt(urls, reqSchema, request.prompt ?? "");
+  } = await analyzeSchemaAndPrompt(urls, reqSchema, request.prompt ?? "", logger);
 
   logger.debug("Analyzed schema.", {
     isMultiEntity,