foliko 1.0.0

package/src/core/agent.js

@@ -0,0 +1,573 @@
+/**
+ * Agent 类
+ * 负责对话和推理
+ */
+
+const { EventEmitter } = require('../utils/event-emitter')
+const { AgentChatHandler } = require('./agent-chat')
+const os = require('os')
+
+class Agent extends EventEmitter {
+  /**
+   * @param {Framework} framework - 框架实例
+   * @param {Object} config - 配置
+   * @param {string} [config.name] - Agent 名称
+   * @param {string} [config.systemPrompt] - 系统提示
+   * @param {string} [config.sharedPrompt] - 共享提示模板，支持 {{VAR}} 占位符
+   * @param {Object} [config.metadata] - 元数据，注入到 sharedPrompt
+   * @param {boolean} [config.enableToolRouting=true] - 是否启用工具路由
+   */
+  constructor(framework, config = {}) {
+    super()
+
+    this.framework = framework
+    this.config = config
+
+    this.name = config.name || 'Agent'
+    this.model = config.model || 'deepseek-chat'
+    this.apiKey = config.apiKey
+    this.baseURL = config.baseURL
+    this.provider = config.provider || 'deepseek'
+    this.providerOptions = config.providerOptions || {}
+
+    // 原始 system prompt
+    this._originalPrompt = config.systemPrompt || 'You are a helpful assistant.'
+
+    // 共享提示模板
+    this._sharedPrompt = config.sharedPrompt || ''
+
+    // 元数据
+    const metadata = config.metadata || {}
+    this._metadata = new Map(Object.entries(metadata))
+
+    this._chatHandler = null
+    this._tools = new Map()
+    this._status = 'idle'
+
+    // 子Agent管理
+    this._subAgents = new Map()
+
+    // 处理后的 system prompt (带上下文)
+    this.systemPrompt = this._buildSystemPrompt()
+
+    // 初始化聊天处理器
+    this._initChatHandler()
+  }
+
+  /**
+   * 获取元数据值
+   */
+  _getMetadataValue(key) {
+    // 优先从 metadata 获取
+    if (this._metadata.has(key)) {
+      return this._metadata.get(key)
+    }
+
+    // 内置变量
+    switch (key) {
+      case 'WORK_DIR':
+      case 'CWD':
+        return process.cwd()
+      case 'HOME_DIR':
+        return os.homedir()
+      case 'HOST_NAME':
+        return os.hostname()
+      case 'PLATFORM':
+        return process.platform
+      case 'TIME':
+        return new Date().toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai' })
+      case 'DATE':
+        return new Date().toISOString().split('T')[0]
+      default:
+        return `{{${key}}}`
+    }
+  }
+
+  /**
+   * 替换 sharedPrompt 中的占位符
+   */
+  _replacePlaceholders(template) {
+    return template.replace(/\{\{(\w+)\}\}/g, (match, key) => {
+      return this._getMetadataValue(key)
+    })
+  }
+
+  /**
+   * 构建工具描述
+   */
+  _buildToolsDescription() {
+    if (this._tools.size === 0) {
+      return ''
+    }
+
+    let desc = '【可用工具】\n'
+    for (const [name, tool] of this._tools) {
+      desc += `- ${name}: ${tool.description || '无描述'}\n`
+    }
+    return desc.trim()
+  }
+
+  /**
+   * 构建技能描述
+   */
+  _buildSkillsDescription() {
+    const skillManager = this.framework.pluginManager.get('skill-manager')
+    if (!skillManager) {
+      return ''
+    }
+
+    const skills = skillManager.getAllSkills()
+    if (!skills || skills.length === 0) {
+      return ''
+    }
+
+    let desc = '【可用技能】\n'
+    for (const skill of skills) {
+      const name = skill.metadata?.name || skill.name || 'unknown'
+      const descText = skill.metadata?.description || ''
+      desc += `- ${name}: ${descText}\n`
+    }
+    desc += '\n重要：当需要开发插件、执行专业任务时，必须先使用 loadSkill 工具加载对应技能，获取专业指导。'
+    return desc.trim()
+  }
+
+  /**
+   * 构建系统能力描述
+   */
+  _buildCapabilitiesDescription() {
+    const plugins = this.framework.pluginManager.getAll()
+    if (!plugins || plugins.length === 0) {
+      return ''
+    }
+
+    // 过滤出有描述的关键插件（排除内部插件）
+    const keyPlugins = plugins.filter(p => {
+      const name = p.name
+      // 排除默认配置插件和内部插件
+      return name !== 'defaults' && name !== 'agent'
+    })
+
+    if (keyPlugins.length === 0) {
+      return ''
+    }
+
+    let desc = '【系统能力】\n'
+    for (const plugin of keyPlugins) {
+      const name = plugin.instance?.name || plugin.name || 'unknown'
+      const description = plugin.instance?.description || '无描述'
+      desc += `- ${name}: ${description}\n`
+    }
+
+    return desc.trim()
+  }
+
+  /**
+   * 构建子Agent描述
+   */
+  _buildSubAgentsDescription() {
+    if (this._subAgents.size === 0) {
+      return ''
+    }
+
+    let desc = '【子 Agent 分配规则 - 必须遵守】\n'
+    desc += '1. 每个子Agent有固定的专业领域，必须委托给对应的专家：\n'
+    for (const [name, { role, goal }] of this._subAgents) {
+      desc += `  - ${name}: ${role || goal || '子代理'}\n`
+    }
+    desc += '2. 你需要对任务进行拆分，传递给对应的子Agent执行。\n'
+    desc += '3. 多个不同类型的任务必须分别委托给不同的子Agent，每个子Agent只处理自己专业领域的任务。\n'
+    desc += '\n【可委托的子 Agent】\n'
+    for (const [name, { role, goal }] of this._subAgents) {
+      desc += `  - ${name}: ${role || goal || '子代理'}\n`
+    }
+    desc += '\n使用相应子Agent名称的工具将任务委托给相应子代理。'
+
+    return desc.trim()
+  }
+
+  /**
+   * 构建系统提示（带上下文）
+   */
+  _buildSystemPrompt() {
+    const parts = []
+
+    // 1. 原始 system prompt
+    parts.push(this._originalPrompt)
+
+    // 2. 共享提示模板（带占位符替换）
+    if (this._sharedPrompt) {
+      const replaced = this._replacePlaceholders(this._sharedPrompt)
+      parts.push(replaced)
+    }
+
+    // 3. 元数据
+    if (this._metadata.size > 0) {
+      const metaParts = ['【元数据】']
+      for (const [key, value] of this._metadata) {
+        if (typeof value === 'object') {
+          metaParts.push(`- ${key}: ${JSON.stringify(value)}`)
+        } else {
+          metaParts.push(`- ${key}: ${value}`)
+        }
+      }
+      parts.push(metaParts.join('\n'))
+    }
+
+    // 4. 工具列表
+    const toolsDesc = this._buildToolsDescription()
+    if (toolsDesc) {
+      parts.push(toolsDesc)
+    }
+
+    // 5. 技能列表
+    const skillsDesc = this._buildSkillsDescription()
+    if (skillsDesc) {
+      parts.push(skillsDesc)
+    }
+
+    // 6. 子Agent列表
+    const subAgentsDesc = this._buildSubAgentsDescription()
+    if (subAgentsDesc) {
+      parts.push(subAgentsDesc)
+    }
+
+    // 7. 系统能力
+    const capabilitiesDesc = this._buildCapabilitiesDescription()
+    if (capabilitiesDesc) {
+      parts.push(capabilitiesDesc)
+    }
+
+    return parts.join('\n\n')
+  }
+
+  /**
+   * 刷新上下文
+   */
+  _refreshContext() {
+    this.systemPrompt = this._buildSystemPrompt()
+    if (this._chatHandler) {
+      this._chatHandler.setSystemPrompt(this.systemPrompt)
+    }
+  }
+
+  /**
+   * 初始化聊天处理器
+   * @private
+   */
+  _initChatHandler() {
+    let aiClient = null
+    const aiPlugin = this.framework.pluginManager.get('ai')
+    if (aiPlugin) {
+      aiClient = aiPlugin.getAIClient()
+    }
+
+    this._chatHandler = new AgentChatHandler(this, {
+      model: this.model,
+      provider: this.provider,
+      apiKey: this.apiKey,
+      baseURL: this.baseURL,
+      providerOptions: this.providerOptions
+    })
+
+    if (aiClient) {
+      this._chatHandler.setAIClient(aiClient)
+    }
+
+    // 转发事件
+    this._chatHandler.on('message', (msg) => this.emit('message', msg))
+    this._chatHandler.on('chunk', (chunk) => this.emit('chunk', chunk))
+    this._chatHandler.on('tool-call', (tool) => this.emit('tool-call', tool))
+    this._chatHandler.on('tool-result', (result) => this.emit('tool-result', result))
+    this._chatHandler.on('error', (err) => this.emit('error', err))
+
+    this._syncTools()
+  }
+
+  /**
+   * 设置系统提示
+   */
+  setSystemPrompt(prompt) {
+    this._originalPrompt = prompt
+    this.systemPrompt = this._buildSystemPrompt()
+    if (this._chatHandler) {
+      this._chatHandler.setSystemPrompt(this.systemPrompt)
+    }
+    return this
+  }
+
+  /**
+   * 注册工具到 Agent
+   */
+  registerTool(tool) {
+    this._tools.set(tool.name, tool)
+    if (this._chatHandler) {
+      this._chatHandler.registerTool(tool)
+    }
+    this._refreshContext()
+    return this
+  }
+
+  /**
+   * 获取已注册工具
+   */
+  getTools() {
+    return Array.from(this._tools.values())
+  }
+
+  /**
+   * 同步框架中的工具到 Agent
+   * @private
+   */
+  _syncTools() {
+    if (this.framework.toolRegistry.size() > 0) {
+      for (const tool of this.framework.getTools()) {
+        if (!this._tools.has(tool.name)) {
+          this.registerTool(tool)
+        }
+      }
+    }
+  }
+
+  /**
+   * 注册子Agent
+   * @param {string} name - 子Agent名称
+   * @param {Agent} agent - 子Agent实例
+   * @param {string} role - 角色描述
+   * @param {string} goal - 目标描述
+   */
+  registerSubAgent(name, agent, role, goal) {
+    this._subAgents.set(name, { agent, role, goal })
+    this._refreshContext()
+    return this
+  }
+
+  /**
+   * 注销子Agent
+   */
+  unregisterSubAgent(name) {
+    this._subAgents.delete(name)
+    this._refreshContext()
+    return this
+  }
+
+  /**
+   * 获取所有子Agent
+   */
+  getSubAgents() {
+    return this._subAgents
+  }
+
+  /**
+   * 获取待处理的调度通知并清除（下次不再重复显示）
+   */
+  _getAndClearSchedulerNotifications() {
+    try {
+      const scheduler = this.framework.pluginManager.get('scheduler')
+      if (scheduler && scheduler.instance && scheduler.instance.getPendingNotifications) {
+        const results = scheduler.instance.getPendingNotifications()
+        if (results.length > 0) {
+          const notifications = results.slice(-5).reverse() // 最多5条，最新的在前
+          // 清除已发送的通知
+          if (scheduler.instance.clearDeliveredNotifications) {
+            scheduler.instance.clearDeliveredNotifications(notifications.length)
+          }
+          return notifications
+        }
+      }
+    } catch (err) {
+      // 忽略错误，避免影响主流程
+    }
+    return []
+  }
+
+  /**
+   * 获取待处理的思考结果并清除
+   */
+  _getAndClearThinkNotifications() {
+    try {
+      const think = this.framework.pluginManager.get('think')
+      if (think && think.instance && think.instance.getPendingThoughts) {
+        const thoughts = think.instance.getPendingThoughts()
+        if (thoughts.length > 0) {
+          return thoughts.slice(-5).reverse() // 最多5条，最新的在前
+        }
+      }
+    } catch (err) {
+      // 忽略错误，避免影响主流程
+    }
+    return []
+  }
+
+  /**
+   * 获取所有待处理的系统通知（调度 + 思考）
+   */
+  _getAllPendingNotifications() {
+    const notifications = []
+
+    // 调度通知
+    const schedulerNotifs = this._getAndClearSchedulerNotifications()
+    for (const n of schedulerNotifs) {
+      notifications.push(`【定时任务通知】\n任务: ${n.taskName || n.taskId}\n执行时间: ${n.executedAt}\n结果: ${n.result || n.action || '执行完成'}`)
+    }
+
+    // 思考通知
+    const thinkNotifs = this._getAndClearThinkNotifications()
+    for (const t of thinkNotifs) {
+      notifications.push(`【主动思考】\n模式: ${t.mode}\n主题: ${t.topic}\n结果: ${t.result || '思考完成'}`)
+    }
+
+    return notifications
+  }
+
+  /**
+   * 发送消息
+   */
+  async chat(message, options = {}) {
+    if (this._status === 'busy') {
+      throw new Error('Agent is busy')
+    }
+
+    if (this._status === 'error') {
+      this._status = 'idle'
+    }
+
+    this._status = 'busy'
+    this.emit('status', { status: 'busy' })
+
+    try {
+      // 检查是否有待处理的系统通知（调度 + 思考）
+      const notifications = this._getAllPendingNotifications()
+      let enhancedMessage = message
+      if (notifications.length > 0) {
+        enhancedMessage = `【系统通知】\n${notifications.join('\n\n')}\n\n---\n用户消息: ${message}`
+      }
+
+      this._syncTools()
+
+      const result = await this._chatHandler.chat(enhancedMessage, options)
+      this._status = 'idle'
+      this.emit('status', { status: 'idle' })
+      return result
+    } catch (err) {
+      this._status = 'error'
+      this.emit('status', { status: 'error', error: err.message })
+      throw err
+    }
+  }
+
+  /**
+   * 发送消息（流式）
+   */
+  async *chatStream(message, options = {}) {
+    if (this._status === 'busy') {
+      throw new Error('Agent is busy')
+    }
+
+    // 允许从 error 状态重试
+    if (this._status === 'error') {
+      this._status = 'idle'
+    }
+
+    this._status = 'busy'
+    this.emit('status', { status: 'busy' })
+
+    try {
+      // 检查是否有待处理的系统通知（调度 + 思考）
+      const notifications = this._getAllPendingNotifications()
+      let enhancedMessage = message
+      if (notifications.length > 0) {
+        enhancedMessage = `【系统通知】\n${notifications.join('\n\n')}\n\n---\n用户消息: ${message}`
+      }
+
+      this._syncTools()
+
+      yield* this._chatHandler.chatStream(enhancedMessage, options)
+      this._status = 'idle'
+      this.emit('status', { status: 'idle' })
+    } catch (err) {
+      this._status = 'error'
+      this.emit('status', { status: 'error', error: err.message })
+      throw err
+    }
+  }
+
+  /**
+   * 设置元数据
+   */
+  setMetadata(keyOrObj, value) {
+    if (typeof keyOrObj === 'string') {
+      this._metadata.set(keyOrObj, value)
+    } else if (keyOrObj && typeof keyOrObj === 'object') {
+      for (const [key, val] of Object.entries(keyOrObj)) {
+        this._metadata.set(key, val)
+      }
+    }
+    this._refreshContext()
+    return this
+  }
+
+  /**
+   * 获取元数据
+   */
+  getMetadata(key) {
+    return this._metadata.get(key)
+  }
+
+  /**
+   * 删除元数据
+   */
+  deleteMetadata(key) {
+    this._metadata.delete(key)
+    this._refreshContext()
+    return this
+  }
+
+  /**
+   * 清空元数据
+   */
+  clearMetadata() {
+    this._metadata.clear()
+    this._refreshContext()
+    return this
+  }
+
+  /**
+   * 清空对话历史
+   */
+  clearHistory() {
+    if (this._chatHandler) {
+      this._chatHandler.clearHistory()
+    }
+    return this
+  }
+
+  /**
+   * 获取状态
+   */
+  getStatus() {
+    return this._status
+  }
+
+  /**
+   * 重置状态（从卡住状态恢复）
+   */
+  resetStatus() {
+    this._status = 'idle'
+    this.emit('status', { status: 'idle' })
+    return this
+  }
+
+  /**
+   * 销毁 Agent
+   */
+  destroy() {
+    if (this._chatHandler) {
+      this._chatHandler.destroy()
+    }
+    this._tools.clear()
+    this.removeAllListeners()
+    this.emit('destroyed')
+  }
+}
+
+module.exports = { Agent }