Update sentiment_analysis.py

sentiment_analysis.py

@@ -12,7 +12,7 @@ class SentimentAnalysisTool(Tool):
             "description": "The text to analyze for sentiment"
         }
     }
-    output_type = "list"
+    output_type = "json"
     
     # Available sentiment analysis models
     models = {
@@ -38,13 +38,13 @@ class SentimentAnalysisTool(Tool):
         return self.predict(text)
         
     def _parse_output(self, output_json):  
-        """Parse model output into a list of (label, score) tuples."""
-        list_pred = []
+        """Parse model output into a dictionary of scores by label."""
+        result = {}
         for i in range(len(output_json[0])):
             label = output_json[0][i]['label']
             score = output_json[0][i]['score']
-            list_pred.append((label, score))
-        return list_pred
+            result[label] = score
+        return result
     
     def _get_classifier(self, model_id):
         """Get or create a classifier for the given model ID."""