Update agent.py

agent.py

@@ -17,6 +17,57 @@ llm = HuggingFaceInferenceAPI(
     token=os.getenv("HF_TOKEN"),
 )
 
+@tool
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers.
+    Args:
+        a: first int
+        b: second int
+    """
+    return a * b
+
+@tool
+def add(a: int, b: int) -> int:
+    """Add two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
+    return a + b
+
+@tool
+def subtract(a: int, b: int) -> int:
+    """Subtract two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
+    return a - b
+
+@tool
+def divide(a: int, b: int) -> int:
+    """Divide two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
+    if b == 0:
+        raise ValueError("Cannot divide by zero.")
+    return a / b
+
+@tool
+def modulus(a: int, b: int) -> int:
+    """Get the modulus of two numbers.
+    
+    Args:
+        a: first int
+        b: second int
+    """
+    return a % b
+    
 @tool
 def web_search(query: str) -> list:
     """Search Tavily for a query and return a maximum 3 results as LlamaIndex Documents.
@@ -74,6 +125,38 @@ wiki_search = FunctionTool(
     metadata=ToolMetadata(name="wiki_search", description="Wikipedia 2-hit search")
 )
 
-tools = [web_search, wiki_search]
+multiply_tool = FunctionTool(
+    multiply,
+    metadata=ToolMetadata(name="multiply", description="Multiply two numbers.")
+)
+add_tool = FunctionTool(
+    add,
+    metadata=ToolMetadata(name="add", description="Add two numbers.")
+)
+subtract_tool = FunctionTool(
+    subtract,
+    metadata=ToolMetadata(name="subtract", description="Subtract two numbers.")
+)
+divide_tool = FunctionTool(
+    divide,
+    metadata=ToolMetadata(name="divide", description="Divide two numbers.")
+)
+modulus_tool = FunctionTool(
+    modulus,
+    metadata=ToolMetadata(name="modulus", description="Modulus operation on two numbers.")
+)
+
+# Then include all tools in the list
+tools = [
+    web_search,
+    wiki_search,
+    multiply_tool,
+    add_tool,
+    subtract_tool,
+    divide_tool,
+    modulus_tool
+]
+
+
 
 agent = AgentWorkflow.from_tools_or_functions(tools, llm=llm)