axon-code 2.4.0

package/dist/context/index.js

@@ -0,0 +1,1043 @@
+/**
+ * 上下文管理系统
+ * 处理对话历史、上下文窗口和自动摘要
+ *
+ * 功能：
+ * 1. 智能消息摘要（使用 Claude 生成摘要）
+ * 2. 保留最近 N 条消息不压缩
+ * 3. 工具调用结果智能压缩
+ * 4. 代码块智能保留
+ * 5. 文件内容引用压缩
+ * 6. 精确的压缩比计算
+ * 7. 增量压缩支持
+ */
+export { estimateTokens } from '../utils/token-estimate.js';
+import { estimateTokens } from '../utils/token-estimate.js';
+// Token 估算常量
+const CHARS_PER_TOKEN = 3.5; // 更精确的估算（英文约4，中文约2）
+const MAX_CONTEXT_TOKENS = 180000; // Claude 3.5 的上下文窗口
+const RESERVE_TOKENS = 32000; // 保留给输出
+// 压缩配置常量
+const CODE_BLOCK_MAX_LINES = 50; // 代码块最大保留行数
+const TOOL_OUTPUT_MAX_CHARS = 2000; // 工具输出最大字符数
+const FILE_CONTENT_MAX_CHARS = 1500; // 文件内容最大字符数
+const SUMMARY_TARGET_RATIO = 0.3; // 摘要目标压缩比
+/**
+ * 估算 tool_result.content 的 token 数
+ * 正确处理 content 为数组（包含 image/text block）的情况，
+ * 避免对 base64 图片数据做 JSON.stringify 导致 token 估算膨胀
+ */
+function estimateToolResultTokens(content) {
+    if (!content)
+        return 0;
+    if (typeof content === 'string') {
+        return estimateTokens(content);
+    }
+    // content 是数组，逐个 block 处理
+    let total = 0;
+    for (const item of content) {
+        if (item.type === 'image') {
+            total += 2000; // 图片固定常量
+        }
+        else if (item.type === 'text') {
+            total += estimateTokens(item.text || '');
+        }
+        else {
+            total += estimateTokens(JSON.stringify(item));
+        }
+    }
+    return total;
+}
+/**
+ * 从 tool_result.content 中提取纯文本内容
+ * 对数组 content 中的 image 块用占位符替代，避免将 base64 混入文本流
+ */
+function extractToolResultText(content) {
+    if (!content)
+        return '';
+    if (typeof content === 'string')
+        return content;
+    const parts = [];
+    let imageCount = 0;
+    for (const item of content) {
+        if (item.type === 'image') {
+            imageCount++;
+        }
+        else if (item.type === 'text') {
+            parts.push(item.text || '');
+        }
+    }
+    if (imageCount > 0) {
+        parts.push(`[${imageCount} image${imageCount > 1 ? 's' : ''}]`);
+    }
+    return parts.join('\n');
+}
+/**
+ * 估算消息 token 数
+ */
+export function estimateMessageTokens(message) {
+    if (typeof message.content === 'string') {
+        return estimateTokens(message.content) + 10; // 额外的消息开销
+    }
+    let total = 10; // 消息开销
+    for (const block of message.content) {
+        if (block.type === 'text') {
+            total += estimateTokens(block.text || '');
+        }
+        else if (block.type === 'tool_use') {
+            total += estimateTokens(block.name || '') + estimateTokens(JSON.stringify(block.input));
+        }
+        else if (block.type === 'tool_result') {
+            total += estimateToolResultTokens(block.content);
+        }
+        else if (block.type === 'image') {
+            // 图片按固定大小估算（对齐官方 Nr4=2000）
+            total += 2000;
+        }
+    }
+    return total;
+}
+/**
+ * 估算消息数组的总 token 数
+ */
+export function estimateTotalTokens(messages) {
+    return messages.reduce((sum, msg) => sum + estimateMessageTokens(msg), 0);
+}
+/**
+ * 检测代码块
+ */
+function extractCodeBlocks(text) {
+    const blocks = [];
+    const regex = /```(\w+)?\n([\s\S]*?)```/g;
+    let match;
+    while ((match = regex.exec(text)) !== null) {
+        blocks.push({
+            language: match[1],
+            code: match[2],
+            start: match.index,
+            end: match.index + match[0].length,
+        });
+    }
+    return blocks;
+}
+/**
+ * 压缩代码块（保留关键部分）
+ */
+function compressCodeBlock(code, maxLines = CODE_BLOCK_MAX_LINES) {
+    const lines = code.split('\n');
+    if (lines.length <= maxLines) {
+        return code;
+    }
+    // 保留开头和结尾
+    const keepHead = Math.floor(maxLines * 0.6);
+    const keepTail = Math.floor(maxLines * 0.4);
+    const head = lines.slice(0, keepHead).join('\n');
+    const tail = lines.slice(-keepTail).join('\n');
+    const omitted = lines.length - maxLines;
+    return `${head}\n\n... [${omitted} lines omitted] ...\n\n${tail}`;
+}
+/**
+ * 压缩工具输出
+ */
+function compressToolOutput(content, maxChars = TOOL_OUTPUT_MAX_CHARS) {
+    if (content.length <= maxChars) {
+        return content;
+    }
+    // 检测是否包含代码块
+    const codeBlocks = extractCodeBlocks(content);
+    if (codeBlocks.length > 0) {
+        // 如果有代码块，优先保留代码
+        let result = content;
+        for (const block of codeBlocks) {
+            const compressed = compressCodeBlock(block.code);
+            const marker = block.language ? `\`\`\`${block.language}` : '```';
+            result = result.replace(`${marker}\n${block.code}\`\`\``, `${marker}\n${compressed}\`\`\``);
+        }
+        if (result.length <= maxChars) {
+            return result;
+        }
+    }
+    // 检测是否是文件内容
+    if (content.includes('→') || /^\s*\d+\s*[│|]/.test(content)) {
+        // 看起来是文件列表或文件内容，保留头尾
+        const lines = content.split('\n');
+        const keepHead = 20;
+        const keepTail = 10;
+        if (lines.length > keepHead + keepTail) {
+            const head = lines.slice(0, keepHead).join('\n');
+            const tail = lines.slice(-keepTail).join('\n');
+            const omitted = lines.length - keepHead - keepTail;
+            return `${head}\n... [${omitted} lines omitted] ...\n${tail}`;
+        }
+    }
+    // 默认：简单截断
+    const keepHead = Math.floor(maxChars * 0.7);
+    const keepTail = Math.floor(maxChars * 0.3);
+    const head = content.slice(0, keepHead);
+    const tail = content.slice(-keepTail);
+    const omitted = content.length - maxChars;
+    return `${head}\n\n... [~${omitted} chars omitted] ...\n\n${tail}`;
+}
+/**
+ * 检测文件路径引用
+ */
+function extractFileReferences(text) {
+    // 匹配绝对路径（Unix 和 Windows）
+    // Unix: /path/to/file.ext
+    // Windows: C:\path\to\file.ext 或 C:/path/to/file.ext
+    const pathRegex = /(?:(?:\/[\w\-_.]+)+\.\w+|[A-Za-z]:[\\/](?:[\w\-_.]+[\\/])*[\w\-_.]+\.\w+)/g;
+    const matches = text.match(pathRegex);
+    if (!matches) {
+        return [];
+    }
+    // 去重
+    const seen = new Set();
+    const refs = [];
+    for (const match of matches) {
+        if (!seen.has(match)) {
+            seen.add(match);
+            refs.push(match);
+        }
+    }
+    return refs;
+}
+/**
+ * 提取消息的核心内容用于摘要
+ */
+function extractMessageCore(message) {
+    if (typeof message.content === 'string') {
+        return message.content;
+    }
+    const parts = [];
+    for (const block of message.content) {
+        if (block.type === 'text') {
+            parts.push(block.text || '');
+        }
+        else if (block.type === 'tool_use') {
+            const inputStr = JSON.stringify(block.input, null, 2);
+            parts.push(`[Tool: ${block.name || 'unknown'}]\nInput: ${inputStr.slice(0, 200)}`);
+        }
+        else if (block.type === 'tool_result') {
+            const textContent = extractToolResultText(block.content);
+            const compressed = compressToolOutput(textContent, 300);
+            parts.push(`[Result: ${block.tool_use_id || 'unknown'}]\n${compressed}`);
+        }
+        else if (block.type === 'image') {
+            parts.push('[image]');
+        }
+    }
+    return parts.join('\n\n');
+}
+/**
+ * 创建对话摘要（简单版本，不使用 AI）
+ */
+export function createSummary(turns) {
+    const summaryParts = ['=== Previous Conversation Summary ===\n'];
+    for (const turn of turns) {
+        const userContent = extractMessageCore(turn.user);
+        const assistantContent = extractMessageCore(turn.assistant);
+        // 提取关键信息
+        const userSummary = userContent.slice(0, 300);
+        const assistantSummary = assistantContent.slice(0, 400);
+        // 提取文件引用
+        const fileRefs = [
+            ...extractFileReferences(userContent),
+            ...extractFileReferences(assistantContent),
+        ];
+        const timestamp = new Date(turn.timestamp).toLocaleTimeString();
+        summaryParts.push(`[${timestamp}]`);
+        summaryParts.push(`User: ${userSummary}${userContent.length > 300 ? '...' : ''}`);
+        summaryParts.push(`Assistant: ${assistantSummary}${assistantContent.length > 400 ? '...' : ''}`);
+        if (fileRefs.length > 0) {
+            summaryParts.push(`Files: ${fileRefs.slice(0, 5).join(', ')}`);
+        }
+        summaryParts.push(''); // 空行分隔
+    }
+    summaryParts.push('=== End of Summary ===\n');
+    return summaryParts.join('\n');
+}
+/**
+ * 使用 AI 生成智能摘要（需要 API 客户端）
+ */
+export async function createAISummary(turns, apiClient) {
+    if (!apiClient) {
+        return createSummary(turns);
+    }
+    try {
+        // 构建摘要请求
+        const conversationText = turns.map((turn) => {
+            const user = extractMessageCore(turn.user);
+            const assistant = extractMessageCore(turn.assistant);
+            return `User: ${user}\n\nAssistant: ${assistant}`;
+        }).join('\n\n---\n\n');
+        // 调用 API 生成摘要
+        const response = await apiClient.createMessage([{
+                role: 'user',
+                content: `Please create a concise summary of the following conversation, preserving key information, decisions made, and important code/file references. Focus on what matters for future context:\n\n${conversationText}\n\nProvide a structured summary in 3-5 bullet points.`,
+            }], undefined, 'You are a helpful assistant that creates concise conversation summaries.');
+        // 提取文本内容
+        const summaryText = response.content
+            .filter((block) => block.type === 'text')
+            .map((block) => block.text)
+            .join('\n');
+        return `=== AI-Generated Summary ===\n\n${summaryText}\n\n=== End of Summary ===`;
+    }
+    catch (error) {
+        console.warn('Failed to generate AI summary, falling back to simple summary:', error);
+        return createSummary(turns);
+    }
+}
+/**
+ * 压缩消息内容（应用于单个消息）
+ */
+export function compressMessage(message, config) {
+    if (typeof message.content === 'string') {
+        // 简单字符串内容，直接压缩
+        const maxChars = config?.toolOutputMaxChars || TOOL_OUTPUT_MAX_CHARS;
+        if (message.content.length > maxChars) {
+            return {
+                ...message,
+                content: compressToolOutput(message.content, maxChars),
+            };
+        }
+        return message;
+    }
+    // 压缩数组内容中的每个块
+    const compressedBlocks = [];
+    for (const block of message.content) {
+        if (block.type === 'tool_result') {
+            const maxChars = config?.toolOutputMaxChars || TOOL_OUTPUT_MAX_CHARS;
+            if (typeof block.content === 'string') {
+                // string content：直接压缩
+                if (block.content.length > maxChars) {
+                    compressedBlocks.push({
+                        ...block,
+                        content: compressToolOutput(block.content, maxChars),
+                    });
+                }
+                else {
+                    compressedBlocks.push(block);
+                }
+            }
+            else if (Array.isArray(block.content)) {
+                // 数组 content（如 Browser screenshot 的 [TextBlock, ImageBlock]）
+                // 逐个 block 处理：压缩 text，保留 image 不动
+                const compressedContent = [];
+                for (const item of block.content) {
+                    if (item.type === 'image') {
+                        // 图片块原样保留，不做任何压缩
+                        compressedContent.push(item);
+                    }
+                    else if (item.type === 'text') {
+                        const text = item.text || '';
+                        if (text.length > maxChars) {
+                            compressedContent.push({ ...item, text: compressToolOutput(text, maxChars) });
+                        }
+                        else {
+                            compressedContent.push(item);
+                        }
+                    }
+                    else {
+                        compressedContent.push(item);
+                    }
+                }
+                compressedBlocks.push({ ...block, content: compressedContent });
+            }
+            else {
+                compressedBlocks.push(block);
+            }
+        }
+        else if (block.type === 'text') {
+            // 压缩文本块中的代码
+            const text = block.text || '';
+            const codeBlocks = extractCodeBlocks(text);
+            if (codeBlocks.length > 0) {
+                let compressed = text;
+                const maxLines = config?.codeBlockMaxLines || CODE_BLOCK_MAX_LINES;
+                for (const cb of codeBlocks) {
+                    const compressedCode = compressCodeBlock(cb.code, maxLines);
+                    const marker = cb.language ? `\`\`\`${cb.language}` : '```';
+                    compressed = compressed.replace(`${marker}\n${cb.code}\`\`\``, `${marker}\n${compressedCode}\`\`\``);
+                }
+                compressedBlocks.push({
+                    ...block,
+                    text: compressed,
+                });
+            }
+            else {
+                compressedBlocks.push(block);
+            }
+        }
+        else {
+            compressedBlocks.push(block);
+        }
+    }
+    return {
+        ...message,
+        content: compressedBlocks,
+    };
+}
+/**
+ * 上下文管理器
+ */
+export class ContextManager {
+    config;
+    turns = [];
+    systemPrompt = '';
+    compressionCount = 0;
+    savedTokens = 0;
+    apiClient; // 用于 AI 摘要的 API 客户端
+    compressing = false; // 防止并发压缩
+    constructor(config = {}) {
+        this.config = {
+            maxTokens: config.maxTokens ?? MAX_CONTEXT_TOKENS,
+            reserveTokens: config.reserveTokens ?? RESERVE_TOKENS,
+            summarizeThreshold: config.summarizeThreshold ?? 0.7, // 70% 时开始摘要
+            keepRecentMessages: config.keepRecentMessages ?? 10,
+            enableAISummary: config.enableAISummary ?? false,
+            codeBlockMaxLines: config.codeBlockMaxLines ?? CODE_BLOCK_MAX_LINES,
+            toolOutputMaxChars: config.toolOutputMaxChars ?? TOOL_OUTPUT_MAX_CHARS,
+            enableIncrementalCompression: config.enableIncrementalCompression ?? true,
+        };
+    }
+    /**
+     * 设置 API 客户端（用于 AI 摘要）
+     */
+    setApiClient(client) {
+        this.apiClient = client;
+    }
+    /**
+     * 设置系统提示
+     */
+    setSystemPrompt(prompt) {
+        this.systemPrompt = prompt;
+    }
+    /**
+     * 添加对话轮次（支持真实 API usage）
+     */
+    addTurn(user, assistant, apiUsage) {
+        const originalUserTokens = estimateMessageTokens(user);
+        const originalAssistantTokens = estimateMessageTokens(assistant);
+        const originalTokens = originalUserTokens + originalAssistantTokens;
+        // 应用增量压缩（如果启用）
+        let processedUser = user;
+        let processedAssistant = assistant;
+        let compressed = false;
+        if (this.config.enableIncrementalCompression) {
+            processedUser = compressMessage(user, this.config);
+            processedAssistant = compressMessage(assistant, this.config);
+            const compressedUserTokens = estimateMessageTokens(processedUser);
+            const compressedAssistantTokens = estimateMessageTokens(processedAssistant);
+            const compressedTokens = compressedUserTokens + compressedAssistantTokens;
+            if (compressedTokens < originalTokens) {
+                compressed = true;
+                this.savedTokens += originalTokens - compressedTokens;
+            }
+        }
+        const tokenEstimate = estimateMessageTokens(processedUser) + estimateMessageTokens(processedAssistant);
+        this.turns.push({
+            user: processedUser,
+            assistant: processedAssistant,
+            timestamp: Date.now(),
+            tokenEstimate,
+            originalTokens,
+            compressed,
+            apiUsage, // ✅ 保存真实 API usage
+        });
+        // 检查是否需要摘要压缩
+        this.maybeCompress();
+    }
+    /**
+     * 获取当前上下文的消息
+     */
+    getMessages() {
+        const messages = [];
+        // 添加摘要消息（如果有）
+        const summarizedTurns = this.turns.filter((t) => t.summarized);
+        if (summarizedTurns.length > 0) {
+            const summary = createSummary(summarizedTurns);
+            messages.push({
+                role: 'user',
+                content: summary,
+            });
+            messages.push({
+                role: 'assistant',
+                content: 'I understand. I\'ll keep this context in mind.',
+            });
+        }
+        // 添加非摘要的消息
+        const recentTurns = this.turns.filter((t) => !t.summarized);
+        for (const turn of recentTurns) {
+            messages.push(turn.user);
+            messages.push(turn.assistant);
+        }
+        return messages;
+    }
+    /**
+     * 获取可用的 token 数
+     */
+    getAvailableTokens() {
+        const used = this.getUsedTokens();
+        return this.config.maxTokens - this.config.reserveTokens - used;
+    }
+    /**
+     * 获取已使用的 token 数
+     *
+     * 混合策略（对齐官方 Fv 函数）：
+     * 1. 从末尾向前找最后一个有真实 apiUsage 的 turn
+     * 2. 用其 inputTokens + cacheTokens + outputTokens 作为基准（inputTokens 已包含之前所有消息）
+     * 3. 该 turn 之后的新消息用文本 length/4 估算追加
+     * 4. 如果没有任何真实 apiUsage，回退到纯估算
+     */
+    getUsedTokens() {
+        // 从后往前找最后一个有 apiUsage 的 turn
+        let lastApiTurnIndex = -1;
+        for (let i = this.turns.length - 1; i >= 0; i--) {
+            if (this.turns[i].apiUsage && !this.turns[i].summarized) {
+                lastApiTurnIndex = i;
+                break;
+            }
+        }
+        if (lastApiTurnIndex >= 0) {
+            const apiTurn = this.turns[lastApiTurnIndex];
+            const usage = apiTurn.apiUsage;
+            // 基准：该轮 API 调用的 input + cache + output（input 已包含之前所有上下文）
+            let total = usage.inputTokens +
+                (usage.cacheCreationTokens ?? 0) +
+                (usage.cacheReadTokens ?? 0) +
+                usage.outputTokens +
+                (usage.thinkingTokens ?? 0);
+            // 加上该 turn 之后新增消息的估算
+            for (let i = lastApiTurnIndex + 1; i < this.turns.length; i++) {
+                const turn = this.turns[i];
+                if (turn.summarized && turn.summary) {
+                    total += estimateTokens(turn.summary);
+                }
+                else {
+                    total += turn.tokenEstimate;
+                }
+            }
+            return total;
+        }
+        // 回退：没有真实 apiUsage，纯估算
+        let total = estimateTokens(this.systemPrompt);
+        for (const turn of this.turns) {
+            if (turn.summarized && turn.summary) {
+                total += estimateTokens(turn.summary);
+            }
+            else {
+                total += turn.tokenEstimate;
+            }
+        }
+        return total;
+    }
+    /**
+     * 检查并执行压缩
+     */
+    async maybeCompress() {
+        // 防止并发压缩
+        if (this.compressing)
+            return;
+        this.compressing = true;
+        try {
+            await this.doCompress();
+        }
+        finally {
+            this.compressing = false;
+        }
+    }
+    async doCompress() {
+        const threshold = this.config.maxTokens * this.config.summarizeThreshold;
+        const used = this.getUsedTokens();
+        if (used < threshold) {
+            return;
+        }
+        // 标记旧消息为需要摘要
+        const recentCount = this.config.keepRecentMessages;
+        const toSummarize = this.turns.slice(0, -recentCount);
+        if (toSummarize.length === 0) {
+            return;
+        }
+        const beforeTokens = toSummarize.reduce((sum, t) => sum + t.tokenEstimate, 0);
+        // 生成摘要
+        let summary;
+        if (this.config.enableAISummary && this.apiClient) {
+            try {
+                summary = await createAISummary(toSummarize, this.apiClient);
+            }
+            catch (error) {
+                console.warn('AI summary failed, using simple summary:', error);
+                summary = createSummary(toSummarize);
+            }
+        }
+        else {
+            summary = createSummary(toSummarize);
+        }
+        // 标记为已摘要
+        for (const turn of toSummarize) {
+            if (!turn.summarized) {
+                turn.summarized = true;
+                turn.summary = summary;
+            }
+        }
+        const afterTokens = estimateTokens(summary);
+        this.savedTokens += beforeTokens - afterTokens;
+        this.compressionCount++;
+    }
+    /**
+     * 强制压缩
+     */
+    async compact() {
+        const recentCount = this.config.keepRecentMessages;
+        if (this.turns.length <= recentCount) {
+            return;
+        }
+        const toSummarize = this.turns.slice(0, -recentCount);
+        if (toSummarize.length === 0) {
+            return;
+        }
+        const beforeTokens = toSummarize.reduce((sum, t) => sum + t.tokenEstimate, 0);
+        // 生成摘要
+        let summary;
+        if (this.config.enableAISummary && this.apiClient) {
+            try {
+                summary = await createAISummary(toSummarize, this.apiClient);
+            }
+            catch (error) {
+                console.warn('AI summary failed, using simple summary:', error);
+                summary = createSummary(toSummarize);
+            }
+        }
+        else {
+            summary = createSummary(toSummarize);
+        }
+        for (const turn of toSummarize) {
+            if (!turn.summarized) {
+                turn.summarized = true;
+                turn.summary = summary;
+            }
+        }
+        const afterTokens = estimateTokens(summary);
+        this.savedTokens += beforeTokens - afterTokens;
+        this.compressionCount++;
+    }
+    /**
+     * v2.1.32: Summarize from here - 从指定消息开始的部分对话摘要
+     * 对齐官方 "Summarize from here" 功能
+     *
+     * @param fromMessage 起始消息（之前的消息将被摘要）
+     * @param additionalContext 可选的额外上下文
+     */
+    async summarizeFromHere(fromMessage, additionalContext) {
+        // 找到消息在 turns 中的位置
+        let splitIndex = -1;
+        for (let i = 0; i < this.turns.length; i++) {
+            const turn = this.turns[i];
+            if (turn.user === fromMessage || turn.assistant === fromMessage) {
+                splitIndex = i;
+                break;
+            }
+        }
+        if (splitIndex <= 0) {
+            // 如果没找到或在开头，对所有历史进行摘要
+            splitIndex = Math.max(1, this.turns.length - (this.config.keepRecentMessages || 5));
+        }
+        const toSummarize = this.turns.slice(0, splitIndex);
+        const toKeep = this.turns.slice(splitIndex);
+        if (toSummarize.length === 0) {
+            return {
+                preCompactTokenCount: this.getUsedTokens(),
+                postCompactTokenCount: this.getUsedTokens(),
+                messagesKept: toKeep.length,
+                messagesSummarized: 0,
+            };
+        }
+        const preCompactTokenCount = this.getUsedTokens();
+        // 生成摘要
+        let summary;
+        const contextSuffix = additionalContext ? `\n\nAdditional context: ${additionalContext}` : '';
+        if (this.config.enableAISummary && this.apiClient) {
+            try {
+                summary = await createAISummary(toSummarize, this.apiClient) + contextSuffix;
+            }
+            catch {
+                summary = createSummary(toSummarize) + contextSuffix;
+            }
+        }
+        else {
+            summary = createSummary(toSummarize) + contextSuffix;
+        }
+        // 标记已摘要的 turns
+        for (const turn of toSummarize) {
+            turn.summarized = true;
+            turn.summary = summary;
+        }
+        const postCompactTokenCount = this.getUsedTokens();
+        this.savedTokens += preCompactTokenCount - postCompactTokenCount;
+        this.compressionCount++;
+        return {
+            preCompactTokenCount,
+            postCompactTokenCount,
+            messagesKept: toKeep.length,
+            messagesSummarized: toSummarize.length,
+        };
+    }
+    /**
+     * 获取统计信息
+     */
+    getStats() {
+        const summarized = this.turns.filter((t) => t.summarized).length;
+        const originalTokens = this.turns.reduce((sum, t) => sum + t.originalTokens, 0);
+        const currentTokens = this.getUsedTokens();
+        return {
+            totalMessages: this.turns.length * 2, // user + assistant
+            estimatedTokens: currentTokens,
+            summarizedMessages: summarized * 2,
+            compressionRatio: originalTokens > 0 ? currentTokens / originalTokens : 1,
+            savedTokens: this.savedTokens,
+            compressionCount: this.compressionCount,
+        };
+    }
+    /**
+     * 获取压缩详情
+     */
+    getCompressionDetails() {
+        const summarized = this.turns.filter((t) => t.summarized).length;
+        const compressed = this.turns.filter((t) => t.compressed).length;
+        const recent = Math.min(this.config.keepRecentMessages, this.turns.length);
+        const originalTokens = this.turns.reduce((sum, t) => sum + t.originalTokens, 0);
+        const currentTokens = this.getUsedTokens();
+        return {
+            totalTurns: this.turns.length,
+            summarizedTurns: summarized,
+            compressedTurns: compressed,
+            recentTurns: recent,
+            compressionRatio: originalTokens > 0 ? currentTokens / originalTokens : 1,
+            savedTokens: this.savedTokens,
+        };
+    }
+    /**
+     * 清除所有历史
+     */
+    clear() {
+        this.turns = [];
+        this.compressionCount = 0;
+        this.savedTokens = 0;
+    }
+    /**
+     * 导出为可序列化格式
+     */
+    export() {
+        return {
+            systemPrompt: this.systemPrompt,
+            turns: this.turns,
+            config: this.config,
+            compressionCount: this.compressionCount,
+            savedTokens: this.savedTokens,
+        };
+    }
+    /**
+     * 从导出数据恢复
+     */
+    import(data) {
+        this.systemPrompt = data.systemPrompt;
+        this.turns = data.turns;
+        this.compressionCount = data.compressionCount ?? 0;
+        this.savedTokens = data.savedTokens ?? 0;
+        if (data.config) {
+            this.config = { ...this.config, ...data.config };
+        }
+    }
+    /**
+     * 分析压缩效果
+     */
+    analyzeCompression() {
+        const originalTokens = this.turns.reduce((sum, t) => sum + t.originalTokens, 0);
+        const compressedTokens = this.getUsedTokens();
+        return {
+            originalTokens,
+            compressedTokens,
+            ratio: originalTokens > 0 ? compressedTokens / originalTokens : 1,
+            method: this.config.enableAISummary ? 'ai_summary' : 'simple_summary',
+        };
+    }
+    /**
+     * 获取上下文使用率
+     */
+    getContextUsage() {
+        const used = this.getUsedTokens();
+        const total = this.config.maxTokens - this.config.reserveTokens;
+        const available = total - used;
+        return {
+            used,
+            available,
+            total,
+            percentage: (used / total) * 100,
+        };
+    }
+    /**
+     * 检查是否接近上下文限制
+     */
+    isNearLimit() {
+        const usage = this.getContextUsage();
+        return usage.percentage >= this.config.summarizeThreshold * 100;
+    }
+    /**
+     * 获取格式化的统计报告
+     */
+    getFormattedReport() {
+        const stats = this.getStats();
+        const details = this.getCompressionDetails();
+        const usage = this.getContextUsage();
+        const lines = [
+            '=== Context Manager Report ===',
+            '',
+            `Total Messages: ${stats.totalMessages}`,
+            `Estimated Tokens: ${stats.estimatedTokens.toLocaleString()}`,
+            `Context Usage: ${usage.percentage.toFixed(1)}% (${usage.used.toLocaleString()}/${usage.total.toLocaleString()})`,
+            '',
+            `Compression:`,
+            `  - Summarized Turns: ${details.summarizedTurns}`,
+            `  - Compressed Turns: ${details.compressedTurns}`,
+            `  - Compression Ratio: ${(details.compressionRatio * 100).toFixed(1)}%`,
+            `  - Saved Tokens: ${details.savedTokens.toLocaleString()}`,
+            `  - Compression Count: ${stats.compressionCount}`,
+            '',
+            `Configuration:`,
+            `  - Keep Recent Messages: ${this.config.keepRecentMessages}`,
+            `  - Summarize Threshold: ${(this.config.summarizeThreshold * 100).toFixed(0)}%`,
+            `  - AI Summary: ${this.config.enableAISummary ? 'Enabled' : 'Disabled'}`,
+            `  - Incremental Compression: ${this.config.enableIncrementalCompression ? 'Enabled' : 'Disabled'}`,
+            '',
+            '==============================',
+        ];
+        return lines.join('\n');
+    }
+}
+/**
+ * 智能裁剪消息数组以适应 token 限制
+ */
+export function truncateMessages(messages, maxTokens, keepFirst = 2, keepLast = 10) {
+    let totalTokens = estimateTotalTokens(messages);
+    if (totalTokens <= maxTokens) {
+        return messages;
+    }
+    // 保护首尾消息
+    const firstMessages = messages.slice(0, keepFirst);
+    const lastMessages = messages.slice(-keepLast);
+    const middleMessages = messages.slice(keepFirst, -keepLast);
+    // 按 user/assistant 配对保留中间消息（保持交替）
+    const result = [...firstMessages];
+    let currentTokens = estimateTotalTokens(firstMessages) + estimateTotalTokens(lastMessages);
+    for (let i = 0; i < middleMessages.length; i++) {
+        const msg = middleMessages[i];
+        // 找到配对：user+assistant 一起保留或一起丢弃
+        if (msg.role === 'user' && i + 1 < middleMessages.length && middleMessages[i + 1].role === 'assistant') {
+            const pair = middleMessages[i + 1];
+            const pairTokens = estimateMessageTokens(msg) + estimateMessageTokens(pair);
+            if (currentTokens + pairTokens <= maxTokens) {
+                result.push(msg, pair);
+                currentTokens += pairTokens;
+            }
+            i++; // 跳过 assistant
+        }
+        else if (msg.role === 'assistant' && result.length > 0 && result[result.length - 1].role === 'user') {
+            // 单独的 assistant 消息，前面已有 user，可以保留
+            const msgTokens = estimateMessageTokens(msg);
+            if (currentTokens + msgTokens <= maxTokens) {
+                result.push(msg);
+                currentTokens += msgTokens;
+            }
+        }
+        // 其他情况（连续相同 role）跳过，避免破坏交替
+    }
+    result.push(...lastMessages);
+    // 最终验证：确保 first 和 last 的连接处也保持交替
+    // 如果 result 中最后一个 first/middle 消息与第一个 last 消息角色相同，移除冲突的
+    if (lastMessages.length > 0) {
+        const insertPoint = result.length - lastMessages.length;
+        if (insertPoint > 0 && result[insertPoint - 1].role === result[insertPoint].role) {
+            result.splice(insertPoint - 1, 1);
+        }
+    }
+    return result;
+}
+/**
+ * 智能压缩消息数组
+ */
+export function compressMessages(messages, config) {
+    return messages.map((msg) => compressMessage(msg, config));
+}
+/**
+ * 裁剪单条消息的内容
+ */
+export function truncateMessageContent(message, maxTokens) {
+    if (typeof message.content === 'string') {
+        const maxChars = maxTokens * CHARS_PER_TOKEN;
+        if (message.content.length <= maxChars) {
+            return message;
+        }
+        return {
+            ...message,
+            content: message.content.slice(0, maxChars) + '\n[Content truncated...]',
+        };
+    }
+    // 对于数组内容，裁剪每个块
+    const truncatedBlocks = [];
+    let remainingTokens = maxTokens;
+    for (const block of message.content) {
+        if (remainingTokens <= 0) {
+            break;
+        }
+        if (block.type === 'text') {
+            const maxChars = remainingTokens * CHARS_PER_TOKEN;
+            const blockText = block.text || '';
+            if (blockText.length <= maxChars) {
+                truncatedBlocks.push(block);
+                remainingTokens -= estimateTokens(blockText);
+            }
+            else {
+                truncatedBlocks.push({
+                    type: 'text',
+                    text: blockText.slice(0, maxChars) + '\n[Content truncated...]',
+                });
+                remainingTokens = 0;
+            }
+        }
+        else if (block.type === 'tool_result') {
+            const content = typeof block.content === 'string'
+                ? block.content
+                : JSON.stringify(block.content);
+            const maxChars = remainingTokens * CHARS_PER_TOKEN;
+            if (content.length <= maxChars) {
+                truncatedBlocks.push(block);
+                remainingTokens -= estimateTokens(content);
+            }
+            else {
+                truncatedBlocks.push({
+                    ...block,
+                    content: content.slice(0, maxChars) + '\n[Output truncated...]',
+                });
+                remainingTokens = 0;
+            }
+        }
+        else {
+            // 保留其他类型的块
+            truncatedBlocks.push(block);
+            remainingTokens -= 100; // 估算
+        }
+    }
+    return {
+        ...message,
+        content: truncatedBlocks,
+    };
+}
+/**
+ * 批量压缩工具输出
+ */
+export function batchCompressToolResults(messages, maxChars = TOOL_OUTPUT_MAX_CHARS) {
+    return messages.map((msg) => {
+        if (typeof msg.content === 'string') {
+            return msg;
+        }
+        const compressedBlocks = msg.content.map((block) => {
+            if (block.type === 'tool_result') {
+                const content = typeof block.content === 'string'
+                    ? block.content
+                    : JSON.stringify(block.content);
+                if (content.length > maxChars) {
+                    return {
+                        ...block,
+                        content: compressToolOutput(content, maxChars),
+                    };
+                }
+            }
+            return block;
+        });
+        return {
+            ...msg,
+            content: compressedBlocks,
+        };
+    });
+}
+/**
+ * 计算消息数组的压缩比
+ */
+export function calculateCompressionRatio(originalMessages, compressedMessages) {
+    const originalTokens = estimateTotalTokens(originalMessages);
+    const compressedTokens = estimateTotalTokens(compressedMessages);
+    return originalTokens > 0 ? compressedTokens / originalTokens : 1;
+}
+/**
+ * 优化上下文窗口（综合压缩策略）
+ */
+export function optimizeContext(messages, maxTokens, config) {
+    const originalTokens = estimateTotalTokens(messages);
+    if (originalTokens <= maxTokens) {
+        return {
+            messages,
+            compressionRatio: 1,
+            savedTokens: 0,
+        };
+    }
+    // 步骤 1: 压缩工具输出
+    let optimized = compressMessages(messages, config);
+    let currentTokens = estimateTotalTokens(optimized);
+    // 步骤 2: 如果仍然超限，裁剪消息
+    if (currentTokens > maxTokens) {
+        optimized = truncateMessages(optimized, maxTokens);
+        currentTokens = estimateTotalTokens(optimized);
+    }
+    return {
+        messages: optimized,
+        compressionRatio: currentTokens / originalTokens,
+        savedTokens: originalTokens - currentTokens,
+    };
+}
+/**
+ * 提取上下文关键信息
+ */
+export function extractContextKeyInfo(messages) {
+    const files = new Set();
+    const tools = new Set();
+    const keywords = new Set();
+    for (const msg of messages) {
+        const content = typeof msg.content === 'string'
+            ? msg.content
+            : msg.content.map((block) => {
+                if (block.type === 'text')
+                    return block.text || '';
+                if (block.type === 'tool_use') {
+                    tools.add(block.name || 'unknown');
+                    return JSON.stringify(block.input);
+                }
+                if (block.type === 'tool_result') {
+                    return extractToolResultText(block.content);
+                }
+                return '';
+            }).join(' ');
+        // 提取文件路径
+        const fileRefs = extractFileReferences(content);
+        fileRefs.forEach((f) => files.add(f));
+        // 提取关键词（简单实现）
+        const words = content
+            .split(/\s+/)
+            .filter((w) => w.length > 5 && /^[a-zA-Z]/.test(w))
+            .slice(0, 20);
+        words.forEach((w) => keywords.add(w));
+    }
+    return {
+        files: Array.from(files),
+        tools: Array.from(tools),
+        keywords: Array.from(keywords),
+    };
+}
+// 默认实例
+export const contextManager = new ContextManager();
+// ============ 导出增强功能 ============
+export * from './enhanced.js';
+// ============ 导出新增模块 ============
+export * from './summarizer.js';
+export * from './window.js';
+export * from './session-memory.js';
+//# sourceMappingURL=index.js.map