Update app.py

app.py

@@ -1,6 +1,5 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool,HfApiModel,load_tool,tool
-import datetime 
-from duckduckgo_search import DDGS
+import datetime
 import requests
 import pytz
 import yaml
@@ -9,85 +8,54 @@ from tools.final_answer import FinalAnswerTool
 
 
 @tool
-def DuckDuckGoSearchTool (web_search: str) -> str:
-    """"Performs a DuckDuckGo web search based on your query (think a Google search) then returns the top search results.Perform a DuckDuckGo search and return formatted results.
+def add_numbers(a: int, b: int) -> int:
+    """
+    Add two numbers together.
 
     Args:
-        web_search (str): The query string to search for.
+        a (int): The first number.
+        b (int): The second number.
 
     Returns:
-        str: A formatted string containing the search results.
+        int: The sum of the two numbers.
     """
-    # your implementation here
-    ddgs = DDGS()
-    results = DuckDuckGoSearchTool.text(web_search, max_results=10)
-    if not results:
-        raise Exception("No results found! Try a less restrictive/shorter query.")
-
-    postprocessed_results = [
-        f"[{r['title']}]({r['href']})\n{r['body']}" for r in results
-    ]
-    return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
+    return a + b
        
-@tool 
+@tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
+    """
+    Get the current time in a given timezone.
+
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        timezone (str): A valid timezone string (e.g., 'America/New_York').
+
+    Returns:
+        str: The current time in that timezone, formatted as a string.
     """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+    from datetime import datetime
+    import pytz
 
+    tz = pytz.timezone(timezone)
+    now = datetime.now(tz)
+    return now.strftime("%Y-%m-%d %H:%M:%S %Z")
 
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
-model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
-custom_role_conversions=None,
-)
+model = HfApiModel(model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded)
 
-
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-# Tools list (no registry object needed)
 tool=[
-    DuckDuckGoSearchTool,
+    add_numbers,
     get_current_time_in_timezone,
     FinalAnswerTool(),
-    load_tool("agents-course/text-to-image", trust_remote_code=True),
 ]
 
-model = HfApiModel(
-model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-max_tokens=2048,
-temperature=0.5
-)
-with open("prompts.yaml", "r") as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-# --- Agent setup ---
-
 agent = CodeAgent(
-    tools=tool,
+    tool=tool
     model= model, 
     add_base_tools=True,
-    name="MyAgent",
-    description="Agent with web search and time lookup tools"
 )