Update tools/polite_guard.py

tools/polite_guard.py

@@ -44,7 +44,8 @@ class PoliteGuardTool(Tool):
             classifier = pipeline("text-classification", "Intel/polite-guard")
             result = classifier(input_text)
             print(f"return {str(result)}")
-            return (result[0]['label'], result[0]['score'])
+            output = result[0]
+            return output['label'], output['score']
             
         except Exception as e:
             return f"Error fetching classification for text '{input_text}': {str(e)}"