# /// script
# requires-python = ">=3.12"
# dependencies = ["openai==3.24.0"]
# ///
import os

from openai import OpenAI

with OpenAI(
    base_url=os.environ["AMPERSAND_BASE_URL"],
    api_key=os.environ["AMPERSAND_API_KEY"],
    max_retries=0,
) as client:
    with client.beta.agents.sessions.create(
        agent={
            "model": os.environ["AMPERSAND_MODEL"],
            "instructions": "Help the team plan a reliable launch.",
        },
        environment={"type": "none"},
        input="Suggest three checks before a product launch.",
        stream=True,
    ).with_result_collection() as stream:
        for event in stream:
            print(event.to_json(indent=None), flush=True)
            if event.type in {
                "agent.session.idle",
                "agent.session.failed",
                "agent.session.requires_action",
            }:
                break
        result = stream.get_final_result()
    print(result.output_text)
    session_id = result.session_id