From eb2e51e50bbd65a35514f1240b65daeb3b6bed31 Mon Sep 17 00:00:00 2001
From: Nicolas <nicolascamara29@gmail.com>
Date: Tue, 3 Dec 2024 12:08:15 -0300
Subject: [PATCH] Nick: fixed /extract without a schema

---
 apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts b/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts
index 64073eee..1c6adcd1 100644
--- a/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts
+++ b/apps/api/src/scraper/scrapeURL/transformers/llmExtract.ts
@@ -162,7 +162,7 @@ export async function generateOpenAICompletions(logger: Logger, options: Extract
                 extract = JSON.parse(jsonCompletion.choices[0].message.content);
             } else {
                 const extractData = JSON.parse(jsonCompletion.choices[0].message.content);
-                extract = extractData.data.extract;
+                extract = options.schema ? extractData.data.extract : extractData;
             }
         } catch (e) {
             logger.error("Failed to parse returned JSON, no schema specified.", { error: e });