Skip to content

Commit a97b01b

Browse files
committed
fix: addressing rebase issues
1 parent b3a7e3a commit a97b01b

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

src/agents/run.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1025,9 +1025,9 @@ async def run(
10251025
)
10261026
if call_id in output_call_ids and item not in items_to_save:
10271027
items_to_save.append(item)
1028-
1029-
# Don't save original_user_input again - it was already saved at the start
1030-
await self._save_result_to_session(session, [], items_to_save, run_state)
1028+
1029+
# Don't save original_user_input again - already saved at start
1030+
await self._save_result_to_session(session, [], items_to_save)
10311031

10321032
return result
10331033
elif isinstance(turn_result.next_step, NextStepInterruption):
@@ -1058,7 +1058,7 @@ async def run(
10581058
for guardrail_result in input_guardrail_results
10591059
):
10601060
await self._save_result_to_session(
1061-
session, [], turn_result.new_step_items, run_state
1061+
session, [], turn_result.new_step_items
10621062
)
10631063
else:
10641064
raise AgentsException(

tests/test_result_cast.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ def test_run_result_release_agents_breaks_strong_refs() -> None:
9292
tool_output_guardrail_results=[],
9393
_last_agent=agent,
9494
context_wrapper=RunContextWrapper(context=None),
95+
interruptions=[],
9596
)
9697
assert item.agent is not None
9798
assert item.agent.name == "leak-test-agent"
@@ -122,6 +123,7 @@ def build_item() -> tuple[MessageOutputItem, weakref.ReferenceType[RunResult]]:
122123
tool_input_guardrail_results=[],
123124
tool_output_guardrail_results=[],
124125
_last_agent=agent,
126+
interruptions=[],
125127
context_wrapper=RunContextWrapper(context=None),
126128
)
127129
return item, weakref.ref(result)
@@ -172,6 +174,7 @@ def test_run_result_repr_and_asdict_after_release_agents() -> None:
172174
tool_input_guardrail_results=[],
173175
tool_output_guardrail_results=[],
174176
_last_agent=agent,
177+
interruptions=[],
175178
context_wrapper=RunContextWrapper(context=None),
176179
)
177180

@@ -199,6 +202,7 @@ def test_run_result_release_agents_without_releasing_new_items() -> None:
199202
tool_input_guardrail_results=[],
200203
tool_output_guardrail_results=[],
201204
_last_agent=last_agent,
205+
interruptions=[],
202206
context_wrapper=RunContextWrapper(context=None),
203207
)
204208

@@ -230,6 +234,7 @@ def test_run_result_release_agents_is_idempotent() -> None:
230234
tool_output_guardrail_results=[],
231235
_last_agent=agent,
232236
context_wrapper=RunContextWrapper(context=None),
237+
interruptions=[],
233238
)
234239

235240
result.release_agents()
@@ -264,6 +269,7 @@ def test_run_result_streaming_release_agents_releases_current_agent() -> None:
264269
max_turns=1,
265270
_current_agent_output_schema=None,
266271
trace=None,
272+
interruptions=[],
267273
)
268274

269275
streaming_result.release_agents(release_new_items=False)

0 commit comments

Comments
 (0)