ai-agents.py

from agents import Agent, ModelSettings, function_tool, Runner

@function_tool
def get_pending_payments_ammount() -> int:
    """
    Return the current amount that needs to be paid for the current expenses.
    """
    return 200

@function_tool
def get_account_balance() -> int:
    """
    Return the amount balance for the checking amount.
    """
    return 1000

finanace_agent = Agent(
    name="Finance agent",
    instructions="You are helpfule agent that helps user with managing money.",
    model="o3-mini",
    tools=[get_pending_payments_ammount,get_account_balance],
)

helper_agent = Agent(
    name="helper agent",
    instructions="You are helpfule agent that undestanding finance concept. Only answer question about finance related question and nothing else.",
    model="o3-mini"
)


triage_agent = Agent(
    name="Triage Agent",
    instructions="You determine which agent to use based on the user's homework question",
    handoffs=[finanace_agent, helper_agent]
)


result = Runner.run_sync(triage_agent, "What is Roth IRA.")
print(result.final_output)