import asyncio
import random

from agents import Agent, ItemHelpers, Runner, function_tool


@function_tool
def how_many_jokes() -> int:
    return random.randint(1, 10)

from agents import Agent, Runner,RunConfig,OpenAIProvider
run_config = RunConfig(model_provider = OpenAIProvider(
    api_key="your api key",
    base_url="https://open.bigmodel.cn/api/paas/v4/",
    use_responses=False,
    )
)


async def main():
    agent = Agent(
        name="Joker",
        instructions="First call the `how_many_jokes` tool, then tell that many jokes.",
        tools=[how_many_jokes],
        model="glm-4"
    )

    result = Runner.run_streamed(
        agent,
        input="Hello",
        run_config=run_config
    )
    print("=== Run starting ===")
    async for event in result.stream_events():
        # We'll ignore the raw responses event deltas
        if event.type == "raw_response_event":
            continue
        elif event.type == "agent_updated_stream_event":
            print(f"Agent updated: {event.new_agent.name}")
            continue
        elif event.type == "run_item_stream_event":
            if event.item.type == "tool_call_item":
                print("-- Tool was called")
            elif event.item.type == "tool_call_output_item":
                print(f"-- Tool output: {event.item.output}")
            elif event.item.type == "message_output_item":
                print(f"-- Message output:\n {ItemHelpers.text_message_output(event.item)}")
            else:
                pass  # Ignore other event types

    print("=== Run complete ===")


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

"""
$python .My_test\stream_itemstest.py  
=== Run starting ===
Agent updated: Joker
Agent updated: Joker
-- Tool was called
-- Message output:
 Here are 9 jokes for you:
1. Why don't scientists trust atoms? Because they make up everything!
2. Why did the scarecrow win an award? Because he was outstanding in his field!
3. Why did the math book look sad? Because it had too many problems!
4. Why did the cookie go to the doctor? Because it was feeling crummy!
5. Why did the bicycle fall over? Because it was two-tired!
6. Why did the cat sit on the computer? Because it wanted to keep an eye on the mouse!
7. Why did the teacher wear sunglasses? Because her students were so bright!
8. Why did the chicken cross the playground? To get to the other slide!
9. Why did the bookworm go to the gym? Because it heard the exercise was good for the mind!     
=== Run complete ===
OPENAI_API_KEY is not set, skipping trace export
"""