From bfd1ee69510c84cd99fd6d3af76f783e0fec625b Mon Sep 17 00:00:00 2001 From: jhpiedrahitao Date: Wed, 9 Jul 2025 11:11:51 -0500 Subject: [PATCH] minor change in shield validation --- llama_stack/providers/remote/safety/sambanova/sambanova.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/safety/sambanova/sambanova.py b/llama_stack/providers/remote/safety/sambanova/sambanova.py index 84c8267ae..0e034935f 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 ): - 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( self, shield_id: str, messages: list[Message], params: dict[str, Any] | None = None