foliko 1.0.74 → 1.0.75

package/src/core/agent-chat.js

@@ -3,8 +3,9 @@
  * 使用 AI SDK 的 ToolLoopAgent 处理工具调用循环
  */
 
-const { EventEmitter } = require('../utils/event-emitter')
-const tiktoken = require('tiktoken')
+const { EventEmitter } = require('../utils/event-emitter');
+const { logger } = require('../utils/logger');
+const { generateText, pruneMessages } = require('ai');
 
 // 模型上下文限制表（留 15-20% 余量给 system prompt 和输出）
 const MODEL_CONTEXT_LIMITS = {
@@ -22,60 +23,109 @@ const MODEL_CONTEXT_LIMITS = {
   'claude-3-5-sonnet': 150000,
   'claude-3-opus': 150000,
   'claude-3-sonnet': 150000,
+};
+
+/**
+ * 纯 JavaScript token 计数器
+ * 基于字符数估算，兼容中英文
+ * 估算规则：
+ * - 英文：约 4 字符 = 1 token
+ * - 中文：约 2 字符 = 1 token
+ * - 混合文本取加权平均
+ */
+class SimpleTokenizer {
+  constructor() {
+    // 无需初始化
+  }
+
+  /**
+   * 估算文本的 token 数
+   * @param {string} text
+   * @returns {number}
+   */
+  encode(text) {
+    if (!text || typeof text !== 'string') {
+      return 0;
+    }
+
+    // 清理文本
+    const cleanText = text
+      .replace(/\0+/g, '')
+      .replace(/[\x00-\x08\x0B\x0C\x0E-\x1F\x7F]/g, '')
+      .slice(0, 100000);
+
+    if (!cleanText) {
+      return 0;
+    }
+
+    // 分离中英文分别计数
+    const chineseChars = (cleanText.match(/[\u4e00-\u9fff\u3400-\u4dbf]/g) || []).length;
+    const englishChars = cleanText.length - chineseChars;
+
+    // 中英文分开估算后相加
+    // 中文约 2 字符/token，英文约 4 字符/token
+    const chineseTokens = chineseChars / 2;
+    const englishTokens = englishChars / 4;
+
+    return Math.ceil(chineseTokens + englishTokens);
+  }
 }
 
+// 全局 tokenizer 实例
+const _globalTokenizer = new SimpleTokenizer();
+
+// 压缩超时时间（毫秒）
+const COMPRESSION_TIMEOUT = 30000;
+
 class AgentChatHandler extends EventEmitter {
   /**
    * @param {Agent} agent - Agent 实例
    * @param {Object} config - 配置
    */
   constructor(agent, config = {}) {
-    super()
+    super();
 
-    this.agent = agent
-    this.config = config
+    this.agent = agent;
+    this.config = config;
 
-    this.model = config.model || 'deepseek-chat'
-    this.provider = config.provider || 'deepseek'
-    this.apiKey = config.apiKey
-    this.baseURL = config.baseURL
-    this.providerOptions = config.providerOptions || {}
+    this.model = config.model || 'deepseek-chat';
+    this.provider = config.provider || 'deepseek';
+    this.apiKey = config.apiKey;
+    this.baseURL = config.baseURL;
+    this.providerOptions = config.providerOptions || {};
 
-    this._systemPrompt = config.systemPrompt || 'You are a helpful assistant.'
-    this._messages = []
-    this._tools = new Map()
-    this._maxSteps = 5  // 降低默认步骤数，减少上下文消耗
+    this._systemPrompt = config.systemPrompt || 'You are a helpful assistant.';
+    this._messages = [];
+    this._tools = new Map();
+    this._maxSteps = 5; // 降低默认步骤数，减少上下文消耗
 
     // 上下文压缩配置：根据模型自动设置限制
-    const modelKey = Object.keys(MODEL_CONTEXT_LIMITS).find(k =>
+    const modelKey = Object.keys(MODEL_CONTEXT_LIMITS).find((k) =>
       this.model.toLowerCase().includes(k.toLowerCase())
-    )
-    const defaultLimit = modelKey ? MODEL_CONTEXT_LIMITS[modelKey] : 40000
-    this._maxContextTokens = config.maxContextTokens || defaultLimit
-    this._compressionThreshold = config.compressionThreshold || 0.6  // 60% 就压缩，早触发
-    this._keepRecentMessages = config.keepRecentMessages || 20        // 保留最近 20 条
-    this._enableSmartCompress = config.enableSmartCompress !== false // 默认开启智能摘要
-    this._encoder = null
-    this._compressionCount = 0 // 压缩次数统计
+    );
+    const defaultLimit = modelKey ? MODEL_CONTEXT_LIMITS[modelKey] : 40000;
+    this._maxContextTokens = config.maxContextTokens || defaultLimit;
+    this._compressionThreshold = config.compressionThreshold || 0.6; // 60% 就压缩，早触发
+    this._keepRecentMessages = config.keepRecentMessages || 20; // 保留最近 20 条
+    this._enableSmartCompress = config.enableSmartCompress !== false; // 默认开启智能摘要
+    this._encoder = null;
+    this._compressionCount = 0; // 压缩次数统计
+    this._compressionInProgress = false; // 压缩锁，防止并发压缩
+    this._compressionPromise = null; // 正在进行的压缩 Promise
 
     // 工具结果压缩配置
-    this._maxToolResultSize = config.maxToolResultSize || 4000  // 工具结果超过此大小则压缩（字节）
+    this._maxToolResultSize = config.maxToolResultSize || 4000; // 工具结果超过此大小则压缩（字节）
 
     // 初始化编码器
-    this._initEncoder()
+    // 使用纯 JS tokenizer
+    this._encoder = _globalTokenizer;
   }
 
   /**
-   * 初始化 tiktoken 编码器
-   * @private
+   * 静态方法：保留接口兼容性（无实际作用）
    */
-  _initEncoder() {
-    try {
-      // cl100k_base 是 GPT-4/ChatGPT 使用的编码
-      this._encoder = tiktoken.get_encoding('cl100k_base')
-    } catch (err) {
-      console.warn('[AgentChat] Failed to initialize tiktoken encoder:', err.message)
-    }
+  static clearEncoderCache() {
+    // 不再需要清理
   }
 
   /**
@@ -85,12 +135,13 @@ class AgentChatHandler extends EventEmitter {
    * @private
    */
   _countTokens(text) {
-    if (!this._encoder || !text) return 0
+    if (!text || typeof text !== 'string') return 0;
+    // SimpleTokenizer.encode 已经处理了清理逻辑
     try {
-      return this._encoder.encode(text).length
+      return this._encoder.encode(text);
     } catch (err) {
-      // 粗略估算：约 4 字符 = 1 token
-      return Math.ceil(text.length / 4)
+      // 估算失败时使用字符计数
+      return Math.ceil(text.length / 4);
     }
   }
 
@@ -101,21 +152,30 @@ class AgentChatHandler extends EventEmitter {
    * @private
    */
   _countMessagesTokens(messages) {
-    let total = 0
+    let total = 0;
     for (const msg of messages) {
-      total += 4 // role 标记
+      if (!msg) continue;
+      total += 4; // role 标记
       if (typeof msg.content === 'string') {
-        total += this._countTokens(msg.content)
+        total += this._countTokens(msg.content);
       } else if (Array.isArray(msg.content)) {
         for (const part of msg.content) {
-          if (part.text) {
-            total += this._countTokens(part.text)
+          if (part && typeof part === 'object' && part.text) {
+            total += this._countTokens(String(part.text));
           }
         }
+      } else if (msg.content && typeof msg.content === 'object') {
+        // 处理工具结果等对象类型
+        try {
+          const str = JSON.stringify(msg.content);
+          total += this._countTokens(str);
+        } catch (e) {
+          // 无法序列化的内容，忽略
+        }
       }
     }
-    total += 4 // 结尾标记
-    return total
+    total += 4; // 结尾标记
+    return total;
   }
 
   /**
@@ -124,8 +184,8 @@ class AgentChatHandler extends EventEmitter {
    * @private
    */
   _shouldCompress() {
-    const totalTokens = this._countMessagesTokens(this._messages)
-    return totalTokens > this._maxContextTokens * this._compressionThreshold
+    const totalTokens = this._countMessagesTokens(this._messages);
+    return totalTokens > this._maxContextTokens * this._compressionThreshold;
   }
 
   /**
@@ -134,21 +194,22 @@ class AgentChatHandler extends EventEmitter {
    * @private
    */
   _countToolsTokens() {
-    let total = 0
+    let total = 0;
     for (const toolDef of this._tools.values()) {
       // 工具名 + 描述
-      total += this._countTokens(toolDef.name || '')
-      total += this._countTokens(toolDef.description || '')
+      total += this._countTokens(String(toolDef.name || ''));
+      total += this._countTokens(String(toolDef.description || ''));
 
       // 工具参数 schema
       if (toolDef.inputSchema) {
-        const schemaStr = typeof toolDef.inputSchema === 'string'
-          ? toolDef.inputSchema
-          : JSON.stringify(toolDef.inputSchema)
-        total += this._countTokens(schemaStr)
+        const schemaStr =
+          typeof toolDef.inputSchema === 'string'
+            ? toolDef.inputSchema
+            : JSON.stringify(toolDef.inputSchema);
+        total += this._countTokens(schemaStr);
       }
     }
-    return total
+    return total;
   }
 
   /**
@@ -157,61 +218,137 @@ class AgentChatHandler extends EventEmitter {
    * @private
    */
   _shouldCompressWithTools() {
-    const messagesTokens = this._countMessagesTokens(this._messages)
-    const toolsTokens = this._countToolsTokens()
-    const systemPromptTokens = this._countTokens(this._systemPrompt)
-    const total = messagesTokens + toolsTokens + systemPromptTokens
+    const messagesTokens = this._countMessagesTokens(this._messages);
+    const toolsTokens = this._countToolsTokens();
+    const systemPromptTokens = this._countTokens(this._systemPrompt);
+    const total = messagesTokens + toolsTokens + systemPromptTokens;
 
     // 如果总token数超过上下文限制的 85%，就压缩
-    return total > this._maxContextTokens * 0.85
+    return total > this._maxContextTokens * 0.85;
   }
 
   /**
-   * 压缩上下文消息（智能摘要模式）
+   * 压缩上下文消息（智能摘要模式，支持超时控制）
    * 策略：
    * 1. 如果启用了智能摘要且有 AI 客户端，对早期消息进行 AI 总结
    * 2. 否则使用简单裁剪 + 标记
+   * 3. 支持超时控制，防止压缩阻塞太久
    * @private
    */
   async _compressContext() {
+    // 如果已经有压缩在进行，返回现有的 Promise
+    if (this._compressionInProgress && this._compressionPromise) {
+      logger.debug('Compression already in progress, waiting for it...');
+      return this._compressionPromise;
+    }
+
     if (this._messages.length <= this._keepRecentMessages) {
-      return
+      return;
+    }
+
+    this._compressionInProgress = true;
+
+    // 创建压缩 Promise，带超时控制
+    this._compressionPromise = this._executeCompressionWithTimeout().finally(() => {
+      this._compressionInProgress = false;
+      this._compressionPromise = null;
+    });
+
+    return this._compressionPromise;
+  }
+
+  /**
+   * 执行压缩（带超时）
+   * @private
+   */
+  async _executeCompressionWithTimeout() {
+    try {
+      return await Promise.race([this._doCompress(), this._createTimeoutPromise()]);
+    } catch (err) {
+      logger.warn('Compression failed:', err.message);
+      // 压缩失败时使用简单的截断策略
+      this._simpleCompress();
     }
+  }
+
+  /**
+   * 创建超时 Promise
+   * @private
+   */
+  _createTimeoutPromise() {
+    return new Promise((_, reject) => {
+      setTimeout(() => {
+        reject(new Error('Compression timeout'));
+      }, COMPRESSION_TIMEOUT);
+    });
+  }
 
-    const systemMessages = this._messages.filter(m => m.role === 'system')
-    const otherMessages = this._messages.filter(m => m.role !== 'system')
+  /**
+   * 执行实际压缩逻辑
+   * @private
+   */
+  async _doCompress() {
+    const systemMessages = this._messages.filter((m) => m.role === 'system');
+    const otherMessages = this._messages.filter((m) => m.role !== 'system');
 
     // 保留最近的 N 条非系统消息
-    const recentMessages = otherMessages.slice(-this._keepRecentMessages)
-    const messagesToSummarize = otherMessages.slice(0, -this._keepRecentMessages)
+    const recentMessages = otherMessages.slice(-this._keepRecentMessages);
+    const messagesToSummarize = otherMessages.slice(0, -this._keepRecentMessages);
 
-    const compressedCount = messagesToSummarize.length
-    let summaryContent = ''
+    const compressedCount = messagesToSummarize.length;
+    let summaryContent = '';
 
     // 尝试使用 AI 总结
     if (this._enableSmartCompress && this._aiClient) {
       try {
-        const summaryText = await this._summarizeMessages(messagesToSummarize)
-        summaryContent = `[早期对话摘要]: ${summaryText}`
-        console.log(`[AgentChat] AI 摘要生成成功 (${summaryText.length} chars)`)
+        const summaryText = await this._summarizeMessages(messagesToSummarize);
+        summaryContent = `[早期对话摘要]: ${summaryText}`;
+        logger.info(`AI summary generated (${summaryText.length} chars)`);
       } catch (err) {
-        console.warn('[AgentChat] AI 摘要失败，使用简单压缩:', err.message)
-        summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`
+        logger.warn('AI summary failed, using simple compression:', err.message);
+        summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`;
       }
     } else {
-      summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`
+      summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`;
     }
 
     const summary = {
       role: 'system',
-      content: summaryContent
-    }
+      content: summaryContent,
+    };
 
-    this._messages = [...systemMessages, summary, ...recentMessages]
-    this._compressionCount++
+    this._messages = [...systemMessages, summary, ...recentMessages];
+    this._compressionCount++;
 
-    const totalTokens = this._countMessagesTokens(this._messages)
-    console.log(`[AgentChat] Context compressed (${this._compressionCount} times). Messages: ${this._messages.length}, Est. tokens: ${totalTokens}`)
+    const totalTokens = this._countMessagesTokens(this._messages);
+    logger.info(
+      `Context compressed (${this._compressionCount} times). Messages: ${this._messages.length}, Est. tokens: ${totalTokens}`
+    );
+  }
+
+  /**
+   * 简单压缩（当 AI 压缩失败时使用）
+   * @private
+   */
+  _simpleCompress() {
+    const systemMessages = this._messages.filter((m) => m.role === 'system');
+    const otherMessages = this._messages.filter((m) => m.role !== 'system');
+    const recentMessages = otherMessages.slice(-this._keepRecentMessages);
+    const compressedCount = otherMessages.length - this._keepRecentMessages;
+
+    const summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`;
+
+    const summary = {
+      role: 'system',
+      content: summaryContent,
+    };
+
+    this._messages = [...systemMessages, summary, ...recentMessages];
+    this._compressionCount++;
+
+    logger.info(
+      `Context simple compressed (${this._compressionCount} times). Messages: ${this._messages.length}`
+    );
   }
 
   /**
@@ -222,15 +359,17 @@ class AgentChatHandler extends EventEmitter {
    */
   async _summarizeMessages(messages) {
     if (!this._aiClient || messages.length === 0) {
-      return '（无早期对话）'
+      return '（无早期对话）';
     }
 
     // 构建总结提示
-    const conversationText = messages.map(m => {
-      const role = m.role === 'user' ? '用户' : '助手'
-      const content = typeof m.content === 'string' ? m.content : JSON.stringify(m.content)
-      return `${role}: ${content}`
-    }).join('\n')
+    const conversationText = messages
+      .map((m) => {
+        const role = m.role === 'user' ? '用户' : '助手';
+        const content = typeof m.content === 'string' ? m.content : JSON.stringify(m.content);
+        return `${role}: ${content}`;
+      })
+      .join('\n');
 
     const summarizePrompt = `请简洁地总结以下对话的要点，保留关键信息和用户需求：
 
@@ -240,17 +379,16 @@ ${conversationText}
 1. 提取用户的主要需求和意图
 2. 保留关键的技术细节或决策
 3. 不要超过 1000 字
-4. 用中文回复`
+4. 用中文回复`;
 
     // 使用 AI SDK 6.x 的 generateText
-    const { generateText } = require('ai')
     const { text } = await generateText({
       model: this._aiClient,
       prompt: summarizePrompt,
-      ...this.providerOptions
-    })
+      ...this.providerOptions,
+    });
 
-    return text || '（总结生成失败）'
+    return text || '（总结生成失败）';
   }
 
   /**
@@ -260,21 +398,21 @@ ${conversationText}
    * @private
    */
   _shouldCompressToolResult(result) {
-    if (!result || this._maxToolResultSize <= 0) return false
+    if (!result || this._maxToolResultSize <= 0) return false;
 
     // 计算结果的大小
-    let size = 0
+    let size = 0;
     if (typeof result === 'string') {
-      size = result.length
+      size = result.length;
     } else if (typeof result === 'object') {
       try {
-        size = JSON.stringify(result).length
+        size = JSON.stringify(result).length;
       } catch {
-        size = String(result).length
+        size = String(result).length;
       }
     }
 
-    return size > this._maxToolResultSize
+    return size > this._maxToolResultSize;
   }
 
   /**
@@ -285,28 +423,30 @@ ${conversationText}
    */
   async _compressToolResult(result) {
     if (!this._shouldCompressToolResult(result)) {
-      return result
+      return result;
     }
 
     if (!this._aiClient) {
-      console.warn('[AgentChat] Cannot compress tool result: no AI client')
-      return this._fallbackCompress(result)
+      logger.warn('Cannot compress tool result: no AI client');
+      return this._fallbackCompress(result);
     }
 
     try {
-      const originalSize = typeof result === 'string' ? result.length : JSON.stringify(result).length
-      const resultStr = typeof result === 'string' ? result : JSON.stringify(result, null, 2)
+      const originalSize =
+        typeof result === 'string' ? result.length : JSON.stringify(result).length;
+      const resultStr = typeof result === 'string' ? result : JSON.stringify(result, null, 2);
 
       // 对于超大型内容（如网页），采用更好的截断策略
-      const maxInputSize = 6000  // 保留给 AI 处理的输入大小
-      const shouldTruncate = resultStr.length > maxInputSize
-      const truncatedContent = resultStr.substring(0, maxInputSize)
-      const truncatedNote = shouldTruncate ? `\n\n[内容已截断，原始长度 ${originalSize} 字符]` : ''
+      const maxInputSize = 6000; // 保留给 AI 处理的输入大小
+      const shouldTruncate = resultStr.length > maxInputSize;
+      const truncatedContent = resultStr.substring(0, maxInputSize);
+      const truncatedNote = shouldTruncate ? `\n\n[内容已截断，原始长度 ${originalSize} 字符]` : '';
 
       // 检测内容类型
-      const isHTML = resultStr.startsWith('<') || resultStr.includes('<html') || resultStr.includes('<!DOCTYPE')
-      const isJSON = !isHTML && (resultStr.startsWith('{') || resultStr.startsWith('['))
-      const contentTypeHint = isHTML ? '（HTML 网页内容）' : isJSON ? '（JSON 数据）' : ''
+      const isHTML =
+        resultStr.startsWith('<') || resultStr.includes('<html') || resultStr.includes('<!DOCTYPE');
+      const isJSON = !isHTML && (resultStr.startsWith('{') || resultStr.startsWith('['));
+      const contentTypeHint = isHTML ? '（HTML 网页内容）' : isJSON ? '（JSON 数据）' : '';
 
       // 构建压缩提示
       const compressPrompt = `以下是一个工具执行结果${contentTypeHint}，长度 ${originalSize} 字符。请简洁地总结其核心内容：
@@ -318,25 +458,24 @@ ${truncatedContent}${truncatedNote}
 2. 关键信息点（不超过 5 个）
 3. 重要数据或结论
 
-用简洁的中文总结，不超过 400 字：`
+用简洁的中文总结，不超过 400 字：`;
 
       // 使用 AI SDK 6.x 的 generateText
-      const { generateText } = require('ai')
       const { text } = await generateText({
         model: this._aiClient,
         prompt: compressPrompt,
         ...this.providerOptions,
-        maxTokens: 500
-      })
+        maxTokens: 500,
+      });
 
-      const summary = text || '（总结生成失败）'
-      const compressed = `[工具结果已压缩${contentTypeHint}: ${originalSize} → ${summary.length} 字符]\n\n${summary}`
+      const summary = text || '（总结生成失败）';
+      const compressed = `[工具结果已压缩${contentTypeHint}: ${originalSize} -> ${summary.length} 字符]\n\n${summary}`;
 
-      console.log(`[AgentChat] Tool result compressed: ${originalSize} → ${summary.length} chars`)
-      return compressed
+      logger.info(`Tool result compressed: ${originalSize} -> ${summary.length} chars`);
+      return compressed;
     } catch (err) {
-      console.warn('[AgentChat] Tool result compression failed:', err.message)
-      return this._fallbackCompress(result)
+      logger.warn('Tool result compression failed:', err.message);
+      return this._fallbackCompress(result);
     }
   }
 
@@ -347,18 +486,18 @@ ${truncatedContent}${truncatedNote}
    * @private
    */
   _fallbackCompress(result) {
-    const originalSize = typeof result === 'string' ? result.length : JSON.stringify(result).length
-    const resultStr = typeof result === 'string' ? result : JSON.stringify(result, null, 2)
+    const originalSize = typeof result === 'string' ? result.length : JSON.stringify(result).length;
+    const resultStr = typeof result === 'string' ? result : JSON.stringify(result, null, 2);
 
     // 简单截断策略：保留前 2000 字符
-    const maxSize = 2000
+    const maxSize = 2000;
     if (resultStr.length <= maxSize) {
-      return result
+      return result;
     }
 
-    const compressed = `[工具结果已压缩（简单截断）: ${originalSize} → ${maxSize} 字符]\n\n${resultStr.substring(0, maxSize)}\n\n...[内容已截断，原文 ${originalSize} 字符]`
-    console.log(`[AgentChat] Tool result fallback compressed: ${originalSize} → ${maxSize} chars`)
-    return compressed
+    const compressed = `[工具结果已压缩（简单截断）: ${originalSize} -> ${maxSize} 字符]\n\n${resultStr.substring(0, maxSize)}\n\n...[内容已截断，原文 ${originalSize} 字符]`;
+    logger.info(`Tool result fallback compressed: ${originalSize} -> ${maxSize} chars`);
+    return compressed;
   }
 
   /**
@@ -366,8 +505,8 @@ ${truncatedContent}${truncatedNote}
    * @param {Object} client - AI 模型客户端
    */
   setAIClient(client) {
-    this._aiClient = client
-    return this
+    this._aiClient = client;
+    return this;
   }
 
   /**
@@ -375,8 +514,8 @@ ${truncatedContent}${truncatedNote}
    * @param {string} prompt
    */
   setSystemPrompt(prompt) {
-    this._systemPrompt = prompt
-    return this
+    this._systemPrompt = prompt;
+    return this;
   }
 
   /**
@@ -384,17 +523,17 @@ ${truncatedContent}${truncatedNote}
    * @param {Object} toolDef - 工具定义
    */
   registerTool(toolDef) {
-    this._tools.set(toolDef.name, toolDef)
-    return this
+    this._tools.set(toolDef.name, toolDef);
+    return this;
   }
 
   /**
    * 清空对话历史
    */
   clearHistory() {
-    this._messages = []
-    this._compressionCount = 0
-    return this
+    this._messages = [];
+    this._compressionCount = 0;
+    return this;
   }
 
   /**
@@ -402,7 +541,7 @@ ${truncatedContent}${truncatedNote}
    * @returns {Array}
    */
   getTools() {
-    return Array.from(this._tools.values())
+    return Array.from(this._tools.values());
   }
 
   /**
@@ -411,13 +550,13 @@ ${truncatedContent}${truncatedNote}
    */
   async _importAI() {
     try {
-      const ai = require('ai')
+      const ai = require('ai');
       return {
         tool: ai.tool,
-        ToolLoopAgent: ai.ToolLoopAgent
-      }
+        ToolLoopAgent: ai.ToolLoopAgent,
+      };
     } catch (err) {
-      throw new Error('AI SDK not found. Please install: npm install ai')
+      throw new Error('AI SDK not found. Please install: npm install ai');
     }
   }
 
@@ -427,83 +566,90 @@ ${truncatedContent}${truncatedNote}
    * @param {Object} options - 选项
    */
   async chat(message, options = {}) {
-    const context = { sessionId: options.sessionId || null, isStream: false }
-    const framework = this.agent.framework
-    const self = this  // 保存引用用于回调
+    const context = { sessionId: options.sessionId || null, isStream: false };
+    const framework = this.agent.framework;
+    const self = this; // 保存引用用于回调
 
+    // 关键：每次 chat 调用时刷新系统提示词，确保包含最新的工具/技能描述
+    // 解决上下文过长时 LLM 不调用工具的问题
+    this._systemPrompt = this.agent._buildSystemPrompt();
     // 动态导入 AI SDK
-    const { tool, ToolLoopAgent } = await this._importAI()
+    const { tool, ToolLoopAgent } = await this._importAI();
 
-    const userMessage = typeof message === 'string'
-      ? { role: 'user', content: message }
-      : message
+    const userMessage = typeof message === 'string' ? { role: 'user', content: message } : message;
 
-    this._messages.push(userMessage)
+    this._messages.push(userMessage);
 
     // 检查是否需要压缩上下文（包括工具定义）
-    const messagesTokens = this._countMessagesTokens(this._messages)
-    const toolsTokens = this._countToolsTokens()
-    const systemPromptTokens = this._countTokens(this._systemPrompt)
-    const totalTokens = messagesTokens + toolsTokens + systemPromptTokens
-    const limit = this._maxContextTokens * 0.7  // 降低到 70%，更早压缩
+    const messagesTokens = this._countMessagesTokens(this._messages);
+    const toolsTokens = this._countToolsTokens();
+    const systemPromptTokens = this._countTokens(this._systemPrompt);
+    const totalTokens = messagesTokens + toolsTokens + systemPromptTokens;
+    const limit = this._maxContextTokens * 0.7; // 降低到 70%，更早压缩
 
     if (totalTokens > limit) {
-      console.log(`[AgentChat] Context large (${totalTokens}/${this._maxContextTokens} tokens = msgs:${messagesTokens} + tools:${toolsTokens} + sys:${systemPromptTokens}), compressing...`)
-      await this._compressContext()
+      logger.info(
+        `Context large (${totalTokens}/${this._maxContextTokens} tokens = msgs:${messagesTokens} + tools:${toolsTokens} + sys:${systemPromptTokens}), compressing...`
+      );
+      // 使用带超时控制的压缩
+      await this._compressContext();
     } else {
-      console.log(`[AgentChat] Context OK: ${totalTokens}/${this._maxContextTokens} tokens`)
+      logger.info(`Context OK: ${totalTokens}/${this._maxContextTokens} tokens`);
     }
 
-    const maxSteps = options.maxSteps || this._maxSteps
-    const tools = this._getAITools(tool)
+    const maxSteps = options.maxSteps || this._maxSteps;
+    const tools = this._getAITools(tool);
 
     if (!this._aiClient) {
-      throw new Error('AI client not configured.')
+      throw new Error('AI client not configured.');
     }
 
     const agent = new ToolLoopAgent({
       model: this._aiClient,
       instructions: this._systemPrompt,
       tools: tools,
-      stopWhen: (step) => step.stepCount >= maxSteps
-    })
+      stopWhen: (step) => step.stepCount >= maxSteps,
+    });
 
-    const messages = [
-      ...this._cleanMessages(this._messages)
-    ]
+    const messages = this._cleanMessages(this._messages);
+    const prunedMessages = pruneMessages({
+      messages,
+      reasoning: 'all', // Remove all reasoning parts
+      toolCalls: 'before-last-2-messages', // Remove tool calls except those in the last message
+    });
 
     try {
       // 使用 runWithContext 让工具执行时能获取 sessionId
       const result = await framework.runWithContext(context, async () => {
-        return agent.generate({ messages, ...this.providerOptions })
-      })
+        return agent.generate({ messages: prunedMessages, ...this.providerOptions });
+      });
 
       // 处理返回的消息历史：只保留 user 和 assistant 消息，让 SDK 自动处理 tool 消息
       if (result.messages && Array.isArray(result.messages)) {
         // 只保留 user 和 assistant 消息，SDK 会自动维护 tool 消息
-        this._messages = result.messages.filter(m => m.role === 'user' || m.role === 'assistant')
+        this._messages = result.messages.filter((m) => m.role === 'user' || m.role === 'assistant');
       } else if (result.text) {
         this._messages.push({
           role: 'assistant',
-          content: result.text
-        })
+          content: result.text,
+        });
       }
 
       // 生成后检查：如果消息太长，下次需要压缩
-      const afterTokens = this._countMessagesTokens(this._messages)
+      const afterTokens = this._countMessagesTokens(this._messages);
       if (afterTokens > this._maxContextTokens * 0.8) {
-        console.log(`[AgentChat] After generation: ${afterTokens} tokens, will compress on next turn`)
+        logger.info(`After generation: ${afterTokens} tokens, will compress on next turn`);
       }
 
       return {
         success: true,
         message: result.text || '',
-        stepCount: result.stepCount || 1
-      }
+        stepCount: result.stepCount || 1,
+      };
     } catch (err) {
-      this.emit('error', { error: err.message })
+      this.emit('error', { error: err.message });
       // 抛出错误而不是返回错误响应，让 Agent 能捕获
-      throw err
+      throw err;
     }
   }
 
@@ -513,98 +659,96 @@ ${truncatedContent}${truncatedNote}
    * @param {Object} options - 选项
    */
   async *chatStream(message, options = {}) {
-    const context = { sessionId: options.sessionId || null, isStream: true }
-    const framework = this.agent.framework
+    const context = { sessionId: options.sessionId || null, isStream: true };
+    const framework = this.agent.framework;
 
+    // 关键：每次 chat 调用时刷新系统提示词，确保包含最新的工具/技能描述
+    // 解决上下文过长时 LLM 不调用工具的问题
+    this._systemPrompt = this.agent._buildSystemPrompt();
     // 动态导入 AI SDK
-    const { tool, ToolLoopAgent } = await this._importAI()
-
-    const userMessage = typeof message === 'string'
-      ? { role: 'user', content: message }
-      : message
-
-    this._messages.push(userMessage)
+    const { tool, ToolLoopAgent } = await this._importAI();
 
+    const userMessage = typeof message === 'string' ? { role: 'user', content: message } : message;
+    // console.log('System Prompt:', this._systemPrompt);
+    this._messages.push(userMessage);
     // 检查是否需要压缩上下文（包括工具定义）
-    const messagesTokens = this._countMessagesTokens(this._messages)
-    const toolsTokens = this._countToolsTokens()
-    const systemPromptTokens = this._countTokens(this._systemPrompt)
-    const totalTokens = messagesTokens + toolsTokens + systemPromptTokens
-    const limit = this._maxContextTokens * 0.7  // 降低到 70%
+    const messagesTokens = this._countMessagesTokens(this._messages);
+    const toolsTokens = this._countToolsTokens();
+    const systemPromptTokens = this._countTokens(this._systemPrompt);
+    const totalTokens = messagesTokens + toolsTokens + systemPromptTokens;
+    const limit = this._maxContextTokens * 0.7; // 降低到 70%
 
     // 对于流式调用，如果上下文太大，先压缩再开始
     if (totalTokens > limit) {
-      console.log(`[AgentChat] Context large (${totalTokens}/${this._maxContextTokens} tokens), compressing...`)
-      // 流式调用时等待压缩完成
-      await this._compressContext()
+      logger.info(
+        `Context large (${totalTokens}/${this._maxContextTokens} tokens), compressing...`
+      );
+      // 流式调用时等待压缩完成（使用带超时控制的压缩）
+      await this._compressContext();
     }
 
-    const maxSteps = options.maxSteps || this._maxSteps
-    const tools = this._getAITools(tool)
-    const self = this  // 保存引用用于回调
+    const maxSteps = options.maxSteps || this._maxSteps;
+    const tools = this._getAITools(tool);
+    const self = this; // 保存引用用于回调
 
     if (!this._aiClient) {
-      throw new Error('AI client not configured.')
+      throw new Error('AI client not configured.');
     }
 
     const agent = new ToolLoopAgent({
       model: this._aiClient,
       instructions: this._systemPrompt,
       tools: tools,
-      stopWhen: (step) => step.stepCount >= maxSteps
-    })
-
-    const messages = [
-      ...this._cleanMessages(this._messages)
-    ]
-
+      stopWhen: (step) => step.stepCount >= maxSteps,
+    });
+
+    const messages = this._cleanMessages(this._messages);
+    const prunedMessages = pruneMessages({
+      messages,
+      reasoning: 'all', // Remove all reasoning parts
+      toolCalls: 'before-last-2-messages', // Remove tool calls except those in the last message
+    });
     try {
       // 使用 runWithContext 让工具执行时能获取 sessionId（支持并行）
       const result = await framework.runWithContext(context, async () => {
-        return agent.stream({ messages, ...this.providerOptions })
-      })
+        return agent.stream({ messages: prunedMessages, ...this.providerOptions });
+      });
 
-      const stream = result.fullStream
-      let fullText = ''
+      const stream = result.fullStream;
+      let fullText = '';
 
       // 流式迭代器
-      const iterator = stream[Symbol.asyncIterator] ? stream : stream.fullStream
-      const finalMessages = []
+      const iterator = stream[Symbol.asyncIterator] ? stream : stream.fullStream;
+      const finalMessages = [];
 
-      for await (const part of (iterator || stream)) {
+      for await (const part of iterator || stream) {
         if (part.type === 'text-delta') {
-          const text = part.text || part.textDelta || ''
-          fullText += text
-          yield { type: 'text', text }
+          const text = part.text || part.textDelta || '';
+          fullText += text;
+          yield { type: 'text', text };
         } else if (part.type === 'reasoning') {
-          this.emit('thinking', { content: part.text })
+          finalMessages.push(part);
+          this.emit('thinking', { content: part.text });
         } else if (part.type === 'tool-call') {
-          yield { type: 'tool-call', toolName: part.toolName, args: part.input }
+          finalMessages.push(part);
+          yield { type: 'tool-call', toolName: part.toolName, args: part.input };
         } else if (part.type === 'tool-result') {
-          // 保存到临时消息列表
-          finalMessages.push({ role: 'tool', content: part.output, toolName: part.toolName })
-          yield { type: 'tool-result', toolName: part.toolName, result: part.output }
+          // AI SDK 6.x 要求 tool 消息格式为:
+          // { role: 'tool', content: [{ type: 'tool-result', toolCallId, toolName, output }] }
+          finalMessages.push(part); // 先保存到 finalMessages，等生成结束后统一添加到历史
+          yield { type: 'tool-result', toolName: part.toolName, result: part.output };
         } else if (part.type === 'error') {
-          yield { type: 'error', error: part.error }
+          yield { type: 'error', error: part.error };
         }
       }
 
-      // 暂时禁用压缩以调试 schema 问题
-      // for (const msg of finalMessages) {
-      //   if (msg.content) {
-      //     const contentSize = typeof msg.content === 'string' ? msg.content.length : JSON.stringify(msg.content).length
-      //     if (contentSize > self._maxToolResultSize) {
-      //       msg.content = await self._compressToolResult(msg.content)
-      //     }
-      //   }
-      // }
-
-      // 只保留 user 和 assistant 消息，让 SDK 自动处理 tool 消息
-      const assistantMsg = { role: 'assistant', content: fullText }
-      this._messages.push(assistantMsg)
+      this._messages.push({ role: 'tool', content: finalMessages });
+      // 添加 assistant 消息
+      const assistantMsg = { role: 'assistant', content: fullText };
+      this._messages.push(assistantMsg);
     } catch (err) {
-      this.emit('error', { error: err.message })
-      yield { type: 'error', error: err.message }
+      this.emit('error', { error: err.message });
+      yield { type: 'error', error: err.message };
     }
   }
 
@@ -615,10 +759,10 @@ ${truncatedContent}${truncatedNote}
    * @private
    */
   _getAITools(toolFn) {
-    const tools = {}
+    const tools = {};
 
     for (const [name, toolDef] of this._tools) {
-      const toolName = toolDef.name || name
+      const toolName = toolDef.name || name;
 
       // 使用 AI SDK 的 tool() 格式
       // 支持 inputSchema (zod schema) 或 parameters (旧格式)
@@ -627,33 +771,33 @@ ${truncatedContent}${truncatedNote}
         description: toolDef.description || '',
         execute: async (args) => {
           // 清理参数：移除 undefined、function 等无效值
-          const cleanedArgs = this._cleanToolArgs(args)
+          const cleanedArgs = this._cleanToolArgs(args);
 
           // 执行工具
-          this.emit('tool-call', { name: toolName, args: cleanedArgs })
+          this.emit('tool-call', { name: toolName, args: cleanedArgs });
           try {
-            const result = await toolDef.execute(cleanedArgs, this.agent.framework)
-            this.emit('tool-result', { name: toolName, args: cleanedArgs, result })
-            return result
+            const result = await toolDef.execute(cleanedArgs, this.agent.framework);
+            this.emit('tool-result', { name: toolName, args: cleanedArgs, result });
+            return result;
           } catch (err) {
-            this.emit('tool-error', { name: toolName, args: cleanedArgs, error: err.message })
-            return { error: err.message }
+            this.emit('tool-error', { name: toolName, args: cleanedArgs, error: err.message });
+            return { error: err.message };
           }
-        }
-      }
+        },
+      };
 
       // 支持 inputSchema (zod) 或 parameters (旧格式)
       if (toolDef.inputSchema) {
-        toolConfig.inputSchema = toolDef.inputSchema
+        toolConfig.inputSchema = toolDef.inputSchema;
       } else if (toolDef.parameters) {
-        toolConfig.parameters = toolDef.parameters
+        toolConfig.parameters = toolDef.parameters;
       }
 
       // AI SDK 6.x 使用对象形式，键为工具名
-      tools[toolName] = toolFn(toolConfig)
+      tools[toolName] = toolFn(toolConfig);
     }
 
-    return tools
+    return tools;
   }
 
   /**
@@ -664,32 +808,32 @@ ${truncatedContent}${truncatedNote}
    */
   _cleanToolArgs(args) {
     if (!args || typeof args !== 'object') {
-      return {}
+      return {};
     }
 
-    const cleaned = {}
+    const cleaned = {};
     for (const [key, value] of Object.entries(args)) {
       // 跳过 undefined、function、symbol 等无效值
       if (value === undefined || value === null) {
-        continue
+        continue;
       }
       if (typeof value === 'function' || typeof value === 'symbol') {
-        continue
+        continue;
       }
       // 递归清理嵌套对象
       if (typeof value === 'object' && !Array.isArray(value)) {
-        cleaned[key] = this._cleanToolArgs(value)
+        cleaned[key] = this._cleanToolArgs(value);
       } else if (Array.isArray(value)) {
-        cleaned[key] = value.map(item =>
-          typeof item === 'object' && item !== null
-            ? this._cleanToolArgs(item)
-            : item
-        ).filter(item => item !== undefined && typeof item !== 'function')
-      } else {
         cleaned[key] = value
+          .map((item) =>
+            typeof item === 'object' && item !== null ? this._cleanToolArgs(item) : item
+          )
+          .filter((item) => item !== undefined && typeof item !== 'function');
+      } else {
+        cleaned[key] = value;
       }
     }
-    return cleaned
+    return cleaned;
   }
 
   /**
@@ -697,39 +841,42 @@ ${truncatedContent}${truncatedNote}
    * @private
    */
   _cleanMessages(messages) {
-    return messages.map(msg => {
+    return messages.map((msg) => {
       if (!msg || typeof msg !== 'object') {
-        return { role: 'user', content: '' }
+        return { role: 'user', content: '' };
       }
 
       const cleaned = {
-        role: msg.role || 'user'
-      }
+        role: msg.role || 'user',
+      };
 
       if (Array.isArray(msg.content)) {
-        cleaned.content = msg.content
-      } else if (msg.content !== undefined) {
-        cleaned.content = msg.content
+        cleaned.content = msg.content;
+      } else if (typeof msg.content === 'string') {
+        cleaned.content = msg.content;
+      } else if (typeof msg.content === 'object' && msg.content !== null) {
+        // 对象类型的 content（如 tool result），转为字符串
+        cleaned.content =
+          typeof msg.content === 'object' && msg.content.content !== undefined
+            ? String(msg.content.content)
+            : JSON.stringify(msg.content);
       } else {
-        cleaned.content = msg.text || msg.input || ''
+        cleaned.content = msg.text || msg.input || '';
       }
 
-      return cleaned
-    })
+      return cleaned;
+    });
   }
 
   /**
    * 销毁
    */
   destroy() {
-    this._messages = []
-    this._tools.clear()
-    if (this._encoder) {
-      this._encoder.free()
-      this._encoder = null
-    }
-    this.removeAllListeners()
+    this._messages = [];
+    this._tools.clear();
+    this._encoder = null;
+    this.removeAllListeners();
   }
 }
 
-module.exports = { AgentChatHandler }
+module.exports = { AgentChatHandler };