foliko 1.1.93 → 2.0.1

package/src/core/context-compressor.js

@@ -1,959 +0,0 @@
-/**
- * ContextCompressor - 上下文压缩器
- * Ported from pi's compaction/compaction.ts with enhanced functionality
- */
-
-const path = require('path');
-const fs = require('fs');
-const { logger } = require('../utils/logger');
-const {
-  estimateTokens,
-  estimateContextTokens,
-  shouldCompact,
-  getLastAssistantUsage,
-  calculateContextTokens,
-  safeJsonStringify,
-  TokenCounter
-} = require('./token-counter');
-const {
-  validateMessagesPairing,
-} = require('../utils/message-validator');
-const {
-  EntryTypes,
-  createBranchSummaryMessage,
-  createCompactionSummaryMessage,
-  createCustomMessage,
-  convertToLlm,
-  buildSessionContext
-} = require('./session-entry');
-
-// Default compaction settings
-const DEFAULT_COMPACTION_SETTINGS = {
-  enabled: true,
-  reserveTokens: 16384,
-  keepRecentTokens: 20000
-};
-
-// Model context limits
-const MODEL_CONTEXT_LIMITS = {
-  'deepseek-chat': 100000,
-  'deepseek-coder': 100000,
-  'deepseek-v4-pro': 800000,
-  'deepseek-v4-flash': 800000,
-  'deepseek-reasoner': 100000,
-  'MiniMax-M2.7': 100000,
-  'gpt-4': 100000,
-  'gpt-4o': 100000,
-  'gpt-4o-mini': 100000,
-  'gpt-4-turbo': 100000,
-  'claude-3-5-sonnet': 110000,
-  'claude-3-opus': 110000,
-  'claude-3-sonnet': 110000,
-  'claude-3-haiku': 110000,
-  'claude-4-sonnet': 110000,
-  'claude-4-opus': 110000,
-  'glm-5.1': 110000
-};
-
-// Compression timeout
-const COMPRESSION_TIMEOUT_MS = 120000;
-
-/**
- * CompactionDetails - File operation details stored on generated compaction entries
- */
-class CompactionDetails {
-  constructor(readFiles = [], modifiedFiles = []) {
-    this.readFiles = readFiles;
-    this.modifiedFiles = modifiedFiles;
-  }
-}
-
-/**
- * Result - Result of a fallible operation
- */
-class Result {
-  constructor(ok, value, error) {
-    this.ok = ok;
-    this.value = value;
-    this.error = error;
-  }
-
-  static success(value) {
-    return new Result(true, value, null);
-  }
-
-  static failure(error) {
-    return new Result(false, null, error);
-  }
-}
-
-/**
- * CompactionError
- */
-class CompactionError extends Error {
-  constructor(code, message, cause) {
-    super(message, cause === undefined ? undefined : { cause });
-    this.name = 'CompactionError';
-    this.code = code;
-  }
-}
-
-/**
- * BranchSummaryError
- */
-class BranchSummaryError extends Error {
-  constructor(code, message, cause) {
-    super(message, cause === undefined ? undefined : { cause });
-    this.name = 'BranchSummaryError';
-    this.code = code;
-  }
-}
-
-// File operations utilities
-function createFileOps() {
-  return {
-    read: new Set(),
-    written: new Set(),
-    edited: new Set()
-  };
-}
-
-function extractFileOpsFromMessage(message, fileOps) {
-  if (message.role !== 'assistant') return;
-  if (!message.content || !Array.isArray(message.content)) return;
-
-  for (const block of message.content) {
-    if (!block || block.type !== 'toolCall') continue;
-
-    const args = block.arguments;
-    if (!args) continue;
-
-    const filePath = typeof args.path === 'string' ? args.path : undefined;
-    if (!filePath) continue;
-
-    switch (block.name) {
-      case 'read':
-        fileOps.read.add(filePath);
-        break;
-      case 'write':
-        fileOps.written.add(filePath);
-        break;
-      case 'edit':
-        fileOps.edited.add(filePath);
-        break;
-    }
-  }
-}
-
-function computeFileLists(fileOps) {
-  const modified = new Set([...fileOps.edited, ...fileOps.written]);
-  const readOnly = [...fileOps.read].filter(f => !modified.has(f)).sort();
-  const modifiedFiles = [...modified].sort();
-  return { readFiles: readOnly, modifiedFiles };
-}
-
-function formatFileOperations(readFiles, modifiedFiles) {
-  const sections = [];
-  if (readFiles.length > 0) {
-    sections.push(`<read-files>\n${readFiles.join('\n')}\n</read-files>`);
-  }
-  if (modifiedFiles.length > 0) {
-    sections.push(`<modified-files>\n${modifiedFiles.join('\n')}\n</modified-files>`);
-  }
-  if (sections.length === 0) return '';
-  return `\n\n${sections.join('\n\n')}`;
-}
-
-const TOOL_RESULT_MAX_CHARS = 2000;
-
-function truncateForSummary(text, maxChars) {
-  if (text.length <= maxChars) return text;
-  const truncatedChars = text.length - maxChars;
-  return `${text.slice(0, maxChars)}\n\n[... ${truncatedChars} more characters truncated]`;
-}
-
-function serializeConversation(messages) {
-  const parts = [];
-
-  for (const msg of messages) {
-    if (msg.role === 'user') {
-      const content = Array.isArray(msg.content)
-        ? msg.content.filter(c => c.type === 'text').map(c => c.text).join('')
-        : msg.content;
-      if (content) parts.push(`[User]: ${content}`);
-    } else if (msg.role === 'assistant') {
-      const textParts = [];
-      const thinkingParts = [];
-      const toolCalls = [];
-
-      for (const block of msg.content || []) {
-        if (block.type === 'text') {
-          textParts.push(block.text);
-        } else if (block.type === 'thinking') {
-          thinkingParts.push(block.thinking);
-        } else if (block.type === 'toolCall') {
-          const argsStr = Object.entries(block.arguments || {})
-            .map(([k, v]) => `${k}=${safeJsonStringify(v)}`)
-            .join(', ');
-          toolCalls.push(`${block.name}(${argsStr})`);
-        }
-      }
-
-      if (thinkingParts.length > 0) {
-        parts.push(`[Assistant thinking]: ${thinkingParts.join('\n')}`);
-      }
-      if (textParts.length > 0) {
-        parts.push(`[Assistant]: ${textParts.join('\n')}`);
-      }
-      if (toolCalls.length > 0) {
-        parts.push(`[Assistant tool calls]: ${toolCalls.join('; ')}`);
-      }
-    } else if (msg.role === 'toolResult') {
-      const content = Array.isArray(msg.content)
-        ? msg.content.filter(c => c.type === 'text').map(c => c.text).join('')
-        : msg.content;
-      if (content) {
-        parts.push(`[Tool result]: ${truncateForSummary(content, TOOL_RESULT_MAX_CHARS)}`);
-      }
-    }
-  }
-
-  return parts.join('\n\n');
-}
-
-function getMessageFromEntry(entry) {
-  switch (entry.type) {
-    case EntryTypes.MESSAGE:
-      if (entry.message.role === 'toolResult') return undefined;
-      return entry.message;
-    case EntryTypes.CUSTOM_MESSAGE:
-      return createCustomMessage(entry.customType, entry.content, entry.display, entry.details, entry.timestamp);
-    case EntryTypes.BRANCH_SUMMARY:
-      return createBranchSummaryMessage(entry.summary, entry.fromId, entry.timestamp);
-    case EntryTypes.COMPACTION:
-      return createCompactionSummaryMessage(entry.summary, entry.tokensBefore, entry.timestamp);
-    default:
-      return undefined;
-  }
-}
-
-function getMessageFromEntryForCompaction(entry) {
-  if (entry.type === EntryTypes.COMPACTION) {
-    return undefined;
-  }
-  return getMessageFromEntry(entry);
-}
-
-function findValidCutPoints(entries, startIndex, endIndex) {
-  const cutPoints = [];
-  for (let i = startIndex; i < endIndex; i++) {
-    const entry = entries[i];
-    if (entry.type === EntryTypes.MESSAGE) {
-      const role = entry.message.role;
-      switch (role) {
-        case 'bashExecution':
-        case 'custom':
-        case 'branchSummary':
-        case 'compactionSummary':
-        case 'user':
-        case 'assistant':
-          cutPoints.push(i);
-          break;
-        case 'toolResult':
-          break;
-      }
-    }
-    if (entry.type === EntryTypes.BRANCH_SUMMARY || entry.type === EntryTypes.CUSTOM_MESSAGE) {
-      cutPoints.push(i);
-    }
-  }
-  return cutPoints;
-}
-
-function findTurnStartIndex(entries, entryIndex, startIndex) {
-  for (let i = entryIndex; i >= startIndex; i--) {
-    const entry = entries[i];
-    if (entry.type === EntryTypes.BRANCH_SUMMARY || entry.type === EntryTypes.CUSTOM_MESSAGE) {
-      return i;
-    }
-    if (entry.type === EntryTypes.MESSAGE) {
-      const role = entry.message.role;
-      if (role === 'user' || role === 'bashExecution') {
-        return i;
-      }
-    }
-  }
-  return -1;
-}
-
-/**
- * Find the compaction cut point that keeps approximately the requested recent-token budget
- */
-function findCutPoint(entries, startIndex, endIndex, keepRecentTokens) {
-  const cutPoints = findValidCutPoints(entries, startIndex, endIndex);
-
-  if (cutPoints.length === 0) {
-    return { firstKeptEntryIndex: startIndex, turnStartIndex: -1, isSplitTurn: false };
-  }
-  let accumulatedTokens = 0;
-  let cutIndex = cutPoints[0];
-
-  for (let i = endIndex - 1; i >= startIndex; i--) {
-    const entry = entries[i];
-    if (entry.type !== EntryTypes.MESSAGE) continue;
-    const messageTokens = estimateTokens(entry.message);
-    accumulatedTokens += messageTokens;
-    if (accumulatedTokens >= keepRecentTokens) {
-      for (let c = 0; c < cutPoints.length; c++) {
-        if (cutPoints[c] >= i) {
-          cutIndex = cutPoints[c];
-          break;
-        }
-      }
-      break;
-    }
-  }
-  while (cutIndex > startIndex) {
-    const prevEntry = entries[cutIndex - 1];
-    if (prevEntry.type === EntryTypes.COMPACTION) break;
-    if (prevEntry.type === EntryTypes.MESSAGE) break;
-    cutIndex--;
-  }
-  const cutEntry = entries[cutIndex];
-  const isUserMessage = cutEntry.type === EntryTypes.MESSAGE && cutEntry.message.role === 'user';
-  const turnStartIndex = isUserMessage ? -1 : findTurnStartIndex(entries, cutIndex, startIndex);
-
-  return {
-    firstKeptEntryIndex: cutIndex,
-    turnStartIndex,
-    isSplitTurn: !isUserMessage && turnStartIndex !== -1
-  };
-}
-
-const SUMMARIZATION_SYSTEM_PROMPT = `You are a context summarization assistant. Your task is to read a conversation between a user and an AI coding assistant, then produce a structured summary following the exact format specified.
-
-Do NOT continue the conversation. Do NOT respond to any questions in the conversation. ONLY output the structured summary.`;
-
-const SUMMARIZATION_PROMPT = `The messages above are a conversation to summarize. Create a structured context checkpoint summary that another LLM will use to continue the work.
-
-Use this EXACT format:
-
-## Goal
-[What is the user trying to accomplish? Can be multiple items if the session covers different tasks.]
-
-## Constraints & Preferences
-- [Any constraints, preferences, or requirements mentioned by user]
-- [Or "(none)" if none were mentioned]
-
-## Progress
-### Done
-- [x] [Completed tasks/changes]
-
-### In Progress
-- [ ] [Current work]
-
-### Blocked
-- [Issues preventing progress, if any]
-
-## Key Decisions
-- **[Decision]**: [Brief rationale]
-
-## Next Steps
-1. [Ordered list of what should happen next]
-
-## Critical Context
-- [Any data, examples, or references needed to continue]
-- [Or "(none)" if not applicable]
-
-Keep each section concise. Preserve exact file paths, function names, and error messages.`;
-
-const UPDATE_SUMMARIZATION_PROMPT = `The messages above are NEW conversation messages to incorporate into the existing summary provided in <previous-summary> tags.
-
-Update the existing structured summary with new information. RULES:
-- PRESERVE all existing information from the previous summary
-- ADD new progress, decisions, and context from the new messages
-- UPDATE the Progress section: move items from "In Progress" to "Done" when completed
-- UPDATE "Next Steps" based on what was accomplished
-- PRESERVE exact file paths, function names, and error messages
-- If something is no longer relevant, you may remove it
-
-Use this EXACT format:
-
-## Goal
-[Preserve existing goals, add new ones if the task expanded]
-
-## Constraints & Preferences
-- [Preserve existing, add new ones discovered]
-
-## Progress
-### Done
-- [x] [Include previously done items AND newly completed items]
-
-### In Progress
-- [ ] [Current work - update based on progress]
-
-### Blocked
-- [Current blockers - remove if resolved]
-
-## Key Decisions
-- **[Decision]**: [Brief rationale] (preserve all previous, add new)
-
-## Next Steps
-1. [Update based on current state]
-
-## Critical Context
-- [Preserve important context, add new if needed]
-
-Keep each section concise. Preserve exact file paths, function names, and error messages.`;
-
-const TURN_PREFIX_SUMMARIZATION_PROMPT = `This is the PREFIX of a turn that was too large to keep. The SUFFIX (recent work) is retained.
-
-Summarize the prefix to provide context for the retained suffix:
-
-## Original Request
-[What did the user ask for in this turn?]
-
-## Early Progress
-- [Key decisions and work done in the prefix]
-
-## Context for Suffix
-- [Information needed to understand the retained recent work]
-
-Be concise. Focus on what's needed to understand the kept suffix.`;
-
-/**
- * Extract file operations from messages and previous compaction
- */
-function extractFileOperations(messages, entries, prevCompactionIndex) {
-  const fileOps = createFileOps();
-  if (prevCompactionIndex >= 0) {
-    const prevCompaction = entries[prevCompactionIndex];
-    if (!prevCompaction.fromHook && prevCompaction.details) {
-      const details = prevCompaction.details;
-      if (Array.isArray(details.readFiles)) {
-        for (const f of details.readFiles) fileOps.read.add(f);
-      }
-      if (Array.isArray(details.modifiedFiles)) {
-        for (const f of details.modifiedFiles) {
-          fileOps.edited.add(f);
-        }
-      }
-    }
-  }
-  for (const msg of messages) {
-    extractFileOpsFromMessage(msg, fileOps);
-  }
-  return fileOps;
-}
-
-/**
- * CompactionResult - Prepared compaction data ready to be persisted
- */
-class CompactionResult {
-  constructor(summary, firstKeptEntryId, tokensBefore, details) {
-    this.summary = summary;
-    this.firstKeptEntryId = firstKeptEntryId;
-    this.tokensBefore = tokensBefore;
-    this.details = details;
-  }
-}
-
-/**
- * Prepare session entries for compaction
- */
-function prepareCompaction(pathEntries, settings) {
-  if (pathEntries.length === 0 || pathEntries[pathEntries.length - 1].type === EntryTypes.COMPACTION) {
-    return Result.success(undefined);
-  }
-
-  let prevCompactionIndex = -1;
-  for (let i = pathEntries.length - 1; i >= 0; i--) {
-    if (pathEntries[i].type === EntryTypes.COMPACTION) {
-      prevCompactionIndex = i;
-      break;
-    }
-  }
-
-  let previousSummary;
-  let boundaryStart = 0;
-  if (prevCompactionIndex >= 0) {
-    const prevCompaction = pathEntries[prevCompactionIndex];
-    previousSummary = prevCompaction.summary;
-    const firstKeptEntryIndex = pathEntries.findIndex(entry => entry.id === prevCompaction.firstKeptEntryId);
-    boundaryStart = firstKeptEntryIndex >= 0 ? firstKeptEntryIndex : prevCompactionIndex + 1;
-  }
-  const boundaryEnd = pathEntries.length;
-
-  const tokensBefore = estimateContextTokens(buildSessionContext(pathEntries).messages).tokens;
-
-  const cutPoint = findCutPoint(pathEntries, boundaryStart, boundaryEnd, settings.keepRecentTokens);
-  const firstKeptEntry = pathEntries[cutPoint.firstKeptEntryIndex];
-  if (!firstKeptEntry?.id) {
-    return Result.failure(new CompactionError('invalid_session', 'First kept entry has no UUID'));
-  }
-  const firstKeptEntryId = firstKeptEntry.id;
-
-  const historyEnd = cutPoint.isSplitTurn ? cutPoint.turnStartIndex : cutPoint.firstKeptEntryIndex;
-  const messagesToSummarize = [];
-  for (let i = boundaryStart; i < historyEnd; i++) {
-    const msg = getMessageFromEntryForCompaction(pathEntries[i]);
-    if (msg) messagesToSummarize.push(msg);
-  }
-  const turnPrefixMessages = [];
-  if (cutPoint.isSplitTurn) {
-    for (let i = cutPoint.turnStartIndex; i < cutPoint.firstKeptEntryIndex; i++) {
-      const msg = getMessageFromEntryForCompaction(pathEntries[i]);
-      if (msg) turnPrefixMessages.push(msg);
-    }
-  }
-  const fileOps = extractFileOperations(messagesToSummarize, pathEntries, prevCompactionIndex);
-  if (cutPoint.isSplitTurn) {
-    for (const msg of turnPrefixMessages) {
-      extractFileOpsFromMessage(msg, fileOps);
-    }
-  }
-
-  return Result.success({
-    firstKeptEntryId,
-    messagesToSummarize,
-    turnPrefixMessages,
-    isSplitTurn: cutPoint.isSplitTurn,
-    tokensBefore,
-    previousSummary,
-    fileOps,
-    settings
-  });
-}
-
-// Branch summarization utilities
-const BRANCH_SUMMARY_PREAMBLE = `The user explored a different conversation branch before returning here.
-Summary of that exploration:
-
-`;
-
-const BRANCH_SUMMARY_PROMPT = `Create a structured summary of this conversation branch for context when returning later.
-
-Use this EXACT format:
-
-## Goal
-[What was the user trying to accomplish in this branch?]
-
-## Constraints & Preferences
-- [Any constraints, preferences, or requirements mentioned]
-- [Or "(none)" if none were mentioned]
-
-## Progress
-### Done
-- [x] [Completed tasks/changes]
-
-### In Progress
-- [ ] [Work that was started but not finished]
-
-### Blocked
-- [Issues preventing progress, if any]
-
-## Key Decisions
-- **[Decision]**: [Brief rationale]
-
-## Next Steps
-1. [What should happen next to continue this work]
-
-Keep each section concise. Preserve exact file paths, function names, and error messages.`;
-
-/**
- * BranchPreparation - Prepared branch content for summarization
- */
-class BranchPreparation {
-  constructor(messages, fileOps, totalTokens) {
-    this.messages = messages;
-    this.fileOps = fileOps;
-    this.totalTokens = totalTokens;
-  }
-}
-
-/**
- * Prepare branch entries for summarization within an optional token budget
- */
-function prepareBranchEntries(entries, tokenBudget = 0) {
-  const messages = [];
-  const fileOps = createFileOps();
-  let totalTokens = 0;
-
-  for (const entry of entries) {
-    if (entry.type === EntryTypes.BRANCH_SUMMARY && !entry.fromHook && entry.details) {
-      const details = entry.details;
-      if (Array.isArray(details.readFiles)) {
-        for (const f of details.readFiles) fileOps.read.add(f);
-      }
-      if (Array.isArray(details.modifiedFiles)) {
-        for (const f of details.modifiedFiles) {
-          fileOps.edited.add(f);
-        }
-      }
-    }
-  }
-
-  for (let i = entries.length - 1; i >= 0; i--) {
-    const entry = entries[i];
-    const message = getMessageFromEntry(entry);
-    if (!message) continue;
-    extractFileOpsFromMessage(message, fileOps);
-
-    const tokens = estimateTokens(message);
-    if (tokenBudget > 0 && totalTokens + tokens > tokenBudget) {
-      if (entry.type === EntryTypes.COMPACTION || entry.type === EntryTypes.BRANCH_SUMMARY) {
-        if (totalTokens < tokenBudget * 0.9) {
-          messages.unshift(message);
-          totalTokens += tokens;
-        }
-      }
-      break;
-    }
-
-    messages.unshift(message);
-    totalTokens += tokens;
-  }
-
-  return new BranchPreparation(messages, fileOps, totalTokens);
-}
-
-/**
- * Collect entries that should be summarized before navigating to a different session tree entry
- */
-function collectEntriesForBranchSummary(session, oldLeafId, targetId) {
-  if (!oldLeafId) {
-    return { entries: [], commonAncestorId: null };
-  }
-  const oldPath = new Set(session.getBranch(oldLeafId).map(e => e.id));
-  const targetPath = session.getBranch(targetId);
-  let commonAncestorId = null;
-  for (let i = targetPath.length - 1; i >= 0; i--) {
-    if (oldPath.has(targetPath[i].id)) {
-      commonAncestorId = targetPath[i].id;
-      break;
-    }
-  }
-  const entries = [];
-  let current = oldLeafId;
-
-  while (current && current !== commonAncestorId) {
-    const entry = session.getEntry(current);
-    if (!entry) break;
-    entries.push(entry);
-    current = entry.parentId;
-  }
-  entries.reverse();
-
-  return { entries, commonAncestorId };
-}
-
-/**
- * ContextCompressor - Main compression class
- */
-class ContextCompressor {
-  constructor(config = {}) {
-    this.config = config;
-    this.agent = config.agent;
-    this.framework = config.framework;
-
-    this.model = config.model || 'deepseek-chat';
-    this._maxContextTokens = config.maxContextTokens || this._getDefaultContextLimit();
-    this._keepRecentMessages = config.keepRecentMessages || 20;
-    // 消息数量阈值：超过该数量即触发压缩（与 token 阈值并联触发）
-    this._compressionMessageThreshold = config.compressionMessageThreshold || 200;
-    this._enableSmartCompress = config.enableSmartCompress !== false;
-    this._compactionSettings = config.compactionSettings || DEFAULT_COMPACTION_SETTINGS;
-
-    this._compressionCount = 0;
-    this._compressionInProgress = false;
-    this._compressionPromise = null;
-    this._compressionTimeoutId = null;
-    this._maxToolResultSize = config.maxToolResultSize || 4000;
-    this._tokenCounter = new TokenCounter();
-  }
-
-  _getDefaultContextLimit() {
-    const modelKey = Object.keys(MODEL_CONTEXT_LIMITS).find(k =>
-      this.model.toLowerCase().includes(k.toLowerCase())
-    );
-    return modelKey ? MODEL_CONTEXT_LIMITS[modelKey] : 40000;
-  }
-
-  /**
-   * Validate message pairing (delegate to message-validator)
-   */
-  validateMessagesPairing(messages) {
-    return validateMessagesPairing(messages);
-  }
-
-  async compress(sessionId, messages, messageStore) {
-    if (this._compressionInProgress && this._compressionPromise) {
-      logger.debug('Compression already in progress, waiting...');
-      return this._compressionPromise;
-    }
-
-    if (messages.length <= this._keepRecentMessages) {
-      return;
-    }
-
-    this._compressionInProgress = true;
-
-    this._compressionPromise = this._executeWithTimeout(sessionId, messages, messageStore).finally(() => {
-      this._compressionInProgress = false;
-      this._compressionPromise = null;
-      if (this._compressionTimeoutId) {
-        clearTimeout(this._compressionTimeoutId);
-        this._compressionTimeoutId = null;
-      }
-    });
-
-    return this._compressionPromise;
-  }
-
-  async _executeWithTimeout(sessionId, messages, messageStore) {
-    try {
-      return await Promise.race([
-        this._doCompress(sessionId, messages, messageStore),
-        this._createTimeoutPromise()
-      ]);
-    } catch (err) {
-      logger.warn('Compression failed:', err.message);
-      this._simpleCompress(sessionId, messages, messageStore);
-    }
-  }
-
-  _createTimeoutPromise() {
-    return new Promise((_, reject) => {
-      this._compressionTimeoutId = setTimeout(() => {
-        this._compressionTimeoutId = null;
-        reject(new Error(`Compression timeout (${COMPRESSION_TIMEOUT_MS}ms)`));
-      }, COMPRESSION_TIMEOUT_MS);
-    });
-  }
-
-  cancelCompression() {
-    if (this._compressionTimeoutId) {
-      clearTimeout(this._compressionTimeoutId);
-      this._compressionTimeoutId = null;
-    }
-    this._compressionInProgress = false;
-    this._compressionPromise = null;
-  }
-
-  async _doCompress(sessionId, messages, messageStore) {
-    const systemMessages = messages.filter(m => m.role === 'system');
-    const otherMessages = messages.filter(m => m.role !== 'system');
-    const recentMessages = otherMessages.slice(-this._keepRecentMessages);
-    const messagesToSummarize = otherMessages.slice(0, -this._keepRecentMessages);
-
-    const compressedCount = messagesToSummarize.length;
-    let summaryContent = '';
-
-    if (this._enableSmartCompress && this.agent?._chatHandler?._aiClient) {
-      try {
-        const summaryText = await this._summarizeMessages(messagesToSummarize);
-        summaryContent = `[Early conversation summary]: ${summaryText || '(no content)'}`;
-      } catch (err) {
-        logger.warn('AI summary failed, using simple compression:', err.message);
-        summaryContent = `[Context compressed: ${compressedCount} early messages omitted. Kept recent ${this._keepRecentMessages} messages.]`;
-      }
-    } else {
-      summaryContent = `[Context compressed: ${compressedCount} early messages omitted. Kept recent ${this._keepRecentMessages} messages.]`;
-    }
-
-    const summary = {
-      role: 'assistant',
-      content: summaryContent
-    };
-
-    const combined = [...systemMessages, summary, ...recentMessages];
-    this._cleanupOrphanedToolResults(combined);
-
-    messages.length = 0;
-    messages.push(...combined);
-
-    this._compressionCount++;
-    const tokenCount = this._tokenCounter.countMessages(messages);
-    if (messageStore.compressionState) {
-      messageStore.compressionState.count++;
-      messageStore.compressionState.lastCompressedAt = Date.now();
-      messageStore.compressionState.lastTokenCount = tokenCount;
-    }
-    if (typeof messageStore.recordCompression === 'function') {
-      messageStore.recordCompression(tokenCount);
-    }
-
-    logger.info(`Context compressed (${this._compressionCount} times). Messages: ${messages.length}`);
-  }
-
-  _simpleCompress(sessionId, messages, messageStore) {
-    const systemMessages = messages.filter(m => m.role === 'system');
-    const otherMessages = messages.filter(m => m.role !== 'system');
-    const recentMessages = otherMessages.slice(-this._keepRecentMessages);
-    const compressedCount = otherMessages.length - this._keepRecentMessages;
-
-    const summaryContent = `[Context compressed: ${compressedCount} early messages omitted. Kept recent ${this._keepRecentMessages} messages.]`;
-
-    const summary = {
-      role: 'assistant',
-      content: summaryContent
-    };
-
-    const combined = [...systemMessages, summary, ...recentMessages];
-    this._cleanupOrphanedToolResults(combined);
-
-    messages.length = 0;
-    messages.push(...combined);
-
-    this._compressionCount++;
-    if (messageStore.compressionState) {
-      messageStore.compressionState.count++;
-      messageStore.compressionState.lastCompressedAt = Date.now();
-      messageStore.compressionState.lastTokenCount = this._tokenCounter.countMessages(combined);
-    }
-
-    logger.info(`Context simple compressed. Messages: ${messages.length}`);
-  }
-
-  /**
-   * 清理孤立的 tool-result 消息（没有对应 tool-call 的）
-   * 压缩/裁剪后 recentMessages 可能切碎 tool-call/tool-result 配对，
-   * 必须就地清理，否则下游 API 会报 "tool result's tool id not found"
-   */
-  _cleanupOrphanedToolResults(messages) {
-    const validToolCallIds = new Set();
-    for (const msg of messages) {
-      if (msg.role !== 'assistant') continue;
-      if (Array.isArray(msg.content)) {
-        for (const item of msg.content) {
-          if ((item.type === 'tool-call' || item.type === 'tool-use') && item.toolCallId) {
-            validToolCallIds.add(item.toolCallId);
-          }
-        }
-      }
-      if (Array.isArray(msg.tool_calls)) {
-        for (const tc of msg.tool_calls) {
-          if (tc.id) validToolCallIds.add(tc.id);
-        }
-      }
-    }
-
-    let removedItems = 0;
-    let removedMsgs = 0;
-    for (const msg of messages) {
-      if (msg.role !== 'tool' || !Array.isArray(msg.content)) continue;
-      const originalLength = msg.content.length;
-      msg.content = msg.content.filter((item) => {
-        if (
-          item &&
-          (item.type === 'tool-result' || item.type === 'tool_result') &&
-          item.toolCallId &&
-          !validToolCallIds.has(item.toolCallId)
-        ) {
-          removedItems++;
-          return false;
-        }
-        return true;
-      });
-      if (msg.content.length === 0 && originalLength > 0) {
-        msg._orphaned = true;
-      }
-    }
-
-    for (let i = messages.length - 1; i >= 0; i--) {
-      if (messages[i]._orphaned) {
-        messages.splice(i, 1);
-        removedMsgs++;
-      }
-    }
-
-    if (removedItems > 0 || removedMsgs > 0) {
-      logger.debug(
-        `[ContextCompressor] cleanup: removed ${removedItems} orphaned tool-result items, ${removedMsgs} orphaned tool messages`
-      );
-    }
-  }
-
-  async _summarizeMessages(messages) {
-    if (!this.framework) {
-      throw new Error('Framework not available');
-    }
-
-    const msg_str = messages
-      .map(m => {
-        let text = '';
-        if (typeof m.content === 'string') {
-          text = m.content;
-        } else if (Array.isArray(m.content)) {
-          text = m.content
-            .filter(c => c.type === 'text')
-            .map(c => c.text)
-            .join(' ');
-        }
-        if (m.role === 'tool') {
-          return '';
-        }
-        return `${m.role}: ${text}`;
-      })
-      .filter(line => line.length > 0)
-      .join('\n');
-
-    const task = `Summarize the following conversation, keeping only meaningful information (task requirements, decisions, important context). Ignore meaningless chatter and repetitive content. Describe in 1000 characters or less:\n\n${msg_str}`;
-
-    try {
-      const subagent = this.framework.createSubAgent({
-        name: 'context-compressor',
-        role: 'Information extraction expert',
-        systemPrompt: 'You are a conversation summarization assistant. Extract and preserve only meaningful information (task requirements, decisions, important context). Ignore meaningless chatter, repetitive content, and intermediate processes. Keep output concise.',
-        maxRetries: 0,
-        disableTools: true
-      });
-
-      const result = await subagent.chat(task);
-      if (result.success) {
-        return result.message;
-      }
-      throw new Error(result.error || 'Summary generation failed');
-    } catch (err) {
-      logger.warn('Summarize failed:', err.message);
-      throw err;
-    }
-  }
-
-  getStats() {
-    return {
-      compressionCount: this._compressionCount,
-      inProgress: this._compressionInProgress
-    };
-  }
-}
-
-module.exports = {
-  ContextCompressor,
-  CompactionDetails,
-  CompactionResult,
-  CompactionError,
-  BranchSummaryError,
-  Result,
-  DEFAULT_COMPACTION_SETTINGS,
-  createFileOps,
-  extractFileOpsFromMessage,
-  computeFileLists,
-  formatFileOperations,
-  serializeConversation,
-  findCutPoint,
-  findTurnStartIndex,
-  SUMMARIZATION_SYSTEM_PROMPT,
-  SUMMARIZATION_PROMPT,
-  UPDATE_SUMMARIZATION_PROMPT,
-  TURN_PREFIX_SUMMARIZATION_PROMPT,
-  BRANCH_SUMMARY_PREAMBLE,
-  BRANCH_SUMMARY_PROMPT,
-  extractFileOperations,
-  prepareCompaction,
-  prepareBranchEntries,
-  collectEntriesForBranchSummary,
-  getMessageFromEntry,
-  getMessageFromEntryForCompaction,
-  EntryTypes
-};