Update agent.py

agent.py

@@ -138,6 +138,18 @@ class ValidatedExcelReader(Tool):
 class UniversalLoader(Tool):
     name = "universal_loader"
     description = "Loads various file types and web content using appropriate sub-tools."
+    inputs = {
+        'source': {
+            'type': 'string', 
+            'description': 'Type of source to load (web/excel/audio/arxiv)'
+        },
+        'task_id': {
+            'type': 'string', 
+            'description': 'Task ID for attachments',
+            'nullable': True
+        }
+    }
+    output_type = "string"
     
     def __init__(self):
         self.loaders = {
@@ -276,7 +288,7 @@ class EnhancedExcelReader(Tool):
 # --------------------------
 # Cross-Verified Search
 # --------------------------
-class CrossVerifiedSearch:
+class CrossVerifiedSearch(Tool):
     name = "cross_verified_search"
     description = "Searches multiple sources and returns consensus results."
     inputs = {'query': {'type': 'string', 'description': 'Search query'}}