foliko 1.1.63 → 1.1.64

package/src/core/constants.js

@@ -0,0 +1,198 @@
+/**
+ * Foliko Framework 统一常量
+ *
+ * 集中管理所有 magic numbers / strings，消除散落各处的重复值。
+ * 所有模块应引用此文件而非硬编码。
+ */
+
+const path = require('path');
+
+// ============================================================================
+// AI 模型与提供者
+// ============================================================================
+
+/** 默认 AI 提供者 */
+const DEFAULT_PROVIDER = 'deepseek';
+
+/** 默认 AI 模型 */
+const DEFAULT_MODEL = 'deepseek-chat';
+
+/** 默认 Max Output Tokens */
+const DEFAULT_MAX_OUTPUT_TOKENS = 8192;
+
+/** 默认 Temperature */
+const DEFAULT_TEMPERATURE = 0.3;
+
+/** 默认 Max Steps（工具调用最大轮数） */
+const DEFAULT_MAX_STEPS = 20;
+
+/** 需要禁用 temperature 的 thinking mode 模型列表 */
+const THINKING_MODELS = [
+  'deepseek-v4-pro',
+  'deepseek-v4-flash',
+  'deepseek-reasoner',
+];
+
+// ============================================================================
+// Session 上下文
+// ============================================================================
+
+/** Session TTL：默认 30 分钟 */
+const DEFAULT_SESSION_TTL_MS = 30 * 60 * 1000;
+
+/** Session 清理检查间隔 */
+const DEFAULT_SESSION_CLEANUP_INTERVAL_MS = 60 * 1000;
+
+/** 每个 Session 最大消息数 */
+const DEFAULT_MAX_MESSAGES_PER_SESSION = 1000;
+
+// ============================================================================
+// 上下文压缩
+// ============================================================================
+
+/** 保留最近的 N 条消息 */
+const KEEP_RECENT_MESSAGES = 20;
+
+/** 智能压缩启用 */
+const ENABLE_SMART_COMPRESS = true;
+
+/** 压缩超时（毫秒） */
+const COMPRESSION_TIMEOUT_MS = 1_200_000;
+
+/** 工具结果最大字符数 */
+const MAX_TOOL_RESULT_SIZE = 4000;
+
+// ============================================================================
+// 消息队列
+// ============================================================================
+
+/** 默认最大并发数 */
+const DEFAULT_MAX_CONCURRENT = 1;
+
+/** 默认重试次数 */
+const DEFAULT_RETRY_ATTEMPTS = 3;
+
+/** 默认重试延迟（毫秒） */
+const DEFAULT_RETRY_DELAY_MS = 2000;
+
+// ============================================================================
+// System Prompt 优先级
+// ============================================================================
+
+const PROMPT_PRIORITY = {
+  DATETIME: 50,
+  ORIGINAL_PROMPT: 100,
+  SHARED_PROMPT: 200,
+  METADATA: 300,
+  TOOLS: 400,
+  SKILLS: 500,
+  SUB_AGENTS: 600,
+  CAPABILITIES: 700,
+  MCP_TOOLS: 750,
+  EXTENSION_TOOLS: 800,
+  TOOL_CORE_RULES: 95,
+};
+
+// ============================================================================
+// 熔断器
+// ============================================================================
+
+/** 熔断器默认失败阈值 */
+const DEFAULT_CIRCUIT_FAILURE_THRESHOLD = 3;
+
+/** 熔断器默认成功阈值 */
+const DEFAULT_CIRCUIT_SUCCESS_THRESHOLD = 2;
+
+/** 熔断器默认超时（毫秒） */
+const DEFAULT_CIRCUIT_TIMEOUT_MS = 60_000;
+
+// ============================================================================
+// 插件
+// ============================================================================
+
+/** 默认插件优先级 */
+const DEFAULT_PLUGIN_PRIORITY = 100;
+
+/** 系统插件名称集合 */
+const SYSTEM_PLUGINS = new Set([
+  'ai', 'defaults', 'tools', 'skill-manager', 'session',
+  'storage', 'audit', 'rules', 'file-system',
+]);
+
+/** Agent 配置目录名 */
+const AGENT_DIR = '.agent';
+
+// ============================================================================
+// Session 存储
+// ============================================================================
+
+/** 默认 Session 存储类型 */
+const DEFAULT_SESSION_STORAGE_TYPE = 'file';
+
+/** 默认 Session 存储目录 */
+const DEFAULT_SESSION_STORAGE_DIR = path.join(AGENT_DIR, 'sessions');
+
+// ============================================================================
+// Token 计数（估算系数）
+// ============================================================================
+
+/** 每字符对应的 token 数（中文系数更大） */
+const TOKENS_PER_CHAR = 0.25;
+
+/** 每工具对应的 token 数 */
+const TOKENS_PER_TOOL = 500;
+
+/** 每消息固定开销 token */
+const TOKENS_PER_MESSAGE_OVERHEAD = 4;
+
+// ============================================================================
+// 导出
+// ============================================================================
+
+module.exports = {
+  // AI
+  DEFAULT_PROVIDER,
+  DEFAULT_MODEL,
+  DEFAULT_MAX_OUTPUT_TOKENS,
+  DEFAULT_TEMPERATURE,
+  DEFAULT_MAX_STEPS,
+  THINKING_MODELS,
+
+  // Session
+  DEFAULT_SESSION_TTL_MS,
+  DEFAULT_SESSION_CLEANUP_INTERVAL_MS,
+  DEFAULT_MAX_MESSAGES_PER_SESSION,
+
+  // Compression
+  KEEP_RECENT_MESSAGES,
+  ENABLE_SMART_COMPRESS,
+  COMPRESSION_TIMEOUT_MS,
+  MAX_TOOL_RESULT_SIZE,
+
+  // Queue
+  DEFAULT_MAX_CONCURRENT,
+  DEFAULT_RETRY_ATTEMPTS,
+  DEFAULT_RETRY_DELAY_MS,
+
+  // Priority
+  PROMPT_PRIORITY,
+
+  // Circuit Breaker
+  DEFAULT_CIRCUIT_FAILURE_THRESHOLD,
+  DEFAULT_CIRCUIT_SUCCESS_THRESHOLD,
+  DEFAULT_CIRCUIT_TIMEOUT_MS,
+
+  // Plugin
+  DEFAULT_PLUGIN_PRIORITY,
+  SYSTEM_PLUGINS,
+  AGENT_DIR,
+
+  // Storage
+  DEFAULT_SESSION_STORAGE_TYPE,
+  DEFAULT_SESSION_STORAGE_DIR,
+
+  // Token
+  TOKENS_PER_CHAR,
+  TOKENS_PER_TOOL,
+  TOKENS_PER_MESSAGE_OVERHEAD,
+};

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',
   },
 };