Cheshire Cat AI核心项目中WebSocket连接异常关闭问题分析与解决方案

问题背景

在Cheshire Cat AI核心项目中，当用户与AI进行对话时，如果用户在AI尚未完成响应前突然关闭WebSocket连接，会导致系统出现连接状态不一致的问题。具体表现为当用户重新连接时，系统无法正确处理新的会话请求，出现"Unexpected ASGI message 'websocket.close'"运行时错误。

问题现象

当用户通过WebSocket与Cheshire Cat AI建立连接并进行对话时，如果用户在AI仍在生成响应过程中突然关闭连接，系统会记录以下错误：

RuntimeError: Unexpected ASGI message 'websocket.close', after sending 'websocket close'

此后，即使用户重新连接，系统也无法正确处理该用户的会话请求，必须重启整个AI服务才能恢复正常。

技术分析

WebSocket连接状态管理机制

在Cheshire Cat AI项目中，WebSocket连接管理主要涉及以下几个关键组件：

1. StrayCat类：负责维护单个用户的WebSocket连接状态
2. WebSocket端点处理：负责接受和关闭WebSocket连接
3. 消息处理循环：负责处理用户输入和AI响应

问题根源

当用户突然关闭连接时，系统会经历以下异常流程：

1. AI正在通过WebSocket发送响应消息
2. 用户端突然关闭连接
3. 系统尝试继续发送剩余响应时遇到连接已关闭的异常
4. 系统尝试发送错误信息时再次遇到异常
5. 最终导致WebSocket连接状态不一致

异常处理流程缺陷

正常情况下，当用户关闭连接时，系统会捕获WebSocketDisconnect异常并清理连接状态。但在上述异常场景中，由于错误处理逻辑本身又尝试通过已关闭的连接发送错误信息，导致异常被掩盖，连接状态未能正确重置。

解决方案

方案一：增强发送逻辑的异常处理

在StrayCat类的__send_ws_json方法中添加对ConnectionClosedOK异常的处理：

from websockets.exceptions import ConnectionClosedOK

def __send_ws_json(self, data: Any):
    try:
        asyncio.run_coroutine_threadsafe(
            self.__ws.send_json(data), loop=self.__main_loop
        ).result()
    except ConnectionClosedOK as ex:
        if ex.code == 1000:
            log.warning(ex)
            if self.__ws:
                del self.__ws
                self.__ws = None

此方案能有效捕获连接关闭异常并清理无效连接状态。

方案二：完善全局异常处理机制

在消息处理循环中添加对错误发送逻辑的保护：

def run(self, user_message_json):
    try:
        cat_message = self.loop.run_until_complete(self.__call__(user_message_json))
        self.send_chat_message(cat_message)
    except Exception as e:
        log.error(e)
        traceback.print_exc()
        try:
            self.send_error(e)
        except Exception as send_error_ex:
            log.warning(f"Failed to send error: {send_error_ex}")

此方案更全面地保护了系统免受异常影响，同时保持了原有错误处理流程的完整性。

实施建议

基于技术分析，推荐采用以下最佳实践：

1. 优先使用方案二：它提供了更全面的异常保护，同时保持了代码结构的清晰性
2. 添加连接状态检查：在执行任何发送操作前检查连接状态
3. 完善日志记录：为异常情况添加详细的日志记录，便于问题诊断
4. 实现连接健康检查：定期检查连接状态，自动清理无效连接

总结

WebSocket连接异常处理是实时交互系统中的关键问题。通过对Cheshire Cat AI核心项目中这一特定问题的分析，我们不仅找到了有效的解决方案，还总结出了一套通用的WebSocket连接管理最佳实践。这些经验对于开发高可靠的实时AI对话系统具有重要参考价值。

在实际应用中，开发者应当特别注意异常处理逻辑本身可能引发的二次异常，并确保系统在任何异常情况下都能保持一致的连接状态。通过合理的架构设计和细致的异常处理，可以显著提升系统的稳定性和用户体验。