cherrypick

UmstadAI · Jul 6, 2024 · b15d71c · b15d71c
1 parent df4fa3b
commit b15d71c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/app/api/demosearch/route.ts b/app/api/demosearch/route.ts
@@ -7,7 +7,6 @@ import { NextResponse } from 'next/server'
 import { SEARCHER_PROMPT } from './prompt'
 import { validateApiKey } from '@/lib/utils'
 import { demoSearchRunnable } from '@/lib/tools'
-import { demoSearchToolRunnable } from '@/lib/tools/demoSearch'
 
 export const runtime = 'edge'
 
@@ -64,7 +63,7 @@ export async function POST(req: Request) {
       })
     }
 
-    model = 'gpt-4o'
+    model = 'gpt-4-turbo'
     const openai = new OpenAI(configuration)
 
     const runner = openai.beta.chat.completions.runTools({
@@ -81,7 +80,7 @@ export async function POST(req: Request) {
           content: message
         }
       ],
-      tools: [demoSearchToolRunnable(message)]
+      tools: demoSearchRunnable
     })
 
     const stream = OpenAIStream(runner)

diff --git a/lib/tools/index.ts b/lib/tools/index.ts
@@ -54,4 +54,5 @@ export const searchRunnable: RunnableToolFunction<any>[] = [
 ]
 
 export const demoSearchRunnable:  RunnableToolFunction<any>[] = [
+  demoSearchToolRunnable
 ]