@myassis/gateway 1.0.84 → 1.0.85

package/dist/routes/agent.js

@@ -488,7 +488,9 @@ router.post('/sessions/:sessionId/reset', async (req, res) => {
     try {
         const userId = req.userId;
         const { sessionId } = req.params;
-        const stopped = (0, index_js_2.getSessionManager)(userId).stopSession(sessionId);
+        // 终端标识：停止事件广播给其他终端时排除发起端
+        const clientId = req.body?.clientId || req.headers['x-client-id'] || undefined;
+        const stopped = (0, index_js_2.getSessionManager)(userId).stopSession(sessionId, clientId);
         res.json({ success: true, data: { stopped } });
     }
     catch (error) {

package/dist/services/session/Session.js

@@ -255,16 +255,29 @@ class Session {
     addUserMessage(content, userMessageId, attachments) {
         const message = this.getUserMessage(content, userMessageId, attachments);
         this.messages.push(message);
-        this.toInsertMessages.push(message);
+        this.enqueueMessageSave(message);
         return message;
     }
     /**
      * Add assistant message
      */
     addAssistantMessage(content, toolCalls, modelName, id) {
+        const messageId = id ? id : (0, uuid_1.v4)();
+        // 本轮可能已经通过 persistAssistantProgress 落过盘并入过 this.messages，
+        // 这里必须复用同一条，否则 messages 里会出现两条同 id 的助手消息，
+        // 上下文重复、前端也会渲染出两个气泡。
+        const existing = this.messages.find(m => m.id === messageId && m.role === 'assistant');
+        if (existing) {
+            existing.content = content;
+            existing.toolCalls = toolCalls;
+            existing.modelName = modelName;
+            existing.updatedAt = Date.now();
+            this.enqueueMessageSave(existing);
+            return existing;
+        }
         const message = {
             sessionId: this.id,
-            id: id ? id : (0, uuid_1.v4)(),
+            id: messageId,
             role: 'assistant',
             content,
             toolCalls,
@@ -272,26 +285,72 @@ class Session {
             modelName
         };
         this.messages.push(message);
-        this.toInsertMessages.push(message);
+        this.enqueueMessageSave(message);
         return message;
     }
     /**
-     * Save a single message to store
+     * 把消息加入待落库队列（按 id 去重）。
+     *
+     * 同一条助手消息在一轮里会被反复标记为待保存（每个工具轮次、正文就绪、收尾各一次），
+     * 队列里只保留一份引用即可——落库时读的是对象的最新字段。
+     */
+    enqueueMessageSave(message) {
+        if (!this.toInsertMessages.some(m => m.id === message.id)) {
+            this.toInsertMessages.push(message);
+        }
+    }
+    /**
+     * 冲刷待落库队列。
+     *
+     * 原先只在 streamChat 的 finally 里调用一次，意味着一轮对话（可能持续几分钟、
+     * 跑十几个工具）期间进程崩溃/被杀，用户消息和已经产出的助手内容会全部丢失。
+     * 现在改为在流程中的关键节点多次调用，每次都是幂等 UPSERT。
      */
     async saveMessage() {
         const store = this.store;
+        if (this.toInsertMessages.length === 0) {
+            return;
+        }
+        // 取出快照后立刻清空：落库期间若有新消息入队，属于下一次冲刷的范围，
+        // 不能因为本次失败/成功而被连带清掉或重复写入。
+        const pending = this.toInsertMessages;
+        this.toInsertMessages = [];
         this.updatedAt = Date.now();
         try {
             store.transaction(() => {
-                for (let message of this.toInsertMessages) {
-                    store.insertMessage(this.id, message);
+                for (const message of pending) {
+                    store.upsertMessage(this.id, message);
                 }
-                this.toInsertMessages = [];
                 store.updateSession(this.toStoreData());
             });
         }
         catch (error) {
             logger.error('数据保存错误：', error);
+            // 落库失败的消息重新排回队列头部，等下一个节点重试，避免直接丢数据
+            this.toInsertMessages = pending.concat(this.toInsertMessages);
+        }
+    }
+    /**
+     * 会话进行中增量保存助手消息。
+     *
+     * 在每个工具轮次结束、以及正文流式输出完成后调用，让数据库里始终有一份
+     * 接近最新的助手消息。这样即使进程异常退出，用户也能看到已经完成的部分，
+     * 而不是整轮消失。
+     *
+     * @param content 当前已产出的正文
+     * @param toolCalls 当前已完成的工具调用
+     * @param modelName 参与本轮的模型名
+     */
+    persistAssistantProgress(content, toolCalls, modelName) {
+        if (!this.currentMessageId)
+            return;
+        try {
+            this.addAssistantMessage(content, toolCalls, modelName, this.currentMessageId);
+            void this.saveMessage();
+        }
+        catch (error) {
+            // 增量保存是尽力而为的容灾手段，失败不能中断正在进行的对话
+            logger.warn('助手消息增量保存失败:', error?.message);
         }
     }
     /**
@@ -739,6 +798,9 @@ class Session {
         if (!childAgent) {
             this.addUserMessage(content, userMessageId, attachments);
             messages.push(this.messages.at(-1));
+            // 用户消息一旦确定就立即落库：后面的模型调用可能耗时数分钟，
+            // 期间进程退出不该让用户刚发的话凭空消失。
+            void this.saveMessage();
         }
         else {
             messages.push(this.getUserMessage(content, userMessageId, attachments));
@@ -1081,6 +1143,11 @@ class Session {
                 };
                 await Promise.all(llmResult.toolCalls.map(x => startToolCall(x)));
                 toolCalls.push(toolCall);
+                // 本轮工具已执行完，增量落库一次。工具轮次可能很多且每轮都耗时，
+                // 这里保存能让崩溃后仍保留已完成的工具调用记录。
+                if (!childAgent) {
+                    this.persistAssistantProgress(llmResult.content || llmResult.reasoningContent || '', toolCalls, [...new Set(modelNames)].join(','));
+                }
                 // 计划模式：本轮是否调用了 updatePlan
                 const calledUpdatePlan = llmResult.toolCalls.some((tc) => tc.toolName === 'updatePlan');
                 roundsSincePlanUpdate = calledUpdatePlan ? 0 : roundsSincePlanUpdate + 1;
@@ -1147,6 +1214,9 @@ class Session {
                 }
                 if (!childAgent) {
                     this.addAssistantMessage(llmResult.content || llmResult.reasoningContent, toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
+                    // 正文已完整产出，先落库再发 complete：
+                    // 客户端收到 complete 后就认为这条消息定稿了，此时库里必须已经有它。
+                    await this.saveMessage();
                 }
                 sendSSE(res, { type: 'complete', modelName: [...new Set(modelNames)].join(",") });
                 sendSSE(res, { type: '[DONE]' });
@@ -1161,7 +1231,9 @@ class Session {
             logger.error('stream chat ', error);
             if (!childAgent) {
                 if (toolCalls) {
-                    this.addAssistantMessage('', toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
+                    // 用 '' 覆盖会把已经增量保存的正文清空，这里保留已有内容
+                    const saved = this.messages.find(m => m.id === this.currentMessageId && m.role === 'assistant');
+                    this.addAssistantMessage(saved?.content || '', toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
                 }
                 if (error.message !== 'aborted') {
                     try {
@@ -1242,7 +1314,9 @@ class Session {
     /**
      * 停止当前正在进行的生成
      */
-    stopGenerating() {
+    stopGenerating(clientId) {
+        const wasGenerating = this.isGenerating;
+        const stoppedMessageId = this.currentMessageId;
         this.isGenerating = false;
         this.abortPrecompression();
         // 停止生成后清除计划，避免 Desktop 的进度圈停留在未完成状态
@@ -1251,6 +1325,39 @@ class Session {
             this.abortController.abort();
             this.abortController = null;
         }
+        // 通知其他终端本轮已被停止。
+        //
+        // 中断走的是 abort 分支，streamChat 既不会发 error 也不会发 [DONE]
+        // （见 catch 中的 error.message !== 'aborted' 判断），
+        // 因此镜像终端收不到任何终止事件，会永远卡在「思考中」占位和 loading 态。
+        // 这里必须显式广播一次。
+        if (wasGenerating) {
+            void this.notifySessionStopped(stoppedMessageId, clientId);
+        }
+    }
+    /**
+     * 向其他终端广播「本轮生成已停止」。
+     *
+     * @param assistantMessageId 被停止的助手消息 id，接收端据此定位占位气泡
+     * @param excludeClientId 发起停止的终端（它自己已在本地做过收尾，无需再处理）
+     */
+    async notifySessionStopped(assistantMessageId, excludeClientId) {
+        try {
+            const webSocketService = await this.getWebSocketService();
+            if (!webSocketService)
+                return;
+            webSocketService.sendToUser(String(this.userId), {
+                type: 'session_stopped',
+                payload: {
+                    sessionId: this.id,
+                    agentId: this.agentId,
+                    assistantMessageId,
+                },
+            }, { excludeClientId });
+        }
+        catch (error) {
+            logger.error('推送会话停止事件失败:', error);
+        }
     }
     /**
      * 获取 AbortController 用于中断请求

package/dist/services/session/SessionManager.js

@@ -111,14 +111,15 @@ class SessionManager {
     /**
      * Stop session generation
      */
-    stopSession(sessionId) {
+    stopSession(sessionId, clientId) {
         const session = this.getSession(sessionId);
         if (!session) {
             logger.warn(`Session ${sessionId} not found for stop`);
             return false;
         }
         if (session.isGenerating) {
-            session.stopGenerating();
+            // 透传发起端标识：停止事件广播时要排除它，它已在本地收尾
+            session.stopGenerating(clientId);
             logger.info(`Stopped session ${sessionId}`);
             return true;
         }

package/dist/services/session/SessionStore.js

@@ -81,17 +81,17 @@ class SessionStore {
     }
     // ========== Session Operations ==========
     insertSession(session) {
-        this.db.prepare(`
-      INSERT INTO sessions (id, user_id, agent_id, title, select_model_id, voice_state, created_at, updated_at,last_message_summary,last_message_summary_at,unread_count,is_current,message_queue,message_queue_auto_execute, use_system_mode)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?,?,?,?,?,?,?,?)
+        this.db.prepare(`
+      INSERT INTO sessions (id, user_id, agent_id, title, select_model_id, voice_state, created_at, updated_at,last_message_summary,last_message_summary_at,unread_count,is_current,message_queue,message_queue_auto_execute, use_system_mode)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?,?,?,?,?,?,?,?)
     `).run(session.id, session.userId, session.agentId || null, session.title, session.selectModelId || '', JSON.stringify(session.voiceState), session.createdAt, session.updatedAt, session.lastMessageSummary, session.lastMessageSummaryAt, session.unreadCount || 0, session.isCurrent ? 1 : 0, JSON.stringify(session.messageQueue || []), session.messageQueueAutoExecute === false ? 0 : 1, session.useSystemMode ? 1 : 0);
     }
     updateSession(session) {
-        this.db.prepare(`
-      UPDATE sessions 
-      SET user_id = ?, agent_id = ?, title = ?, select_model_id = ?,
-          voice_state = ?, updated_at = ?,last_message_summary=?,last_message_summary_at=?,unread_count=?,is_current=?,message_queue=?,message_queue_auto_execute=?, use_system_mode=?
-      WHERE id = ?
+        this.db.prepare(`
+      UPDATE sessions 
+      SET user_id = ?, agent_id = ?, title = ?, select_model_id = ?,
+          voice_state = ?, updated_at = ?,last_message_summary=?,last_message_summary_at=?,unread_count=?,is_current=?,message_queue=?,message_queue_auto_execute=?, use_system_mode=?
+      WHERE id = ?
     `).run(session.userId, session.agentId || null, session.title, session.selectModelId || '', JSON.stringify(session.voiceState), session.updatedAt, session.lastMessageSummary, session.lastMessageSummaryAt, session.unreadCount || 0, session.isCurrent ? 1 : 0, JSON.stringify(session.messageQueue || []), session.messageQueueAutoExecute === false ? 0 : 1, session.useSystemMode ? 1 : 0, session.id);
     }
     // 支持部分更新的 updateSession
@@ -142,25 +142,48 @@ class SessionStore {
     }
     // ========== Message Operations ==========
     insertMessage(sessionId, msg) {
-        this.db.prepare(`
-      INSERT INTO messages (id, session_id, role, content, attachments, tool_calls, created_at, updated_at, model_name, feedback)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        this.db.prepare(`
+      INSERT INTO messages (id, session_id, role, content, attachments, tool_calls, created_at, updated_at, model_name, feedback)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+    `).run(msg.id, sessionId, msg.role, msg.content, JSON.stringify(msg.attachments || []), msg.toolCalls ? JSON.stringify(msg.toolCalls) : null, msg.createdAt, msg.updatedAt || msg.createdAt, msg.modelName || null, msg.feedback || null);
+    }
+    /**
+     * 写入或更新消息（按 id 幂等）。
+     *
+     * 会话进行中需要多次落库同一条助手消息（工具轮次结束、正文就绪、收尾），
+     * 用 INSERT 会撞主键，用 UPDATE 首次又没有行，因此统一走 UPSERT。
+     *
+     * 冲突时刻意不更新 session_id / role / created_at：
+     * created_at 是消息列表的排序依据（findMessagesBySessionId ORDER BY created_at），
+     * 若每次落库都刷新它，同一轮的用户消息与助手消息顺序会被打乱。
+     */
+    upsertMessage(sessionId, msg) {
+        this.db.prepare(`
+      INSERT INTO messages (id, session_id, role, content, attachments, tool_calls, created_at, updated_at, model_name, feedback)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+      ON CONFLICT(id) DO UPDATE SET
+        content = excluded.content,
+        attachments = excluded.attachments,
+        tool_calls = excluded.tool_calls,
+        updated_at = excluded.updated_at,
+        model_name = excluded.model_name,
+        feedback = excluded.feedback
     `).run(msg.id, sessionId, msg.role, msg.content, JSON.stringify(msg.attachments || []), msg.toolCalls ? JSON.stringify(msg.toolCalls) : null, msg.createdAt, msg.updatedAt || msg.createdAt, msg.modelName || null, msg.feedback || null);
     }
     insertMessages(sessionId, messages) {
-        const stmt = this.db.prepare(`
-      INSERT INTO messages (id, session_id, role, content, attachments,  tool_calls,created_at, updated_at, model_name, feedback)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+        const stmt = this.db.prepare(`
+      INSERT INTO messages (id, session_id, role, content, attachments,  tool_calls,created_at, updated_at, model_name, feedback)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
     `);
         for (const msg of messages) {
             stmt.run(msg.id, sessionId, msg.role, msg.content, JSON.stringify(msg.attachments || []), msg.toolCalls ? JSON.stringify(msg.toolCalls) : null, msg.createdAt, msg.updatedAt || msg.createdAt, msg.modelName || null, msg.feedback || null);
         }
     }
     updateMessage(sessionId, msg) {
-        this.db.prepare(`
-      UPDATE messages 
-      SET content = ?, attachments = ?, tool_calls = ?,  updated_at = ?, model_name = ?, feedback = ?
-      WHERE id = ? AND session_id = ?
+        this.db.prepare(`
+      UPDATE messages 
+      SET content = ?, attachments = ?, tool_calls = ?,  updated_at = ?, model_name = ?, feedback = ?
+      WHERE id = ? AND session_id = ?
     `).run(msg.content, JSON.stringify(msg.attachments || []), msg.toolCalls ? JSON.stringify(msg.toolCalls) : null, msg.updatedAt || msg.createdAt, msg.modelName || null, msg.feedback || null, msg.id, sessionId);
     }
     deleteMessage(messageId) {

package/package.json

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