Update agent.py

agent.py

@@ -136,6 +136,9 @@ class ValidatedExcelReader(Tool):
 # --------------------------
 
 class UniversalLoader(Tool):
+    name = "universal_loader"
+    description = "Loads various file types and web content using appropriate sub-tools."
+    
     def __init__(self):
         self.loaders = {
             'excel': ValidatedExcelReader(),
@@ -274,6 +277,11 @@ class EnhancedExcelReader(Tool):
 # Cross-Verified Search
 # --------------------------
 class CrossVerifiedSearch:
+    name = "cross_verified_search"
+    description = "Searches multiple sources and returns consensus results."
+    inputs = {'query': {'type': 'string', 'description': 'Search query'}}
+    output_type = "string"
+
     SOURCES = [
         DuckDuckGoSearchTool(),
         WikipediaSearchTool(),
@@ -288,7 +296,16 @@ class CrossVerifiedSearch:
             except Exception as e:
                 continue
         return self._consensus(results)
-
+    
+    def forward(self, query: str) -> str:
+        results = []
+        for source in self.SOURCES:
+            try:
+                results.append(source(query))
+            except Exception as e:
+                continue
+        return self._consensus(results)
+        
     def _consensus(self, results):
         # Simple majority voting implementation
         counts = {}
@@ -315,12 +332,11 @@ class MagAgent:
 
         self.tools = [
             UniversalLoader(),
-            EnhancedSearchTool(),  # Replaces individual search tools
+            CrossVerifiedSearch(),  # Replaces individual search tools
             ValidatedExcelReader(),
             VisitWebpageTool(),
             DownloadTaskAttachmentTool(),
             SpeechToTextTool(),
-            CrossVerifiedSearch()
         ]
 
         with open("prompts.yaml") as f: