Update tools/polite_guard.py

tools/polite_guard.py

@@ -32,7 +32,7 @@ class PoliteGuardTool(Tool):
         self.score = None
         
     #@tool
-    def ask_polite_guard(self, input_text: str) -> str:
+    def ask_polite_guard(self, input_text: str) -> tuple[str, float]:
         """ 
     
         Args:
@@ -43,7 +43,10 @@ class PoliteGuardTool(Tool):
         """
         try:
             classifier = pipeline("text-classification", "Intel/polite-guard")
-            return classifier(input_text)
+            result = classifier(input_text)[0]
+            print(result)
+            print(len(result))
+            return result['label'], result['score']
             
         except Exception as e:
             return f"Error fetching classification for text '{input_text}': {str(e)}"