Feat/gemini flash (#1352)

* Remove duplicate item from array * Add references to Gemini Flash * Update documentation to ref Gemini Flash
continuedev · Jun 4, 2024 · 321050d · 321050d
1 parent 5224572
commit 321050d
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/core/config/types.ts b/core/config/types.ts
@@ -562,6 +562,7 @@ declare global {
     // 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
@@ -94,7 +94,6 @@ const PARALLEL_PROVIDERS: ModelProvider[] = [
   "bedrock",
   "deepinfra",
   "gemini",
-  "gemini",
   "huggingface-inference-api",
   "huggingface-tgi",
   "mistral",

diff --git a/docs/docs/reference/Model Providers/geminiapi.md b/docs/docs/reference/Model Providers/geminiapi.md
@@ -1,6 +1,6 @@
 # Gemini API
 
-The Google Gemini API is currently in beta. You can [create an API key in Google AI Studio](https://aistudio.google.com) and use `gemini-1.5-pro-latest`. Change `~/.continue/config.json` to look like this:
+The Google Gemini API is currently in beta. You can [create an API key in Google AI Studio](https://aistudio.google.com) and use `gemini-1.5-pro-latest`. Change `~/.continue/config.json` to include the following entry in the "models" array:
 
 ```json title="~/.continue/config.json"
 {
@@ -15,4 +15,6 @@ The Google Gemini API is currently in beta. You can [create an API key in Google
 }
 ```
 
+Google has also released a more lightweight version of the model that still has a one-million-token context window and multimodal capabilities named Gemini Flash. It can be accessed by adding an entry in the models array similar to the above, but substituting "flash" for "pro" in the `title` and `model` values.
+
 [View the source](https://github.com/continuedev/continue/blob/main/core/llm/llms/Gemini.ts)
diff --git a/docs/docs/setup/select-model.md b/docs/docs/setup/select-model.md
@@ -43,7 +43,7 @@ _You can also use other open-source chat models by adding them to your `config.j
 #### Gemini Pro from Google
 
 - Unlimited budget: `gemini-pro-1.5-latest`
-- Limited budget: `gemini-pro-1.0`
+- Limited budget: `gemini-flash-1.5-latest` or `gemini-pro-1.0`
 
 _You can also use other commercial chat models by adding them to your `config.json`._