Update app.py

app.py

@@ -22,6 +22,20 @@ def sum_tool(a: float, b: float) -> float:
     """
     return a + b
 
+@tool
+def multiply_tool(a: float, b: float) -> float:
+    """
+    Returns the product of two numbers.
+
+    Args:
+        a (float): The first value.
+        b (float): The second value.
+
+    Returns:
+        float: Product of a and b
+    """
+    return a * b
+
 @tool
 def sub_tool(a: float, b: float) -> float:
     """
@@ -43,7 +57,7 @@ model = LiteLLMModel(
 )
 
 agent = ToolCallingAgent(
-    tools=[sum_tool, sub_tool, PythonInterpreterTool()],
+    tools=[sum_tool, sub_tool, multiply_tool, PythonInterpreterTool()],
     model=model,
     max_steps=3
 )