xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-10-24 17:35:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
src/langbot/pkg/agent/runner/invoker.py
+10 -7
View File
@@ -48,12 +48,15 @@ class RunnerInvoker:
context=context,
)
while True:
try:
result_dict = await self._next_with_deadline(gen, descriptor, context)
except StopAsyncIteration:
break
yield result_dict
try:
while True:
try:
result_dict = await self._next_with_deadline(gen, descriptor, context)
except StopAsyncIteration:
break
yield result_dict
finally:
await self._close_generator(gen, descriptor)
except asyncio.TimeoutError as e:
raise RunnerExecutionError(
@@ -128,4 +131,4 @@ class RunnerInvoker:
try:
await gen.aclose()
except Exception as e:
self.ap.logger.warning(f'Failed to close timed-out runner {descriptor.id}: {e}')
self.ap.logger.warning(f'Failed to close runner {descriptor.id}: {e}')