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

Fix: verify remote model when model_list in argument and factory_llms is empty (#17364)

Browse Source
This commit is contained in:
Lynn
2026-07-27 09:53:22 +08:00
committed by GitHub
parent 53e83dcadf
commit c62853bfda
1 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
api/apps/services/provider_api_service.py
View File

@@ -668,7 +668,23 @@ async def verify_api_key(provider_id_or_name: str, api_key: str | dict, base_url
else:
factory_llms = factory_info[0]["llm"]
if not factory_llms:
return False, f"No models found for provider '{provider_id_or_name}'", {}
model_base_url = base_url or factory_info[0].get("url", "")
try:
if provider_name in ModelMeta:
remote_models = await ModelMeta[provider_name](api_key, model_base_url).get_model_list()
if remote_models:
factory_llms = [
{
"model_type": mt,
"llm_name": m["name"],
}
for m in remote_models
for mt in m.get("model_types", [])
]
except Exception:
pass
if not factory_llms:
return False, f"No models found for provider '{provider_id_or_name}'", {}
model_verify_result = {}
# test if api key works