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

minor change in shield validation

Browse source
This commit is contained in:
jhpiedrahitao 2025-07-09 11:11:51 -05:00
parent 9b7eecebcf
commit bfd1ee6951
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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
): ):
raise ValueError(f"Shield {shield.provider_resource_id} not found in SambaNova") logger.warning(f"Shield {shield.provider_resource_id} not found as available in SambaNova 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