@myassis/gateway 1.0.91 → 1.0.92

package/dist/routes/agent.js

@@ -393,6 +393,39 @@ router.get('/sessions/:sessionId/messages/sync', ensureAgentManager, async (req,
         res.status(500).json({ success: false, error: 'Failed to sync messages' });
     }
 });
+/**
+ * GET /api/agent/sessions/:sessionId/state
+ *
+ * 会话的后台运行态。终端在断线期间收不到任何流事件，重连后必须问一次
+ * 「这一轮到底还在跑吗」，否则只能二选一地猜：要么把仍在生成的会话当成
+ * 已结束（加载圆点与计划进度圈凭空消失），要么永远停在加载态。
+ *
+ * 注意要注册在 /messages/:messageId 之前，避免被当成 messageId 匹配。
+ */
+router.get('/sessions/:sessionId/state', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId } = req.params;
+        const session = (0, index_js_2.getSessionManager)(userId).getSession(sessionId);
+        if (!session) {
+            return res.status(404).json({ success: false, error: 'Session not found' });
+        }
+        return res.json({
+            success: true,
+            data: {
+                isGenerating: session.isGenerating,
+                // 只在生成中才回消息 id：结束后回它会让终端以为还有流在推进
+                currentMessageId: session.isGenerating ? session.currentMessageId : null,
+                plan: session.plan,
+                planUpdatedAt: session.planUpdatedAt,
+            },
+        });
+    }
+    catch (error) {
+        logger.error(`Get session state error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to get session state' });
+    }
+});
 /**
  * DELETE /api/agent/sessions/:sessionId/messages/:messageId
  * Delete a specific message

package/dist/services/agent/AgentManager.js

@@ -255,7 +255,9 @@ class AgentManager {
             updatedAt: s.updatedAt,
             messageQueue: s.messageQueue,
             messageQueueAutoExecute: s.messageQueueAutoExecute,
-            plan: s.plan
+            plan: s.plan,
+            isGenerating: s.isGenerating,
+            currentMessageId: s.isGenerating ? s.currentMessageId : null
         }));
     }
     /**

package/dist/services/session/Session.js

@@ -714,7 +714,27 @@ class Session {
         // 这里预先取到 WebSocketService，把每个 SSE 事件同步给该用户的其他连接，
         // 从而让所有终端看到同一份会话流。子 Agent 的内部流不需要同步。
         const wsService = childAgent ? null : await this.getWebSocketService();
-        /** 把 SSE 事件镜像给当前用户的其他终端（排除发起端，避免重复渲染） */
+        // SSE 通道是否还活着。
+        //
+        // 中继模式下隧道一断，这条回环响应会被直接销毁，而本轮生成仍在后台跑。
+        // 此时发起端拿不到任何后续事件：SSE 已断，而它又因 excludeClientId 被
+        // 排除在 WebSocket 镜像之外，界面上的加载圆点与计划进度圈会凭空消失。
+        // 所以 SSE 一断就取消排除，让发起端也从镜像里接着收。
+        let sseAlive = !!res;
+        if (res && typeof res.on === 'function') {
+            res.on('close', () => {
+                if (!sseAlive)
+                    return;
+                sseAlive = false;
+                logger.warn(`会话 ${this.id} 的 SSE 已断开，本轮生成继续，后续事件改由 WebSocket 镜像下发`);
+            });
+        }
+        /**
+         * 把 SSE 事件镜像给当前用户的其他终端。
+         *
+         * SSE 还活着时排除发起端（它已从 SSE 收到同样的事件，不排除会渲染两遍）；
+         * SSE 已断开时不再排除，发起端重连 WebSocket 后就能接着看完本轮。
+         */
         const mirrorToOtherClients = (data) => {
             if (!wsService)
                 return;
@@ -730,7 +750,7 @@ class Session {
                         assistantMessageId,
                         event: data,
                     },
-                }, { excludeClientId: clientId });
+                }, { excludeClientId: sseAlive ? clientId : undefined });
             }
             catch (error) {
                 logger.error('SSE mirror error:', error);
@@ -758,7 +778,8 @@ class Session {
         // SSE 辅助方法：res 为 null 时跳过写入（本地执行模式），但仍同步给其他终端
         const sendSSE = (res, data) => {
             mirrorToOtherClients(data);
-            if (!res)
+            // 连接已断开时不再写入：写只会抛错刷日志，内容已改由镜像下发
+            if (!res || !sseAlive)
                 return;
             try {
                 res.write(`data: ${JSON.stringify(data)}\n\n`);
@@ -770,7 +791,7 @@ class Session {
         };
         // 心跳机制：每 15 秒发送一次心跳，防止连接超时
         const heartbeatInterval = setInterval(() => {
-            if (res && !res.destroyed) {
+            if (res && sseAlive && !res.destroyed) {
                 sendSSE(res, { type: 'heartbeat' });
             }
             else {

package/package.json

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