Refactor tool imports and add calculator functions to the toolset

tools/__init__.py

@@ -1,8 +1,10 @@
 from .wikipedia import wiki_fetch_article, wiki_parse_html
 from .search import web_search
 from .code_interpreter import execute_code_multilang
-from .files import file_management_toolkit
+from .files import *
+from .calculator import *
 from langchain_core.tools import BaseTool
+from . import calculator, files
 
 def get_all_tools() -> list[BaseTool]:
     """
@@ -15,10 +17,16 @@ def get_all_tools() -> list[BaseTool]:
         wiki_fetch_article,
         wiki_parse_html,
         web_search,
-        execute_code_multilang
+        execute_code_multilang,
     ]
-
-    tools.extend(file_management_toolkit.get_tools())  # Add file management tools
+    
+    # Automatically add all calculator functions
+    for name in calculator.__all__:
+        tools.append(getattr(calculator, name))
+    
+    # Automatically add all file management functions
+    for name in files.__all__:
+        tools.append(getattr(files, name))
 
     return tools
 

tools/calculator.py

@@ -0,0 +1,118 @@
+from langchain_core.tools import tool
+import wikipediaapi
+import requests
+from bs4 import BeautifulSoup
+
+__all__ = [
+    'add_numbers',
+    'subtract_numbers',
+    'multiply_numbers',
+    'divide_numbers',
+    'power_numbers',
+    'root_numbers',
+    'modulus_numbers'
+]
+
+@tool
+def add_numbers(a: float, b: float) -> float:
+    """
+    Adds two numbers.
+
+    Args:
+        a (float): The first number.
+        b (float): The second number.
+
+    Returns:
+        float: The sum of the two numbers.
+    """
+    return a + b
+
+@tool
+def subtract_numbers(a: float, b: float) -> float:
+    """
+    Subtracts the second number from the first.
+
+    Args:
+        a (float): The first number.
+        b (float): The second number.
+
+    Returns:
+        float: The result of the subtraction.
+    """
+    return a - b
+
+@tool
+def multiply_numbers(a: float, b: float) -> float:
+    """
+    Multiplies two numbers.
+
+    Args:
+        a (float): The first number.
+        b (float): The second number.
+
+    Returns:
+        float: The product of the two numbers.
+    """
+    return a * b
+
+@tool
+def divide_numbers(a: float, b: float) -> float:
+    """
+    Divides the first number by the second.
+
+    Args:
+        a (float): The first number.
+        b (float): The second number.
+
+    Returns:
+        float: The result of the division.
+    """
+    if b == 0:
+        return float("inf")  # Handle division by zero
+    return a / b
+
+@tool
+def modulus_numbers(a: float, b: float) -> float:
+    """
+    Calculates the modulus of the first number by the second.
+
+    Args:
+        a (float): The first number.
+        b (float): The second number.
+
+    Returns:
+        float: The result of the modulus operation.
+    """
+    if b == 0:
+        return float("inf")  # Handle division by zero
+    return a % b
+
+@tool
+def power_numbers(a: float, b: float) -> float:
+    """
+    Raises the first number to the power of the second.
+
+    Args:
+        a (float): The base number.
+        b (float): The exponent.
+
+    Returns:
+        float: The result of the exponentiation.
+    """
+    return a ** b
+
+@tool
+def root_numbers(a: float, b: float) -> float:
+    """
+    Calculates the nth root of a number.
+
+    Args:
+        a (float): The number.
+        b (float): The root.
+
+    Returns:
+        float: The result of the root operation.
+    """
+    if b == 0:
+        return float("inf")  # Handle division by zero
+    return a ** (1 / b)