Update agent.py

added more tools

agent.py

@@ -1,8 +1,8 @@
-from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, tool
+from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, WikipediaSearchTool, tool
 
 
 @tool
-def add(int: a, int: b) -> int:
+def add(a: int, b: int) -> int:
     """
         This tool adds two integers together and returns an integer.
 
@@ -13,7 +13,7 @@ def add(int: a, int: b) -> int:
     return a + b
 
 @tool
-def subtract(int: a, int: b) -> int:
+def subtract(a: int, b: int) -> int:
     """
         This tool subtracts two integers and returns an integer.
 
@@ -24,7 +24,7 @@ def subtract(int: a, int: b) -> int:
     return a - b
 
 @tool
-def multiply(int: a, int: b) -> int:
+def multiply(a: int, b: int) -> int:
     """
         This tool multiplies two integers together and returns an integer.
 
@@ -35,7 +35,7 @@ def multiply(int: a, int: b) -> int:
     return a * b
 
 @tool
-def divide(int: a, int: b) -> float:
+def divide(a: int, b: int) -> int:
     """
         This tool divides two integers together and returns a float.
 
@@ -47,20 +47,35 @@ def divide(int: a, int: b) -> float:
         raise ValueError("Cannot divide by zero.")
     return a* (1.0) / b
 
+def modulo(a: int, b: int) -> float:
+    """
+        This tool returns the modulus of two integers
+
+        args:
+            a: Integer 1
+            b: Integer 2
+    """
+    return a % b
+
 tools=[
     add,
     subtract,
     multiply,
-    divide
+    divide,
+    modulo,
+    DuckDuckGoSearchTool(),
+    WikipediaSearchTool()
 ]
 
 
 def create_agent():
-    manager_agent = CodeAgent(
+    agent = CodeAgent(
         model = HfApiModel(),
-        
+        tools = tools,
+        max_steps=10,
+        verbosity_level=2
     )
 
-    return manager_agent
+    return agent