Update tools/polite_guard.py

tools/polite_guard.py

@@ -43,10 +43,13 @@ class PoliteGuardTool(Tool):
         """
         try:
             classifier = pipeline("text-classification", "Intel/polite-guard")
-            result = classifier(input_text)[0]
-            print(result)
-            print(len(result))
-            return result['label'], result['score']
+            result = classifier(input_text)
+            print(f"return {str(result)}")
+            print(f"length is {len(result)}")
+            # get the first tuple 
+            label = result[0]['label']
+            score = result[0]['score']
+            return label, score
             
         except Exception as e:
             return f"Error fetching classification for text '{input_text}': {str(e)}"