From 87130bf3dfb2f00eda49590a28f7cf64e004e26b Mon Sep 17 00:00:00 2001 From: David Warshawsky <49329558+davidwarshawsky@users.noreply.github.com> Date: Fri, 20 Mar 2026 13:53:13 +0200 Subject: [PATCH] fix: gemini api keys through openai-compatible endpoint --- pkg/providers/factory_provider.go | 6 +++--- pkg/providers/openai_compat/provider.go | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pkg/providers/factory_provider.go b/pkg/providers/factory_provider.go index a7fef8f5b..f3f327b27 100644 --- a/pkg/providers/factory_provider.go +++ b/pkg/providers/factory_provider.go @@ -113,7 +113,7 @@ func CreateProviderFromConfig(cfg *config.ModelConfig) (LLMProvider, string, err cfg.RequestTimeout, ), modelID, nil - case "litellm", "openrouter", "groq", "zhipu", "gemini", "nvidia", + case "litellm", "openrouter", "groq", "zhipu", "gemini", "google", "nvidia", "ollama", "moonshot", "shengsuanyun", "deepseek", "cerebras", "vivgrid", "volcengine", "vllm", "qwen", "qwen-intl", "qwen-international", "dashscope-intl", "qwen-us", "dashscope-us", "mistral", "avian", "minimax", "longcat", "modelscope", "novita", @@ -241,8 +241,8 @@ func getDefaultAPIBase(protocol string) string { return "https://api.groq.com/openai/v1" case "zhipu": return "https://open.bigmodel.cn/api/paas/v4" - case "gemini": - return "https://generativelanguage.googleapis.com/v1beta" + case "gemini", "google": + return "https://generativelanguage.googleapis.com/v1beta/openai" case "nvidia": return "https://integrate.api.nvidia.com/v1" case "ollama": diff --git a/pkg/providers/openai_compat/provider.go b/pkg/providers/openai_compat/provider.go index 463db83c9..987aba1c6 100644 --- a/pkg/providers/openai_compat/provider.go +++ b/pkg/providers/openai_compat/provider.go @@ -190,7 +190,7 @@ func normalizeModel(model, apiBase string) string { prefix := strings.ToLower(before) switch prefix { - case "litellm", "moonshot", "nvidia", "groq", "ollama", "deepseek", "google", + case "litellm", "moonshot", "nvidia", "groq", "ollama", "deepseek", "google", "gemini", "openrouter", "zhipu", "mistral", "vivgrid", "minimax", "novita": return after default: