Update chat_model.py

This commit is contained in:
Stephen Hu 2025-11-19 11:28:36 +08:00
parent 4c187ce2fb
commit 36dd4610f7

View file

@ -1423,7 +1423,7 @@ class LiteLLMBase(ABC):
elif self.provider == SupportedLiteLLMProvider.OpenRouter:
self.api_key = json.loads(key).get("api_key", "")
self.provider_order = json.loads(key).get("provider_order", "")
elif self.provider == SupportedLiteLLMProvider.Ollama and key is str and key.startswith("{"):
elif self.provider == SupportedLiteLLMProvider.Ollama:
temp_api_key = self.api_key
self.api_key = json.loads(key).get("api_key", temp_api_key)