gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update chat_model.py

Browse source
This commit is contained in:
Stephen Hu 2025-11-19 08:38:39 +08:00
parent 50bc53a1f5
commit 4c187ce2fb
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
rag/llm/chat_model.py
View file

@ -1423,6 +1423,9 @@ 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("{"):
temp_api_key = self.api_key
self.api_key = json.loads(key).get("api_key", temp_api_key)
def _get_delay(self):
"""Calculate retry delay time"""