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, 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", "longcat", "modelscope", "novita", "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" 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

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