fully functional

app.py

@@ -1,22 +1,13 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, HfApiModel, load_tool, tool
 import datetime
-import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
+from tools.web_search import DuckDuckGoSearchTool
+from tools.visit_webpage import VisitWebpageTool
+from tools.calculate_tool import CalculateTool
 from Gradio_UI import GradioUI
-
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
-@tool
-def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
-    #Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
+from tools.get_weather import GetWeatherTool
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +46,15 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+        tools=[
+            final_answer,
+            get_current_time_in_timezone,
+            CalculateTool(),
+            GetWeatherTool(),
+            DuckDuckGoSearchTool(),
+            VisitWebpageTool(),
+            image_generation_tool
+        ],
     max_steps=6,
     verbosity_level=1,
     grammar=None,

requirements.txt

@@ -3,3 +3,4 @@ smolagents==1.13.0
 requests
 duckduckgo_search
 pandas
+pytz

tools/calculate_tool.py

@@ -0,0 +1,20 @@
+from smolagents.tools import Tool
+
+
+class CalculateTool(Tool):
+    name = "calculate"
+    description = "Evaluates a mathematical expression safely."
+    inputs = {
+        "expression": {
+            "type": "string",
+            "description": "A mathematical expression (e.g., '2 * 3 + 5')",
+        }
+    }
+    output_type = "string"
+
+    def forward(self, expression: str) -> str:
+        try:
+            result = eval(expression, {"__builtins__": {}}, {})
+            return f"The result of '{expression}' is: {result}"
+        except Exception as e:
+            return f"Error calculating '{expression}': {str(e)}"

tools/get_weather.py

@@ -0,0 +1,43 @@
+from asyncio import timeout
+from warnings import catch_warnings
+import os
+from smolagents.tools import Tool
+import requests
+
+class GetWeatherTool(Tool):
+    name="get_weather"
+    description="Get weather information for a specific city using OpenWeatherAPI. Use this tool to fetch current weather data for a given city using the OpenWeatherAPI (e.g., 'Delhi')."
+    inputs={'city':{'type':'string',
+                    'description':'The name of the city for which to fetch weather information (e.g., "Delhi").'
+                    }
+            }
+    output_type="string"
+    
+    def __init__(self, api_key:str=None, timeout:int=10):
+        super().__init__()
+        self.api_key=api_key or os.getenv("OPENWEATHER_API_KEY")
+        self.timeout=timeout
+        self.base_url="http://api.openweathermap.org/data/2.5/weather"
+        
+        self.session=requests.Session()
+        
+    
+    def forward(self, city:str) -> str:
+        params={
+            'q':city,
+            'appid':self.api_key,
+            'units':'metric'
+        }
+        try:
+            res=self.session.get(self.base_url, params=params, timeout=self.timeout)
+            res.raise_for_status()
+            data=res.json()
+            temp=data['main']['temp']
+            desc=data['weather'][0]['description']
+            return f"{city}:{temp} C, {desc}"
+
+        except Exception as e:
+            return f"failed to fetch weather for {city}: {str(e)}"
+
+
+        

tools/visit_webpage.py

@@ -1,5 +1,6 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
+import re
 import requests
 import markdownify
 import smolagents