Update app.py

app.py

@@ -19,13 +19,19 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 
 @tool
-def add_two_numbers(n1:float, n2:float)-> float: 
-    """A tool that adds two integer or float numbers.
-    Args:
-        n1: float
-        n2: float
+def add_numbers(a: float, b: float) -> float:
+    """
+    Adds two numbers and returns the result.
+
+    Parameters:
+    - a (float): The first number.
+    - b (float): The second number.
+
+    Returns:
+    - float: The sum of a and b.
     """
-    return n1 + n2    
+    c = a + b
+    return f"The result of adding {a} plus {b} equals {c}"  
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -64,7 +70,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, DuckDuckGoSearchTool(), add_two_numbers], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, DuckDuckGoSearchTool(), add_numbers], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,