xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-07 06:16:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: kill runtime process when exit in stdio mode

Browse Source
This commit is contained in:
Junyan Qin
2025-07-16 22:43:39 +08:00
parent e0abd19636
commit f5b893cfe0
3 changed files with 14 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/plugin/connector.py
View File

@@ -30,6 +30,8 @@ class PluginRuntimeConnector:
stdio_client_controller: stdio_client_controller.StdioClientController
ctrl: stdio_client_controller.StdioClientController | ws_client_controller.WebSocketClientController
runtime_disconnect_callback: typing.Callable[
[PluginRuntimeConnector], typing.Coroutine[typing.Any, typing.Any, None]
]
@@ -73,22 +75,22 @@ class PluginRuntimeConnector:
self.ap.logger.error('Failed to connect to plugin runtime, trying to reconnect...')
await self.runtime_disconnect_callback(self)
ctrl = ws_client_controller.WebSocketClientController(
self.ctrl = ws_client_controller.WebSocketClientController(
ws_url=ws_url,
make_connection_failed_callback=make_connection_failed_callback,
)
task = ctrl.run(new_connection_callback)
task = self.ctrl.run(new_connection_callback)
else: # stdio
self.ap.logger.info('use stdio to connect to plugin runtime')
# cmd: lbp rt -s
python_path = sys.executable
env = os.environ.copy()
ctrl = stdio_client_controller.StdioClientController(
self.ctrl = stdio_client_controller.StdioClientController(
command=python_path,
args=['-m', 'langbot_plugin.cli.__init__', 'rt', '-s'],
env=env,
)
task = ctrl.run(new_connection_callback)
task = self.ctrl.run(new_connection_callback)
asyncio.create_task(task)
@@ -135,3 +137,7 @@ class PluginRuntimeConnector:
cmd_ret = command_context.CommandReturn.model_validate(ret)
yield cmd_ret
def dispose(self):
if isinstance(self.ctrl, stdio_client_controller.StdioClientController):
self.ctrl.process.terminate()