Update route.ts

UmstadAI · Aug 7, 2024 · 50fef62 · 50fef62
1 parent 09c459b
commit 50fef62
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/app/api/talhasearch/route.ts b/app/api/talhasearch/route.ts
@@ -7,6 +7,7 @@ import { NextResponse } from 'next/server'
 import { validateApiKey } from '@/lib/utils'
 import { demoSearcherTool, demoSearchRunnable } from '@/lib/tools'
 import { demoSearchToolRunnable } from '@/lib/tools/demoSearch'
+import { ChatCompletionMessageParam } from 'openai/resources/chat/completions'
 
 export const runtime = 'edge'
 
@@ -67,14 +68,14 @@ export async function POST(req: Request) {
     model = 'gpt-4o-mini'
     const openai = new OpenAI(configuration)
 
-    const messages = [
-        { role: "user", content: message },
+    const messages: ChatCompletionMessageParam[] = [
+        { role: "user", content: message }
     ];
 
 
     const response = await openai.chat.completions.create({
         model: model,
-        messages,
+        messages: messages,
         tools: [demoSearchToolRunnable.function],
         tool_choice: {"type": "function", "function": {"name": demoSearchToolRunnable.function.name}},
     });