tools/arithmetic_tools.py

from crewai.tools import tool

class ArithmeticTools():
    @tool("Addition Tool")
    def add(a: float, b: float) -> float:
        """Add two numbers.
        
        Args:
            a: First number
            b: Second number
        Returns:
            Result number
        """
        return a + b
    
    @tool("Subtraction Tool")
    def subtract(a: float, b: float) -> float:
        """Subtract two numbers.
        
        Args:
            a: First number
            b: Second number
        Returns:
            Result number
        """
        return a - b
    
    @tool("Multiplication Tool")
    def multiply(a: float, b: float) -> float:
        """Multiply two numbers.
        Args:
            a: First number
            b: Second number
        Returns:
            Result number
        """
        return a * b
    
    @tool("Division Tool")
    def divide(a: float, b: float) -> float:
        """Divide two numbers.
        
        Args:
            a: First number
            b: Second number
        Returns:
            Result number
        """
        if b == 0:
            raise ValueError("Cannot divide by zero.")
        return a / b
    
    @tool("Modulus Tool")
    def modulus(a: float, b: float) -> float:
        """Get the modulus of two numbers.
        
        Args:
            a: First number
            b: Second number
        Returns:
            Result number
        """
        return a % b