Update agent.py

agent.py

@@ -64,34 +64,41 @@ def process_tool_output(tool_output):
         return tool_output.get('title', str(tool_output))
     return str(tool_output)
 
-class CustomDuckDuckGoSearchTool(Tool):
-    name = "CustomDuckDuckGoSearchTool"
-    description = "Perform a DuckDuckGo search and return a string result."
-    def __call__(self, query: str) -> str:
-        
-        try:
-            with DDGS() as ddgs:
-                results = list(ddgs.text(query, max_results=3))  # Convert generator to list
+@tool
+def CustomDuckDuckGoSearchTool(query:str): -> str
+    """
+    Perform a DuckDuckGo search and return a string result.
+
+    Args:
+        query: the question you input
+    """
+    try:
+        with DDGS() as ddgs:
+            results = list(ddgs.text(query, max_results=3))  # Convert generator to list
             return process_tool_output(results)
-        except Exception as e:
+    except Exception as e:
             print(f"Error in DuckDuckGoSearchTool: {e}")
             return f"Search error: {e}"
 custome_duck_search = CustomDuckDuckGoSearchTool()
 
-class CustomVisitWebpageTool(Tool):
-    name = "CustomVisitWebpageTool"
-    description = "Visit a webpage and extract content as a string."
-    def __call__(self, url: str) -> str:
-        try:
-            response = requests.get(url, timeout=10)
-            response.raise_for_status()
-            soup = BeautifulSoup(response.text, 'html.parser')
-            # Extract text from the webpage
-            text = soup.get_text(separator=' ', strip=True)
-            return text[:1000]  # Limit output length to avoid overload
-        except Exception as e:
-            print(f"Error in VisitWebpageTool: {e}")
-            return f"Webpage access error: {e}"
+@tool
+def CustomVisitWebpageTool(query:str):->str
+    """
+    Visit a webpage and extract content as a string.
+
+    Args:
+        query: the question you input
+    """
+    try:
+        response = requests.get(url, timeout=10)
+        response.raise_for_status()
+        soup = BeautifulSoup(response.text, 'html.parser')
+        # Extract text from the webpage
+        text = soup.get_text(separator=' ', strip=True)
+        return text[:1000]  # Limit output length to avoid overload
+    except Exception as e:
+        print(f"Error in VisitWebpageTool: {e}")
+        return f"Webpage access error: {e}"
 custome_web_search = CustomVisitWebpageTool()
 
 #=======Agent========#