foliko 1.0.74 → 1.0.76

package/src/core/agent-chat.js

@@ -1,735 +1,882 @@
-/**
- * AgentChatHandler 聊天处理器
- * 使用 AI SDK 的 ToolLoopAgent 处理工具调用循环
- */
-
-const { EventEmitter } = require('../utils/event-emitter')
-const tiktoken = require('tiktoken')
-
-// 模型上下文限制表（留 15-20% 余量给 system prompt 和输出）
-const MODEL_CONTEXT_LIMITS = {
-  // DeepSeek
-  'deepseek-chat': 28000,
-  'deepseek-coder': 28000,
-  // MiniMax
-  'MiniMax-M2.7': 90000,
-  // OpenAI
-  'gpt-4': 100000,
-  'gpt-4o': 100000,
-  'gpt-4o-mini': 100000,
-  'gpt-4-turbo': 100000,
-  // Anthropic
-  'claude-3-5-sonnet': 150000,
-  'claude-3-opus': 150000,
-  'claude-3-sonnet': 150000,
-}
-
-class AgentChatHandler extends EventEmitter {
-  /**
-   * @param {Agent} agent - Agent 实例
-   * @param {Object} config - 配置
-   */
-  constructor(agent, config = {}) {
-    super()
-
-    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._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 =>
-      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 // 压缩次数统计
-
-    // 工具结果压缩配置
-    this._maxToolResultSize = config.maxToolResultSize || 4000  // 工具结果超过此大小则压缩（字节）
-
-    // 初始化编码器
-    this._initEncoder()
-  }
-
-  /**
-   * 初始化 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)
-    }
-  }
-
-  /**
-   * 计算文本的 token 数
-   * @param {string} text
-   * @returns {number}
-   * @private
-   */
-  _countTokens(text) {
-    if (!this._encoder || !text) return 0
-    try {
-      return this._encoder.encode(text).length
-    } catch (err) {
-      // 粗略估算：约 4 字符 = 1 token
-      return Math.ceil(text.length / 4)
-    }
-  }
-
-  /**
-   * 计算消息列表的总 token 数
-   * @param {Array} messages
-   * @returns {number}
-   * @private
-   */
-  _countMessagesTokens(messages) {
-    let total = 0
-    for (const msg of messages) {
-      total += 4 // role 标记
-      if (typeof msg.content === 'string') {
-        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)
-          }
-        }
-      }
-    }
-    total += 4 // 结尾标记
-    return total
-  }
-
-  /**
-   * 检查是否需要压缩上下文
-   * @returns {boolean}
-   * @private
-   */
-  _shouldCompress() {
-    const totalTokens = this._countMessagesTokens(this._messages)
-    return totalTokens > this._maxContextTokens * this._compressionThreshold
-  }
-
-  /**
-   * 计算工具定义的 token 数（估算）
-   * @returns {number}
-   * @private
-   */
-  _countToolsTokens() {
-    let total = 0
-    for (const toolDef of this._tools.values()) {
-      // 工具名 + 描述
-      total += this._countTokens(toolDef.name || '')
-      total += this._countTokens(toolDef.description || '')
-
-      // 工具参数 schema
-      if (toolDef.inputSchema) {
-        const schemaStr = typeof toolDef.inputSchema === 'string'
-          ? toolDef.inputSchema
-          : JSON.stringify(toolDef.inputSchema)
-        total += this._countTokens(schemaStr)
-      }
-    }
-    return total
-  }
-
-  /**
-   * 检查是否需要压缩（包括工具定义）
-   * @returns {boolean}
-   * @private
-   */
-  _shouldCompressWithTools() {
-    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
-  }
-
-  /**
-   * 压缩上下文消息（智能摘要模式）
-   * 策略：
-   * 1. 如果启用了智能摘要且有 AI 客户端，对早期消息进行 AI 总结
-   * 2. 否则使用简单裁剪 + 标记
-   * @private
-   */
-  async _compressContext() {
-    if (this._messages.length <= this._keepRecentMessages) {
-      return
-    }
-
-    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 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)`)
-      } catch (err) {
-        console.warn('[AgentChat] AI 摘要失败，使用简单压缩:', err.message)
-        summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`
-      }
-    } else {
-      summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`
-    }
-
-    const summary = {
-      role: 'system',
-      content: summaryContent
-    }
-
-    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}`)
-  }
-
-  /**
-   * 使用 AI 对消息进行总结
-   * @param {Array} messages - 要总结的消息
-   * @returns {Promise<string>} 总结文本
-   * @private
-   */
-  async _summarizeMessages(messages) {
-    if (!this._aiClient || messages.length === 0) {
-      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 summarizePrompt = `请简洁地总结以下对话的要点，保留关键信息和用户需求：
-
-${conversationText}
-
-总结要求：
-1. 提取用户的主要需求和意图
-2. 保留关键的技术细节或决策
-3. 不要超过 1000 字
-4. 用中文回复`
-
-    // 使用 AI SDK 6.x 的 generateText
-    const { generateText } = require('ai')
-    const { text } = await generateText({
-      model: this._aiClient,
-      prompt: summarizePrompt,
-      ...this.providerOptions
-    })
-
-    return text || '（总结生成失败）'
-  }
-
-  /**
-   * 检查工具结果是否需要压缩
-   * @param {any} result - 工具返回结果
-   * @returns {boolean}
-   * @private
-   */
-  _shouldCompressToolResult(result) {
-    if (!result || this._maxToolResultSize <= 0) return false
-
-    // 计算结果的大小
-    let size = 0
-    if (typeof result === 'string') {
-      size = result.length
-    } else if (typeof result === 'object') {
-      try {
-        size = JSON.stringify(result).length
-      } catch {
-        size = String(result).length
-      }
-    }
-
-    return size > this._maxToolResultSize
-  }
-
-  /**
-   * 压缩工具结果
-   * @param {any} result - 工具返回结果
-   * @returns {Promise<any>} 压缩后的结果
-   * @private
-   */
-  async _compressToolResult(result) {
-    if (!this._shouldCompressToolResult(result)) {
-      return result
-    }
-
-    if (!this._aiClient) {
-      console.warn('[AgentChat] 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 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 compressPrompt = `以下是一个工具执行结果${contentTypeHint}，长度 ${originalSize} 字符。请简洁地总结其核心内容：
-
-${truncatedContent}${truncatedNote}
-
-请提取并保留：
-1. 主要标题和主题
-2. 关键信息点（不超过 5 个）
-3. 重要数据或结论
-
-用简洁的中文总结，不超过 400 字：`
-
-      // 使用 AI SDK 6.x 的 generateText
-      const { generateText } = require('ai')
-      const { text } = await generateText({
-        model: this._aiClient,
-        prompt: compressPrompt,
-        ...this.providerOptions,
-        maxTokens: 500
-      })
-
-      const summary = text || '（总结生成失败）'
-      const compressed = `[工具结果已压缩${contentTypeHint}: ${originalSize} → ${summary.length} 字符]\n\n${summary}`
-
-      console.log(`[AgentChat] Tool result compressed: ${originalSize} → ${summary.length} chars`)
-      return compressed
-    } catch (err) {
-      console.warn('[AgentChat] Tool result compression failed:', err.message)
-      return this._fallbackCompress(result)
-    }
-  }
-
-  /**
-   * 回退压缩方法（当 AI 客户端不可用时）
-   * @param {any} result - 工具返回结果
-   * @returns {string} 压缩后的结果
-   * @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)
-
-    // 简单截断策略：保留前 2000 字符
-    const maxSize = 2000
-    if (resultStr.length <= maxSize) {
-      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
-  }
-
-  /**
-   * 设置 AI 客户端
-   * @param {Object} client - AI 模型客户端
-   */
-  setAIClient(client) {
-    this._aiClient = client
-    return this
-  }
-
-  /**
-   * 设置系统提示
-   * @param {string} prompt
-   */
-  setSystemPrompt(prompt) {
-    this._systemPrompt = prompt
-    return this
-  }
-
-  /**
-   * 注册工具
-   * @param {Object} toolDef - 工具定义
-   */
-  registerTool(toolDef) {
-    this._tools.set(toolDef.name, toolDef)
-    return this
-  }
-
-  /**
-   * 清空对话历史
-   */
-  clearHistory() {
-    this._messages = []
-    this._compressionCount = 0
-    return this
-  }
-
-  /**
-   * 获取已注册的工具
-   * @returns {Array}
-   */
-  getTools() {
-    return Array.from(this._tools.values())
-  }
-
-  /**
-   * 导入 AI SDK（动态导入）
-   * @private
-   */
-  async _importAI() {
-    try {
-      const ai = require('ai')
-      return {
-        tool: ai.tool,
-        ToolLoopAgent: ai.ToolLoopAgent
-      }
-    } catch (err) {
-      throw new Error('AI SDK not found. Please install: npm install ai')
-    }
-  }
-
-  /**
-   * 发送消息（非流式）
-   * @param {string|Object} message - 消息
-   * @param {Object} options - 选项
-   */
-  async chat(message, options = {}) {
-    const context = { sessionId: options.sessionId || null, isStream: false }
-    const framework = this.agent.framework
-    const self = this  // 保存引用用于回调
-
-    // 动态导入 AI SDK
-    const { tool, ToolLoopAgent } = await this._importAI()
-
-    const userMessage = typeof message === 'string'
-      ? { role: 'user', content: message }
-      : message
-
-    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%，更早压缩
-
-    if (totalTokens > limit) {
-      console.log(`[AgentChat] 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`)
-    }
-
-    const maxSteps = options.maxSteps || this._maxSteps
-    const tools = this._getAITools(tool)
-
-    if (!this._aiClient) {
-      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)
-    ]
-
-    try {
-      // 使用 runWithContext 让工具执行时能获取 sessionId
-      const result = await framework.runWithContext(context, async () => {
-        return agent.generate({ messages, ...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')
-      } else if (result.text) {
-        this._messages.push({
-          role: 'assistant',
-          content: result.text
-        })
-      }
-
-      // 生成后检查：如果消息太长，下次需要压缩
-      const afterTokens = this._countMessagesTokens(this._messages)
-      if (afterTokens > this._maxContextTokens * 0.8) {
-        console.log(`[AgentChat] After generation: ${afterTokens} tokens, will compress on next turn`)
-      }
-
-      return {
-        success: true,
-        message: result.text || '',
-        stepCount: result.stepCount || 1
-      }
-    } catch (err) {
-      this.emit('error', { error: err.message })
-      // 抛出错误而不是返回错误响应，让 Agent 能捕获
-      throw err
-    }
-  }
-
-  /**
-   * 发送消息（流式）
-   * @param {string|Object} message - 消息
-   * @param {Object} options - 选项
-   */
-  async *chatStream(message, options = {}) {
-    const context = { sessionId: options.sessionId || null, isStream: true }
-    const framework = this.agent.framework
-
-    // 动态导入 AI SDK
-    const { tool, ToolLoopAgent } = await this._importAI()
-
-    const userMessage = typeof message === 'string'
-      ? { role: 'user', content: message }
-      : message
-
-    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%
-
-    // 对于流式调用，如果上下文太大，先压缩再开始
-    if (totalTokens > limit) {
-      console.log(`[AgentChat] Context large (${totalTokens}/${this._maxContextTokens} tokens), compressing...`)
-      // 流式调用时等待压缩完成
-      await this._compressContext()
-    }
-
-    const maxSteps = options.maxSteps || this._maxSteps
-    const tools = this._getAITools(tool)
-    const self = this  // 保存引用用于回调
-
-    if (!this._aiClient) {
-      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)
-    ]
-
-    try {
-      // 使用 runWithContext 让工具执行时能获取 sessionId（支持并行）
-      const result = await framework.runWithContext(context, async () => {
-        return agent.stream({ messages, ...this.providerOptions })
-      })
-
-      const stream = result.fullStream
-      let fullText = ''
-
-      // 流式迭代器
-      const iterator = stream[Symbol.asyncIterator] ? stream : stream.fullStream
-      const finalMessages = []
-
-      for await (const part of (iterator || stream)) {
-        if (part.type === 'text-delta') {
-          const text = part.text || part.textDelta || ''
-          fullText += text
-          yield { type: 'text', text }
-        } else if (part.type === 'reasoning') {
-          this.emit('thinking', { content: part.text })
-        } else if (part.type === 'tool-call') {
-          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 }
-        } else if (part.type === '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)
-    } catch (err) {
-      this.emit('error', { error: err.message })
-      yield { type: 'error', error: err.message }
-    }
-  }
-
-  /**
-   * 获取 AI SDK 格式的工具
-   * AI SDK 6.x 需要对象形式 { toolName: tool }
-   * @param {Function} toolFn - AI SDK 的 tool 函数
-   * @private
-   */
-  _getAITools(toolFn) {
-    const tools = {}
-
-    for (const [name, toolDef] of this._tools) {
-      const toolName = toolDef.name || name
-
-      // 使用 AI SDK 的 tool() 格式
-      // 支持 inputSchema (zod schema) 或 parameters (旧格式)
-      const toolConfig = {
-        name: toolName,
-        description: toolDef.description || '',
-        execute: async (args) => {
-          // 清理参数：移除 undefined、function 等无效值
-          const cleanedArgs = this._cleanToolArgs(args)
-
-          // 执行工具
-          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
-          } catch (err) {
-            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
-      } else if (toolDef.parameters) {
-        toolConfig.parameters = toolDef.parameters
-      }
-
-      // AI SDK 6.x 使用对象形式，键为工具名
-      tools[toolName] = toolFn(toolConfig)
-    }
-
-    return tools
-  }
-
-  /**
-   * 清理工具参数，移除无效值
-   * @param {Object} args - 原始参数
-   * @returns {Object} 清理后的参数
-   * @private
-   */
-  _cleanToolArgs(args) {
-    if (!args || typeof args !== 'object') {
-      return {}
-    }
-
-    const cleaned = {}
-    for (const [key, value] of Object.entries(args)) {
-      // 跳过 undefined、function、symbol 等无效值
-      if (value === undefined || value === null) {
-        continue
-      }
-      if (typeof value === 'function' || typeof value === 'symbol') {
-        continue
-      }
-      // 递归清理嵌套对象
-      if (typeof value === 'object' && !Array.isArray(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
-      }
-    }
-    return cleaned
-  }
-
-  /**
-   * 清理消息格式
-   * @private
-   */
-  _cleanMessages(messages) {
-    return messages.map(msg => {
-      if (!msg || typeof msg !== 'object') {
-        return { role: 'user', content: '' }
-      }
-
-      const cleaned = {
-        role: msg.role || 'user'
-      }
-
-      if (Array.isArray(msg.content)) {
-        cleaned.content = msg.content
-      } else if (msg.content !== undefined) {
-        cleaned.content = msg.content
-      } else {
-        cleaned.content = msg.text || msg.input || ''
-      }
-
-      return cleaned
-    })
-  }
-
-  /**
-   * 销毁
-   */
-  destroy() {
-    this._messages = []
-    this._tools.clear()
-    if (this._encoder) {
-      this._encoder.free()
-      this._encoder = null
-    }
-    this.removeAllListeners()
-  }
-}
-
-module.exports = { AgentChatHandler }
+/**
+ * AgentChatHandler 聊天处理器
+ * 使用 AI SDK 的 ToolLoopAgent 处理工具调用循环
+ */
+
+const { EventEmitter } = require('../utils/event-emitter');
+const { logger } = require('../utils/logger');
+const { generateText, pruneMessages } = require('ai');
+
+// 模型上下文限制表（留 15-20% 余量给 system prompt 和输出）
+const MODEL_CONTEXT_LIMITS = {
+  // DeepSeek
+  'deepseek-chat': 28000,
+  'deepseek-coder': 28000,
+  // MiniMax
+  'MiniMax-M2.7': 90000,
+  // OpenAI
+  'gpt-4': 100000,
+  'gpt-4o': 100000,
+  'gpt-4o-mini': 100000,
+  'gpt-4-turbo': 100000,
+  // Anthropic
+  '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();
+
+    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._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) =>
+      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; // 压缩次数统计
+    this._compressionInProgress = false; // 压缩锁，防止并发压缩
+    this._compressionPromise = null; // 正在进行的压缩 Promise
+
+    // 工具结果压缩配置
+    this._maxToolResultSize = config.maxToolResultSize || 4000; // 工具结果超过此大小则压缩（字节）
+
+    // 初始化编码器
+    // 使用纯 JS tokenizer
+    this._encoder = _globalTokenizer;
+  }
+
+  /**
+   * 静态方法：保留接口兼容性（无实际作用）
+   */
+  static clearEncoderCache() {
+    // 不再需要清理
+  }
+
+  /**
+   * 计算文本的 token 数
+   * @param {string} text
+   * @returns {number}
+   * @private
+   */
+  _countTokens(text) {
+    if (!text || typeof text !== 'string') return 0;
+    // SimpleTokenizer.encode 已经处理了清理逻辑
+    try {
+      return this._encoder.encode(text);
+    } catch (err) {
+      // 估算失败时使用字符计数
+      return Math.ceil(text.length / 4);
+    }
+  }
+
+  /**
+   * 计算消息列表的总 token 数
+   * @param {Array} messages
+   * @returns {number}
+   * @private
+   */
+  _countMessagesTokens(messages) {
+    let total = 0;
+    for (const msg of messages) {
+      if (!msg) continue;
+      total += 4; // role 标记
+      if (typeof msg.content === 'string') {
+        total += this._countTokens(msg.content);
+      } else if (Array.isArray(msg.content)) {
+        for (const part of msg.content) {
+          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;
+  }
+
+  /**
+   * 检查是否需要压缩上下文
+   * @returns {boolean}
+   * @private
+   */
+  _shouldCompress() {
+    const totalTokens = this._countMessagesTokens(this._messages);
+    return totalTokens > this._maxContextTokens * this._compressionThreshold;
+  }
+
+  /**
+   * 计算工具定义的 token 数（估算）
+   * @returns {number}
+   * @private
+   */
+  _countToolsTokens() {
+    let total = 0;
+    for (const toolDef of this._tools.values()) {
+      // 工具名 + 描述
+      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);
+      }
+    }
+    return total;
+  }
+
+  /**
+   * 检查是否需要压缩（包括工具定义）
+   * @returns {boolean}
+   * @private
+   */
+  _shouldCompressWithTools() {
+    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;
+  }
+
+  /**
+   * 压缩上下文消息（智能摘要模式，支持超时控制）
+   * 策略：
+   * 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;
+    }
+
+    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);
+    });
+  }
+
+  /**
+   * 执行实际压缩逻辑
+   * @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 compressedCount = messagesToSummarize.length;
+    let summaryContent = '';
+
+    // 尝试使用 AI 总结
+    if (this._enableSmartCompress && this._aiClient) {
+      try {
+        const summaryText = await this._summarizeMessages(messagesToSummarize);
+        summaryContent = `[早期对话摘要]: ${summaryText}`;
+        logger.info(`AI summary generated (${summaryText.length} chars)`);
+      } catch (err) {
+        logger.warn('AI summary failed, using simple compression:', err.message);
+        summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`;
+      }
+    } else {
+      summaryContent = `[上下文已压缩: 省略了 ${compressedCount} 条早期消息。保留了最近 ${this._keepRecentMessages} 条对话记录。]`;
+    }
+
+    const summary = {
+      role: 'system',
+      content: summaryContent,
+    };
+
+    this._messages = [...systemMessages, summary, ...recentMessages];
+    this._compressionCount++;
+
+    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}`
+    );
+  }
+
+  /**
+   * 使用 AI 对消息进行总结
+   * @param {Array} messages - 要总结的消息
+   * @returns {Promise<string>} 总结文本
+   * @private
+   */
+  async _summarizeMessages(messages) {
+    if (!this._aiClient || messages.length === 0) {
+      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 summarizePrompt = `请简洁地总结以下对话的要点，保留关键信息和用户需求：
+
+${conversationText}
+
+总结要求：
+1. 提取用户的主要需求和意图
+2. 保留关键的技术细节或决策
+3. 不要超过 1000 字
+4. 用中文回复`;
+
+    // 使用 AI SDK 6.x 的 generateText
+    const { text } = await generateText({
+      model: this._aiClient,
+      prompt: summarizePrompt,
+      ...this.providerOptions,
+    });
+
+    return text || '（总结生成失败）';
+  }
+
+  /**
+   * 检查工具结果是否需要压缩
+   * @param {any} result - 工具返回结果
+   * @returns {boolean}
+   * @private
+   */
+  _shouldCompressToolResult(result) {
+    if (!result || this._maxToolResultSize <= 0) return false;
+
+    // 计算结果的大小
+    let size = 0;
+    if (typeof result === 'string') {
+      size = result.length;
+    } else if (typeof result === 'object') {
+      try {
+        size = JSON.stringify(result).length;
+      } catch {
+        size = String(result).length;
+      }
+    }
+
+    return size > this._maxToolResultSize;
+  }
+
+  /**
+   * 压缩工具结果
+   * @param {any} result - 工具返回结果
+   * @returns {Promise<any>} 压缩后的结果
+   * @private
+   */
+  async _compressToolResult(result) {
+    if (!this._shouldCompressToolResult(result)) {
+      return result;
+    }
+
+    if (!this._aiClient) {
+      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 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 compressPrompt = `以下是一个工具执行结果${contentTypeHint}，长度 ${originalSize} 字符。请简洁地总结其核心内容：
+
+${truncatedContent}${truncatedNote}
+
+请提取并保留：
+1. 主要标题和主题
+2. 关键信息点（不超过 5 个）
+3. 重要数据或结论
+
+用简洁的中文总结，不超过 400 字：`;
+
+      // 使用 AI SDK 6.x 的 generateText
+      const { text } = await generateText({
+        model: this._aiClient,
+        prompt: compressPrompt,
+        ...this.providerOptions,
+        maxTokens: 500,
+      });
+
+      const summary = text || '（总结生成失败）';
+      const compressed = `[工具结果已压缩${contentTypeHint}: ${originalSize} -> ${summary.length} 字符]\n\n${summary}`;
+
+      logger.info(`Tool result compressed: ${originalSize} -> ${summary.length} chars`);
+      return compressed;
+    } catch (err) {
+      logger.warn('Tool result compression failed:', err.message);
+      return this._fallbackCompress(result);
+    }
+  }
+
+  /**
+   * 回退压缩方法（当 AI 客户端不可用时）
+   * @param {any} result - 工具返回结果
+   * @returns {string} 压缩后的结果
+   * @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);
+
+    // 简单截断策略：保留前 2000 字符
+    const maxSize = 2000;
+    if (resultStr.length <= maxSize) {
+      return result;
+    }
+
+    const compressed = `[工具结果已压缩（简单截断）: ${originalSize} -> ${maxSize} 字符]\n\n${resultStr.substring(0, maxSize)}\n\n...[内容已截断，原文 ${originalSize} 字符]`;
+    logger.info(`Tool result fallback compressed: ${originalSize} -> ${maxSize} chars`);
+    return compressed;
+  }
+
+  /**
+   * 设置 AI 客户端
+   * @param {Object} client - AI 模型客户端
+   */
+  setAIClient(client) {
+    this._aiClient = client;
+    return this;
+  }
+
+  /**
+   * 设置系统提示
+   * @param {string} prompt
+   */
+  setSystemPrompt(prompt) {
+    this._systemPrompt = prompt;
+    return this;
+  }
+
+  /**
+   * 注册工具
+   * @param {Object} toolDef - 工具定义
+   */
+  registerTool(toolDef) {
+    this._tools.set(toolDef.name, toolDef);
+    return this;
+  }
+
+  /**
+   * 清空对话历史
+   */
+  clearHistory() {
+    this._messages = [];
+    this._compressionCount = 0;
+    return this;
+  }
+
+  /**
+   * 获取已注册的工具
+   * @returns {Array}
+   */
+  getTools() {
+    return Array.from(this._tools.values());
+  }
+
+  /**
+   * 导入 AI SDK（动态导入）
+   * @private
+   */
+  async _importAI() {
+    try {
+      const ai = require('ai');
+      return {
+        tool: ai.tool,
+        ToolLoopAgent: ai.ToolLoopAgent,
+      };
+    } catch (err) {
+      throw new Error('AI SDK not found. Please install: npm install ai');
+    }
+  }
+
+  /**
+   * 发送消息（非流式）
+   * @param {string|Object} message - 消息
+   * @param {Object} options - 选项
+   */
+  async chat(message, options = {}) {
+    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 userMessage = typeof message === 'string' ? { role: 'user', content: message } : message;
+
+    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%，更早压缩
+
+    if (totalTokens > limit) {
+      logger.info(
+        `Context large (${totalTokens}/${this._maxContextTokens} tokens = msgs:${messagesTokens} + tools:${toolsTokens} + sys:${systemPromptTokens}), compressing...`
+      );
+      // 使用带超时控制的压缩
+      await this._compressContext();
+    } else {
+      logger.info(`Context OK: ${totalTokens}/${this._maxContextTokens} tokens`);
+    }
+
+    const maxSteps = options.maxSteps || this._maxSteps;
+    const tools = this._getAITools(tool);
+
+    if (!this._aiClient) {
+      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);
+    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: 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');
+      } else if (result.text) {
+        this._messages.push({
+          role: 'assistant',
+          content: result.text,
+        });
+      }
+
+      // 生成后检查：如果消息太长，下次需要压缩
+      const afterTokens = this._countMessagesTokens(this._messages);
+      if (afterTokens > this._maxContextTokens * 0.8) {
+        logger.info(`After generation: ${afterTokens} tokens, will compress on next turn`);
+      }
+
+      return {
+        success: true,
+        message: result.text || '',
+        stepCount: result.stepCount || 1,
+      };
+    } catch (err) {
+      this.emit('error', { error: err.message });
+      // 抛出错误而不是返回错误响应，让 Agent 能捕获
+      throw err;
+    }
+  }
+
+  /**
+   * 发送消息（流式）
+   * @param {string|Object} message - 消息
+   * @param {Object} options - 选项
+   */
+  async *chatStream(message, options = {}) {
+    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;
+    // 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%
+
+    // 对于流式调用，如果上下文太大，先压缩再开始
+    if (totalTokens > limit) {
+      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; // 保存引用用于回调
+
+    if (!this._aiClient) {
+      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);
+    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: prunedMessages, ...this.providerOptions });
+      });
+
+      const stream = result.fullStream;
+      let fullText = '';
+
+      // 流式迭代器
+      const iterator = stream[Symbol.asyncIterator] ? stream : stream.fullStream;
+      const finalMessages = [];
+
+      for await (const part of iterator || stream) {
+        if (part.type === 'text-delta') {
+          const text = part.text || part.textDelta || '';
+          fullText += text;
+          yield { type: 'text', text };
+        } else if (part.type === 'reasoning') {
+          finalMessages.push(part);
+          this.emit('thinking', { content: part.text });
+        } else if (part.type === 'tool-call') {
+          finalMessages.push(part);
+          yield { type: 'tool-call', toolName: part.toolName, args: part.input };
+        } else if (part.type === 'tool-result') {
+          // 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 };
+        }
+      }
+
+      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 };
+    }
+  }
+
+  /**
+   * 获取 AI SDK 格式的工具
+   * AI SDK 6.x 需要对象形式 { toolName: tool }
+   * @param {Function} toolFn - AI SDK 的 tool 函数
+   * @private
+   */
+  _getAITools(toolFn) {
+    const tools = {};
+
+    for (const [name, toolDef] of this._tools) {
+      const toolName = toolDef.name || name;
+
+      // 使用 AI SDK 的 tool() 格式
+      // 支持 inputSchema (zod schema) 或 parameters (旧格式)
+      const toolConfig = {
+        name: toolName,
+        description: toolDef.description || '',
+        execute: async (args) => {
+          // 清理参数：移除 undefined、function 等无效值
+          const cleanedArgs = this._cleanToolArgs(args);
+
+          // 执行工具
+          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;
+          } catch (err) {
+            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;
+      } else if (toolDef.parameters) {
+        toolConfig.parameters = toolDef.parameters;
+      }
+
+      // AI SDK 6.x 使用对象形式，键为工具名
+      tools[toolName] = toolFn(toolConfig);
+    }
+
+    return tools;
+  }
+
+  /**
+   * 清理工具参数，移除无效值
+   * @param {Object} args - 原始参数
+   * @returns {Object} 清理后的参数
+   * @private
+   */
+  _cleanToolArgs(args) {
+    if (!args || typeof args !== 'object') {
+      return {};
+    }
+
+    const cleaned = {};
+    for (const [key, value] of Object.entries(args)) {
+      // 跳过 undefined、function、symbol 等无效值
+      if (value === undefined || value === null) {
+        continue;
+      }
+      if (typeof value === 'function' || typeof value === 'symbol') {
+        continue;
+      }
+      // 递归清理嵌套对象
+      if (typeof value === 'object' && !Array.isArray(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;
+      }
+    }
+    return cleaned;
+  }
+
+  /**
+   * 清理消息格式
+   * @private
+   */
+  _cleanMessages(messages) {
+    return messages.map((msg) => {
+      if (!msg || typeof msg !== 'object') {
+        return { role: 'user', content: '' };
+      }
+
+      const cleaned = {
+        role: msg.role || 'user',
+      };
+
+      if (Array.isArray(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 || '';
+      }
+
+      return cleaned;
+    });
+  }
+
+  /**
+   * 销毁
+   */
+  destroy() {
+    this._messages = [];
+    this._tools.clear();
+    this._encoder = null;
+    this.removeAllListeners();
+  }
+}
+
+module.exports = { AgentChatHandler };