Update ai_tools.py

ai_tools.py

@@ -1,6 +1,6 @@
-from duckduckgo_search import DDGS
 import re
-from typing import Dict, Any
+from duckduckgo_search import DDGS
+from typing import List, Dict, Any
 
 class BaseTool:
     def __init__(self, name: str, description: str):
@@ -14,33 +14,34 @@ class Calculator(BaseTool):
     def __init__(self):
         super().__init__(
             name="Calculator",
-            description="Performs basic arithmetic. Input format: 'expression: <math_expression>'"
+            description="Performs basic arithmetic. Input: math expression as string"
         )
 
     def run(self, expression: str) -> str:
         try:
-            # Safe evaluation for basic operations
             expression = expression.replace(' ', '')
             if not re.match(r'^[\d+\-*/.()]+$', expression):
-                return "Error: Invalid characters"
+                return "Error: Invalid characters in expression"
             result = eval(expression)
             return str(result)
         except Exception as e:
             return f"Calculation error: {str(e)}"
 
 class DocRetriever(BaseTool):
-    def __init__(self, document: str = ""):
+    def __init__(self):
         super().__init__(
             name="DocRetriever",
             description="Searches provided text. Input: 'query: <search_term>'"
         )
-        self.document = document
+        self.document = ""
+
+    def load_document(self, text: str):
+        self.document = text
 
     def run(self, query: str) -> str:
         if not self.document:
-            return "Error: No document loaded"
+            return "No document loaded"
         
-        # Case-insensitive search for sentences containing query
         sentences = [s.strip() for s in self.document.split('.') if s]
         results = [s for s in sentences if query.lower() in s.lower()]
         return '. '.join(results[:3]) + '...' if results else "No matches found"
@@ -49,7 +50,7 @@ class WebSearcher(BaseTool):
     def __init__(self):
         super().__init__(
             name="WebSearcher",
-            description="Searches the web. Input: 'query: <search_term>'"
+            description="Searches the web. Input: search query"
         )
 
     def run(self, query: str) -> str: