Update tools/polite_guard.py

tools/polite_guard.py

@@ -4,20 +4,20 @@ from transformers import pipeline
 
 
 class PoliteGuardTool(Tool):
- """
-    Takes the input text from users and then evaluates it against Polite Guard to return specific information
-    about whether the content is polite. 
-    
-    Args:
-        input_text: Text that the user inputs into the agent and should then be evaluated. 
-        
-    Returns:
-        A classification label about whether the content is polite, somewhat polite, neutral or impolite.
+    """
+        Takes the input text from users and then evaluates it against Polite Guard to return specific information
+        about whether the content is polite. 
         
+        Args:
+            input_text: Text that the user inputs into the agent and should then be evaluated. 
+            
+        Returns:
+            A classification label about whether the content is polite, somewhat polite, neutral or impolite.
+            
     """
 
     name = "polite_guard"
-    description = "Uses Polite guard to classify the user's input text by 4 labels from polite to impolite"
+    description = "Uses Polite guard to classify input text from polite to impolite"
     inputs = {'input_text': {'type': 'any', 'description': 'Enter text for assessing whether it is respectful'}}
     output_type = "any"