From 1d775b342c23cef014f3832825730619d1f42bcf Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Tue, 5 Nov 2024 16:23:00 -0800 Subject: [PATCH] change to return only one violation --- .../providers/adapters/safety/bedrock/bedrock.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/llama_stack/providers/adapters/safety/bedrock/bedrock.py b/llama_stack/providers/adapters/safety/bedrock/bedrock.py index 54ace0cda..18da5e914 100644 --- a/llama_stack/providers/adapters/safety/bedrock/bedrock.py +++ b/llama_stack/providers/adapters/safety/bedrock/bedrock.py @@ -136,13 +136,11 @@ class BedrockSafetyAdapter(Safety, ShieldsProtocolPrivate): metadata = dict(assessment) return RunShieldResponse( - violations=[ - SafetyViolation( - user_message=user_message, - violation_level=ViolationLevel.ERROR, - metadata=metadata, - ) - ] + violation=SafetyViolation( + user_message=user_message, + violation_level=ViolationLevel.ERROR, + metadata=metadata, + ) ) - return RunShieldResponse(violations=[]) + return RunShieldResponse()