Update tools/polite_guard.py

tools/polite_guard.py

@@ -1,6 +1,7 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
 from transformers import pipeline
+import json
 
 
 class PoliteGuardTool(Tool):
@@ -19,7 +20,7 @@ class PoliteGuardTool(Tool):
     name = "polite_guard"
     description = "Uses Polite guard to classify input text from polite to impolite name and it provides a score as well.   Anything over .95 should be considered a significant threshold."
     inputs = {'input_text': {'type': 'any', 'description': 'Enter text for assessing whether it is respectful'}}
-    output_type = "any"
+    output_type = "string"
 
     def forward(self, input_text: Any) -> Any:
         str_return_value = self.ask_polite_guard(input_text)
@@ -50,6 +51,13 @@ class PoliteGuardTool(Tool):
             score = output['score']
             str_output = f"label is {label} with a score of {score}"
             return  str_output
+
+            # Ensure fixed keys + float score
+            payload = {
+                "label": str(output.get("label", "")),
+                "score": float(output.get("score", 0.5)),
+            }
+            return json.dumps(payload)
             
         except Exception as e:
             return f"Error fetching classification for text '{input_text}': {str(e)}"