mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2026-06-10 01:12:20 +00:00
64bb16a884
- Decouple WebSocket connection from session lifecycle: workflows continue running after disconnect
- Message buffering with ring buffer (max 1000) for chat history replay on reconnect
- Session garbage collection: 24-hour TTL for terminal sessions via background asyncio task
- Multi-tab support: last tab wins, old WebSocket closed on new connection for same session
- Cancel now sends explicit WebSocket message instead of relying on disconnect detection
- Replace hardcoded API keys and BASE_URL with ${API_KEY}/${BASE_URL} placeholders in yaml configs
92 lines
3.6 KiB
Python
Executable File
92 lines
3.6 KiB
Python
Executable File
import logging
|
|
from typing import Any, Dict
|
|
|
|
from utils.exceptions import ValidationError
|
|
|
|
from server.services.session_execution import SessionExecutionController
|
|
from server.services.session_store import WorkflowSessionStore
|
|
|
|
|
|
class MessageHandler:
|
|
"""Routes WebSocket messages to the appropriate handlers."""
|
|
|
|
def __init__(
|
|
self,
|
|
session_store: WorkflowSessionStore,
|
|
session_controller: SessionExecutionController,
|
|
workflow_run_service=None,
|
|
) -> None:
|
|
self.session_store = session_store
|
|
self.session_controller = session_controller
|
|
self.workflow_run_service = workflow_run_service
|
|
self.logger = logging.getLogger(__name__)
|
|
|
|
async def handle_message(self, session_id: str, data: Dict[str, Any], websocket_manager):
|
|
message_type = data.get("type")
|
|
if message_type == "human_input":
|
|
await self._handle_human_input(session_id, data, websocket_manager)
|
|
elif message_type == "ping":
|
|
await self._handle_ping(session_id, websocket_manager)
|
|
elif message_type == "get_status":
|
|
await self._handle_get_status(session_id, websocket_manager)
|
|
elif message_type == "cancel":
|
|
await self._handle_cancel(session_id, websocket_manager)
|
|
else:
|
|
await websocket_manager.send_message(
|
|
session_id,
|
|
{"type": "error", "data": {"message": f"Unknown message type: {message_type}"}},
|
|
)
|
|
|
|
async def _handle_cancel(self, session_id: str, websocket_manager):
|
|
if self.workflow_run_service:
|
|
self.workflow_run_service.request_cancel(session_id, reason="User requested cancellation")
|
|
await websocket_manager.send_message(
|
|
session_id,
|
|
{"type": "input_received", "data": {"message": "Cancellation requested"}},
|
|
)
|
|
|
|
async def _handle_human_input(self, session_id: str, data: Dict[str, Any], websocket_manager):
|
|
try:
|
|
payload = data.get("data", {}) or {}
|
|
user_input = payload.get("input", "")
|
|
attachments = payload.get("attachments") or []
|
|
|
|
if not user_input and not attachments:
|
|
await websocket_manager.send_message(
|
|
session_id,
|
|
{"type": "error", "data": {"message": "Empty input"}},
|
|
)
|
|
return
|
|
|
|
self.session_controller.provide_human_input(
|
|
session_id,
|
|
{"text": user_input, "attachments": attachments},
|
|
)
|
|
|
|
await websocket_manager.send_message(
|
|
session_id,
|
|
{"type": "input_received", "data": {"message": "Input received"}},
|
|
)
|
|
|
|
except ValidationError as exc:
|
|
await websocket_manager.send_message(
|
|
session_id,
|
|
{"type": "error", "data": {"message": str(exc)}},
|
|
)
|
|
except Exception as exc:
|
|
self.logger.error("Error handling human input for session %s: %s", session_id, exc)
|
|
await websocket_manager.send_message(
|
|
session_id,
|
|
{"type": "error", "data": {"message": str(exc)}},
|
|
)
|
|
|
|
async def _handle_ping(self, session_id: str, websocket_manager):
|
|
await websocket_manager.handle_heartbeat(session_id)
|
|
|
|
async def _handle_get_status(self, session_id: str, websocket_manager):
|
|
session_info = self.session_store.get_session_info(session_id)
|
|
await websocket_manager.send_message(
|
|
session_id,
|
|
{"type": "status", "data": session_info or {"message": "Session not found"}},
|
|
)
|