fix: skip provider prefix rules for vLLM/OpenRouter/AiHubMix endpoints

This commit is contained in:
Re-bin 2026-02-08 04:29:51 +00:00
parent 438ec66fd8
commit 3b61ae4fff

View File

@ -107,11 +107,12 @@ class LiteLLMProvider(LLMProvider):
(("moonshot", "kimi"), "moonshot", ("moonshot/", "openrouter/")), (("moonshot", "kimi"), "moonshot", ("moonshot/", "openrouter/")),
(("gemini",), "gemini", ("gemini/",)), (("gemini",), "gemini", ("gemini/",)),
] ]
model_lower = model.lower() if not (self.is_vllm or self.is_openrouter or self.is_aihubmix):
for keywords, prefix, skip in _prefix_rules: model_lower = model.lower()
if any(kw in model_lower for kw in keywords) and not any(model.startswith(s) for s in skip): for keywords, prefix, skip in _prefix_rules:
model = f"{prefix}/{model}" if any(kw in model_lower for kw in keywords) and not any(model.startswith(s) for s in skip):
break model = f"{prefix}/{model}"
break
# Gateway/endpoint-specific prefixes (detected by api_base/api_key, not model name) # Gateway/endpoint-specific prefixes (detected by api_base/api_key, not model name)
if self.is_openrouter and not model.startswith("openrouter/"): if self.is_openrouter and not model.startswith("openrouter/"):