Update tools/polite_guard.py

tools/polite_guard.py

@@ -23,7 +23,7 @@ class PoliteGuardTool(Tool):
 
     def forward(self, input_text: Any) -> Any:
         self.label, self.score = self.ask_polite_guard(input_text)
-        print(f"forward sets the following: {self.label } and {self.score}" )
+        print(f"forward sets the following: {self.label } and {self.score}")
         return self.label 
            
     def __init__(self, *args, **kwargs):
@@ -32,18 +32,18 @@ class PoliteGuardTool(Tool):
         self.score = None
         
     #@tool
-        def ask_polite_guard(self, input_text: str) -> tuple[str,int]:
-        """ 
-    
-        Args:
-            input_text: The text to classify.
+    def ask_polite_guard(self, input_text: str) -> tuple[str,int]:
+    """ 
 
-        Returns:
-            tuple: with classification label and the score 
-        """
-        try:
-            classifier = pipeline("text-classification", "Intel/polite-guard")
-            return classifier(input_text)
-            
-        except Exception as e:
-            return f"Error fetching classification for text '{input_text}': {str(e)}"
+    Args:
+        input_text: The text to classify.
+
+    Returns:
+        tuple: with classification label and the score 
+    """
+    try:
+        classifier = pipeline("text-classification", "Intel/polite-guard")
+        return classifier(input_text)
+        
+    except Exception as e:
+        return f"Error fetching classification for text '{input_text}': {str(e)}"