@myassis/gateway 1.0.36 → 1.0.38

package/dist/services/session/Session.js

@@ -474,9 +474,23 @@ class Session {
         const sendSSE = (res, data) => {
             if (!res)
                 return;
-            res.write(`data: ${JSON.stringify(data)}\n\n`);
-            res.flush?.();
+            try {
+                res.write(`data: ${JSON.stringify(data)}\n\n`);
+                res.flush?.();
+            }
+            catch (error) {
+                logger.error('SSE write error:', error);
+            }
         };
+        // 心跳机制：每 15 秒发送一次心跳，防止连接超时
+        const heartbeatInterval = setInterval(() => {
+            if (res && !res.destroyed) {
+                sendSSE(res, { type: 'heartbeat' });
+            }
+            else {
+                clearInterval(heartbeatInterval);
+            }
+        }, 15000);
         // Send message start event
         sendSSE(res, { type: 'message_start' });
         // Build messages for API call (保留 tool_call_id 等必要字段)
@@ -918,17 +932,29 @@ class Session {
                     this.addAssistantMessage('', toolCalls, [...new Set(modelNames)].join(','), this.currentMessageId);
                 }
                 if (error.message !== 'aborted') {
-                    sendSSE(res, { type: 'error', error: error.message });
+                    try {
+                        sendSSE(res, { type: 'error', error: error.message });
+                    }
+                    catch (e) {
+                        logger.error('Failed to send error event:', e);
+                    }
                 }
             }
             return { success: false, error: error.message };
         }
         finally {
+            // 清理心跳定时器
+            clearInterval(heartbeatInterval);
             if (!childAgent) {
                 this.isGenerating = false;
                 this.abortController = null;
-                if (res) {
-                    res.end();
+                if (res && !res.destroyed) {
+                    try {
+                        res.end();
+                    }
+                    catch (e) {
+                        logger.error('Error closing SSE connection:', e);
+                    }
                 }
                 // 非当前查看的 session，助手回复完成后增加未读计数
                 (0, SessionManager_js_1.getSessionManager)(this.userId).onMessageComplete(this.id);

package/dist/services/tools/exec.js

@@ -140,28 +140,39 @@ exports.execTool = {
                 type: 'number',
                 description: '命令超时时间（毫秒），默认 60000',
                 default: 60000,
+            },
+            isUserAsk: {
+                type: 'boolean',
+                description: '是否为用户主动要求的命令（用于绕过危险命令拦截）。如果用户明确要求执行此命令（如"帮我执行shutdown"），设置为true。',
+                default: false,
             }
         },
         required: ['command'],
     },
     handler: async (args, sessionId) => {
         return new Promise(async (resolve) => {
-            const { cwd, timeout = 60000, approved } = args;
+            const { cwd, timeout = 60000, approved, isUserAsk = false } = args;
             let command = args.command;
-            // 检查危险命令（如果已批准则跳过）
+            // 检查危险命令（如果已批准或用户主动要求则跳过拦截）
             if (!approved && isDangerousCommand(command)) {
-                if (!sessionId) {
-                    resolve({ success: false, errorMessage: '危险命令需要用户确认，但缺少会话ID' });
+                if (isUserAsk) {
+                    // 用户主动要求，跳过拦截
+                    logger.info(`用户主动要求执行危险命令，跳过拦截: ${command}`);
+                }
+                else {
+                    if (!sessionId) {
+                        resolve({ success: false, errorMessage: '危险命令需要用户确认，但缺少会话ID' });
+                        return;
+                    }
+                    const token = addPendingApproval(command, cwd, timeout, sessionId);
+                    resolve({
+                        success: false,
+                        needsApproval: true,
+                        approvalToken: token,
+                        errorMessage: `危险命令需要用户确认：${command}`,
+                    });
                     return;
                 }
-                const token = addPendingApproval(command, cwd, timeout, sessionId);
-                resolve({
-                    success: false,
-                    needsApproval: true,
-                    approvalToken: token,
-                    errorMessage: `危险命令需要用户确认：${command}`,
-                });
-                return;
             }
             const options = {
                 cwd,

package/package.json

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