mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-07-17 01:46:07 +00:00
fix(debug-chat): preserve websocket pipeline routing
This commit is contained in:
@@ -922,17 +922,31 @@ class RuntimeBot:
|
||||
)
|
||||
|
||||
async def initialize(self):
|
||||
def websocket_pipeline_uuid(event, adapter):
|
||||
if adapter.__class__.__name__ != 'WebSocketAdapter':
|
||||
return None
|
||||
value = getattr(event, '_langbot_pipeline_uuid', None)
|
||||
return value if isinstance(value, str) and value else None
|
||||
|
||||
async def on_friend_message(
|
||||
event: platform_events.FriendMessage,
|
||||
adapter: abstract_platform_adapter.AbstractMessagePlatformAdapter,
|
||||
):
|
||||
await self._handle_legacy_message_event(event, adapter)
|
||||
await self._handle_legacy_message_event(
|
||||
event,
|
||||
adapter,
|
||||
pipeline_uuid_override=websocket_pipeline_uuid(event, adapter),
|
||||
)
|
||||
|
||||
async def on_group_message(
|
||||
event: platform_events.GroupMessage,
|
||||
adapter: abstract_platform_adapter.AbstractMessagePlatformAdapter,
|
||||
):
|
||||
await self._handle_legacy_message_event(event, adapter)
|
||||
await self._handle_legacy_message_event(
|
||||
event,
|
||||
adapter,
|
||||
pipeline_uuid_override=websocket_pipeline_uuid(event, adapter),
|
||||
)
|
||||
|
||||
self.adapter.register_listener(platform_events.FriendMessage, on_friend_message)
|
||||
self.adapter.register_listener(platform_events.GroupMessage, on_group_message)
|
||||
|
||||
@@ -447,6 +447,8 @@ class WebSocketAdapter(abstract_platform_adapter.AbstractMessagePlatformAdapter)
|
||||
sender=sender, message_chain=message_chain, time=datetime.now().timestamp()
|
||||
)
|
||||
|
||||
object.__setattr__(event, '_langbot_pipeline_uuid', pipeline_uuid)
|
||||
|
||||
# 设置流水线UUID (proxy bot always needs it for reply_message routing)
|
||||
self.ap.platform_mgr.websocket_proxy_bot.bot_entity.use_pipeline_uuid = pipeline_uuid
|
||||
if owner_bot is not None:
|
||||
|
||||
Reference in New Issue
Block a user