Update tools/polite_guard.py

tools/polite_guard.py

@@ -22,7 +22,8 @@ class PoliteGuardTool(Tool):
     output_type = "any"
 
     def forward(self, input_text: Any) -> Any:
-        self.label = self.ask_polite_guard(input_text)
+        self.label, self.score = self.ask_polite_guard(input_text)
+        print(f"forward sets the following: {self.label } and {self.score}" )
         return self.label 
            
     def __init__(self, *args, **kwargs):
@@ -31,14 +32,14 @@ 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,int]:
         """ 
     
         Args:
             input_text: The text to classify.
 
         Returns:
-            string with classification label
+            tuple: with classification label and the score 
         """
         try:
             classifier = pipeline("text-classification", "Intel/polite-guard")