Skip to content

Commit 5e32283

Browse files
committed
remove prints in the test file
1 parent 5daa8e1 commit 5e32283

File tree

1 file changed

+2
-15
lines changed

1 file changed

+2
-15
lines changed

tests/test_stream_events.py

Lines changed: 2 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import pytest
55

6-
from agents import Agent, ItemHelpers, Runner, function_tool
6+
from agents import Agent, Runner, function_tool
77

88
from .fake_model import FakeModel
99
from .test_responses import get_function_tool_call, get_text_message
@@ -39,28 +39,15 @@ async def test_stream_events_main():
3939
agent,
4040
input="Hello",
4141
)
42-
print("=== Run starting ===")
4342
tool_call_start_time = -1
4443
tool_call_end_time = -1
4544
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":
5246
if event.item.type == "tool_call_item":
5347
tool_call_start_time = time.time_ns()
54-
print(f"-- Tool was called at {tool_call_start_time}")
5548
elif event.item.type == "tool_call_output_item":
5649
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-
)
6250

63-
print("=== Run complete ===")
6451
assert tool_call_start_time > 0, "tool_call_item was not observed"
6552
assert tool_call_end_time > 0, "tool_call_output_item was not observed"
6653
assert tool_call_start_time < tool_call_end_time, "Tool call ended before or equals it started?"

0 commit comments

Comments
 (0)