foliko 1.1.63 → 1.1.65

package/src/core/context-compressor.js

@@ -9,7 +9,7 @@
 
 const { logger } = require('../utils/logger');
 const { TokenCounter } = require('./token-counter');
-const { Subagent } = require('./subagent');
+const { validateMessagesPairing, filterPairedMessages } = require('../utils/message-validator');
 
 // 模型上下文限制表
 const MODEL_CONTEXT_LIMITS = {
@@ -30,7 +30,7 @@ const MODEL_CONTEXT_LIMITS = {
 };
 
 // 压缩超时
-const COMPRESSION_TIMEOUT = 1200000;
+const COMPRESSION_TIMEOUT_MS = 1200000;
 
 class ContextCompressor {
   /**
@@ -128,8 +128,8 @@ class ContextCompressor {
     return new Promise((_, reject) => {
       this._compressionTimeoutId = setTimeout(() => {
         this._compressionTimeoutId = null;
-        reject(new Error(`Compression timeout (${COMPRESSION_TIMEOUT}ms)`));
-      }, COMPRESSION_TIMEOUT);
+        reject(new Error(`Compression timeout (${COMPRESSION_TIMEOUT_MS}ms)`));
+      }, COMPRESSION_TIMEOUT_MS);
     });
   }
 
@@ -242,122 +242,7 @@ class ContextCompressor {
    * @private
    */
   _filterPairedMessages(messages) {
-    // 工具类型检查
-    const isToolCall = (item) =>
-      item && (item.type === 'tool-call' || item.type === 'tool-use') && item.toolCallId;
-    const isToolResult = (item) =>
-      item && (item.type === 'tool-result' || item.type === 'tool_result') && item.toolCallId;
-
-    // 第一遍：找出所有 tool call 和它们的 results
-    const assistantToolCalls = new Map(); // toolCallId -> assistant message index
-    const toolResults = new Map(); // toolCallId -> tool result indices
-
-    for (let i = 0; i < messages.length; i++) {
-      const msg = messages[i];
-      if (msg.role === 'assistant') {
-        // 格式1: msg.content 中的 tool-call 块
-        if (msg.content) {
-          const content = Array.isArray(msg.content) ? msg.content : [msg.content];
-          for (const item of content) {
-            if (isToolCall(item)) {
-              assistantToolCalls.set(item.toolCallId, i);
-            }
-          }
-        }
-        // 格式2: msg.tool_calls 数组 (OpenAI 格式)
-        if (Array.isArray(msg.tool_calls)) {
-          for (const tc of msg.tool_calls) {
-            if (tc.id) {
-              assistantToolCalls.set(tc.id, i);
-            }
-          }
-        }
-      }
-      if (msg.role === 'tool' && Array.isArray(msg.content)) {
-        for (const item of msg.content) {
-          if (isToolResult(item)) {
-            if (!toolResults.has(item.toolCallId)) {
-              toolResults.set(item.toolCallId, []);
-            }
-            toolResults.get(item.toolCallId).push(i);
-          }
-        }
-      }
-    }
-
-    // 第二遍：找出哪些 assistant 消息需要保留
-    const assistantIndicesToKeep = new Set();
-    for (let i = 0; i < messages.length; i++) {
-      const msg = messages[i];
-      if (msg.role === 'assistant') {
-        let hasToolCall = false;
-        // 检查 msg.content 格式
-        if (msg.content) {
-          const content = Array.isArray(msg.content) ? msg.content : [msg.content];
-          for (const item of content) {
-            if (isToolCall(item)) {
-              hasToolCall = true;
-              break;
-            }
-          }
-        }
-        // 检查 msg.tool_calls 格式 (OpenAI)
-        if (!hasToolCall && Array.isArray(msg.tool_calls)) {
-          for (const tc of msg.tool_calls) {
-            if (tc.id) {
-              hasToolCall = true;
-              break;
-            }
-          }
-        }
-        if (hasToolCall) {
-          assistantIndicesToKeep.add(i);
-        } else if (i >= messages.length - 3) {
-          // 保留最近几条 assistant 消息
-          assistantIndicesToKeep.add(i);
-        }
-      }
-    }
-
-    // 如果有孤儿的 tool call（没有 result），保留该 assistant 消息
-    for (const [toolCallId, assistantIdx] of assistantToolCalls) {
-      if (!toolResults.has(toolCallId)) {
-        assistantIndicesToKeep.add(assistantIdx);
-      }
-    }
-
-    // 第三遍：确定哪些 tool result 要保留
-    const indicesToKeep = new Set();
-    for (let i = 0; i < messages.length; i++) {
-      const msg = messages[i];
-      if (msg.role === 'tool') {
-        let hasPairedAssistant = false;
-        const content = Array.isArray(msg.content) ? msg.content : [msg.content];
-        for (const item of content) {
-          if (isToolResult(item)) {
-            const assistantIdx = assistantToolCalls.get(item.toolCallId);
-            if (assistantIdx !== undefined && assistantIndicesToKeep.has(assistantIdx)) {
-              hasPairedAssistant = true;
-              break;
-            }
-          }
-        }
-        if (hasPairedAssistant) {
-          indicesToKeep.add(i);
-        }
-      } else if (msg.role === 'assistant') {
-        if (assistantIndicesToKeep.has(i)) {
-          indicesToKeep.add(i);
-        }
-      } else {
-        // user, system 等角色默认保留
-        indicesToKeep.add(i);
-      }
-    }
-
-    // 按索引排序并返回
-    const sortedIndices = Array.from(indicesToKeep).sort((a, b) => a - b);
-    return sortedIndices.map((i) => messages[i]);
+    return filterPairedMessages(messages);
   }
 
   /**
@@ -366,67 +251,7 @@ class ContextCompressor {
    * @returns {Array} 验证后的消息
    */
   validateMessagesPairing(messages) {
-    // 收集所有 assistant 的 tool-call IDs（兼容多种类型名）
-    const assistantToolCallIds = new Set();
-    for (const msg of messages) {
-      if (msg.role === 'assistant') {
-        // 格式1: msg.content 中的 tool-call 块
-        if (Array.isArray(msg.content)) {
-          for (const item of msg.content) {
-            if ((item.type === 'tool-call' || item.type === 'tool-use') && item.toolCallId) {
-              assistantToolCallIds.add(item.toolCallId);
-            }
-          }
-        }
-        // 格式2: msg.tool_calls 数组 (OpenAI 格式)
-        if (Array.isArray(msg.tool_calls)) {
-          for (const tc of msg.tool_calls) {
-            if (tc.id) {
-              assistantToolCallIds.add(tc.id);
-            }
-          }
-        }
-      }
-    }
-
-    // 检查并删除没有配对的 tool result
-    let removedCount = 0;
-    for (const msg of messages) {
-      if (msg.role === 'tool' && Array.isArray(msg.content)) {
-        const originalLength = msg.content.length;
-        msg.content = msg.content.filter((item) => {
-          // 兼容 tool-result 和 tool_result 两种类型
-          if (
-            item &&
-            (item.type === 'tool-result' || item.type === 'tool_result') &&
-            item.toolCallId
-          ) {
-            if (!assistantToolCallIds.has(item.toolCallId)) {
-              removedCount++;
-              return false; // 删除没有配对的 tool result
-            }
-          }
-          return true;
-        });
-
-        // 如果所有 content 都被删除了，标记整个消息待删除
-        if (msg.content.length === 0 && originalLength > 0) {
-          msg._orphaned = true;
-        }
-      }
-    }
-
-    // 移除被标记为 orphaned 的 tool 消息
-    const originalLength = messages.length;
-    const filtered = messages.filter((msg) => !(msg.role === 'tool' && msg._orphaned));
-
-    if (removedCount > 0 || filtered.length !== originalLength) {
-      logger.debug(
-        `Removed ${removedCount} orphaned tool-results, ${originalLength - filtered.length} orphaned tool messages`
-      );
-    }
-
-    return filtered;
+    return validateMessagesPairing(messages);
   }
 
   /**

package/src/core/framework.js

@@ -324,6 +324,55 @@ class Framework extends EventEmitter {
     return this.toolRegistry.getAll();
   }
 
+  // ============================================================================
+  // 公开 API — 减少外部直接访问私有属性
+  // ============================================================================
+
+  /**
+   * 获取主 Agent
+   * @returns {Agent|null}
+   */
+  getMainAgent() {
+    return this._mainAgent;
+  }
+
+  /**
+   * 获取主 Agent 的 System Prompt
+   * @returns {string}
+   */
+  getSystemPrompt() {
+    if (this._mainAgent) {
+      return this._mainAgent._buildSystemPrompt();
+    }
+    return '';
+  }
+
+  /**
+   * 获取插件实例（安全访问）
+   * @param {string} name - 插件名称
+   * @returns {Object|null}
+   */
+  getPluginInstance(name) {
+    const entry = this.pluginManager.get(name);
+    return entry || null;
+  }
+
+  /**
+   * 获取所有 Agent 列表
+   * @returns {Array<Agent>}
+   */
+  getAgents() {
+    return [...this._agents];
+  }
+
+  /**
+   * 获取所有 SessionContext 的 ID 列表
+   * @returns {string[]}
+   */
+  listSessionContexts() {
+    return Array.from(this._sessionContexts.keys());
+  }
+
   // ============================================================================
   // 事件描述注册（供 Ambient Agent 使用）
   // ============================================================================
@@ -551,6 +600,68 @@ class Framework extends EventEmitter {
     }
   }
 
+  // Session TTL 清理配置
+  _sessionTTL = 30 * 60 * 1000; // 默认 30 分钟
+  _sessionCleanupInterval = null;
+
+  /**
+   * 配置 Session TTL
+   * @param {number} ttlMs - TTL 毫秒数
+   */
+  setSessionTTL(ttlMs) {
+    this._sessionTTL = ttlMs;
+    return this;
+  }
+
+  /**
+   * 启动 Session 自动清理
+   * @param {number} [intervalMs=60000] - 检查间隔
+   */
+  startSessionCleanup(intervalMs = 60000) {
+    if (this._sessionCleanupInterval) {
+      clearInterval(this._sessionCleanupInterval);
+    }
+    this._sessionCleanupInterval = setInterval(() => {
+      this._cleanupStaleSessions();
+    }, intervalMs);
+    // 允许进程退出时不阻止
+    if (this._sessionCleanupInterval.unref) {
+      this._sessionCleanupInterval.unref();
+    }
+    return this;
+  }
+
+  /**
+   * 停止 Session 自动清理
+   */
+  stopSessionCleanup() {
+    if (this._sessionCleanupInterval) {
+      clearInterval(this._sessionCleanupInterval);
+      this._sessionCleanupInterval = null;
+    }
+    return this;
+  }
+
+  /**
+   * 清理过期的 Session
+   * @private
+   */
+  _cleanupStaleSessions() {
+    const now = Date.now();
+    const staleIds = [];
+    for (const [sessionId, ctx] of this._sessionContexts) {
+      if (ctx.metadata && (now - ctx.metadata.lastActive) > this._sessionTTL) {
+        staleIds.push(sessionId);
+      }
+    }
+    for (const id of staleIds) {
+      this.destroySessionContext(id);
+    }
+    if (staleIds.length > 0) {
+      this.logger.debug(`Cleaned up ${staleIds.length} stale sessions`);
+    }
+  }
+
   /**
    * 列出所有活跃的 Session ID
    * @returns {string[]}
@@ -737,9 +848,9 @@ class Framework extends EventEmitter {
 
   /**
    * 同步主Agent的提示词部分到所有子Agent
-   * @private
+   * @public — 插件可能需要调用
    */
-  _syncPromptPartsToSubagents() {
+  syncPromptPartsToSubagents() {
     if (!this._mainAgent || !this._mainAgent._systemPromptBuilder) return;
     const mainParts = this._mainAgent._systemPromptBuilder._parts;
     if (!mainParts) return;
@@ -784,8 +895,9 @@ class Framework extends EventEmitter {
     lines.push('');
 
     // 2. 主Agent的系统提示词（基础部分）
-    if (this._mainAgent) {
-      const mainPrompt = this._mainAgent._originalPrompt;
+    const mainAgent = this.getMainAgent();
+    if (mainAgent) {
+      const mainPrompt = mainAgent.getOriginalPrompt();
       if (mainPrompt) {
         lines.push('## 主Agent的系统提示词');
         lines.push(mainPrompt);
@@ -988,10 +1100,17 @@ class Framework extends EventEmitter {
       this._toolRegistryListeners = [];
     }
 
-    // 卸载所有插件
+    // 停止 Session 清理
+    this.stopSessionCleanup();
+
+    // 卸载所有插件（逐个保护，防止一个失败阻断后续）
     const plugins = this.pluginManager.getAll();
     for (const { name } of plugins) {
-      await this.pluginManager.unload(name);
+      try {
+        await this.pluginManager.unload(name);
+      } catch (err) {
+        this.logger.warn(`Failed to unload plugin '${name}': ${err.message}`);
+      }
     }
 
     // 清空工具

package/src/core/plugin-base.js

@@ -286,10 +286,11 @@ class Plugin {
     this._promptParts.push({ name, priority, provider });
 
     // 如果主 agent 已存在，直接注册
-    if (this._framework._mainAgent) {
-      this._framework._mainAgent.registerPromptPart(name, priority, provider);
+    const mainAgent = this._framework.getMainAgent();
+    if (mainAgent) {
+      mainAgent.registerPromptPart(name, priority, provider);
       // 同步到所有子Agent
-      this._framework._syncPromptPartsToSubagents();
+      this._framework.syncPromptPartsToSubagents();
       return;
     }
 
@@ -307,14 +308,15 @@ class Plugin {
         if (this._deferredPromptParts) {
           for (const part of this._deferredPromptParts) {
             try {
-              this._framework._mainAgent?.registerPromptPart(part.name, part.priority, part.provider);
+              const mainAgent = this._framework.getMainAgent();
+              mainAgent?.registerPromptPart(part.name, part.priority, part.provider);
             } catch (err) {
               log.error(` Failed to register prompt part ${part.name}:`, err.message);
             }
           }
           this._deferredPromptParts = null;
           // 同步到所有子Agent
-          this._framework._syncPromptPartsToSubagents();
+          this._framework.syncPromptPartsToSubagents();
         }
       }, 100);
     });

package/src/core/provider.js

@@ -14,26 +14,32 @@ const DEFAULT_PROVIDERS = {
   openai: {
     name: 'OpenAI',
     baseURL: 'https://api.openai.com/v1',
+    defaultModel: 'gpt-4o',
   },
   ollama: {
     name: 'Ollama',
     baseURL: 'http://localhost:11434/v1',
+    defaultModel: 'llama3',
   },
   lmstudio: {
     name: 'LM Studio',
     baseURL: 'http://localhost:1234/v1',
+    defaultModel: 'local-model',
   },
   deepseek: {
     name: 'DeepSeek',
     baseURL: 'https://api.deepseek.com/v1',
+    defaultModel: 'deepseek-chat',
   },
   anthropic: {
     name: 'Anthropic',
     baseURL: 'https://api.anthropic.com/v1',
+    defaultModel: 'claude-sonnet-4-20250514',
   },
   minimax: {
     name: 'MiniMax',
     baseURL: 'https://api.minimaxi.com/v1',
+    defaultModel: 'MiniMax-M2.7',
   },
 };
 

package/src/core/subagent.js

@@ -8,8 +8,9 @@ const { EventEmitter } = require('../utils/event-emitter');
 const { cleanResponse } = require('../utils');
 const { generateText, tool, stepCountIs, isLoopFinished, RetryError, ToolLoopAgent } = require('ai');
 const { z } = require('zod');
-const fs=require('fs/promises')
 const { logger } = require('../utils/logger');
+const { validateAll } = require('../utils/message-validator');
+const { isNetworkError, calculateDelay } = require('../utils/retry');
 
 class Subagent extends EventEmitter {
   /**
@@ -94,64 +95,7 @@ class Subagent extends EventEmitter {
    * @private
    */
   _validateMessagesPairing(messages) {
-    // 收集所有 assistant 的 tool-call IDs
-    const assistantToolCallIds = new Set();
-    for (const msg of messages) {
-      if (msg.role === 'assistant') {
-        // 格式1: msg.content 中的 tool-call 块
-        if (Array.isArray(msg.content)) {
-          for (const item of msg.content) {
-            if ((item.type === 'tool-call' || item.type === 'tool-use') && item.toolCallId) {
-              assistantToolCallIds.add(item.toolCallId);
-            }
-          }
-        }
-        // 格式2: msg.tool_calls 数组 (OpenAI 格式)
-        if (Array.isArray(msg.tool_calls)) {
-          for (const tc of msg.tool_calls) {
-            if (tc.id) {
-              assistantToolCallIds.add(tc.id);
-            }
-          }
-        }
-      }
-    }
-
-    // 检查并删除没有配对的 tool result
-    let removedCount = 0;
-    for (const msg of messages) {
-      if (msg.role === 'tool' && Array.isArray(msg.content)) {
-        const originalLength = msg.content.length;
-        msg.content = msg.content.filter((item) => {
-          if (
-            item &&
-            (item.type === 'tool-result' || item.type === 'tool_result') &&
-            item.toolCallId
-          ) {
-            if (!assistantToolCallIds.has(item.toolCallId)) {
-              removedCount++;
-              return false;
-            }
-          }
-          return true;
-        });
-
-        if (msg.content.length === 0 && originalLength > 0) {
-          msg._orphaned = true;
-        }
-      }
-    }
-
-    const originalLength = messages.length;
-    const filtered = messages.filter((msg) => !(msg.role === 'tool' && msg._orphaned));
-
-    if (removedCount > 0 || filtered.length !== originalLength) {
-      logger.debug(
-        `[Subagent:${this.name}] _validateMessagesPairing: removed ${removedCount} orphaned tool-results, ${originalLength - filtered.length} orphaned tool messages`
-      );
-    }
-
-    return filtered;
+    return validateAll(messages);
   }
 
   /**
@@ -159,64 +103,8 @@ class Subagent extends EventEmitter {
    * @private
    */
   _validateToolCalls(messages) {
-    let fixedCount = 0;
-    const invalidatedToolCallIds = new Set();
-
-    for (const msg of messages) {
-      if (msg.role === 'assistant' && Array.isArray(msg.content)) {
-        for (const item of msg.content) {
-          if (item.type !== 'tool-call' && item.type !== 'tool-use') {
-            continue;
-          }
-
-          const input = item.input;
-          if (typeof input !== 'string') {
-            continue;
-          }
-
-          const trimmed = input.trim();
-          if (trimmed === '{' || trimmed === '' || !trimmed.startsWith('{')) {
-            if (item.toolCallId) {
-              invalidatedToolCallIds.add(item.toolCallId);
-            }
-            logger.warn(
-              `[Subagent:${this.name}] _validateToolCalls: invalid tool-call input="${input}", toolCallId=${item.toolCallId}`
-            );
-            item.type = 'text';
-            item.text = `（工具调用 ${item.toolName} 参数不完整，已跳过）`;
-            delete item.toolCallId;
-            delete item.toolName;
-            delete item.input;
-            fixedCount++;
-          }
-        }
-      }
-    }
-
-    if (invalidatedToolCallIds.size > 0) {
-      for (const msg of messages) {
-        if (msg.role === 'tool' && Array.isArray(msg.content)) {
-          const oldLen = msg.content.length;
-          msg.content = msg.content.filter((item) => {
-            if (item.type !== 'tool-result' && item.type !== 'tool_result') {
-              return true;
-            }
-            if (item.toolCallId && invalidatedToolCallIds.has(item.toolCallId)) {
-              logger.warn(
-                `[Subagent:${this.name}] _validateToolCalls: removing orphaned tool-result with toolCallId=${item.toolCallId}`
-              );
-              fixedCount++;
-              return false;
-            }
-            return true;
-          });
-        }
-      }
-    }
-
-    if (fixedCount > 0) {
-      logger.info(`[Subagent:${this.name}] _validateToolCalls: Fixed ${fixedCount} incomplete tool calls/results`);
-    }
+    // validateAll 内部已包含 validateMessagesPairing + validateToolCalls
+    // Subagent 的 chat() 中 _validateMessagesPairing 已调用 validateAll，所以这里不再重复
   }
 
   /**
@@ -327,8 +215,9 @@ class Subagent extends EventEmitter {
     }
 
     // 2. 主Agent的系统提示词
-    if (this.framework && this.framework._mainAgent) {
-      const mainPrompt = this.framework._mainAgent._originalPrompt;
+    const mainAgent = this.framework?.getMainAgent?.();
+    if (mainAgent) {
+      const mainPrompt = mainAgent.getOriginalPrompt();
       if (mainPrompt) {
         lines.push('## 主Agent的系统提示词');
         lines.push(mainPrompt);
@@ -472,18 +361,8 @@ class Subagent extends EventEmitter {
       } catch (err) {
         logger.warn(`[Subagent:${this.name}] generateText error: ${err.message}`);
         lastError = err;
-        const errName = err?.name || '';
-        const errMessage = err?.message || '';
-        const errCode = err?.code || err?.status;
-
-        // 判断是否是重试错误
-        const isRetryError =
-          RetryError.isInstance?.(err) ||
-          errName === 'AI_RetryError' ||
-          errName === 'RetryError' ||
-          err?.reason === 'maxRetriesExceeded';
-
-        if (isRetryError && attempt < maxRetries) {
+          // 统一使用 retry.js 的错误判断
+        if (isNetworkError(err) && attempt < maxRetries) {
           logger.warn(
             `[Subagent:${this.name}] AI 服务暂时不可用，正在重试 (${attempt + 1}/${maxRetries})`
           );
@@ -511,11 +390,13 @@ class Subagent extends EventEmitter {
    */
   _getFriendlyError(err) {
     const errName = err?.name || '';
+
+    // 先用 isNetworkError 统一判断
+    if (isNetworkError(err)) {
+      return 'AI 服务暂时不可用，请稍后重试';
+    }
+
     const errorMessages = {
-      AI_RetryError: 'AI 服务暂时不可用，请稍后重试',
-      AI_APICallError: err?.isRetryable
-        ? 'AI 服务暂时不可用，请稍后重试'
-        : err.message?.split('\n')[0] || 'AI 请求失败',
       AI_NoContentGeneratedError: 'AI 未生成有效内容，请重试',
       AI_NoSuchModelError: '指定的 AI 模型不存在',
       AI_NoSuchProviderError: 'AI 提供商配置错误',