Update decision_maker.py

decision_maker.py

@@ -1,5 +1,5 @@
 from typing import Dict, List, Any, Optional
-from dataclasses import dataclass
+from dataclasses import dataclass, asdict
 from enum import Enum
 import json
 import time
@@ -19,6 +19,17 @@ class Tool:
     optional_params: List[str]
     confidence_threshold: float = 0.7
 
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert Tool object to a dictionary for JSON serialization."""
+        return {
+            "name": self.name,
+            "type": self.type.value,
+            "description": self.description,
+            "required_params": self.required_params,
+            "optional_params": self.optional_params,
+            "confidence_threshold": self.confidence_threshold
+        }
+
 class DecisionMaker:
     def __init__(self):
         self.tools = self._initialize_tools()
@@ -66,15 +77,15 @@ class DecisionMaker:
         
         # Determine required tools based on intent
         if "search" in analysis["intent"]:
-            analysis["required_tools"].append(self.tools["simple_search"])
+            analysis["required_tools"].append(self.tools["simple_search"].to_dict())
         if "code" in analysis["intent"]:
-            analysis["required_tools"].append(self.tools["code_analysis"])
+            analysis["required_tools"].append(self.tools["code_analysis"].to_dict())
         if "file" in analysis["intent"]:
-            analysis["required_tools"].append(self.tools["file_operation"])
+            analysis["required_tools"].append(self.tools["file_operation"].to_dict())
             
         # Calculate confidence based on tool requirements
         if analysis["required_tools"]:
-            analysis["confidence"] = min(tool.confidence_threshold for tool in analysis["required_tools"])
+            analysis["confidence"] = min(tool["confidence_threshold"] for tool in analysis["required_tools"])
             
         # Generate suggested actions
         analysis["suggested_actions"] = self._generate_actions(analysis)
@@ -105,11 +116,11 @@ class DecisionMaker:
         
         for tool in analysis["required_tools"]:
             action = {
-                "tool": tool.name,
-                "type": tool.type.value,
-                "confidence": tool.confidence_threshold,
-                "required_params": tool.required_params,
-                "optional_params": tool.optional_params
+                "tool": tool["name"],
+                "type": tool["type"],
+                "confidence": tool["confidence_threshold"],
+                "required_params": tool["required_params"],
+                "optional_params": tool["optional_params"]
             }
             actions.append(action)