|
3 | 3 |
|
4 | 4 | import pytest |
5 | 5 |
|
6 | | -from agents import Agent, ItemHelpers, Runner, function_tool |
| 6 | +from agents import Agent, Runner, function_tool |
7 | 7 |
|
8 | 8 | from .fake_model import FakeModel |
9 | 9 | from .test_responses import get_function_tool_call, get_text_message |
@@ -39,28 +39,15 @@ async def test_stream_events_main(): |
39 | 39 | agent, |
40 | 40 | input="Hello", |
41 | 41 | ) |
42 | | - print("=== Run starting ===") |
43 | 42 | tool_call_start_time = -1 |
44 | 43 | tool_call_end_time = -1 |
45 | 44 | async for event in result.stream_events(): |
46 | | - # We'll ignore the raw responses event deltas |
47 | | - if event.type == "raw_response_event": |
48 | | - continue |
49 | | - elif event.type == "agent_updated_stream_event": |
50 | | - print(f"Agent updated: {event.new_agent.name}") |
51 | | - elif event.type == "run_item_stream_event": |
| 45 | + if event.type == "run_item_stream_event": |
52 | 46 | if event.item.type == "tool_call_item": |
53 | 47 | tool_call_start_time = time.time_ns() |
54 | | - print(f"-- Tool was called at {tool_call_start_time}") |
55 | 48 | elif event.item.type == "tool_call_output_item": |
56 | 49 | tool_call_end_time = time.time_ns() |
57 | | - print(f"-- Tool output: {event.item.output} at {tool_call_end_time}") |
58 | | - elif event.item.type == "message_output_item": |
59 | | - print( |
60 | | - f"-- Message output:\n {ItemHelpers.text_message_output(event.item)}" |
61 | | - ) |
62 | 50 |
|
63 | | - print("=== Run complete ===") |
64 | 51 | assert tool_call_start_time > 0, "tool_call_item was not observed" |
65 | 52 | assert tool_call_end_time > 0, "tool_call_output_item was not observed" |
66 | 53 | assert tool_call_start_time < tool_call_end_time, "Tool call ended before or equals it started?" |
0 commit comments