Update agent.py

added mathematical functions

agent.py

@@ -1,5 +1,66 @@
-from smolagents import HfApiModel, CodeAgent
+from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, tool
 
-manager_agent = CodeAgent(
+
+@tool
+def add(int: a, int: b) -> int:
+    """
+        This tool adds two integers together and returns an integer.
+
+        args:
+            a: Integer 1
+            b: Integer 2
+    """
+    return a + b
+
+@tool
+def subtract(int: a, int: b) -> int:
+    """
+        This tool subtracts two integers and returns an integer.
+
+        args:
+            a: Integer 1
+            b: Integer 2
+    """
+    return a - b
+
+@tool
+def multiply(int: a, int: b) -> int:
+    """
+        This tool multiplies two integers together and returns an integer.
+
+        args:
+            a: Integer 1
+            b: Integer 2
+    """
+    return a * b
+
+@tool
+def divide(int: a, int: b) -> float:
+    """
+        This tool divides two integers together and returns a float.
+
+        args:
+            a: Integer 1
+            b: Integer 2
+    """
+    if b == 0:
+        raise ValueError("Cannot divide by zero.")
+    return a* (1.0) / b
+
+tools=[
+    add,
+    subtract,
+    multiply,
+    divide
+]
+
+
+def create_agent():
+    manager_agent = CodeAgent(
+        model = HfApiModel(),
+        
+    )
+
+    return manager_agent
     
-)
+