From 547b287da838f9e728802e6d4c3dae7013e7869b Mon Sep 17 00:00:00 2001
From: Jared Palmer <jared@jaredpalmer.com>
Date: Fri, 31 May 2024 10:01:40 -0700
Subject: [PATCH] Fix models

---
 lib/chat/actions.tsx | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/lib/chat/actions.tsx b/lib/chat/actions.tsx
index 15291b1..edff8b6 100644
--- 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.',