Skip to content

Commit 3a823e4

Browse files
chore: ruff format/isort fixes after rebase
1 parent ece469c commit 3a823e4

3 files changed

Lines changed: 30 additions & 21 deletions

File tree

src/agents/run_internal/run_loop.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,14 @@
1919
ResponseOutputItemDoneEvent,
2020
)
2121
from openai.types.responses.response_output_item import McpCall, McpListTools, ResponseOutputItem
22+
from openai.types.responses.response_prompt_param import ResponsePromptParam
23+
from openai.types.responses.response_reasoning_item import ResponseReasoningItem
2224
from openai.types.responses.response_reasoning_summary_text_delta_event import (
2325
ResponseReasoningSummaryTextDeltaEvent,
2426
)
2527
from openai.types.responses.response_reasoning_text_delta_event import (
2628
ResponseReasoningTextDeltaEvent,
2729
)
28-
from openai.types.responses.response_prompt_param import ResponsePromptParam
29-
from openai.types.responses.response_reasoning_item import ResponseReasoningItem
3030

3131
from .._mcp_tool_metadata import collect_mcp_list_tools_metadata
3232
from .._tool_identity import (

tests/test_reasoning_delta_stream_event.py

Lines changed: 27 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,10 @@
33
from __future__ import annotations
44

55
import pytest
6+
from openai.types.responses.response_reasoning_item import ResponseReasoningItem, Summary
67

78
from 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

1211
from .fake_model import FakeModel
1312
from .test_responses import get_text_message
@@ -25,10 +24,12 @@ def _make_reasoning_item(text: str) -> ResponseReasoningItem:
2524
async 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:
4849
async 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:
99102
async 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:
120125
async 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() ->
146153
async 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

tests/test_stream_events.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -367,6 +367,7 @@ async def test_complete_streaming_events():
367367

368368
# Event 6: ReasoningDeltaEvent (emitted alongside the raw delta)
369369
from agents.stream_events import ReasoningDeltaEvent
370+
370371
assert events[6].type == "reasoning_delta"
371372
assert isinstance(events[6], ReasoningDeltaEvent)
372373

0 commit comments

Comments
 (0)