autogen/python/packages/autogen-studio/autogenstudio/gallery/tools/calculator.py

from autogen_core.tools import FunctionTool


def calculator(a: float, b: float, operator: str) -> str:
    try:
        if operator == "+":
            return str(a + b)
        elif operator == "-":
            return str(a - b)
        elif operator == "*":
            return str(a * b)
        elif operator == "/":
            if b == 0:
                return "Error: Division by zero"
            return str(a / b)
        else:
            return "Error: Invalid operator. Please use +, -, *, or /"
    except Exception as e:
        return f"Error: {str(e)}"


# Create calculator tool
calculator_tool = FunctionTool(
    name="calculator",
    description="A simple calculator that performs basic arithmetic operations",
    func=calculator,
    global_imports=[],
)