fix(runner): keep installation context within each stream resume

This commit is contained in:
Hyu
2026-09-15 13:51:27 +08:00
parent 9db097399c
commit d497defbf7
4 changed files with 203 additions and 24 deletions
+6 -4
View File
@@ -2933,8 +2933,9 @@ class RuntimeConnectionHandler(handler.Handler):
timeout=timeout,
)
async for ret in gen:
yield ret
async with contextlib.aclosing(gen):
async for ret in gen:
yield ret
def _get_runner_action_timeout(self, context: dict[str, Any]) -> float:
"""Use the run deadline as the transport idle timeout when available."""
@@ -3143,8 +3144,9 @@ class RuntimeConnectionHandler(handler.Handler):
timeout=180,
)
async for ret in gen:
yield ret
async with contextlib.aclosing(gen):
async for ret in gen:
yield ret
async def retrieve_knowledge(
self,