zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-06-29 15:31:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix OpenRouter key JSONDecodeError (#15776)

Browse Source 
Fix OpenRouter key JSONDecodeError
This commit is contained in:
Wang Qi
2026-06-08 19:19:10 +08:00
committed by GitHub
parent 0c5245e454
commit 8e4fba6cd2
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rag/llm/chat_model.py
View File

@@ -1371,8 +1371,12 @@ class LiteLLMBase(ABC):
# Factory specific fields
if self.provider == SupportedLiteLLMProvider.OpenRouter:
self.api_key = json.loads(key).get("api_key", "")
self.provider_order = json.loads(key).get("provider_order", "")
try:
self.api_key = json.loads(key).get("api_key", "")
self.provider_order = json.loads(key).get("provider_order", "")
except JSONDecodeError:
self.api_key = key
self.provider_order = ""
elif self.provider == SupportedLiteLLMProvider.Azure_OpenAI:
self.api_key = json.loads(key).get("api_key", "")
self.api_version = json.loads(key).get("api_version", "2024-02-01")

8
rag/llm/cv_model.py
View File

@@ -635,13 +635,17 @@ class OpenRouterCV(GptV4):
def __init__(self, key, model_name, lang="Chinese", base_url="https://openrouter.ai/api/v1", **kwargs):
if not base_url:
base_url = "https://openrouter.ai/api/v1"
api_key = json.loads(key).get("api_key", "")
try:
api_key = json.loads(key).get("api_key", "")
provider_order = json.loads(key).get("provider_order", "")
except JSONDecodeError:
api_key = key
provider_order = ""
self.client = OpenAI(api_key=api_key, base_url=base_url)
self.async_client = AsyncOpenAI(api_key=api_key, base_url=base_url)
self.model_name = model_name
self.lang = lang
Base.__init__(self, **kwargs)
provider_order = json.loads(key).get("provider_order", "")
self.extra_body = {}
if provider_order: