diff --git a/.changeset/brown-dingos-teach.md b/.changeset/brown-dingos-teach.md new file mode 100644 index 000000000..7f5246369 --- /dev/null +++ b/.changeset/brown-dingos-teach.md @@ -0,0 +1,5 @@ +--- +"@browserbasehq/stagehand": patch +--- + +Add support for Gemini 2.5 Flash, 2.5 Flash Lite, 2.5 Pro, and corresponding preview models. diff --git a/packages/core/lib/v3/llm/LLMProvider.ts b/packages/core/lib/v3/llm/LLMProvider.ts index 7c16f2118..92349d7db 100644 --- a/packages/core/lib/v3/llm/LLMProvider.ts +++ b/packages/core/lib/v3/llm/LLMProvider.ts @@ -89,7 +89,12 @@ const modelToProviderMap: { [key in AvailableModel]: ModelProvider } = { "gemini-1.5-flash-8b": "google", "gemini-2.0-flash-lite": "google", "gemini-2.0-flash": "google", + "gemini-2.5-flash": "google", "gemini-2.5-flash-preview-04-17": "google", + "gemini-2.5-flash-preview-09-2025": "google", + "gemini-2.5-flash-lite": "google", + "gemini-2.5-flash-lite-preview-09-2025": "google", + "gemini-2.5-pro": "google", "gemini-2.5-pro-preview-03-25": "google", }; diff --git a/packages/core/lib/v3/types/public/model.ts b/packages/core/lib/v3/types/public/model.ts index ea8aa57da..ae66c4c87 100644 --- a/packages/core/lib/v3/types/public/model.ts +++ b/packages/core/lib/v3/types/public/model.ts @@ -54,7 +54,12 @@ export type AvailableModel = | "gemini-1.5-flash-8b" | "gemini-2.0-flash-lite" | "gemini-2.0-flash" + | "gemini-2.5-flash" | "gemini-2.5-flash-preview-04-17" + | "gemini-2.5-flash-preview-09-2025" + | "gemini-2.5-flash-lite" + | "gemini-2.5-flash-lite-preview-09-2025" + | "gemini-2.5-pro" | "gemini-2.5-pro-preview-03-25" | string;