phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Small cleanup for together safety implementation

Browse source
This commit is contained in:
Ashwin Bharambe 2024-09-28 15:47:35 -07:00
parent 940968ee3f
commit ced5fb6388
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/adapters/safety/together/together.py
View file

@ -83,9 +83,7 @@ async def get_safety_response(
response_text = response.choices[0].message.content
if response_text == "safe":
return SafetyViolation(
violation_level=ViolationLevel.INFO, user_message="safe", metadata={}
)
return None
parts = response_text.split("\n")
if len(parts) != 2: