diff --git a/packages/core/lib/v3/llm/LLMProvider.ts b/packages/core/lib/v3/llm/LLMProvider.ts index 4b97f395a..7d7f33423 100644 --- a/packages/core/lib/v3/llm/LLMProvider.ts +++ b/packages/core/lib/v3/llm/LLMProvider.ts @@ -95,6 +95,8 @@ const modelToProviderMap: { [key in AvailableModel]: ModelProvider } = { "gemini-2.0-flash": "google", "gemini-2.5-flash-preview-04-17": "google", "gemini-2.5-pro-preview-03-25": "google", + "gemini-3-flash-preview": "google", + "gemini-3-pro-preview": "google", }; export function getAISDKLanguageModel( diff --git a/packages/core/lib/v3/types/public/model.ts b/packages/core/lib/v3/types/public/model.ts index 17b62dcc5..cc70bcfb3 100644 --- a/packages/core/lib/v3/types/public/model.ts +++ b/packages/core/lib/v3/types/public/model.ts @@ -89,6 +89,8 @@ export type AvailableModel = | "gemini-2.0-flash" | "gemini-2.5-flash-preview-04-17" | "gemini-2.5-pro-preview-03-25" + | "gemini-3-flash-preview" + | "gemini-3-pro-preview" | string; export type ModelProvider = diff --git a/packages/evals/taskConfig.ts b/packages/evals/taskConfig.ts index 65dba8223..8570333d9 100644 --- a/packages/evals/taskConfig.ts +++ b/packages/evals/taskConfig.ts @@ -26,6 +26,8 @@ const ALL_EVAL_MODELS = [ "gemini-1.5-flash-8b", "gemini-2.5-flash-preview-04-17", "gemini-2.5-pro-preview-03-25", + "gemini-3-flash-preview", + "gemini-3-pro-preview", // ANTHROPIC "claude-3-5-sonnet-latest", "claude-3-7-sonnet-latest", diff --git a/packages/server/src/lib/utils.ts b/packages/server/src/lib/utils.ts index e6beb3245..eb0b239e5 100644 --- a/packages/server/src/lib/utils.ts +++ b/packages/server/src/lib/utils.ts @@ -205,6 +205,8 @@ export function mapModelToProvider(model: LegacyModel): LegacyProvider { case "gemini-2.0-flash": case "gemini-2.5-pro-preview-03-25": case "gemini-2.5-flash-preview-04-17": + case "gemini-3-flash-preview": + case "gemini-3-pro-preview": return "google"; case "cerebras-llama-3.3-70b": case "cerebras-llama-3.1-8b": diff --git a/packages/server/src/types/model.ts b/packages/server/src/types/model.ts index 2089d4896..34d34dede 100644 --- a/packages/server/src/types/model.ts +++ b/packages/server/src/types/model.ts @@ -38,6 +38,8 @@ export type LegacyModel = | "gemini-2.0-flash-lite" | "gemini-2.0-flash" | "gemini-2.5-pro-preview-03-25" - | "gemini-2.5-flash-preview-04-17"; + | "gemini-2.5-flash-preview-04-17" + | "gemini-3-flash-preview" + | "gemini-3-pro-preview"; export type LegacyProvider = "openai" | "anthropic" | "google";