axon-code 2.4.0

package/dist/context/index.d.ts

@@ -0,0 +1,249 @@
+/**
+ * 上下文管理系统
+ * 处理对话历史、上下文窗口和自动摘要
+ *
+ * 功能：
+ * 1. 智能消息摘要（使用 Claude 生成摘要）
+ * 2. 保留最近 N 条消息不压缩
+ * 3. 工具调用结果智能压缩
+ * 4. 代码块智能保留
+ * 5. 文件内容引用压缩
+ * 6. 精确的压缩比计算
+ * 7. 增量压缩支持
+ */
+import type { Message } from '../types/index.js';
+export { estimateTokens } from '../utils/token-estimate.js';
+export interface ContextConfig {
+    maxTokens?: number;
+    reserveTokens?: number;
+    summarizeThreshold?: number;
+    keepRecentMessages?: number;
+    enableAISummary?: boolean;
+    codeBlockMaxLines?: number;
+    toolOutputMaxChars?: number;
+    enableIncrementalCompression?: boolean;
+}
+export interface ContextStats {
+    totalMessages: number;
+    estimatedTokens: number;
+    summarizedMessages: number;
+    compressionRatio: number;
+    savedTokens: number;
+    compressionCount: number;
+}
+export interface TokenUsage {
+    inputTokens: number;
+    outputTokens: number;
+    cacheReadTokens?: number;
+    cacheCreationTokens?: number;
+    thinkingTokens?: number;
+}
+export interface ConversationTurn {
+    user: Message;
+    assistant: Message;
+    timestamp: number;
+    tokenEstimate: number;
+    originalTokens: number;
+    summarized?: boolean;
+    summary?: string;
+    compressed?: boolean;
+    compressionMetadata?: {
+        originalSize: number;
+        compressedSize: number;
+        method: 'truncate' | 'ai_summary' | 'code_extract' | 'file_ref';
+    };
+    apiUsage?: TokenUsage;
+}
+export interface CompressionResult {
+    originalTokens: number;
+    compressedTokens: number;
+    ratio: number;
+    method: string;
+}
+/**
+ * 估算消息 token 数
+ */
+export declare function estimateMessageTokens(message: Message): number;
+/**
+ * 估算消息数组的总 token 数
+ */
+export declare function estimateTotalTokens(messages: Message[]): number;
+/**
+ * 创建对话摘要（简单版本，不使用 AI）
+ */
+export declare function createSummary(turns: ConversationTurn[]): string;
+/**
+ * 使用 AI 生成智能摘要（需要 API 客户端）
+ */
+export declare function createAISummary(turns: ConversationTurn[], apiClient?: any): Promise<string>;
+/**
+ * 压缩消息内容（应用于单个消息）
+ */
+export declare function compressMessage(message: Message, config?: ContextConfig): Message;
+/**
+ * 上下文管理器
+ */
+export declare class ContextManager {
+    private config;
+    private turns;
+    private systemPrompt;
+    private compressionCount;
+    private savedTokens;
+    private apiClient?;
+    private compressing;
+    constructor(config?: ContextConfig);
+    /**
+     * 设置 API 客户端（用于 AI 摘要）
+     */
+    setApiClient(client: any): void;
+    /**
+     * 设置系统提示
+     */
+    setSystemPrompt(prompt: string): void;
+    /**
+     * 添加对话轮次（支持真实 API usage）
+     */
+    addTurn(user: Message, assistant: Message, apiUsage?: TokenUsage): void;
+    /**
+     * 获取当前上下文的消息
+     */
+    getMessages(): Message[];
+    /**
+     * 获取可用的 token 数
+     */
+    getAvailableTokens(): number;
+    /**
+     * 获取已使用的 token 数
+     *
+     * 混合策略（对齐官方 Fv 函数）：
+     * 1. 从末尾向前找最后一个有真实 apiUsage 的 turn
+     * 2. 用其 inputTokens + cacheTokens + outputTokens 作为基准（inputTokens 已包含之前所有消息）
+     * 3. 该 turn 之后的新消息用文本 length/4 估算追加
+     * 4. 如果没有任何真实 apiUsage，回退到纯估算
+     */
+    getUsedTokens(): number;
+    /**
+     * 检查并执行压缩
+     */
+    private maybeCompress;
+    private doCompress;
+    /**
+     * 强制压缩
+     */
+    compact(): Promise<void>;
+    /**
+     * v2.1.32: Summarize from here - 从指定消息开始的部分对话摘要
+     * 对齐官方 "Summarize from here" 功能
+     *
+     * @param fromMessage 起始消息（之前的消息将被摘要）
+     * @param additionalContext 可选的额外上下文
+     */
+    summarizeFromHere(fromMessage: Message, additionalContext?: string): Promise<{
+        preCompactTokenCount: number;
+        postCompactTokenCount: number;
+        messagesKept: number;
+        messagesSummarized: number;
+    }>;
+    /**
+     * 获取统计信息
+     */
+    getStats(): ContextStats;
+    /**
+     * 获取压缩详情
+     */
+    getCompressionDetails(): {
+        totalTurns: number;
+        summarizedTurns: number;
+        compressedTurns: number;
+        recentTurns: number;
+        compressionRatio: number;
+        savedTokens: number;
+    };
+    /**
+     * 清除所有历史
+     */
+    clear(): void;
+    /**
+     * 导出为可序列化格式
+     */
+    export(): {
+        systemPrompt: string;
+        turns: ConversationTurn[];
+        config: Required<ContextConfig>;
+        compressionCount: number;
+        savedTokens: number;
+    };
+    /**
+     * 从导出数据恢复
+     */
+    import(data: {
+        systemPrompt: string;
+        turns: ConversationTurn[];
+        config?: ContextConfig;
+        compressionCount?: number;
+        savedTokens?: number;
+    }): void;
+    /**
+     * 分析压缩效果
+     */
+    analyzeCompression(): CompressionResult;
+    /**
+     * 获取上下文使用率
+     */
+    getContextUsage(): {
+        used: number;
+        available: number;
+        total: number;
+        percentage: number;
+    };
+    /**
+     * 检查是否接近上下文限制
+     */
+    isNearLimit(): boolean;
+    /**
+     * 获取格式化的统计报告
+     */
+    getFormattedReport(): string;
+}
+/**
+ * 智能裁剪消息数组以适应 token 限制
+ */
+export declare function truncateMessages(messages: Message[], maxTokens: number, keepFirst?: number, keepLast?: number): Message[];
+/**
+ * 智能压缩消息数组
+ */
+export declare function compressMessages(messages: Message[], config?: ContextConfig): Message[];
+/**
+ * 裁剪单条消息的内容
+ */
+export declare function truncateMessageContent(message: Message, maxTokens: number): Message;
+/**
+ * 批量压缩工具输出
+ */
+export declare function batchCompressToolResults(messages: Message[], maxChars?: number): Message[];
+/**
+ * 计算消息数组的压缩比
+ */
+export declare function calculateCompressionRatio(originalMessages: Message[], compressedMessages: Message[]): number;
+/**
+ * 优化上下文窗口（综合压缩策略）
+ */
+export declare function optimizeContext(messages: Message[], maxTokens: number, config?: ContextConfig): {
+    messages: Message[];
+    compressionRatio: number;
+    savedTokens: number;
+};
+/**
+ * 提取上下文关键信息
+ */
+export declare function extractContextKeyInfo(messages: Message[]): {
+    files: string[];
+    tools: string[];
+    keywords: string[];
+};
+export declare const contextManager: ContextManager;
+export * from './enhanced.js';
+export * from './summarizer.js';
+export * from './window.js';
+export * from './session-memory.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/context/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/context/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAiC,MAAM,mBAAmB,CAAC;AAChF,OAAO,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAc5D,MAAM,WAAW,aAAa;IAC5B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,4BAA4B,CAAC,EAAE,OAAO,CAAC;CACxC;AAED,MAAM,WAAW,YAAY;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;IACxB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,UAAU;IACzB,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,EAAE,OAAO,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,mBAAmB,CAAC,EAAE;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,cAAc,EAAE,MAAM,CAAC;QACvB,MAAM,EAAE,UAAU,GAAG,YAAY,GAAG,cAAc,GAAG,UAAU,CAAC;KACjE,CAAC;IAEF,QAAQ,CAAC,EAAE,UAAU,CAAC;CACvB;AAED,MAAM,WAAW,iBAAiB;IAChC,cAAc,EAAE,MAAM,CAAC;IACvB,gBAAgB,EAAE,MAAM,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB;AAgDD;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAqB9D;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAE/D;AAyJD;;GAEG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,gBAAgB,EAAE,GAAG,MAAM,CAiC/D;AAED;;GAEG;AACH,wBAAsB,eAAe,CACnC,KAAK,EAAE,gBAAgB,EAAE,EACzB,SAAS,CAAC,EAAE,GAAG,GACd,OAAO,CAAC,MAAM,CAAC,CAkCjB;AAED;;GAEG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,aAAa,GAAG,OAAO,CAuFjF;AAED;;GAEG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,KAAK,CAA0B;IACvC,OAAO,CAAC,YAAY,CAAc;IAClC,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,WAAW,CAAa;IAChC,OAAO,CAAC,SAAS,CAAC,CAAM;IACxB,OAAO,CAAC,WAAW,CAAkB;gBAEzB,MAAM,GAAE,aAAkB;IAatC;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,GAAG,GAAG,IAAI;IAI/B;;OAEG;IACH,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAIrC;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,CAAC,EAAE,UAAU,GAAG,IAAI;IAwCvE;;OAEG;IACH,WAAW,IAAI,OAAO,EAAE;IA2BxB;;OAEG;IACH,kBAAkB,IAAI,MAAM;IAK5B;;;;;;;;OAQG;IACH,aAAa,IAAI,MAAM;IA+CvB;;OAEG;YACW,aAAa;YAYb,UAAU;IA6CxB;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAyC9B;;;;;;OAMG;IACG,iBAAiB,CAAC,WAAW,EAAE,OAAO,EAAE,iBAAiB,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;QACjF,oBAAoB,EAAE,MAAM,CAAC;QAC7B,qBAAqB,EAAE,MAAM,CAAC;QAC9B,YAAY,EAAE,MAAM,CAAC;QACrB,kBAAkB,EAAE,MAAM,CAAC;KAC5B,CAAC;IA+DF;;OAEG;IACH,QAAQ,IAAI,YAAY;IAgBxB;;OAEG;IACH,qBAAqB,IAAI;QACvB,UAAU,EAAE,MAAM,CAAC;QACnB,eAAe,EAAE,MAAM,CAAC;QACxB,eAAe,EAAE,MAAM,CAAC;QACxB,WAAW,EAAE,MAAM,CAAC;QACpB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB;IAkBD;;OAEG;IACH,KAAK,IAAI,IAAI;IAMb;;OAEG;IACH,MAAM,IAAI;QACR,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,gBAAgB,EAAE,CAAC;QAC1B,MAAM,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC;QAChC,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB;IAUD;;OAEG;IACH,MAAM,CAAC,IAAI,EAAE;QACX,YAAY,EAAE,MAAM,CAAC;QACrB,KAAK,EAAE,gBAAgB,EAAE,CAAC;QAC1B,MAAM,CAAC,EAAE,aAAa,CAAC;QACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,IAAI;IAWR;;OAEG;IACH,kBAAkB,IAAI,iBAAiB;IAYvC;;OAEG;IACH,eAAe,IAAI;QACjB,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,MAAM,CAAC;KACpB;IAaD;;OAEG;IACH,WAAW,IAAI,OAAO;IAKtB;;OAEG;IACH,kBAAkB,IAAI,MAAM;CA8B7B;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,OAAO,EAAE,EACnB,SAAS,EAAE,MAAM,EACjB,SAAS,GAAE,MAAU,EACrB,QAAQ,GAAE,MAAW,GACpB,OAAO,EAAE,CAkDX;AAED;;GAEG;AACH,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,CAAC,EAAE,aAAa,GACrB,OAAO,EAAE,CAEX;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,GAChB,OAAO,CA6DT;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,GAAE,MAA8B,GACvC,OAAO,EAAE,CA2BX;AAED;;GAEG;AACH,wBAAgB,yBAAyB,CACvC,gBAAgB,EAAE,OAAO,EAAE,EAC3B,kBAAkB,EAAE,OAAO,EAAE,GAC5B,MAAM,CAKR;AAED;;GAEG;AACH,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,OAAO,EAAE,EACnB,SAAS,EAAE,MAAM,EACjB,MAAM,CAAC,EAAE,aAAa,GACrB;IACD,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;CACrB,CA0BA;AAED;;GAEG;AACH,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG;IAC1D,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB,CAqCA;AAGD,eAAO,MAAM,cAAc,gBAAuB,CAAC;AAGnD,cAAc,eAAe,CAAC;AAG9B,cAAc,iBAAiB,CAAC;AAChC,cAAc,aAAa,CAAC;AAC5B,cAAc,qBAAqB,CAAC"}