foliko 1.0.0

package/plugins/storage-plugin.js

@@ -0,0 +1,237 @@
+/**
+ * Storage 存储插件
+ * 提供数据持久化存储，支持 JSON 文件存储
+ */
+
+const { Plugin } = require('../src/core/plugin-base')
+const { z } = require('zod')
+const fs = require('fs')
+const path = require('path')
+
+class StoragePlugin extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = 'storage'
+    this.version = '1.0.0'
+    this.description = '数据持久化存储插件，支持键值对存储'
+    this.priority = 3
+
+    this.config = {
+      type: config.type || 'json',  // json 或 memory
+      path: config.path || '.agent/data',
+      namespace: config.namespace || 'default'
+    }
+
+    this._framework = null
+    this._store = new Map()
+    this._filePath = null
+  }
+
+  install(framework) {
+    this._framework = framework
+    return this
+  }
+
+  start(framework) {
+    // 初始化存储
+    if (this.config.type === 'json') {
+      this._initJsonStore()
+    }
+
+    // 注册存储工具
+    framework.registerTool({
+      name: 'storage_set',
+      description: '存储数据到键值存储',
+      inputSchema: z.object({
+        key: z.string().describe('存储键名'),
+        value: z.any().describe('存储值'),
+        namespace: z.string().optional().describe('命名空间，默认使用全局命名空间')
+      }),
+      execute: async (args) => {
+        try {
+          const ns = args.namespace || this.config.namespace
+          const fullKey = `${ns}:${args.key}`
+          this._store.set(fullKey, {
+            value: args.value,
+            updatedAt: new Date()
+          })
+
+          if (this.config.type === 'json') {
+            await this._persistToFile()
+          }
+
+          return { success: true, key: args.key, namespace: ns }
+        } catch (err) {
+          return { success: false, error: err.message }
+        }
+      }
+    })
+
+    framework.registerTool({
+      name: 'storage_get',
+      description: '从键值存储获取数据',
+      inputSchema: z.object({
+        key: z.string().describe('存储键名'),
+        namespace: z.string().optional().describe('命名空间')
+      }),
+      execute: async (args) => {
+        const ns = args.namespace || this.config.namespace
+        const fullKey = `${ns}:${args.key}`
+        const entry = this._store.get(fullKey)
+
+        if (!entry) {
+          return { success: false, error: 'Key not found', key: args.key }
+        }
+
+        return {
+          success: true,
+          key: args.key,
+          namespace: ns,
+          value: entry.value,
+          updatedAt: entry.updatedAt
+        }
+      }
+    })
+
+    framework.registerTool({
+      name: 'storage_delete',
+      description: '删除存储的数据',
+      inputSchema: z.object({
+        key: z.string().describe('存储键名'),
+        namespace: z.string().optional().describe('命名空间')
+      }),
+      execute: async (args) => {
+        const ns = args.namespace || this.config.namespace
+        const fullKey = `${ns}:${args.key}`
+        const existed = this._store.has(fullKey)
+        this._store.delete(fullKey)
+
+        if (this.config.type === 'json') {
+          await this._persistToFile()
+        }
+
+        return { success: true, deleted: existed, key: args.key }
+      }
+    })
+
+    framework.registerTool({
+      name: 'storage_list',
+      description: '列出命名空间下的所有键',
+      inputSchema: z.object({
+        namespace: z.string().optional().describe('命名空间，默认全局')
+      }),
+      execute: async (args) => {
+        const ns = args.namespace || this.config.namespace
+        const prefix = `${ns}:`
+        const keys = []
+
+        for (const key of this._store.keys()) {
+          if (key.startsWith(prefix)) {
+            keys.push(key.substring(prefix.length))
+          }
+        }
+
+        return {
+          success: true,
+          namespace: ns,
+          keys,
+          count: keys.length
+        }
+      }
+    })
+
+    framework.registerTool({
+      name: 'storage_clear',
+      description: '清空命名空间下的所有数据',
+      inputSchema: z.object({
+        namespace: z.string().optional().describe('命名空间，默认全局')
+      }),
+      execute: async (args) => {
+        const ns = args.namespace || this.config.namespace
+        const prefix = `${ns}:`
+        let count = 0
+
+        for (const key of this._store.keys()) {
+          if (key.startsWith(prefix)) {
+            this._store.delete(key)
+            count++
+          }
+        }
+
+        if (this.config.type === 'json') {
+          await this._persistToFile()
+        }
+
+        return { success: true, namespace: ns, cleared: count }
+      }
+    })
+
+    return this
+  }
+
+  /**
+   * 初始化 JSON 文件存储
+   */
+  _initJsonStore() {
+    this._filePath = path.resolve(process.cwd(), this.config.path, `${this.config.namespace}.json`)
+
+    // 确保目录存在
+    const dir = path.dirname(this._filePath)
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true })
+    }
+
+    // 加载已有数据
+    if (fs.existsSync(this._filePath)) {
+      try {
+        const content = fs.readFileSync(this._filePath, 'utf-8')
+        const data = JSON.parse(content)
+
+        for (const [key, entry] of Object.entries(data)) {
+          this._store.set(key, entry)
+        }
+
+        console.log(`[Storage] Loaded ${this._store.size} entries from ${this._filePath}`)
+      } catch (err) {
+        console.warn(`[Storage] Failed to load storage file: ${err.message}`)
+      }
+    }
+  }
+
+  /**
+   * 持久化到文件
+   */
+  async _persistToFile() {
+    if (!this._filePath) return
+
+    try {
+      const data = Object.fromEntries(this._store)
+      fs.writeFileSync(this._filePath, JSON.stringify(data, null, 2), 'utf-8')
+    } catch (err) {
+      console.error(`[Storage] Failed to persist: ${err.message}`)
+    }
+  }
+
+  /**
+   * 获取存储实例（供其他插件使用）
+   */
+  getStore() {
+    return this._store
+  }
+
+  reload(framework) {
+    this._framework = framework
+    this._store.clear()
+
+    if (this.config.type === 'json') {
+      this._initJsonStore()
+    }
+  }
+
+  uninstall(framework) {
+    this._store.clear()
+    this._framework = null
+  }
+}
+
+module.exports = { StoragePlugin }

package/plugins/subagent-plugin.js

@@ -0,0 +1,395 @@
+/**
+ * SubAgent 子Agent插件
+ * 创建具有独立工具集的子Agent，用于分工处理不同领域的任务
+ */
+
+const { Plugin } = require('../src/core/plugin-base')
+const { z } = require('zod')
+const { Agent } = require('../src/core/agent')
+
+class SubAgentPlugin extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = config.name
+    this.version = '1.0.0'
+    this.description = config.description || `子Agent: ${config.name}`
+    this.priority = 10
+
+    // 子Agent配置
+    // tools: { toolName: toolFn } 自定义工具（只属于此子Agent）
+    // parentTools: ['read_file'] 从父Agent继承的工具名称列表
+    this.config = {
+      name: config.name,                      // 子Agent名称
+      role: config.role || config.name,       // 角色描述
+      description: config.description || '',  // 供主Agent智能选择
+      tools: config.tools || {},              // 自定义工具 { name: toolDef }
+      parentTools: config.parentTools || [],  // 从父Agent继承的工具名称列表
+      llmConfig: config.llmConfig || null     // 独立LLM配置
+    }
+
+    this._framework = null
+    this._agent = null
+    this._parentAgent = null
+  }
+
+  install(framework) {
+    this._framework = framework
+    return this
+  }
+
+  start(framework) {
+    // 创建子Agent
+    this._createSubAgent()
+
+    // 注册委托工具到主Agent（通过framework获取父agent）
+    this._registerDelegateTool()
+
+    return this
+  }
+
+  /**
+   * 创建子Agent
+   */
+  _createSubAgent() {
+    // 获取父Agent（主Agent）
+    const parentAgent = this._getParentAgent()
+    if (!parentAgent) {
+      console.warn(`[SubAgent:${this.config.name}] Parent agent not found`)
+      return
+    }
+
+    // 获取从父Agent继承的工具
+    const parentTools = this._getParentTools(parentAgent)
+
+    // 确定LLM配置
+    const aiPlugin = this._framework.pluginManager.get('ai')
+    const llmConfig = this.config.llmConfig || (aiPlugin ? aiPlugin.getConfig() : {})
+
+    // 创建子Agent
+    this._agent = new Agent(this._framework, {
+      name: this.config.name,
+      systemPrompt: this._buildSystemPrompt(),
+      model: llmConfig.model,
+      provider: llmConfig.provider,
+      apiKey: llmConfig.apiKey,
+      baseURL: llmConfig.baseURL
+    })
+
+    // 注册从父Agent继承的工具
+    for (const tool of parentTools) {
+      this._agent.registerTool(tool)
+    }
+
+    // 注册自定义工具（只属于此子Agent，不污染全局）
+    // tools 格式: { toolName: toolDef }
+    if (this.config.tools && typeof this.config.tools === 'object') {
+      for (const [toolName, toolDef] of Object.entries(this.config.tools)) {
+        const tool = typeof toolDef === 'function' ? toolDef() : toolDef
+        if (typeof tool === 'object' && tool.name) {
+          this._agent.registerTool(tool)
+        } else {
+          // 假设是简化的工具定义，需要补充 name
+          this._agent.registerTool({ name: toolName, ...tool })
+        }
+      }
+    }
+
+    // 保存父Agent引用
+    this._parentAgent = parentAgent
+
+    // 注册子Agent到父Agent
+    parentAgent.registerSubAgent(this.config.name, this._agent, this.config.role, this.config.description)
+
+    // 监听子Agent的chunk事件并转发给父Agent
+    this._agent.on('chunk', (chunk) => {
+      chunk.fromSubAgent = this.config.name
+      parentAgent.emit('chunk', chunk)
+    })
+
+    const parentToolNames = parentTools.map(t => t.name)
+    const customToolNames = Object.keys(this.config.tools || {})
+    console.log(`[SubAgent:${this.config.name}] Created with ${parentToolNames.length} parent tools + ${customToolNames.length} custom tools`)
+    if (parentToolNames.length > 0) {
+      console.log(`  Parent tools: ${parentToolNames.join(', ')}`)
+    }
+    if (customToolNames.length > 0) {
+      console.log(`  Custom tools: ${customToolNames.join(', ')}`)
+    }
+  }
+
+  /**
+   * 获取父Agent
+   */
+  _getParentAgent() {
+    // 优先使用外部传入的获取器
+    if (this.config._parentAgentGetter) {
+      return this.config._parentAgentGetter()
+    }
+
+    // 尝试从framework获取主agent
+    // 在vb-agent中，主agent通常是通过framework.createAgent创建的
+    if (this._framework._mainAgent) {
+      return this._framework._mainAgent
+    }
+
+    // 查找最后一个创建的agent作为父agent
+    const agents = this._framework._agents || []
+    if (agents.length > 0) {
+      return agents[agents.length - 1]
+    }
+
+    return null
+  }
+
+  /**
+   * 获取从父Agent继承的工具
+   */
+  _getParentTools(parentAgent) {
+    const allTools = parentAgent.getTools ? parentAgent.getTools() : []
+    const toolMap = new Map(allTools.map(t => [t.name, t]))
+
+    // 如果没有指定 parentTools，返回全部
+    if (!this.config.parentTools || this.config.parentTools.length === 0) {
+      return allTools
+    }
+
+    // 过滤指定工具
+    const filtered = []
+    for (const toolName of this.config.parentTools) {
+      const tool = toolMap.get(toolName)
+      if (tool) {
+        filtered.push(tool)
+      } else {
+        console.warn(`[SubAgent:${this.config.name}] Parent tool not found: ${toolName}`)
+      }
+    }
+
+    return filtered
+  }
+
+  /**
+   * 构建系统提示
+   */
+  _buildSystemPrompt() {
+    return `你是 ${this.config.role}。
+
+角色描述：${this.config.description || '一个专业的子Agent'}
+
+当你被调用时，你应该：
+1. 仔细理解任务要求
+2. 使用你的工具集完成任务
+3. 返回完整的操作结果
+
+重要：返回结果要简洁明确，只包含最终结果，不需要解释过程。`
+  }
+
+  /**
+   * 注册委托工具到父Agent
+   */
+  _registerDelegateTool() {
+    const parentAgent = this._getParentAgent()
+    if (!parentAgent) return
+
+    const agentName = this.config.name
+    const role = this.config.role
+
+    // 注册工具到父Agent
+    parentAgent.registerTool({
+      name: agentName,
+      description: `${role}：当需要${role}时，调用此工具执行任务。如：编译代码、运行测试、代码重构等。`,
+      inputSchema: z.object({
+        task: z.string().describe('给子Agent的具体任务描述')
+      }),
+      execute: async (args, framework) => {
+        if (!this._agent) {
+          return { error: `SubAgent ${agentName} not initialized` }
+        }
+
+        // 发射子Agent开始处理事件
+        parentAgent.emit('subagent:chat:start', {
+          parentAgent,
+          subAgent: this._agent,
+          subAgentName: agentName,
+          task: args.task
+        })
+
+        try {
+          const result = await this._agent.chat(args.task)
+          // 发射子Agent完成处理事件
+          parentAgent.emit('subagent:chat:end', {
+            parentAgent,
+            subAgent: this._agent,
+            subAgentName: agentName,
+            result
+          })
+          return {
+            agent: agentName,
+            result: result.message || result.error || result,
+            success: result.success !== false
+          }
+        } catch (err) {
+          // 发射子Agent错误事件
+          parentAgent.emit('subagent:error', {
+            parentAgent,
+            subAgent: this._agent,
+            subAgentName: agentName,
+            error: err.message
+          })
+          return {
+            agent: agentName,
+            error: err.message,
+            success: false
+          }
+        }
+      }
+    })
+
+    console.log(`[SubAgent:${this.config.name}] Delegate tool registered on parent agent`)
+  }
+
+  /**
+   * 获取子Agent实例
+   */
+  getAgent() {
+    return this._agent
+  }
+
+  /**
+   * 调用子Agent处理任务
+   */
+  async chat(task) {
+    if (!this._agent) {
+      throw new Error(`SubAgent ${this.config.name} not initialized`)
+    }
+    return this._agent.chat(task)
+  }
+
+  reload(framework) {
+    this._framework = framework
+    this._createSubAgent()
+  }
+
+  uninstall(framework) {
+    if (this._parentAgent) {
+      this._parentAgent.unregisterSubAgent(this.config.name)
+    }
+    if (this._agent) {
+      this._agent.destroy()
+      this._agent = null
+    }
+    this._parentAgent = null
+    this._framework = null
+  }
+}
+
+/**
+ * SubAgentManager - 管理多个子Agent
+ */
+class SubAgentManagerPlugin extends Plugin {
+  constructor(config = {}) {
+    super()
+    this.name = 'subagent-manager'
+    this.version = '1.0.0'
+    this.description = '子Agent管理器，统一管理多个子Agent'
+    this.priority = 10
+
+    this.config = {
+      agents: config.agents || []  // 子Agent配置列表
+    }
+
+    this._framework = null
+    this._subAgents = new Map()
+  }
+
+  install(framework) {
+    this._framework = framework
+    return this
+  }
+
+  start(framework) {
+    // 创建所有配置的子Agent
+    for (const agentConfig of this.config.agents) {
+      const plugin = new SubAgentPlugin(agentConfig)
+      plugin.install(framework)
+      plugin.start(framework)
+      this._subAgents.set(agentConfig.name, plugin)
+    }
+
+    // 注册管理工具
+    framework.registerTool({
+      name: 'subagent_list',
+      description: '列出所有子Agent',
+      inputSchema: z.object({}),
+      execute: async () => {
+        const agents = Array.from(this._subAgents.values()).map(p => ({
+          name: p.config.name,
+          role: p.config.role,
+          description: p.config.description,
+          toolCount: p._agent ? p._agent.getTools().length : 0
+        }))
+        return { success: true, agents }
+      }
+    })
+
+    framework.registerTool({
+      name: 'subagent_call',
+      description: '调用指定的子Agent处理任务',
+      inputSchema: z.object({
+        agentName: z.string().describe('子Agent名称'),
+        task: z.string().describe('任务描述')
+      }),
+      execute: async (args) => {
+        const plugin = this._subAgents.get(args.agentName)
+        if (!plugin) {
+          return { success: false, error: `SubAgent ${args.agentName} not found` }
+        }
+
+        try {
+          const result = await plugin.chat(args.task)
+          return {
+            success: true,
+            agent: args.agentName,
+            result: result.message || result,
+            success: result.success !== false
+          }
+        } catch (err) {
+          return { success: false, error: err.message }
+        }
+      }
+    })
+
+    return this
+  }
+
+  /**
+   * 获取子Agent
+   */
+  getSubAgent(name) {
+    const plugin = this._subAgents.get(name)
+    return plugin ? plugin.getAgent() : null
+  }
+
+  reload(framework) {
+    // 销毁旧的子Agent
+    for (const plugin of this._subAgents.values()) {
+      plugin.uninstall(framework)
+    }
+    this._subAgents.clear()
+
+    // 重新创建
+    this.start(framework)
+  }
+
+  uninstall(framework) {
+    for (const plugin of this._subAgents.values()) {
+      plugin.uninstall(framework)
+    }
+    this._subAgents.clear()
+    this._framework = null
+  }
+}
+
+module.exports = {
+  SubAgentPlugin,
+  SubAgentManagerPlugin
+}