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

feat(plugin): basic communication

Browse Source
This commit is contained in:
Junyan Qin
2025-06-14 17:27:21 +08:00
parent 2d06f1cadb
commit 6f2fd72af6
4 changed files with 150 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/plugin/connector.py
View File

@@ -29,17 +29,23 @@ class PluginRuntimeConnector:
async def initialize(self):
async def new_connection_callback(connection: stdio_connection.StdioConnection):
self.ap.logger.info('Connected to plugin runtime.')
self.handler = handler.RuntimeConnectionHandler(connection)
self.handler = handler.RuntimeConnectionHandler(connection, self.ap)
self.handler_task = asyncio.create_task(self.handler.run())
_ = await self.handler.ping()
self.ap.logger.info('Connected to plugin runtime.')
await self.handler_task
task: asyncio.Task | None = None
if platform.get_platform() == 'docker': # use websocket
self.ap.logger.info('use websocket to connect to plugin runtime')
ws_url = self.ap.instance_config.data['plugin']['runtime_ws_url']
ctrl = ws_client_controller.WebSocketClientController(
ws_url=ws_url,
)
await ctrl.run(new_connection_callback)
task = 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()
@@ -48,7 +54,9 @@ class PluginRuntimeConnector:
args=['-m', 'langbot_plugin.cli.__init__', 'rt', '-s'],
env=env,
)
await ctrl.run(new_connection_callback)
task = ctrl.run(new_connection_callback)
asyncio.create_task(task)
async def run(self):
pass