simple_agent = Agent("openai:gpt-4o", instrument=True)

response = simple_agent.run_stream_sync("Tell me a short joke")
for chunk in response.stream_text():
    print(chunk, end="", flush=True)

class CityInfo(BaseModel):
    name: str
    country: str
    population: int
    famous_landmark: str


structured_agent = Agent(
    "openai:gpt-4o",
    output_type=CityInfo,
    instrument=True,
)

response = structured_agent.run_stream_sync("Tell me about Paris")
result = response.get_output()

calculator_agent = Agent(
    "openai:gpt-4o",
    system_prompt="You are a calculator assistant. Use the tools to perform calculations.",
    instrument=True,
)


@calculator_agent.tool
def add(ctx: RunContext[None], a: float, b: float) -> float:
    """Add two numbers together."""
    return a + b


@calculator_agent.tool
def multiply(ctx: RunContext[None], a: float, b: float) -> float:
    """Multiply two numbers together."""
    return a * b


@calculator_agent.tool
def subtract(ctx: RunContext[None], a: float, b: float) -> float:
    """Subtract b from a."""
    return a - b


response = calculator_agent.run_stream_sync(
    "Calculate (5 + 3) * 2 - 1. Show your work step by step."
)
for chunk in response.stream_text():
    print(chunk, end="", flush=True)

basic_agent = Agent("openai:gpt-4o", instrument=True)
response = basic_agent.run_stream_sync("What is 2 + 2?")
print(f"Direct output: {response.get_output()}")

simple_agent = Agent("openai:gpt-4o", instrument=True)

    async with simple_agent.run_stream("Tell me a short joke") as response:
        async for chunk in response.stream_text(delta=True):
            print(chunk, end="", flush=True)

class CityInfo(BaseModel):
    name: str
    country: str
    population: int
    famous_landmark: str

structured_agent = Agent(
        "openai:gpt-4o",
        output_type=CityInfo,
        instrument=True,
    )

    async with structured_agent.run_stream("Tell me about Paris") as response:
        result = await response.get_output()

calculator_agent = Agent(
        "openai:gpt-4o",
        system_prompt="You are a calculator assistant. Use the tools to perform calculations.",
        instrument=True,
    )

    @calculator_agent.tool
    def add(ctx: RunContext[None], a: float, b: float) -> float:
        """Add two numbers together."""
        return a + b

    @calculator_agent.tool
    def multiply(ctx: RunContext[None], a: float, b: float) -> float:
        """Multiply two numbers together."""
        return a * b

    @calculator_agent.tool
    def subtract(ctx: RunContext[None], a: float, b: float) -> float:
        """Subtract b from a."""
        return a - b

    print("Streaming calculation response:")
    async with calculator_agent.run_stream(
        "Calculate (5 + 3) * 2 - 1. Show your work step by step."
    ) as response:
        async for chunk in response.stream_text(delta=True):
            print(chunk, end="", flush=True)