remove unwanted

Author: slekkala1

llama_stack/providers/inline/safety/code_scanner/code_scanner.py

@@ -6,9 +6,10 @@
 
 import logging
 import uuid
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
-from codeshield.cs import CodeShield, CodeShieldScanResult
+if TYPE_CHECKING:
+    from codeshield.cs import CodeShieldScanResult
 
 from llama_stack.apis.inference import Message
 from llama_stack.apis.safety import (
@@ -59,6 +60,8 @@ async def run_shield(
         if not shield:
             raise ValueError(f"Shield {shield_id} not found")
 
+        from codeshield.cs import CodeShield
+
         text = "\n".join([interleaved_content_as_str(m.content) for m in messages])
         log.info(f"Running CodeScannerShield on {text[50:]}")
         result = await CodeShield.scan_code(text)
@@ -72,7 +75,7 @@ async def run_shield(
             )
         return RunShieldResponse(violation=violation)
 
-    def get_moderation_object_results(self, scan_result: CodeShieldScanResult) -> ModerationObjectResults:
+    def get_moderation_object_results(self, scan_result: "CodeShieldScanResult") -> ModerationObjectResults:
         categories = {}
         category_scores = {}
         category_applied_input_types = {}
@@ -102,6 +105,8 @@ async def run_moderation(self, input: str | list[str], model: str) -> Moderation
         inputs = input if isinstance(input, list) else [input]
         results = []
 
+        from codeshield.cs import CodeShield
+
         for text_input in inputs:
             log.info(f"Running CodeScannerShield moderation on input: {text_input[:100]}...")
             scan_result = await CodeShield.scan_code(text_input)

llama_stack/providers/inline/safety/llama_guard/llama_guard.py

@@ -455,7 +455,7 @@ def create_moderation_object(self, model: str, unsafe_code: str | None = None) -
 
     def is_content_safe(self, response: str, unsafe_code: str | None = None) -> bool:
         """Check if content is safe based on response and unsafe code."""
-        if response.strip() == SAFE_RESPONSE:
+        if response.strip().lower().startswith(SAFE_RESPONSE):
             return True
 
         if unsafe_code: