import sys
from pathlib import Path

_EXAMPLES_ROOT = Path(__file__).resolve().parents[1]
if str(_EXAMPLES_ROOT) not in sys.path:
    sys.path.insert(0, str(_EXAMPLES_ROOT))

from _bootstrap import ensure_local_sdk_src, runtime_config

ensure_local_sdk_src()

import asyncio

from openai_codex import AsyncCodex


async def main() -> None:
    async with AsyncCodex(config=runtime_config()) as codex:
        thread = await codex.thread_start(
            model="gpt-5.4", config={"model_reasoning_effort": "high"}
        )
        steer_turn = await thread.turn("Count from 1 to 40 with commas, then one summary sentence.")
        steer_result = await steer_turn.steer("Keep it brief and stop after 10 numbers.")

        steer_event_count = 0
        steer_completed_status = None
        steer_deltas = []
        async for event in steer_turn.stream():
            steer_event_count += 1
            if event.method == "item/agentMessage/delta":
                steer_deltas.append(event.payload.delta)
                continue
            if event.method == "turn/completed":
                steer_completed_status = event.payload.turn.status.value

        if steer_completed_status is None:
            raise RuntimeError("stream ended without turn/completed")
        steer_preview = "".join(steer_deltas).strip()

        interrupt_turn = await thread.turn(
            "Count from 1 to 200 with commas, then one summary sentence."
        )
        interrupt_result = await interrupt_turn.interrupt()

        interrupt_event_count = 0
        interrupt_completed_status = None
        interrupt_deltas = []
        async for event in interrupt_turn.stream():
            interrupt_event_count += 1
            if event.method == "item/agentMessage/delta":
                interrupt_deltas.append(event.payload.delta)
                continue
            if event.method == "turn/completed":
                interrupt_completed_status = event.payload.turn.status.value

        if interrupt_completed_status is None:
            raise RuntimeError("stream ended without turn/completed")
        interrupt_preview = "".join(interrupt_deltas).strip()

        print("steer.result:", steer_result.model_dump(mode="json", by_alias=True))
        print("steer.final.status:", steer_completed_status)
        print("steer.events.count:", steer_event_count)
        print("steer.assistant.preview:", steer_preview)
        print("interrupt.result:", interrupt_result.model_dump(mode="json", by_alias=True))
        print("interrupt.final.status:", interrupt_completed_status)
        print("interrupt.events.count:", interrupt_event_count)
        print("interrupt.assistant.preview:", interrupt_preview)


if __name__ == "__main__":
    asyncio.run(main())