Fix models

vercel-labs · May 31, 2024 · 547b287 · 547b287
1 parent 3965598
commit 547b287
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/lib/chat/actions.tsx b/lib/chat/actions.tsx
@@ -162,21 +162,9 @@ async function submitUserMessage(content: string) {
   ;(async () => {
     try {
       const result = await experimental_streamText({
-        model: google.generativeAI('models/gemini-1.0-pro-001'),
+        model: google.generativeAI('models/gemini-1.5-flash'),
         temperature: 0,
         tools: {
-          listDestinations: {
-            description: 'List destination cities, max 5.',
-            parameters: z.object({
-              destinations: z.array(
-                z
-                  .string()
-                  .describe(
-                    'List of destination cities. Include rome as one of the cities.'
-                  )
-              )
-            })
-          },
           showFlights: {
             description:
               "List available flights in the UI. List 3 that match user's query.",
@@ -192,6 +180,18 @@ async function submitUserMessage(content: string) {
                 )
             })
           },
+          listDestinations: {
+            description: 'List destinations to travel cities, max 5.',
+            parameters: z.object({
+              destinations: z.array(
+                z
+                  .string()
+                  .describe(
+                    'List of destination cities. Include rome as one of the cities.'
+                  )
+              )
+            })
+          },
           showSeatPicker: {
             description:
               'Show the UI to choose or change seat for the selected flight.',