Reasoning Agent Events

"""
Reasoning Agent Events
=============================

Demonstrates how to capture reasoning events during streaming.
"""

import asyncio

from agno.agent import RunEvent
from agno.agent.agent import Agent
from agno.models.openai import OpenAIResponses

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------
finance_agent = Agent(
    model=OpenAIResponses(id="gpt-5.6"),
    reasoning_model=OpenAIResponses(id="o3-mini"),
)


async def run_agent_with_events(prompt: str):
    content_started = False
    async for run_output_event in finance_agent.arun(
        prompt,
        stream=True,
        stream_events=True,
    ):
        if run_output_event.event in [RunEvent.run_started, RunEvent.run_completed]:
            print(f"\nEVENT: {run_output_event.event}")

        if run_output_event.event in [RunEvent.reasoning_started]:
            print(f"\nEVENT: {run_output_event.event}")

        if run_output_event.event in [RunEvent.reasoning_step]:
            print(f"\nEVENT: {run_output_event.event}")
            print(f"REASONING CONTENT: {run_output_event.reasoning_content}")  # type: ignore

        if run_output_event.event in [RunEvent.reasoning_completed]:
            print(f"\nEVENT: {run_output_event.event}")

        if run_output_event.event in [RunEvent.run_content]:
            if not content_started:
                print("\nCONTENT:")
                content_started = True
            else:
                print(run_output_event.content, end="")


# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
    task = (
        "Analyze the key factors that led to the signing of the Treaty of Versailles in 1919. "
        "Discuss the political, economic, and social impacts of the treaty on Germany and how it "
        "contributed to the onset of World War II. Provide a nuanced assessment that includes "
        "multiple historical perspectives."
    )
    asyncio.run(
        run_agent_with_events(
            task,
        )
    )

async def run_agent_with_events(prompt: str):
    content_started = False
    async for event in finance_agent.arun(prompt, stream=True, stream_events=True):
        if event.event in (
            RunEvent.run_started, RunEvent.run_completed,
            RunEvent.reasoning_started, RunEvent.reasoning_completed,
        ):
            print(f"\nEVENT: {event.event}")
        elif event.event == RunEvent.reasoning_content_delta:
            print(event.reasoning_content, end="", flush=True)
        elif event.event == RunEvent.reasoning_step:
            print(event.reasoning_content, end="", flush=True)
        elif event.event == RunEvent.run_content and event.content:
            if not content_started:
                print("\nCONTENT:")
                content_started = True
            print(event.content, end="", flush=True)
        elif event.event == RunEvent.run_error:
            print(f"\nERROR: {event.content}")