33from __future__ import annotations
44
55import pytest
6+ from openai .types .responses .response_reasoning_item import ResponseReasoningItem , Summary
67
78from agents import Agent , Runner
8- from agents .stream_events import ReasoningDeltaEvent , RawResponsesStreamEvent
9-
10- from openai .types .responses .response_reasoning_item import ResponseReasoningItem , Summary
9+ from agents .stream_events import RawResponsesStreamEvent , ReasoningDeltaEvent
1110
1211from .fake_model import FakeModel
1312from .test_responses import get_text_message
@@ -25,10 +24,12 @@ def _make_reasoning_item(text: str) -> ResponseReasoningItem:
2524async def test_reasoning_delta_event_emitted_during_streaming () -> None :
2625 """ReasoningDeltaEvent is emitted when the model streams a reasoning summary delta."""
2726 model = FakeModel ()
28- model .set_next_output ([
29- _make_reasoning_item ("Let me think..." ),
30- get_text_message ("Answer" ),
31- ])
27+ model .set_next_output (
28+ [
29+ _make_reasoning_item ("Let me think..." ),
30+ get_text_message ("Answer" ),
31+ ]
32+ )
3233
3334 agent = Agent (name = "A" , model = model )
3435 result = Runner .run_streamed (agent , input = "hi" )
@@ -48,10 +49,12 @@ async def test_reasoning_delta_event_emitted_during_streaming() -> None:
4849async def test_reasoning_delta_snapshot_accumulates () -> None :
4950 """The snapshot field grows monotonically across delta events."""
5051 model = FakeModel ()
51- model .set_next_output ([
52- _make_reasoning_item ("Hello world" ),
53- get_text_message ("done" ),
54- ])
52+ model .set_next_output (
53+ [
54+ _make_reasoning_item ("Hello world" ),
55+ get_text_message ("done" ),
56+ ]
57+ )
5558
5659 agent = Agent (name = "A" , model = model )
5760 result = Runner .run_streamed (agent , input = "hi" )
@@ -99,10 +102,12 @@ async def test_no_reasoning_delta_event_without_reasoning() -> None:
99102async def test_reasoning_delta_event_type_field () -> None :
100103 """ReasoningDeltaEvent.type is always 'reasoning_delta'."""
101104 model = FakeModel ()
102- model .set_next_output ([
103- _make_reasoning_item ("some reasoning" ),
104- get_text_message ("answer" ),
105- ])
105+ model .set_next_output (
106+ [
107+ _make_reasoning_item ("some reasoning" ),
108+ get_text_message ("answer" ),
109+ ]
110+ )
106111
107112 agent = Agent (name = "A" , model = model )
108113 result = Runner .run_streamed (agent , input = "hi" )
@@ -120,10 +125,12 @@ async def test_reasoning_delta_event_type_field() -> None:
120125async def test_raw_response_events_still_emitted_alongside_reasoning_delta () -> None :
121126 """RawResponsesStreamEvent is still emitted even when ReasoningDeltaEvent is also emitted."""
122127 model = FakeModel ()
123- model .set_next_output ([
124- _make_reasoning_item ("thinking" ),
125- get_text_message ("result" ),
126- ])
128+ model .set_next_output (
129+ [
130+ _make_reasoning_item ("thinking" ),
131+ get_text_message ("result" ),
132+ ]
133+ )
127134
128135 agent = Agent (name = "A" , model = model )
129136 result = Runner .run_streamed (agent , input = "hi" )
@@ -146,6 +153,7 @@ async def test_raw_response_events_still_emitted_alongside_reasoning_delta() ->
146153async def test_reasoning_delta_event_importable_from_agents () -> None :
147154 """ReasoningDeltaEvent can be imported directly from the agents package."""
148155 from agents import ReasoningDeltaEvent as RDE
156+
149157 assert RDE is ReasoningDeltaEvent
150158
151159
0 commit comments