foliko 1.1.93 → 2.0.1

package/plugins/{subagent-plugin.js → core/sub-agent/index.js}

@@ -5,11 +5,10 @@
 
 const fs = require('fs')
 const path = require('path')
-const { Plugin } = require('../src/core/plugin-base')
-const { logger } = require('../src/utils/logger')
+const { Plugin } = require('../../../src/plugin/base')
+const { logger } = require('../../../src/common/logger')
 const log = logger.child('SubAgent')
 const { z } = require('zod')
-const { Agent } = require('../src/core/agent')
 
 class SubAgentPlugin extends Plugin {
   constructor(config = {}) {
@@ -19,16 +18,13 @@ class SubAgentPlugin extends Plugin {
     this.description = config.description || `子Agent: ${config.name}`
     this.priority = 10
 
-    this.system = true 
+    this.system = true
 
     this.role= config.role || config.name
     this.tools = config.tools || {}
     this.parentTools = config.parentTools || []
     this.llmConfig = config.llmConfig || null
 
-    // 子Agent配置
-    // tools: { toolName: toolFn } 自定义工具（只属于此子Agent）
-    // parentTools: ['read_file'] 从父Agent继承的工具名称列表
     this.config = {}
 
     this._framework = null
@@ -42,31 +38,19 @@ class SubAgentPlugin extends Plugin {
   }
 
   start(framework) {
-    // 创建子Agent（可能需要等待父agent准备好）
     this._createSubAgent()
-
-    // 注册委托工具到主Agent（通过framework获取父agent）
     this._registerDelegateTool()
-
     return this
   }
 
-  /**
-   * 创建子Agent
-   */
   _createSubAgent() {
-    // 获取父Agent（主Agent）
     const parentAgent = this._getParentAgent()
     if (!parentAgent) {
-      // 父agent还没创建，监听事件等其创建
       if (this._framework) {
         this._framework.on('agent:created', (agent) => {
           if (this._framework._mainAgent && agent === this._framework._mainAgent) {
-            // 父agent已创建，更新父agent引用
             this._parentAgent = agent
-            // 重新尝试创建子agent
             this._doCreateSubAgent(agent)
-            // 注册委托工具
             this._registerDelegateTool()
           }
         })
@@ -77,19 +61,11 @@ class SubAgentPlugin extends Plugin {
     this._doCreateSubAgent(parentAgent)
   }
 
-  /**
-   * 实际执行创建子Agent
-   */
   _doCreateSubAgent(parentAgent) {
-    // 避免重复创建
     if (this._agent) {
       return
     }
 
-    // 获取从父Agent继承的工具
-    // const parentTools = this._getParentTools(parentAgent)
-
-    // 确定LLM配置
     const aiPlugin = this._framework.pluginManager.get('ai')
     const llmConfig = this.llmConfig || (aiPlugin ? aiPlugin.getConfig() : {})
     const all_tools=['ext_call']
@@ -99,7 +75,6 @@ class SubAgentPlugin extends Plugin {
       const list=this.tools.split(',').map(a=>a.trim())
       all_tools.push(...list)
     }
-    // 创建子Agent，使用完整的 md 文件内容作为系统提示词
     this._agent = this._framework.createSubAgent({
       name: this.name,
       systemPrompt: this._getFullSystemPrompt(),
@@ -107,46 +82,29 @@ class SubAgentPlugin extends Plugin {
       provider: llmConfig.provider,
       apiKey: llmConfig.apiKey,
       baseURL: llmConfig.baseURL,
-      tools:{},  // 自定义工具
+      tools:{},
       parentTools: all_tools
     })
 
-
-    // 注册从父Agent继承的工具
-    // for (const tool of parentTools) {
-    //   this._framework.registerTool(tool)
-    // }
-
-    // Register custom tools (only for this subAgent, won't pollute global)
-    // tools format: { toolName: toolDef }
-    
     this._parentAgent = parentAgent
 
-    // 注册子Agent到父Agent
     parentAgent.registerSubAgent(this.name, this._agent, this.role, this.description)
 
-    // 监听子Agent的chunk事件并转发给父Agent
     this._agent.on('chunk', (chunk) => {
       chunk.fromSubAgent = this.name
       parentAgent.emit('chunk', chunk)
     })
   }
 
-  /**
-   * 获取父Agent
-   */
   _getParentAgent() {
-    // 优先使用外部传入的获取器
     if (this.config._parentAgentGetter) {
       return this.config._parentAgentGetter()
     }
 
-    // 尝试从framework获取主agent
     if (this._framework._mainAgent) {
       return this._framework._mainAgent
     }
 
-    // 查找最后一个创建的agent作为父agent
     const agents = this._framework._agents || []
     if (agents.length > 0) {
       return agents[agents.length - 1]
@@ -155,22 +113,16 @@ class SubAgentPlugin extends Plugin {
     return null
   }
 
-  /**
-   * 获取从父Agent继承的工具
-   */
   _getParentTools(parentAgent) {
     const allTools = parentAgent.getTools ? parentAgent.getTools() : []
     const toolMap = new Map(allTools.map(t => [t.name, t]))
 
-    // 如果没有指定 parentTools，返回全部
     if (this.parentTools === undefined) {
       return allTools
     }
-    // 如果是空数组，明确返回空（不要任何工具）
     if (this.parentTools.length === 0) {
       return []
     }
-    // 过滤指定工具
     const filtered = []
     for (const toolName of this.parentTools) {
       const tool = toolMap.get(toolName)
@@ -182,9 +134,6 @@ class SubAgentPlugin extends Plugin {
     return filtered
   }
 
-  /**
-   * 构建系统提示
-   */
   _buildSystemPrompt() {
     return `你是 ${this.role}。
 
@@ -209,9 +158,6 @@ class SubAgentPlugin extends Plugin {
 - 不先调用 ext_skill/loadSkill 就直接使用 ext_call`
   }
 
-  /**
-   * 注册委托工具到框架和主Agent
-   */
   _registerDelegateTool() {
     const framework = this._framework
     if (!framework) return
@@ -235,7 +181,6 @@ class SubAgentPlugin extends Plugin {
           return { error: `Parent agent not found` }
         }
 
-        // 发射子Agent开始处理事件
         parentAgent.emit('subagent:chat:start', {
           parentAgent,
           subAgent: this._agent,
@@ -245,7 +190,6 @@ class SubAgentPlugin extends Plugin {
 
         try {
           const result = await this._agent.chat(args.task)
-          // 发射子Agent完成处理事件
           parentAgent.emit('subagent:chat:end', {
             parentAgent,
             subAgent: this._agent,
@@ -258,7 +202,6 @@ class SubAgentPlugin extends Plugin {
             success: result.success !== false
           }
         } catch (err) {
-          // 发射子Agent错误事件
           parentAgent.emit('subagent:error', {
             parentAgent,
             subAgent: this._agent,
@@ -274,57 +217,37 @@ class SubAgentPlugin extends Plugin {
       }
     }
 
-    // 注册到框架（供后续创建的 Agent 同步）
     framework.registerTool(toolDef)
 
-    // // 如果主 Agent 已存在，直接注册到它
     if (framework._mainAgent) {
       framework._mainAgent.registerTool(toolDef)
     }
-
-    //console.log(`[SubAgent:${this.config.name}] Delegate tool registered`)
   }
 
-  /**
-   * 获取子Agent实例
-   */
   getAgent() {
     return this._agent
   }
 
-  /**
-   * 调用子Agent处理任务
-   * 延迟创建 Agent，使用完整的 md 文件内容作为系统提示词
-   */
   async chat(task) {
-    // 延迟创建 Agent（如果尚未创建）
     if (!this._agent) {
       await this._createSubAgentForChat();
     } else {
-      // 每次调用时更新系统提示词，确保使用最新的 md 文件内容
       this._updateAgentSystemPrompt();
     }
     return this._agent.chat(task)
   }
 
-  /**
-   * 创建用于聊天的高完整度 Agent
-   * 使用 SubAgentConfig 读取完整的 md 文件内容
-   */
   async _createSubAgentForChat() {
     const parentAgent = this._getParentAgent();
     if (!parentAgent) {
       throw new Error(`SubAgent ${this.name}: parent agent not found`);
     }
 
-    // 使用 SubAgentConfig 获取完整的系统提示词
     const systemPrompt = this._getFullSystemPrompt();
     const parentTools = this._getParentTools(parentAgent)
-    // 获取 LLM 配置
     const aiPlugin = this._framework.pluginManager.get('ai');
     const llmConfig = this.llmConfig || (aiPlugin ? aiPlugin.getConfig() : {});
 
-    // 创建 Agent，使用完整的 md 内容作为系统提示词
     this._agent = this._framework.createSubAgent({
       name: this.name,
       systemPrompt: systemPrompt,
@@ -332,35 +255,26 @@ class SubAgentPlugin extends Plugin {
       provider: llmConfig.provider,
       apiKey: llmConfig.apiKey,
       baseURL: llmConfig.baseURL,
-      tools: this.tools||{},  // 自定义工具
+      tools: this.tools||{},
       parentTools: parentTools
     });
 
-    // 注册到父 Agent
     this._parentAgent = parentAgent;
     parentAgent.registerSubAgent(this.name, this._agent, this.role, this.description);
   }
 
-  /**
-   * 获取完整的系统提示词（从 md 文件读取）
-   */
   _getFullSystemPrompt() {
-    // 尝试从 SubAgentConfigManager 获取完整内容
     const configManager = this._framework._subAgentConfigManager;
     if (configManager) {
-      const config = configManager.get(this.name); 
+      const config = configManager.get(this.name);
       if (config) {
         return config.getSystemPrompt();
       }
     }
 
-    // 回退到默认的 _buildSystemPrompt
     return this._buildSystemPrompt();
   }
 
-  /**
-   * 更新 Agent 的系统提示词
-   */
   _updateAgentSystemPrompt() {
     if (!this._agent) return;
     const systemPrompt = this._getFullSystemPrompt();
@@ -385,9 +299,6 @@ class SubAgentPlugin extends Plugin {
   }
 }
 
-/**
- * SubAgentManager - 管理多个子Agent
- */
 class SubAgentManagerPlugin extends Plugin {
   constructor(config = {}) {
     super()
@@ -398,9 +309,9 @@ class SubAgentManagerPlugin extends Plugin {
     this.systemPrompt= `你是一个子Agent管理器，负责管理多个子Agent。每个子Agent都有独立的工具集和角色描述。当需要处理特定任务时，你会选择合适的子Agent来执行。`
 
     this.priority = 10
-    this.agents = config.agents || []  // 预定义的子Agent配置列表
+    this.agents = config.agents || []
     this.config = {
-      agentsDir: config.agentsDir,  // 子Agent配置目录
+      agentsDir: config.agentsDir,
     }
 
     this._framework = null
@@ -413,10 +324,8 @@ class SubAgentManagerPlugin extends Plugin {
   }
 
   start(framework) {
-    // 从 agentsDir 加载子Agent配置
     this._loadAgentsFromDir()
 
-    // 创建所有配置的子Agent
     for (const agentConfig of this.agents) {
       const plugin = new SubAgentPlugin(agentConfig)
       plugin.install(framework)
@@ -424,30 +333,22 @@ class SubAgentManagerPlugin extends Plugin {
       this._subAgents.set(agentConfig.name, plugin)
     }
 
-    // 注册管理工具
-    // 如果主 Agent 已存在，直接注册到它（因为它不会自动同步框架工具）
     if (framework._mainAgent) {
       this._registerToolsToAgent(framework._mainAgent)
     }
-    // 同时注册到框架（供后续创建的 Agent 同步）
     this._registerTools(framework)
 
-    // 监听主 Agent 创建事件，为后续创建的主 Agent 注册工具
     framework.on('agent:created', (agent) => {
       if (framework._mainAgent && agent === framework._mainAgent) {
         this._registerToolsToAgent(agent)
       }
     })
 
-    // 注册子 Agent 列表提示词
     this.registerPromptPart('subagents', 85, () => this._buildDescription());
 
     return this
   }
 
-  /**
-   * 注册管理工具到指定 Agent
-   */
   _registerToolsToAgent(agent) {
     agent.registerTool({
       name: 'subagent_list',
@@ -510,13 +411,8 @@ class SubAgentManagerPlugin extends Plugin {
         }
       }
     })
-
-    //log.info(' Management tools registered to agent')
   }
 
-  /**
-   * 注册管理工具到框架工具注册表
-   */
   _registerTools(framework) {
     framework.registerTool({
       name: 'subagent_list',
@@ -578,31 +474,21 @@ class SubAgentManagerPlugin extends Plugin {
         }
       }
     })
-
-    //log.info(' Management tools registered to framework')
   }
 
-  /**
-   * 从 agentsDir 目录加载子Agent配置
-   */
   _loadAgentsFromDir() {
-    // 默认使用 .foliko/agents 目录
     const agentsDir = this.config.agentsDir || path.join(this._framework?.getCwd?.() ?? process.cwd(), '.foliko', 'agents')
-    //log.info(' _loadAgentsFromDir called, agentsDir:', agentsDir)
     if (!fs.existsSync(agentsDir)) {
-      //log.info(' agentsDir not found or does not exist')
       return
     }
 
     const entries = fs.readdirSync(agentsDir, { withFileTypes: true })
-    //log.info(' Found entries:', entries.length)
 
     for (const entry of entries) {
       if (entry.isFile() && (entry.name.endsWith('.js') || entry.name.endsWith('.json') || entry.name.endsWith('.md'))) {
         const baseName = entry.name.replace(/\.(js|json|md)$/, '')
         const filePath = path.join(agentsDir, entry.name)
 
-        // 跳过与已有配置同名的
         if (this.agents.some(a => a.name === baseName)) {
           continue
         }
@@ -613,11 +499,8 @@ class SubAgentManagerPlugin extends Plugin {
             const content = fs.readFileSync(filePath, 'utf-8')
             agentConfig = JSON.parse(content)
           } else if (entry.name.endsWith('.md')) {
-            // 解析 markdown 文件，提取 JSON 配置
             agentConfig = this._parseMarkdownConfig(filePath, baseName)
-            //log.info(' Parsed md:', baseName, agentConfig)
           } else {
-            // 清除缓存并加载
             delete require.cache[require.resolve(filePath)]
             const mod = require(filePath)
             agentConfig = typeof mod === 'function' ? mod() : mod
@@ -626,7 +509,6 @@ class SubAgentManagerPlugin extends Plugin {
           if (agentConfig && agentConfig.name) {
             agentConfig._fromDir = true
             this.agents.push(agentConfig)
-            //log.info(' Loaded agent:', agentConfig.name)
           } else {
             log.warn(' Agent config has no name:', baseName, agentConfig)
           }
@@ -637,78 +519,51 @@ class SubAgentManagerPlugin extends Plugin {
     }
   }
 
-  /**
-   * 解析 markdown 文件中的配置
-   * 支持格式：
-   * ```json
-   * { "name": "xxx", "role": "xxx" }
-   * ```
-   * 或直接用 frontmatter 格式
-   */
   _parseMarkdownConfig(filePath, defaultName) {
     const content = fs.readFileSync(filePath, 'utf-8')
-    //log.info(' _parseMarkdownConfig:', filePath)
 
-    // 尝试从 code block 中提取 JSON
     const jsonMatch = content.match(/```(?:json)?\s*\n([\s\S]*?)\n\s*```/)
     if (jsonMatch) {
       try {
-        const config = JSON.parse(jsonMatch[1].trim())
-        //log.info(' Found JSON in code block')
-        return config
+        return JSON.parse(jsonMatch[1].trim())
       } catch (err) {
-        // JSON 解析失败，继续
       }
     }
 
-    // 尝试从 frontmatter 格式中提取
     const frontmatterMatch = content.match(/^---\n([\s\S]*?)\n---/)
     if (frontmatterMatch) {
-      //log.info(' Found frontmatter')
       const frontmatterContent = frontmatterMatch[1]
       const parsed = this._parseYamlLike(frontmatterContent)
       if (parsed && parsed.name) {
-        //log.info(' Parsed frontmatter:', parsed)
         return { name: parsed.name || defaultName, ...parsed }
       }
     }
 
-    // 从 markdown 内容中提取配置（key: value 格式）
     const config = { name: defaultName }
     const yamlMatch = content.match(/^(name|role|description|parentTools|tools):\s*(.+)$/m)
     if (yamlMatch) {
       config[yamlMatch[1]] = yamlMatch[2].trim()
     }
 
-    // 如果没有找到配置，使用文件名作为 name
     return config
   }
 
-
-  /**
-   * 构建子Agent描述
-   * 使用数组收集 + join 模式优化字符串拼接
-   * 动态从配置文件读取每个子Agent的专业领域
-   */
   _buildDescription() {
     const allSubAgents = this.getAllSubAgents();
     if (!allSubAgents || allSubAgents.length === 0) {
       return '';
     }
 
-    // 使用数组收集 + join 模式优化字符串拼接
     const lines = ['## 子 Agent 匹配表', ''];
     lines.push('');
     lines.push('> **提示**：根据任务类型选择最匹配的子 Agent 处理');
     lines.push('');
 
-    // 动态从配置文件读取每个子Agent的核心能力
     for (const plugin of allSubAgents) {
       const name = plugin.name;
       const role = plugin.role;
       const goal = plugin.description || '';
 
-      // 优先使用 SubAgentConfigManager 获取详细信息
       const config = this._framework._subAgentConfigManager?.get(name);
       if (config) {
         const agentDesc = config.getDescription();
@@ -719,12 +574,10 @@ class SubAgentManagerPlugin extends Plugin {
         }
         lines.push(`- **${name}**：${info}`);
       } else {
-        // 回退到旧逻辑
         lines.push(`- **${name}**：${role || goal || '子代理'}`);
       }
     }
 
-    // 通用规则
     lines.push('');
     lines.push('### 子 Agent 调用规则');
     lines.push('');
@@ -738,16 +591,10 @@ class SubAgentManagerPlugin extends Plugin {
     return lines.join('\n');
   }
 
-   /**
-   * 获取所有 subAgent 的基本信息列表
-   */
   getAllSubAgents() {
     return Array.from(this._subAgents.values());
   }
 
-  /**
-   * 解析类似 YAML 的配置
-   */
   _parseYamlLike(content) {
     const result = {}
     const lines = content.split('\n')
@@ -758,7 +605,6 @@ class SubAgentManagerPlugin extends Plugin {
         const key = match[1]
         let value = match[2].trim()
 
-        // 处理数组格式 [a, b, c] 或 [a b c]
         if (value.startsWith('[') && value.endsWith(']')) {
           value = value.slice(1, -1).split(/[,\s]+/).filter(Boolean)
         }
@@ -770,30 +616,23 @@ class SubAgentManagerPlugin extends Plugin {
     return result
   }
 
-  /**
-   * 获取子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.agents = this.agents.filter(a => !a._fromDir)
 
-    // 刷新 SubAgentConfigManager（如果存在）
     if (framework._subAgentConfigManager) {
       framework._subAgentConfigManager.refresh()
     }
 
-    // 重新创建
     this.start(framework)
   }
 
@@ -809,4 +648,4 @@ class SubAgentManagerPlugin extends Plugin {
 module.exports = {
   SubAgentManagerPlugin,
   SubAgentPlugin
-}
+}