feat: 🔄 update Google Gemini model versions

microsoft · Dec 10, 2024 · 05a818b · 05a818b
1 parent b2e86a8
commit 05a818b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/packages/core/src/constants.ts b/packages/core/src/constants.ts
@@ -60,7 +60,7 @@ export const DEFAULT_MODEL_CANDIDATES = [
     "azure:gpt-4o",
     "azure_serverless:gpt-4o",
     DEFAULT_MODEL,
-    "google:gemini-1.5-pro-002",
+    "google:gemini-1.5-pro-latest",
     "anthropic:claude-2.1",
     "mistral:mistral-large-latest",
     "github:gpt-4o",
@@ -71,7 +71,7 @@ export const DEFAULT_VISION_MODEL_CANDIDATES = [
     "azure:gpt-4o",
     "azure_serverless:gpt-4o",
     DEFAULT_MODEL,
-    "google:gemini-1.5-pro-002",
+    "google:gemini-1.5-flash-latest",
     "anthropic:claude-2.1",
     "github:gpt-4o",
 ]
@@ -80,7 +80,7 @@ export const DEFAULT_SMALL_MODEL_CANDIDATES = [
     "azure:gpt-4o-mini",
     "azure_serverless:gpt-4o-mini",
     DEFAULT_SMALL_MODEL,
-    "google:gemini-1.5-flash-002",
+    "google:gemini-1.5-flash-latest",
     "anthropic:claude-instant-1.2",
     "mistral:mistral-small-latest",
     "github:gpt-4o-mini",

diff --git a/packages/core/src/types/prompt_template.d.ts b/packages/core/src/types/prompt_template.d.ts
@@ -151,11 +151,11 @@ type ModelType = OptionsOrString<
     | "anthropic:claude-instant-1.2"
     | "huggingface:microsoft/Phi-3-mini-4k-instruct"
     | "google:gemini-1.5-flash"
+    | "google:gemini-1.5-flash-latest"
     | "google:gemini-1.5-flash-8b"
-    | "google:gemini-1.5-flash-002"
+    | "google:gemini-1.5-flash-8b-latest"
     | "google:gemini-1.5-pro"
-    | "google:gemini-1.5-pro-002"
-    | "google:gemini-1-pro"
+    | "google:gemini-1.5-pro-latest"
     | "mistral:mistral-large-latest"
     | "mistral:mistral-small-latest"
     | "mistral:pixtral-large-latest"