From 8a95cb42f0ad898d08473c1968ec2b4d3d4988d7 Mon Sep 17 00:00:00 2001
From: Nicolas <nicolascamara29@gmail.com>
Date: Tue, 30 Apr 2024 18:36:21 -0700
Subject: [PATCH] Update models.ts

---
 apps/api/src/lib/LLM-extraction/models.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/apps/api/src/lib/LLM-extraction/models.ts b/apps/api/src/lib/LLM-extraction/models.ts
index ec8a710..ff805bb 100644
--- a/apps/api/src/lib/LLM-extraction/models.ts
+++ b/apps/api/src/lib/LLM-extraction/models.ts
@@ -59,7 +59,7 @@ export async function generateOpenAICompletions({
         },
       },
     ],
-    tool_choice: "auto",
+    tool_choice: { "type": "function", "function": {"name": "extract_content"}},
     temperature,
   });
 
@@ -74,3 +74,4 @@ export async function generateOpenAICompletions({
     llm_extraction: llmExtraction,
   };
 }
+