foliko 1.1.93 → 2.0.0

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

@@ -1,210 +0,0 @@
-/**
- * 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
-};