Update app.py

app.py

@@ -50,6 +50,22 @@ def get_weather(city: str) -> dict:
     else:
         return {"Error": "Unable to fetch weather data."}
 
+@tool
+def multiply_integers(a: int, b: int) -> str:
+    """
+    A tool that multiplies two integers and returns the result as a string.
+
+    Args:
+        a (int): The first integer.
+        b (int): The second integer.
+
+    Returns:
+        str: A string describing the multiplication result.
+    """
+    result = a * b
+    return f"The result of multiplying {a} by {b} is {result}."
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -87,7 +103,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_weather], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_weather, multiply_integers], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,