Skip to content

Commit db934ac

Browse files
committed
reraise external cancellations
1 parent 5b449b3 commit db934ac

File tree

1 file changed

+5
-0
lines changed
  • src/strands/experimental/bidi/agent

1 file changed

+5
-0
lines changed

src/strands/experimental/bidi/agent/agent.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -399,9 +399,14 @@ async def run_outputs(inputs_task: asyncio.Task) -> None:
399399
inputs_task.cancel()
400400
outputs_task.cancel()
401401
await asyncio.gather(inputs_task, outputs_task, return_exceptions=True)
402+
402403
if not isinstance(error, asyncio.CancelledError):
403404
raise
404405

406+
run_task = asyncio.current_task()
407+
if run_task and run_task.cancelling() > 0: # externally cancelled
408+
raise
409+
405410
finally:
406411
input_stops = [input_.stop for input_ in inputs if isinstance(input_, BidiInput)]
407412
output_stops = [output.stop for output in outputs if isinstance(output, BidiOutput)]

0 commit comments

Comments
 (0)