in src/databao_context_engine/llm/service.py [0:0]
def _is_model_available(self, *, model_name, timeout: float = 5.0) -> bool:
url = f"{self._base}/api/tags"
try:
r = self._session.get(url, headers=self._headers, timeout=timeout)
if 200 <= r.status_code < 300:
models = r.json().get("models")
if models and isinstance(models, list):
local_model = next((model for model in models if model.get("name") == model_name), None)
return local_model is not None
return False
except requests.RequestException:
return False