mirror of
https://github.com/openai/codex.git
synced 2026-05-23 20:44:50 +00:00
72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
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())
|