From 4f0e8a298230bdca080c3872b5e19d07365fe2ea Mon Sep 17 00:00:00 2001 From: lintsinghua Date: Tue, 9 Dec 2025 21:13:14 +0800 Subject: [PATCH] refactor(llm): simplify model name formatting logic in LiteLLMAdapter - Remove redundant check for CUSTOM_BASE_URL_PROVIDERS - Consolidate model name prefix logic into single code path - Move prefix retrieval after model name validation - Improve code clarity by eliminating unnecessary conditional branches - Maintain backward compatibility with existing model name formats --- backend/app/services/llm/adapters/litellm_adapter.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/backend/app/services/llm/adapters/litellm_adapter.py b/backend/app/services/llm/adapters/litellm_adapter.py index c1fded9..81a9d90 100644 --- a/backend/app/services/llm/adapters/litellm_adapter.py +++ b/backend/app/services/llm/adapters/litellm_adapter.py @@ -61,16 +61,12 @@ class LiteLLMAdapter(BaseLLMAdapter): provider = self.config.provider model = self.config.model - # 对于使用 OpenAI 兼容模式的提供商,直接使用模型名 - if provider in self.CUSTOM_BASE_URL_PROVIDERS: - return model - - # 对于原生支持的提供商,添加前缀 - prefix = self.PROVIDER_PREFIX_MAP.get(provider, "openai") - # 检查模型名是否已经包含前缀 if "/" in model: return model + + # 获取 provider 前缀 + prefix = self.PROVIDER_PREFIX_MAP.get(provider, "openai") return f"{prefix}/{model}"