From 4db9a4a675a6318d4cc14dd4c32e60f7059351f9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?M=C3=B3ricz=20Gerg=C5=91?= <mo.geryy@gmail.com>
Date: Fri, 24 Jan 2025 11:33:49 +0100
Subject: [PATCH] fix(extraction-service): allow no multiEntityKeys if
 isMultiEntity is false

---
 apps/api/src/lib/extract/extraction-service.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/apps/api/src/lib/extract/extraction-service.ts b/apps/api/src/lib/extract/extraction-service.ts
index 203a5a4b..6f9e99a8 100644
--- a/apps/api/src/lib/extract/extraction-service.ts
+++ b/apps/api/src/lib/extract/extraction-service.ts
@@ -77,10 +77,10 @@ async function analyzeSchemaAndPrompt(
 
   const checkSchema = z.object({
     isMultiEntity: z.boolean(),
-    multiEntityKeys: z.array(z.string()),
+    multiEntityKeys: z.array(z.string()).optional().default([]),
     reasoning: z.string(),
     keyIndicators: z.array(z.string()),
-  });
+  }).refine(x => !x.isMultiEntity || multiEntityKeys.length > 0, "isMultiEntity was true, but no multiEntityKeys were returned");
 
   const model = "gpt-4o";