foliko 1.1.67 → 1.1.69

package/src/core/context-manager.js

@@ -8,7 +8,6 @@
  */
 
 const { RequestContext } = require('./request-context');
-const { SessionContext } = require('./session-context');
 
 class ContextManager {
   /**

package/src/core/enhanced-context-compressor.js

@@ -0,0 +1,210 @@
+/**
+ * EnhancedContextCompressor - 增强的上下文压缩器
+ * 整合 pi 的 compaction 逻辑，提供更智能的压缩策略
+ */
+
+const {
+  ContextCompressor,
+  CompactionDetails,
+  CompactionResult,
+  prepareCompaction,
+  prepareBranchEntries,
+  collectEntriesForBranchSummary,
+  serializeConversation,
+  DEFAULT_COMPACTION_SETTINGS,
+  BRANCH_SUMMARY_PREAMBLE,
+  BRANCH_SUMMARY_PROMPT,
+  EntryTypes
+} = require('./context-compressor');
+
+const { logger } = require('../utils/logger');
+
+class EnhancedContextCompressor {
+  /**
+   * @param {Object} config
+   * @param {Framework} config.framework - Framework 实例
+   * @param {Agent} [config.agent] - Agent 实例
+   * @param {string} [config.model] - 模型名称
+   * @param {Function} [config.onCompaction] - 压缩完成回调
+   * @param {Function} [config.onBranchSummary] - 分支摘要生成回调
+   */
+  constructor(config = {}) {
+    this.framework = config.framework;
+    this.agent = config.agent;
+
+    this.settings = {
+      ...DEFAULT_COMPACTION_SETTINGS,
+      ...config.settings
+    };
+
+    this._onCompaction = config.onCompaction || null;
+    this._onBranchSummary = config.onBranchSummary || null;
+
+    // 创建底层压缩器
+    this._compressor = new ContextCompressor({
+      agent: this.agent,
+      framework: this.framework,
+      model: config.model,
+      compactionSettings: this.settings,
+      enableSmartCompress: config.enableSmartCompress !== false
+    });
+  }
+
+  /**
+   * 执行智能压缩
+   * @param {string} sessionId - Session ID
+   * @param {Array} messages - 消息数组
+   * @param {Object} messageStore - 消息存储
+   * @param {Object} [options] - 选项
+   * @returns {Promise<{didCompact: boolean, reason?: string, summary?: string}>}
+   */
+  async smartCompact(sessionId, messages, messageStore, options = {}) {
+    const {
+      reserveTokens = this.settings.reserveTokens,
+      keepRecentTokens = this.settings.keepRecentTokens,
+      minTokensToCompact = 1000
+    } = options;
+
+    const totalTokens = this._estimateTokens(messages);
+    const effectiveLimit = totalTokens - reserveTokens;
+
+    // 检查是否需要压缩
+    if (totalTokens <= effectiveLimit) {
+      return { didCompact: false, reason: 'below_threshold' };
+    }
+
+    // 计算需要压缩的 token 数
+    const tokensToCompact = totalTokens - keepRecentTokens;
+    if (tokensToCompact < minTokensToCompact) {
+      return { didCompact: false, reason: 'too_small' };
+    }
+
+    // 执行压缩
+    try {
+      await this._compressor.compress(sessionId, messages, messageStore);
+
+      if (this._onCompaction) {
+        await this._onCompaction({
+          sessionId,
+          tokensBefore: totalTokens,
+          tokensAfter: this._estimateTokens(messages),
+          messagesCompressed: tokensToCompact
+        });
+      }
+
+      return {
+        didCompact: true,
+        tokensBefore: totalTokens,
+        tokensAfter: this._estimateTokens(messages)
+      };
+    } catch (err) {
+      logger.error('[EnhancedContextCompressor] Smart compact failed:', err.message);
+      return { didCompact: false, reason: 'error', error: err.message };
+    }
+  }
+
+  /**
+   * 准备分支摘要
+   * 当切换分支时，收集需要摘要的内容
+   * @param {SessionManager} session - Session 管理器
+   * @param {string} oldLeafId - 旧叶子节点 ID
+   * @param {string} targetId - 目标节点 ID
+   * @returns {Promise<{needsSummary: boolean, summary?: string, entriesCount?: number}>}
+   */
+  async prepareBranchSummary(session, oldLeafId, targetId) {
+    const { entries, commonAncestorId } = collectEntriesForBranchSummary(
+      session,
+      oldLeafId,
+      targetId
+    );
+
+    if (entries.length === 0) {
+      return { needsSummary: false };
+    }
+
+    // 检查 token 数量是否足够生成摘要
+    const prep = prepareBranchEntries(entries, 0);
+    if (prep.totalTokens < 500) {
+      return { needsSummary: false, reason: 'too_small' };
+    }
+
+    if (this._onBranchSummary) {
+      await this._onBranchSummary({
+        entries,
+        commonAncestorId,
+        totalTokens: prep.totalTokens
+      });
+    }
+
+    return {
+      needsSummary: true,
+      entriesCount: entries.length,
+      totalTokens: prep.totalTokens
+    };
+  }
+
+  /**
+   * 生成分支摘要
+   * @param {Array} messages - 消息数组
+   * @returns {Promise<string>} 摘要文本
+   */
+  async generateBranchSummary(messages) {
+    const conversation = serializeConversation(messages);
+
+    if (!this.framework) {
+      return `[Branch summary: ${messages.length} messages]`;
+    }
+
+    const subagent = this.framework.createSubAgent({
+      name: 'branch-summarizer',
+      systemPrompt: BRANCH_SUMMARY_PROMPT,
+      disableTools: true,
+      maxRetries: 0
+    });
+
+    try {
+      const result = await subagent.chat(BRANCH_SUMMARY_PREAMBLE + conversation);
+      if (result.success) {
+        return result.message;
+      }
+      throw new Error(result.error || 'Summary generation failed');
+    } catch (err) {
+      logger.warn('[EnhancedContextCompressor] Branch summary failed:', err.message);
+      return `[Branch summary: ${messages.length} messages]`;
+    }
+  }
+
+  /**
+   * 获取压缩统计
+   */
+  getStats() {
+    return this._compressor.getStats();
+  }
+
+  /**
+   * 估算 token 数量
+   */
+  _estimateTokens(messages) {
+    let total = 0;
+    for (const msg of messages) {
+      const content = Array.isArray(msg.content)
+        ? msg.content.map(c => c.text || '').join('')
+        : (msg.content || '');
+      total += Math.ceil(content.length / 4); // 简单估算
+    }
+    return total;
+  }
+
+  /**
+   * 启用/禁用智能压缩
+   */
+  setSmartCompress(enabled) {
+    this._compressor._enableSmartCompress = enabled;
+    return this;
+  }
+}
+
+module.exports = {
+  EnhancedContextCompressor,
+  DEFAULT_COMPACTION_SETTINGS
+};