import asyncio
import random

from agents import (
    Agent,
    ItemHelpers,
    Runner,
    enable_verbose_stdout_logging,
    function_tool,
)

enable_verbose_stdout_logging()


@function_tool
def how_many_jokes() -> int:
    """Return a random integer of jokes to tell between 1 and 10 (inclusive)."""
    return random.randint(1, 10)

ORIGINAL_RS_ID = "rs_07a11282cddc7074006905172505c88198bd41b4889a8d5649"

def mangle_rs_id(original_rs_id: str) -> str:
    return original_rs_id[:-2] + "ff"

async def main():
    agent = Agent(
        name="Joker",
        instructions="Call the `how_many_jokes` tool once, and tell that many jokes.",
        tools=[how_many_jokes],
        model="gpt-5-nano",
    )

    rs_id = mangle_rs_id(ORIGINAL_RS_ID)
    input_list = [
        {"content": "We want to hear jokes", "role": "user"},
        {
            "id": rs_id,
            "summary": [],
            "type": "reasoning",
        },
        {
            "arguments": "{}",
            "call_id": "call_xkkWToQpzfvn0asJfDPo06Wj",
            "name": "how_many_jokes",
            "type": "function_call",
            "id": "fc_07a11282cddc707400690517253a148198a2dad52c744c22ce",
            "status": "completed",
        },
        {
            "call_id": "call_xkkWToQpzfvn0asJfDPo06Wj",
            "output": "7",
            "type": "function_call_output",
        },
    ]

    result = Runner.run_streamed(
        agent,
        input=input_list,
    )
    async for event in result.stream_events():
        if (
            event.type == "run_item_stream_event"
            and event.item.type == "message_output_item"
        ):
            print(f"-- Message output:\n {ItemHelpers.text_message_output(event.item)}")


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