From fee942ed2ddc8dad134e778d3a9b5921cbff0b18 Mon Sep 17 00:00:00 2001
From: Pratik Parmar <steveparmar6nov2011@gmail.com>
Date: Wed, 22 May 2024 23:15:24 +0530
Subject: [PATCH] Add Gemini 1.5 Flash as a model (#1337)

---
 core/index.d.ts                      |  1 +
 core/llm/autodetect.ts               |  1 +
 extensions/vscode/config_schema.json |  4 +++-
 gui/src/util/modelData.ts            | 15 ++++++++++++++-
 4 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/core/index.d.ts b/core/index.d.ts
index 0fd18496a8..64949db765 100644
--- a/core/index.d.ts
+++ b/core/index.d.ts
@@ -568,6 +568,7 @@ export type ModelName =
   // Gemini
   | "gemini-pro"
   | "gemini-1.5-pro-latest"
+  | "gemini-1.5-flash-latest"
   // Mistral
   | "mistral-tiny"
   | "mistral-small"
diff --git a/core/llm/autodetect.ts b/core/llm/autodetect.ts
index 36333326ac..63a7bfe8ea 100644
--- a/core/llm/autodetect.ts
+++ b/core/llm/autodetect.ts
@@ -62,6 +62,7 @@ const MODEL_SUPPORTS_IMAGES: string[] = [
   "claude-3",
   "gemini-ultra",
   "gemini-1.5-pro",
+  "gemini-1.5-flash",
   "sonnet",
   "opus",
   "haiku",
diff --git a/extensions/vscode/config_schema.json b/extensions/vscode/config_schema.json
index fcbcbbd81c..80268de2bc 100644
--- a/extensions/vscode/config_schema.json
+++ b/extensions/vscode/config_schema.json
@@ -610,7 +610,8 @@
                 "enum": [
                   "chat-bison-001",
                   "gemini-pro",
-                  "gemini-1.5-pro-latest"
+                  "gemini-1.5-pro-latest",
+                  "gemini-1.5-flash-latest"
                 ]
               }
             }
@@ -985,6 +986,7 @@
                       "chat-bison-001",
                       "gemini-pro",
                       "gemini-1.5-pro-latest",
+                      "gemini-1.5-flash-latest",
                       "mistral-tiny",
                       "mistral-small",
                       "mistral-medium",
diff --git a/gui/src/util/modelData.ts b/gui/src/util/modelData.ts
index a7797bec7b..fb243df608 100644
--- a/gui/src/util/modelData.ts
+++ b/gui/src/util/modelData.ts
@@ -454,6 +454,18 @@ const gemini15Pro: ModelPackage = {
   icon: "gemini.png",
   providerOptions: ["gemini", "freetrial"],
 };
+const gemini15Flash: ModelPackage = {
+  title: "Gemini 1.5 Flash",
+  description: "Fast and versatile multimodal model for scaling across diverse tasks",
+  params: {
+    title: "Gemini 1.5 Flash",
+    model: "gemini-1.5-flash-latest",
+    contextLength: 1_000_000,
+    apiKey: "<API_KEY>",
+  },
+  icon: "gemini.png",
+  providerOptions: ["gemini"],
+};
 
 const deepseek: ModelPackage = {
   title: "DeepSeek-Coder",
@@ -660,6 +672,7 @@ export const MODEL_INFO: (ModelPackage | string)[] = [
   "Gemini",
   gemini15Pro,
   geminiPro,
+  gemini15Flash,
   "Open Source",
   llama3Chat,
   deepseek,
@@ -863,7 +876,7 @@ export const PROVIDER_INFO: { [key: string]: ModelInfo } = {
         required: true,
       },
     ],
-    packages: [gemini15Pro, geminiPro],
+    packages: [gemini15Pro, geminiPro, gemini15Flash],
   },
   mistral: {
     title: "Mistral API",