Update app.py

app.py

@@ -9,29 +9,43 @@ from tools.final_answer import FinalAnswerTool
 from smolagents import tools
 
 @tool
-def calculator(a: int, b: int, operation: str = "multiply") -> float:
-    """
-    Perform basic arithmetic operations on two numbers.
+class VisitWebpageTool(Tool):
+    name = "visit_webpage"
+    description = "Visits a webpage at the given url and reads its content as a markdown string. Use this to browse webpages."
+    inputs = {'url': {'type': 'string', 'description': 'The url of the webpage to visit.'}}
+    output_type = "string"
 
-    Args:
-        a (int): First number
-        b (int): Second number
-        operation (str): Operation to perform ("add", "subtract", "multiply", "divide")
+    def forward(self, url: str) -> str:
+        try:
+            import requests
+            from markdownify import markdownify
+            from requests.exceptions import RequestException
+
+            from smolagents.utils import truncate_content
+        except ImportError as e:
+            raise ImportError(
+                "You must install packages `markdownify` and `requests` to run this tool: for instance run `pip install markdownify requests`."
+            ) from e
+        try:
+            # Send a GET request to the URL with a 20-second timeout
+            response = requests.get(url, timeout=20)
+            response.raise_for_status()  # Raise an exception for bad status codes
+
+            # Convert the HTML content to Markdown
+            markdown_content = markdownify(response.text).strip()
+
+            # Remove multiple line breaks
+            markdown_content = re.sub(r"\n{3,}", "\n\n", markdown_content)
+
+            return truncate_content(markdown_content, 10000)
+
+        except requests.exceptions.Timeout:
+            return "The request timed out. Please try again later or check the URL."
+        except RequestException as e:
+            return f"Error fetching the webpage: {str(e)}"
+        except Exception as e:
+            return f"An unexpected error occurred: {str(e)}"
 
-    Returns:
-        float: Result of the operation
-    """
-    if operation == "add":
-        return a + b
-    elif operation == "subtract":
-        return a - b
-    elif operation == "multiply":
-        return a * b
-    elif operation == "divide":
-        return a / b if b != 0 else float("inf")
-    else:
-        raise ValueError(f"Unsupported operation: {operation}")
-        
        
 @tool 
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -74,9 +88,9 @@ from tools.final_answer import FinalAnswerTool
 from smolagents import load_tool
 
 tool_registry = {
-    "calculator": {
-        "tool": calculator,
-        "schema": calculator.tool_schema(),
+    "VisitWebPageTool": {
+        "tool": VisitWebpageTool,
+        "schema": VisitWebpageTool(),
         "enabled": True
     },
     "get_current_time_in_timezone": {
@@ -97,7 +111,7 @@ agent_tools = [entry["tool"] for entry in tool_registry.values() if entry["enabl
 ]
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, calculator, get_current_time_in_timezone],
+    tools=[final_answer, VisitWebpageTool, get_current_time_in_timezone],
     max_steps=6,
     verbosity_level=1,
     grammar=None,