phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 07:14:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update log message

Browse source
This commit is contained in:
jhpiedrahitao 2025-07-09 12:27:52 -05:00
parent fc5ad35e0d
commit fce362f901
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/inference/sambanova/sambanova.py
View file

@ -258,7 +258,7 @@ class SambaNovaInferenceAdapter(LiteLLMOpenAIMixin):
raise RuntimeError(f"Request to {list_models_url} failed") from e raise RuntimeError(f"Request to {list_models_url} failed") from e
available_models = [model.get("id") for model in response.json().get("data", {})] 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: 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 return model
async def initialize(self): async def initialize(self):

2
llama_stack/providers/remote/safety/sambanova/sambanova.py
View file

@ -65,7 +65,7 @@ class SambaNovaSafetyAdapter(Safety, ShieldsProtocolPrivate, NeedsRequestProvide
or "guard" not in shield.provider_resource_id.lower() or "guard" not in shield.provider_resource_id.lower()
or shield.provider_resource_id.split("sambanova/")[-1] not in available_models 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( async def run_shield(
self, shield_id: str, messages: list[Message], params: dict[str, Any] | None = None self, shield_id: str, messages: list[Message], params: dict[str, Any] | None = None