foliko 1.0.0

package/src/core/agent-chat.js

@@ -0,0 +1,311 @@
+/**
+ * AgentChatHandler 聊天处理器
+ * 使用 AI SDK 的 ToolLoopAgent 处理工具调用循环
+ */
+
+const { EventEmitter } = require('../utils/event-emitter')
+
+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 = 20
+
+    // AI 客户端
+    this._aiClient = null
+  }
+
+  /**
+   * 设置 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 = []
+    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 = {}) {
+    // 动态导入 AI SDK
+    const { tool, ToolLoopAgent } = await this._importAI()
+
+    const userMessage = typeof message === 'string'
+      ? { role: 'user', content: message }
+      : message
+
+    this._messages.push(userMessage)
+
+    const maxSteps = options.maxSteps || this._maxSteps
+    const tools = this._getAITools(tool)
+
+    // 如果没有 AI 客户端，抛出错误
+    if (!this._aiClient) {
+      throw new Error('AI client not configured. Please set up AI plugin.')
+    }
+
+    try {
+      // 使用 ToolLoopAgent 处理工具调用循环
+      const agent = new ToolLoopAgent({
+        model: this._aiClient,
+        instructions: this._systemPrompt,
+        tools: tools,
+        stopWhen: (step) => step.stepCount >= maxSteps
+      })
+
+      // 构建消息列表
+      const messages = [
+        ...this._cleanMessages(this._messages)
+      ]
+
+      // 生成响应
+      const result = await agent.generate({
+        messages: messages,
+        providerOptions: this.providerOptions
+      })
+
+      // 添加助手消息到历史
+      if (result.text) {
+        this._messages.push({
+          role: 'assistant',
+          content: result.text
+        })
+      }
+
+      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 = {}) {
+    // 动态导入 AI SDK
+    const { tool, ToolLoopAgent } = await this._importAI()
+
+    const userMessage = typeof message === 'string'
+      ? { role: 'user', content: message }
+      : message
+
+    this._messages.push(userMessage)
+
+    const maxSteps = options.maxSteps || this._maxSteps
+    const tools = this._getAITools(tool)
+
+    // 如果没有 AI 客户端，抛出错误
+    if (!this._aiClient) {
+      throw new Error('AI client not configured. Please set up AI plugin.')
+    }
+
+    // 使用 ToolLoopAgent 流式接口
+    const agent = new ToolLoopAgent({
+      model: this._aiClient,
+      instructions: this._systemPrompt,
+      tools: tools,
+      stopWhen: (step) => step.stepCount >= maxSteps
+    })
+
+    const messages = [
+      ...this._cleanMessages(this._messages)
+    ]
+
+    try {
+      const result = await agent.stream({
+        messages: messages,
+        providerOptions: this.providerOptions
+      })
+
+      const stream = result.fullStream
+      let fullText = ''
+
+      for await (const part of 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') {
+          yield { type: 'tool-result', toolName: part.toolName, result: part.output }
+        } else if (part.type === 'error') {
+          yield { type: 'error', error: part.error }
+        }
+      }
+
+      // 保存消息到历史
+      if (fullText) {
+        this._messages.push({
+          role: 'assistant',
+          content: fullText
+        })
+      }
+    } 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) => {
+          // 执行工具
+          this.emit('tool-call', { name: toolName, args })
+          try {
+            const result = await toolDef.execute(args, this.agent.framework)
+            this.emit('tool-result', { name: toolName, args, result })
+            return result
+          } catch (err) {
+            this.emit('tool-error', { name: toolName, args, 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
+  }
+
+  /**
+   * 清理消息格式
+   * @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()
+    this.removeAllListeners()
+  }
+}
+
+module.exports = { AgentChatHandler }