agent.py

from smolagents import HfApiModel, CodeAgent, DuckDuckGoSearchTool, tool


@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