From 4c187ce2fbf652ddc708862032104043b6820a8a Mon Sep 17 00:00:00 2001
From: Stephen Hu <812791840@qq.com>
Date: Wed, 19 Nov 2025 08:38:39 +0800
Subject: [PATCH] Update chat_model.py

---
 rag/llm/chat_model.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/rag/llm/chat_model.py b/rag/llm/chat_model.py
index 856d23b01..50729f036 100644
--- a/rag/llm/chat_model.py
+++ b/rag/llm/chat_model.py
@@ -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"""