@myassis/gateway 1.0.93 → 1.0.94

package/dist/services/WebSocketService.js

@@ -206,6 +206,9 @@ class WebSocketService {
             if (options.excludeClientId && client.clientId === options.excludeClientId) {
                 return;
             }
+            if (options.onlyClientId && client.clientId !== options.onlyClientId) {
+                return;
+            }
             sent = this.sendToConnection(client, message) || sent;
         });
         return sent;

package/dist/services/session/Session.js

@@ -47,6 +47,16 @@ const sleep = (ms) => new Promise(resolve => setTimeout(resolve, ms));
  * 广播出去会让多个终端同时弹窗并重复回传批准结果。
  */
 const NON_MIRRORED_SSE_EVENTS = new Set(['heartbeat', 'approval_pending']);
+/**
+ * 只属于发起终端的事件类型。
+ *
+ * 这些事件不能广播（否则多终端重复弹窗），但也不能丢：
+ * 中继模式下 SSE 一旦被隐式关闭，approval_pending 写不出去，
+ * 发起端永远不弹确认框，而 registerApprovalWaiter 会一直堵到
+ * 5 分钟超时，整轮对话看起来就是“卡住”。
+ * 因此 SSE 不可用时改成定向走 WebSocket 补送给同一个发起终端。
+ */
+const ORIGINATOR_ONLY_SSE_EVENTS = new Set(['approval_pending']);
 // 批准等待缓存：token -> { resolve, reject, expiresAt }
 const approvalWaiters = new Map();
 /**
@@ -790,12 +800,48 @@ class Session {
         // 把镜像函数传进去，使该事件也能同步到其他终端。
         const memoryManager = new MemoryManager_js_1.MemoryManager(this, this.abortController.signal, childAgent, res, mirrorToOtherClients);
         const historyMessages = await memoryManager.getHistoryMessagesAsync();
+        /**
+         * 将只属于发起终端的事件改走 WebSocket 补送。
+         *
+         * 不占用 mirrorSeq：该序号是会话级镜像流的连续编号，
+         * 定向补送若占一个号，其他终端就会把它当成丢包而误报缺口。
+         */
+        const sendToOriginator = (data) => {
+            if (!wsService)
+                return;
+            try {
+                // 没有 clientId（老版终端）时退回广播：审批 token 只能兑付一次，
+                // 重复回传会被忽略，比没人收到而卡 5 分钟更可接受。
+                const delivered = wsService.sendToUser(String(this.userId), {
+                    type: 'session_stream',
+                    payload: {
+                        sessionId: this.id,
+                        agentId: this.agentId,
+                        userMessageId,
+                        assistantMessageId,
+                        event: data,
+                    },
+                }, clientId ? { onlyClientId: clientId } : {});
+                if (!delivered) {
+                    logger.warn(`会话 ${this.id} 的 ${data?.type} 事件无法下发：SSE 已断且无可用 WebSocket 连接`);
+                }
+            }
+            catch (error) {
+                logger.error('SSE originator fallback error:', error);
+            }
+        };
         // SSE 辅助方法：res 为 null 时跳过写入（本地执行模式），但仍同步给其他终端
         const sendSSE = (res, data) => {
             mirrorToOtherClients(data);
             // 连接已断开时不再写入：写只会抛错刷日志，内容已改由镜像下发
-            if (!res || !sseAlive)
+            if (!res || !sseAlive) {
+                // 镜像不负责的那几类事件，在 SSE 不可用时需要定向补送；
+                // heartbeat 是 SSE 保活专用，没了 SSE 也就不需要补送。
+                if (ORIGINATOR_ONLY_SSE_EVENTS.has(data?.type)) {
+                    sendToOriginator(data);
+                }
                 return;
+            }
             try {
                 res.write(`data: ${JSON.stringify(data)}\n\n`);
                 res.flush?.();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@myassis/gateway",
-  "version": "1.0.93",
+  "version": "1.0.94",
   "description": "我的助手 Gateway Service - 本地 AI 网关服务，支持认证、WebSocket 实时通信和任务调度",
   "main": "dist/index.js",
   "bin": {