From ced5fb6388d577520a00076293c47fc06c7aa156 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Sat, 28 Sep 2024 15:47:35 -0700
Subject: [PATCH] Small cleanup for together safety implementation

---
 llama_stack/providers/adapters/safety/together/together.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/llama_stack/providers/adapters/safety/together/together.py b/llama_stack/providers/adapters/safety/together/together.py
index 940d02861..8e552fb6c 100644
--- a/llama_stack/providers/adapters/safety/together/together.py
+++ b/llama_stack/providers/adapters/safety/together/together.py
@@ -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: