fix: skip provider prefix rules for vLLM/OpenRouter/AiHubMix endpoints
This commit is contained in:
parent
438ec66fd8
commit
3b61ae4fff
@ -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/"):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user