main.py

from message import Message
from react_agent import ReactAgent
from tool import tool


@tool
def add(a: int, b: int) -> int:
    """Returns the sum of a and b"""
    return a + b


@tool
def substract(a: int, b: int) -> int:
    """Returns the difference of a and b"""
    return a - b


@tool
def multiply(a: int, b: int) -> int:
    """Multiplies a with b and returns the value"""
    return a * b


@tool
def divide(a: int, b: int) -> float:
    """Divides a with b and returns the value"""
    return a / b


@tool
def exponent(a: int, b: int) -> int:
    """returns the exponent of a with b"""
    return a ** b


tools = [add, substract, multiply, divide, exponent]
agent = ReactAgent(tools=tools, include_internal_logs=True)

message = Message("user",
                  "What is 2 plus 4 and then divided by 3 and multiplied by 4? and what if we also raise it to power of 3?")
print(agent.execute(message))