fix(debug-chat): preserve websocket pipeline routing

tests/unit_tests/platform/test_websocket_adapter_attachments.py

@@ -10,11 +10,17 @@ type and graceful error handling.
 from __future__ import annotations
 
 import base64
+import asyncio
+from types import SimpleNamespace
 from unittest.mock import AsyncMock, Mock
 
 import pytest
 
-from langbot.pkg.platform.sources.websocket_adapter import WebSocketAdapter
+import langbot_plugin.api.entities.builtin.platform.events as platform_events
+import langbot_plugin.api.entities.builtin.platform.entities as platform_entities
+import langbot_plugin.api.entities.builtin.platform.message as platform_message
+from langbot.pkg.platform.botmgr import RuntimeBot
+from langbot.pkg.platform.sources.websocket_adapter import WebSocketAdapter, WebSocketSession
 
 
 def _make_adapter(load_return=b'hello', load_side_effect=None):
@@ -90,3 +96,64 @@ async def test_load_failure_is_logged_not_raised():
     await adapter._process_image_components(chain)
     assert 'base64' not in chain[0]
     adapter.logger.error.assert_awaited_once()
+
+
+@pytest.mark.asyncio
+async def test_handle_websocket_message_marks_event_with_pipeline_uuid():
+    adapter, _ = _make_adapter()
+    adapter.websocket_person_session = WebSocketSession(id='websocketperson')
+    adapter.listeners = {}
+    adapter.listeners[platform_events.FriendMessage] = AsyncMock()
+    adapter.ap.platform_mgr.websocket_proxy_bot.bot_entity.use_pipeline_uuid = ''
+
+    connection = SimpleNamespace(
+        pipeline_uuid='pipeline-123',
+        session_type='person',
+        connection_id='conn-1',
+    )
+
+    await adapter.handle_websocket_message(
+        connection,
+        {'message': [{'type': 'Plain', 'text': 'hello'}], 'stream': True},
+    )
+    await asyncio.sleep(0)
+
+    event = adapter.listeners[platform_events.FriendMessage].await_args.args[0]
+    assert getattr(event, '_langbot_pipeline_uuid') == 'pipeline-123'
+
+
+@pytest.mark.asyncio
+async def test_runtime_bot_websocket_listener_uses_event_pipeline_uuid():
+    app = Mock()
+    app.msg_aggregator.add_message = AsyncMock()
+    app.webhook_pusher = None
+    logger = Mock()
+    logger.info = AsyncMock()
+    logger.warning = AsyncMock()
+    logger.error = AsyncMock()
+    bot_entity = Mock()
+    bot_entity.uuid = 'websocket-proxy-bot'
+    bot_entity.enable = True
+    bot_entity.use_pipeline_uuid = ''
+    adapter = WebSocketAdapter.model_construct(
+        ap=app,
+        logger=Mock(error=AsyncMock()),
+        listeners={},
+        websocket_person_session=WebSocketSession(id='websocketperson'),
+        websocket_group_session=WebSocketSession(id='websocketgroup'),
+    )
+    bot = RuntimeBot(ap=app, bot_entity=bot_entity, adapter=adapter, logger=logger)
+
+    await bot.initialize()
+
+    event = platform_events.FriendMessage(
+        sender=platform_entities.Friend(id='sender-1', nickname='User', remark='User'),
+        message_chain=platform_message.MessageChain([platform_message.Plain(text='hello')]),
+        time=1,
+    )
+    object.__setattr__(event, '_langbot_pipeline_uuid', 'pipeline-123')
+
+    await adapter.listeners[platform_events.FriendMessage](event, adapter)
+
+    app.msg_aggregator.add_message.assert_awaited_once()
+    assert app.msg_aggregator.add_message.await_args.kwargs['pipeline_uuid'] == 'pipeline-123'