This commit is contained in:
David Warshawsky 2026-03-23 17:38:58 +08:00 committed by GitHub
commit 632f47053d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View file

@ -114,7 +114,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", "longcat", "modelscope", "novita",
@ -269,8 +269,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":

View file

@ -398,7 +398,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: