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

Improve error handling and logging in cloud model detection

Browse source
This commit is contained in:
yangdx 2025-11-13 20:41:44 +08:00
parent 67dfd85679
commit 63510478e5
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
lightrag/llm/ollama.py
View file

@ -37,9 +37,13 @@ def _coerce_host_for_cloud_model(host: Optional[str], model: object) -> Optional
return host return host
try: try:
model_name_str = str(model) if model is not None else "" model_name_str = str(model) if model is not None else ""
except Exception: except (TypeError, ValueError, AttributeError) as e:
logger.warning(f"Failed to convert model to string: {e}, using empty string")
model_name_str = "" model_name_str = ""
if _CLOUD_MODEL_SUFFIX_PATTERN.search(model_name_str): if _CLOUD_MODEL_SUFFIX_PATTERN.search(model_name_str):
logger.debug(
f"Detected cloud model '{model_name_str}', using Ollama Cloud host"
)
return _OLLAMA_CLOUD_HOST return _OLLAMA_CLOUD_HOST
return host return host