fix: gemini api keys through openai-compatible endpoint

This commit is contained in:
David Warshawsky 2026-03-20 13:53:13 +02:00
parent 75cfee46de
commit 87130bf3df
2 changed files with 4 additions and 4 deletions

View file

@ -113,7 +113,7 @@ func CreateProviderFromConfig(cfg *config.ModelConfig) (LLMProvider, string, err
cfg.RequestTimeout, cfg.RequestTimeout,
), modelID, nil ), modelID, nil
case "litellm", "openrouter", "groq", "zhipu", "gemini", "nvidia", case "litellm", "openrouter", "groq", "zhipu", "gemini", "google", "nvidia",
"ollama", "moonshot", "shengsuanyun", "deepseek", "cerebras", "ollama", "moonshot", "shengsuanyun", "deepseek", "cerebras",
"vivgrid", "volcengine", "vllm", "qwen", "qwen-intl", "qwen-international", "dashscope-intl", "vivgrid", "volcengine", "vllm", "qwen", "qwen-intl", "qwen-international", "dashscope-intl",
"qwen-us", "dashscope-us", "mistral", "avian", "minimax", "longcat", "modelscope", "novita", "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" return "https://api.groq.com/openai/v1"
case "zhipu": case "zhipu":
return "https://open.bigmodel.cn/api/paas/v4" return "https://open.bigmodel.cn/api/paas/v4"
case "gemini": case "gemini", "google":
return "https://generativelanguage.googleapis.com/v1beta" return "https://generativelanguage.googleapis.com/v1beta/openai"
case "nvidia": case "nvidia":
return "https://integrate.api.nvidia.com/v1" return "https://integrate.api.nvidia.com/v1"
case "ollama": case "ollama":

View file

@ -190,7 +190,7 @@ func normalizeModel(model, apiBase string) string {
prefix := strings.ToLower(before) prefix := strings.ToLower(before)
switch prefix { 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": "openrouter", "zhipu", "mistral", "vivgrid", "minimax", "novita":
return after return after
default: default: