@myassis/gateway 1.0.109 → 1.0.110

package/dist/services/memory/MemoryManager.js

@@ -37,6 +37,10 @@ const DEFAULT_CONFIG = {
     summaryTriggerChars: index_js_2.appConfig.summaryTriggerChars,
     enabled: true,
 };
+/** 单次摘要模型请求超时（毫秒） */
+const SUMMARY_REQUEST_TIMEOUT = 45000;
+/** 整个压缩流程的硬超时（毫秒），超过后放弃压缩、沿用原始历史 */
+const SUMMARY_OVERALL_TIMEOUT = 90000;
 /**
  * 会话级压缩任务表：sessionId -> 进行中的压缩任务。
  *
@@ -298,6 +302,7 @@ class MemoryManager {
             compressionTasks.delete(this.session.id);
         }
         if (!summary) {
+            this.emitSSE({ type: 'context_compress_done' });
             // 摘要失败/被中断：沿用旧摘要或退回原始历史，不写入残缺摘要
             if (plan.action === 'incremental') {
                 return [
@@ -307,6 +312,7 @@ class MemoryManager {
             }
             return plan.prevRetained;
         }
+        this.emitSSE({ type: 'context_compress_done' });
         return [
             this.toSummaryMessage(summary, plan.newBoundaryAt, '', plan.ledger),
             ...plan.retained,
@@ -351,7 +357,7 @@ class MemoryManager {
         const prompt = TURN_COMPACT_PROMPT_HEAD + transcript;
         try {
             const models = await this.getSummaryModels();
-            const llmClient = new LLMClient_js_1.LLMClient(models, [{ role: 'user', content: prompt }], this.signal, [], 600000);
+            const llmClient = new LLMClient_js_1.LLMClient(models, [{ role: 'user', content: prompt }], this.signal, [], SUMMARY_REQUEST_TIMEOUT);
             if (this.session.useSystemMode) {
                 const { getRequestToken, getServerBaseUrl } = await Promise.resolve().then(() => __importStar(require('../../api/index.js')));
                 llmClient.setSystemMode(true, this.session.selectModelId, getServerBaseUrl(), getRequestToken());
@@ -554,6 +560,26 @@ ${conversation}
         if (!this.childAgent) {
             this.emitSSE({ type: 'context_compressing' });
         }
+        try {
+            return await Promise.race([
+                this.doGenerateSummary(messages, lastSummary),
+                this.summaryTimeout(),
+            ]);
+        }
+        catch (error) {
+            logger.warn(`压缩流程异常，保留原始历史: ${error?.message}`);
+            return lastSummary || null;
+        }
+    }
+    summaryTimeout() {
+        return new Promise((resolve) => {
+            setTimeout(() => {
+                logger.warn(`压缩流程超时（${SUMMARY_OVERALL_TIMEOUT}ms），放弃压缩`);
+                resolve(null);
+            }, SUMMARY_OVERALL_TIMEOUT);
+        });
+    }
+    async doGenerateSummary(messages, lastSummary) {
         const MAX_INPUT_CHARS = 30000; // 单次摘要最大输入字符数
         // 格式化所有消息用于估算长度
         const formattedMessages = messages
@@ -565,7 +591,7 @@ ${conversation}
             return await this.hierarchicalSummary(formattedMessages, lastSummary);
         }
         const summaryPrompt = this.buildSummaryPrompt(messages, lastSummary);
-        const llmClient = new LLMClient_js_1.LLMClient(await this.getSummaryModels(), [{ role: 'user', content: summaryPrompt }], this.signal, [], 600000);
+        const llmClient = new LLMClient_js_1.LLMClient(await this.getSummaryModels(), [{ role: 'user', content: summaryPrompt }], this.signal, [], SUMMARY_REQUEST_TIMEOUT);
         if (this.session.useSystemMode) {
             const { getRequestToken, getServerBaseUrl } = await Promise.resolve().then(() => __importStar(require('../../api/index.js')));
             llmClient.setSystemMode(true, this.session.selectModelId, getServerBaseUrl(), getRequestToken());
@@ -598,7 +624,7 @@ ${conversation}
         const models = await this.getSummaryModels();
         const summarizeBatch = async (batch) => {
             const batchPrompt = `请简洁总结以下对话片段的关键信息，特别关注：项目结构、关键决策、错误及修复。\n\n${batch.join('\n\n')}`;
-            const llmClient = new LLMClient_js_1.LLMClient(models, [{ role: 'user', content: batchPrompt }], this.signal, [], 600000);
+            const llmClient = new LLMClient_js_1.LLMClient(models, [{ role: 'user', content: batchPrompt }], this.signal, [], SUMMARY_REQUEST_TIMEOUT);
             if (this.session.useSystemMode) {
                 const { getRequestToken, getServerBaseUrl } = await Promise.resolve().then(() => __importStar(require('../../api/index.js')));
                 llmClient.setSystemMode(true, this.session.selectModelId, getServerBaseUrl(), getRequestToken());
@@ -609,6 +635,7 @@ ${conversation}
         // 保持批次顺序，同时限制并发
         const subSummaries = new Array(batches.length).fill('');
         let aborted = false;
+        let consecutiveFailures = 0;
         for (let i = 0; i < batches.length; i += CONCURRENCY) {
             if (this.signal.aborted) {
                 aborted = true;
@@ -616,14 +643,26 @@ ${conversation}
             }
             const slice = batches.slice(i, i + CONCURRENCY);
             const results = await Promise.allSettled(slice.map((batch) => summarizeBatch(batch)));
+            let groupSuccess = 0;
             results.forEach((result, offset) => {
                 if (result.status === 'fulfilled' && result.value) {
                     subSummaries[i + offset] = result.value;
+                    groupSuccess++;
                 }
                 else if (result.status === 'rejected') {
                     logger.warn(`子摘要生成失败: ${result.reason?.message}`);
                 }
             });
+            if (groupSuccess === 0) {
+                consecutiveFailures++;
+                if (consecutiveFailures >= 2) {
+                    logger.warn(`分层摘要连续失败，提前终止`);
+                    break;
+                }
+            }
+            else {
+                consecutiveFailures = 0;
+            }
         }
         // 被中断时不写入残缺摘要，交由调用方回退到原始历史
         if (aborted || this.signal.aborted) {
@@ -638,7 +677,7 @@ ${conversation}
         }
         // 合并子摘要为最终摘要
         const mergePrompt = this.buildMergePrompt(valid, lastSummary);
-        const mergeLlmClient = new LLMClient_js_1.LLMClient(models, [{ role: 'user', content: mergePrompt }], this.signal, [], 600000);
+        const mergeLlmClient = new LLMClient_js_1.LLMClient(models, [{ role: 'user', content: mergePrompt }], this.signal, [], SUMMARY_REQUEST_TIMEOUT);
         if (this.session.useSystemMode) {
             const { getRequestToken, getServerBaseUrl } = await Promise.resolve().then(() => __importStar(require('../../api/index.js')));
             mergeLlmClient.setSystemMode(true, this.session.selectModelId, getServerBaseUrl(), getRequestToken());

package/package.json

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