diff --git a/llama_stack/providers/remote/inference/sambanova/sambanova.py b/llama_stack/providers/remote/inference/sambanova/sambanova.py index 9b2f7f0e3..9be5763aa 100644 --- a/llama_stack/providers/remote/inference/sambanova/sambanova.py +++ b/llama_stack/providers/remote/inference/sambanova/sambanova.py @@ -258,7 +258,7 @@ class SambaNovaInferenceAdapter(LiteLLMOpenAIMixin): raise RuntimeError(f"Request to {list_models_url} failed") from e available_models = [model.get("id") for model in response.json().get("data", {})] if len(available_models) == 0 or model_id.split("sambanova/")[-1] not in available_models: - logger.warning(f"Model {model_id} not found as available in SambaNova models") + logger.warning(f"Model {model_id} not available in {self.config.url}/models") return model async def initialize(self): diff --git a/llama_stack/providers/remote/safety/sambanova/sambanova.py b/llama_stack/providers/remote/safety/sambanova/sambanova.py index 0e034935f..242da83a6 100644 --- a/llama_stack/providers/remote/safety/sambanova/sambanova.py +++ b/llama_stack/providers/remote/safety/sambanova/sambanova.py @@ -65,7 +65,7 @@ class SambaNovaSafetyAdapter(Safety, ShieldsProtocolPrivate, NeedsRequestProvide or "guard" not in shield.provider_resource_id.lower() or shield.provider_resource_id.split("sambanova/")[-1] not in available_models ): - logger.warning(f"Shield {shield.provider_resource_id} not found as available in SambaNova models") + logger.warning(f"Shield {shield.provider_resource_id} not available in {self.config.url}/models") async def run_shield( self, shield_id: str, messages: list[Message], params: dict[str, Any] | None = None