fix(agent): close reasoning stream on cancellation

This commit is contained in:
KDB
2026-08-30 16:32:08 +08:00
committed by Xubin Ren
parent 1c1b13a3a9
commit bfe041def7
2 changed files with 45 additions and 0 deletions
+4
View File
@@ -1051,6 +1051,10 @@ class AgentRunner:
await coro if outer_timeout_s is None
else await asyncio.wait_for(coro, timeout=outer_timeout_s)
)
except asyncio.CancelledError:
_pause_generation()
await _close_native_reasoning()
raise
except asyncio.TimeoutError:
if outer_timeout_s is None:
response = LLMResponse(