tool

agent.py

@@ -16,7 +16,11 @@ from tools import (
     audio_transcriber_tool,
     excel_tool,
     analyze_code_tool,
-    image_tool
+    image_tool,
+    add_tool,
+    subtract_tool,
+    multiply_tool,
+    divide_tool
 )
 
 # ─────────────────────────── Configuration ───────────────────────────────
@@ -43,7 +47,11 @@ def build_graph():
         audio_transcriber_tool,
         excel_tool,
         analyze_code_tool,
-        image_tool
+        image_tool,
+        add_tool,
+        subtract_tool,
+        multiply_tool,
+        divide_tool
     ]
     
     # Create the react agent - it will use the system prompt from the messages

app.py

@@ -44,8 +44,8 @@ class BasicAgent:
             ]
         }
         
-        # Run the agent
-        out_state = self.graph.invoke(init_state, {"recursion_limit": 8})
+        # Run the agent with increased steps
+        out_state = self.graph.invoke(init_state, {"recursion_limit": 15})
         
         # Extract the final answer from the last message
         if out_state and "messages" in out_state:

tools.py

@@ -405,3 +405,55 @@ def analyze_code_tool(task_id: str) -> str:
 #         "web_search_query": None,
 #         "web_search_result": result_text
 #     }
+
+# ─────────────────────────── Math Tools ───────────────────────────────
+
+@tool
+def add_tool(a: float, b: float) -> str:
+    """
+    Add two numbers.
+    Args:
+        a: First number
+        b: Second number
+    Returns: The sum of a and b
+    """
+    result = a + b
+    return f"Addition result: {a} + {b} = {result}"
+
+@tool
+def subtract_tool(a: float, b: float) -> str:
+    """
+    Subtract two numbers.
+    Args:
+        a: First number (minuend)
+        b: Second number (subtrahend)
+    Returns: The difference of a minus b
+    """
+    result = a - b
+    return f"Subtraction result: {a} - {b} = {result}"
+
+@tool
+def multiply_tool(a: float, b: float) -> str:
+    """
+    Multiply two numbers.
+    Args:
+        a: First number
+        b: Second number
+    Returns: The product of a and b
+    """
+    result = a * b
+    return f"Multiplication result: {a} × {b} = {result}"
+
+@tool
+def divide_tool(a: float, b: float) -> str:
+    """
+    Divide two numbers.
+    Args:
+        a: First number (dividend)
+        b: Second number (divisor)
+    Returns: The quotient of a divided by b
+    """
+    if b == 0:
+        return "Division error: Cannot divide by zero"
+    result = a / b
+    return f"Division result: {a} ÷ {b} = {result}"