foliko 1.0.80 → 1.0.81

package/plugins/extension-executor-plugin.js

@@ -1,326 +1,326 @@
-/**
- * ExtensionExecutorPlugin 扩展插件执行器
- * 统一管理扩展插件（如 gate-trading）的工具，通过 ext_call 调用
- */
-
-const { Plugin } = require('../src/core/plugin-base');
-const { logger } = require('../src/utils/logger');
-const { z } = require('zod');
-const { zodSchemaToMarkdown } = require('@chnak/zod-to-markdown');
-
-const log = logger.child('ExtensionExecutor');
-
-class ExtensionExecutorPlugin extends Plugin {
-  constructor(config = {}) {
-    super();
-    this.name = 'extension-executor';
-    this.version = '1.0.0';
-    this.description = '扩展插件执行器 - 统一管理扩展工具';
-    this.priority = 10;
-    this.system = true;
-
-    this._framework = null;
-    this._extensions = new Map();
-  }
-
-  install(framework) {
-    this._framework = framework;
-
-    // 监听插件加载事件，自动扫描插件的 this.tools
-    framework.on('plugin:loaded', (plugin) => {
-      this._scanPluginTools(plugin);
-    });
-
-    return this;
-  }
-
-  /**
-   * 扫描插件的 this.tools 并注册到扩展系统
-   */
-  _scanPluginTools(plugin) {
-    if (!plugin || !plugin.tools || typeof plugin.tools !== 'object') {
-      return;
-    }
-
-    const pluginName = plugin.name;
-    if (!pluginName) return;
-
-    // 获取插件信息
-    const pluginInfo = {
-      name: plugin.name,
-      description: plugin.description || '',
-      version: plugin.version || '1.0.0',
-    };
-
-    // 遍历插件的所有工具
-    for (const [toolName, toolDef] of Object.entries(plugin.tools)) {
-      if (!toolDef || typeof toolDef !== 'object') continue;
-
-      this.registerTool(pluginName, pluginInfo, {
-        name: toolName,
-        description: toolDef.description || '',
-        inputSchema: toolDef.inputSchema,
-        execute: toolDef.execute,
-      });
-    }
-
-    log.debug(` Scanned ${Object.keys(plugin.tools).length} tools from plugin '${pluginName}'`);
-
-    // 刷新所有 Agent 的扩展提示词
-    if (this._framework && this._framework._ready) {
-      this._refreshAllAgentsExtPrompt(this._framework);
-    }
-  }
-
-  async start(framework) {
-    // 扫描所有已加载插件的 tools
-    const plugins = framework.pluginManager.getAll();
-    for (const { instance: plugin } of plugins) {
-      this._scanPluginTools(plugin);
-    }
-
-    framework.registerTool({
-      name: 'ext_call',
-      description: '调用扩展插件的工具',
-      inputSchema: z.object({
-        plugin: z.string().describe('扩展插件名称'),
-        tool: z.string().describe('工具名称'),
-        args: z.record(z.any()).optional().describe('工具参数'),
-      }),
-      execute: async (args) => {
-        const { plugin, tool, args: toolArgs = {} } = args;
-        log.info(` ext_call: plugin=${plugin}, tool=${tool}`);
-
-        const ext = this._extensions.get(plugin);
-        if (!ext) {
-          return { success: false, error: `扩展插件 '${plugin}' 不存在` };
-        }
-
-        const toolDef = ext.tools.find((t) => t.name === tool);
-        if (!toolDef) {
-          return { success: false, error: `扩展工具 '${tool}' 不存在` };
-        }
-
-        try {
-          // 统一 execute 签名为 (args, framework)
-          const result = await toolDef.execute(toolArgs, framework);
-          return { success: true, result };
-        } catch (err) {
-          log.error(` Tool '${tool}' failed:`, err.message);
-          return { success: false, error: err.message };
-        }
-      },
-    });
-
-    framework.registerTool({
-      name: 'ext_list',
-      description: '列出所有可用的扩展插件及其工具',
-      inputSchema: z.object({}),
-      execute: async () => {
-        const extensions = [];
-        for (const [name, ext] of this._extensions) {
-          extensions.push({
-            name,
-            description: ext.description,
-            version: ext.version,
-            tools: ext.tools.map((t) => ({ name: t.name, description: t.description })),
-          });
-        }
-        return { success: true, extensions };
-      },
-    });
-
-    framework.registerTool({
-      name: 'ext_schema',
-      description: '查询扩展工具的调用参数结构',
-      inputSchema: z.object({
-        plugin: z.string().describe('扩展插件名称'),
-        tool: z.string().describe('工具名称'),
-      }),
-      execute: async (args) => {
-        const { plugin, tool } = args;
-        const ext = this._extensions.get(plugin);
-        if (!ext) {
-          return { success: false, error: `扩展插件 '${plugin}' 不存在` };
-        }
-
-        const toolDef = ext.tools.find((t) => t.name === tool);
-        if (!toolDef) {
-          return { success: false, error: `扩展工具 '${tool}' 不存在` };
-        }
-
-        return {
-          success: true,
-          result: {
-            plugin,
-            tool,
-            description: toolDef.description,
-            inputSchema: toolDef.inputSchema,
-          },
-        };
-      },
-    });
-
-    framework.on('agent:created', (agent) => {
-      this._refreshAgentExtPrompt(agent);
-    });
-
-    if (framework._ready) {
-      this._refreshAllAgentsExtPrompt(framework);
-    } else {
-      framework.once('framework:ready', () => {
-        this._refreshAllAgentsExtPrompt(framework);
-      });
-    }
-
-    return this;
-  }
-
-  registerTool(pluginName, pluginInfo, toolDef) {
-    if (!this._extensions.has(pluginName)) {
-      this._extensions.set(pluginName, {
-        name: pluginInfo.name || pluginName,
-        description: pluginInfo.description || '',
-        version: pluginInfo.version || '1.0.0',
-        tools: [],
-      });
-    }
-
-    const ext = this._extensions.get(pluginName);
-    if (!ext.tools.find((t) => t.name === toolDef.name)) {
-      ext.tools.push({
-        name: toolDef.name,
-        description: toolDef.description || '',
-        inputSchema: toolDef.inputSchema,
-        execute: toolDef.execute,
-      });
-    }
-
-    log.debug(` Registered tool '${toolDef.name}' for extension '${pluginName}'`);
-
-    // 刷新所有 Agent 的扩展提示词
-    if (this._framework && this._framework._ready) {
-      this._refreshAllAgentsExtPrompt(this._framework);
-    }
-  }
-
-  _refreshAllAgentsExtPrompt(framework) {
-    const visited = new Set();
-    const traverse = (agent) => {
-      if (!agent || visited.has(agent)) return;
-      visited.add(agent);
-      this._refreshAgentExtPrompt(agent);
-      const subAgents = agent.getSubAgents?.() || agent._subAgents || new Map();
-      for (const [, subAgentInfo] of subAgents) {
-        traverse(subAgentInfo.agent);
-      }
-    };
-    for (const agent of framework._agents || []) {
-      traverse(agent);
-    }
-  }
-
-  _refreshAgentExtPrompt(agent) {
-    const extDesc = this._buildExtensionsDescription();
-    if (!extDesc) return;
-
-    const existingPrompt = agent._originalPrompt || '';
-
-    // 如果已有 [Extensions] 部分，替换它
-    const extStartIdx = existingPrompt.indexOf('[Extensions]');
-    if (extStartIdx !== -1) {
-      const extEndMarker = '\n\n【';
-      let extEndIdx = existingPrompt.indexOf(extEndMarker, extStartIdx);
-      if (extEndIdx === -1) {
-        // 尝试其他标记
-        extEndIdx = existingPrompt.indexOf('\n\n---\n\n', extStartIdx);
-      }
-      if (extEndIdx === -1) {
-        extEndIdx = existingPrompt.length;
-      } else {
-        extEndIdx += extEndMarker.length;
-      }
-      const newPrompt = existingPrompt.substring(0, extStartIdx) + extDesc + existingPrompt.substring(extEndIdx);
-      agent.setSystemPrompt(newPrompt);
-    } else {
-      agent.setSystemPrompt(existingPrompt + '\n\n' + extDesc);
-    }
-  }
-
-  _buildExtensionsDescription() {
-    if (this._extensions.size === 0) {
-      return '';
-    }
-
-    let desc = '[Extensions]\n\n';
-    desc += '你可以通过 `ext_call` 工具调用以下扩展插件的功能。\n\n';
-
-    for (const [name, ext] of this._extensions) {
-      desc += `### ${ext.name || name}\n\n`;
-      desc += `${ext.description || ''}\n\n`;
-      for (const tool of ext.tools) {
-        desc += `- **${tool.name}**: ${tool.description || '无描述'}\n`;
-        // 添加参数描述
-        if (tool.inputSchema) {
-          try {
-            if (typeof tool.inputSchema.toJSON === 'function') {
-              desc += `**参数:**\n\n`;
-              desc += zodSchemaToMarkdown(tool.inputSchema) + '\n\n';
-            } else if (tool.inputSchema.properties) {
-              // JSON Schema 格式
-              desc += `**参数:**\n\n`;
-              desc += this._schemaToMarkdown(tool.inputSchema) + '\n\n';
-            }
-          } catch (e) {
-            // 忽略转换错误
-          }
-        }
-      }
-      desc += '\n';
-    }
-
-    desc += '**调用格式:**\n';
-    desc += '```\next_call({ plugin: "插件名", tool: "工具名", args: {参数} })\n';
-    desc += '```\n';
-    return desc.trim();
-  }
-
-  /**
-   * 将 JSON Schema 转换为 Markdown
-   */
-  _schemaToMarkdown(schema) {
-    if (!schema || !schema.properties) return '';
-
-    const props = schema.properties || {};
-    const required = schema.required || [];
-    let md = '';
-
-    for (const [key, prop] of Object.entries(props)) {
-      const isRequired = required.includes(key);
-      const type = prop.type || 'any';
-      const descText = prop.description || '';
-      md += `- \`${key}\`${isRequired ? ' (必填)' : ''}: ${type} ${descText}\n`;
-    }
-    return md;
-  }
-
-  getExtensions() {
-    return Array.from(this._extensions.entries()).map(([name, ext]) => ({
-      name,
-      description: ext.description,
-      version: ext.version,
-      tools: ext.tools,
-    }));
-  }
-
-  reload(framework) {
-    this._framework = framework;
-  }
-
-  async uninstall(framework) {
-    this._extensions.clear();
-    this._framework = null;
-  }
-}
-
-module.exports = ExtensionExecutorPlugin
+/**
+ * ExtensionExecutorPlugin 扩展插件执行器
+ * 统一管理扩展插件（如 gate-trading）的工具，通过 ext_call 调用
+ */
+
+const { Plugin } = require('../src/core/plugin-base');
+const { logger } = require('../src/utils/logger');
+const { z } = require('zod');
+const { zodSchemaToMarkdown } = require('@chnak/zod-to-markdown');
+
+const log = logger.child('ExtensionExecutor');
+
+class ExtensionExecutorPlugin extends Plugin {
+  constructor(config = {}) {
+    super();
+    this.name = 'extension-executor';
+    this.version = '1.0.0';
+    this.description = '扩展插件执行器 - 统一管理扩展工具';
+    this.priority = 10;
+    this.system = true;
+
+    this._framework = null;
+    this._extensions = new Map();
+  }
+
+  install(framework) {
+    this._framework = framework;
+
+    // 监听插件加载事件，自动扫描插件的 this.tools
+    framework.on('plugin:loaded', (plugin) => {
+      this._scanPluginTools(plugin);
+    });
+
+    return this;
+  }
+
+  /**
+   * 扫描插件的 this.tools 并注册到扩展系统
+   */
+  _scanPluginTools(plugin) {
+    if (!plugin || !plugin.tools || typeof plugin.tools !== 'object') {
+      return;
+    }
+
+    const pluginName = plugin.name;
+    if (!pluginName) return;
+
+    // 获取插件信息
+    const pluginInfo = {
+      name: plugin.name,
+      description: plugin.description || '',
+      version: plugin.version || '1.0.0',
+    };
+
+    // 遍历插件的所有工具
+    for (const [toolName, toolDef] of Object.entries(plugin.tools)) {
+      if (!toolDef || typeof toolDef !== 'object') continue;
+
+      this.registerTool(pluginName, pluginInfo, {
+        name: toolName,
+        description: toolDef.description || '',
+        inputSchema: toolDef.inputSchema,
+        execute: toolDef.execute,
+      });
+    }
+
+    log.debug(` Scanned ${Object.keys(plugin.tools).length} tools from plugin '${pluginName}'`);
+
+    // 刷新所有 Agent 的扩展提示词
+    if (this._framework && this._framework._ready) {
+      this._refreshAllAgentsExtPrompt(this._framework);
+    }
+  }
+
+  async start(framework) {
+    // 扫描所有已加载插件的 tools
+    const plugins = framework.pluginManager.getAll();
+    for (const { instance: plugin } of plugins) {
+      this._scanPluginTools(plugin);
+    }
+
+    framework.registerTool({
+      name: 'ext_call',
+      description: '调用扩展插件的工具',
+      inputSchema: z.object({
+        plugin: z.string().describe('扩展插件名称'),
+        tool: z.string().describe('工具名称'),
+        args: z.record(z.any()).optional().describe('工具参数'),
+      }),
+      execute: async (args) => {
+        const { plugin, tool, args: toolArgs = {} } = args;
+        log.info(` ext_call: plugin=${plugin}, tool=${tool}`);
+
+        const ext = this._extensions.get(plugin);
+        if (!ext) {
+          return { success: false, error: `扩展插件 '${plugin}' 不存在` };
+        }
+
+        const toolDef = ext.tools.find((t) => t.name === tool);
+        if (!toolDef) {
+          return { success: false, error: `扩展工具 '${tool}' 不存在` };
+        }
+
+        try {
+          // 统一 execute 签名为 (args, framework)
+          const result = await toolDef.execute(toolArgs, framework);
+          return { success: true, result };
+        } catch (err) {
+          log.error(` Tool '${tool}' failed:`, err.message);
+          return { success: false, error: err.message };
+        }
+      },
+    });
+
+    framework.registerTool({
+      name: 'ext_list',
+      description: '列出所有可用的扩展插件及其工具',
+      inputSchema: z.object({}),
+      execute: async () => {
+        const extensions = [];
+        for (const [name, ext] of this._extensions) {
+          extensions.push({
+            name,
+            description: ext.description,
+            version: ext.version,
+            tools: ext.tools.map((t) => ({ name: t.name, description: t.description })),
+          });
+        }
+        return { success: true, extensions };
+      },
+    });
+
+    framework.registerTool({
+      name: 'ext_schema',
+      description: '查询扩展工具的调用参数结构',
+      inputSchema: z.object({
+        plugin: z.string().describe('扩展插件名称'),
+        tool: z.string().describe('工具名称'),
+      }),
+      execute: async (args) => {
+        const { plugin, tool } = args;
+        const ext = this._extensions.get(plugin);
+        if (!ext) {
+          return { success: false, error: `扩展插件 '${plugin}' 不存在` };
+        }
+
+        const toolDef = ext.tools.find((t) => t.name === tool);
+        if (!toolDef) {
+          return { success: false, error: `扩展工具 '${tool}' 不存在` };
+        }
+
+        return {
+          success: true,
+          result: {
+            plugin,
+            tool,
+            description: toolDef.description,
+            inputSchema: toolDef.inputSchema,
+          },
+        };
+      },
+    });
+
+    framework.on('agent:created', (agent) => {
+      this._refreshAgentExtPrompt(agent);
+    });
+
+    if (framework._ready) {
+      this._refreshAllAgentsExtPrompt(framework);
+    } else {
+      framework.once('framework:ready', () => {
+        this._refreshAllAgentsExtPrompt(framework);
+      });
+    }
+
+    return this;
+  }
+
+  registerTool(pluginName, pluginInfo, toolDef) {
+    if (!this._extensions.has(pluginName)) {
+      this._extensions.set(pluginName, {
+        name: pluginInfo.name || pluginName,
+        description: pluginInfo.description || '',
+        version: pluginInfo.version || '1.0.0',
+        tools: [],
+      });
+    }
+
+    const ext = this._extensions.get(pluginName);
+    if (!ext.tools.find((t) => t.name === toolDef.name)) {
+      ext.tools.push({
+        name: toolDef.name,
+        description: toolDef.description || '',
+        inputSchema: toolDef.inputSchema,
+        execute: toolDef.execute,
+      });
+    }
+
+    log.debug(` Registered tool '${toolDef.name}' for extension '${pluginName}'`);
+
+    // 刷新所有 Agent 的扩展提示词
+    if (this._framework && this._framework._ready) {
+      this._refreshAllAgentsExtPrompt(this._framework);
+    }
+  }
+
+  _refreshAllAgentsExtPrompt(framework) {
+    const visited = new Set();
+    const traverse = (agent) => {
+      if (!agent || visited.has(agent)) return;
+      visited.add(agent);
+      this._refreshAgentExtPrompt(agent);
+      const subAgents = agent.getSubAgents?.() || agent._subAgents || new Map();
+      for (const [, subAgentInfo] of subAgents) {
+        traverse(subAgentInfo.agent);
+      }
+    };
+    for (const agent of framework._agents || []) {
+      traverse(agent);
+    }
+  }
+
+  _refreshAgentExtPrompt(agent) {
+    const extDesc = this._buildExtensionsDescription();
+    if (!extDesc) return;
+
+    const existingPrompt = agent._originalPrompt || '';
+
+    // 如果已有 [Extensions] 部分，替换它
+    const extStartIdx = existingPrompt.indexOf('[Extensions]');
+    if (extStartIdx !== -1) {
+      const extEndMarker = '\n\n【';
+      let extEndIdx = existingPrompt.indexOf(extEndMarker, extStartIdx);
+      if (extEndIdx === -1) {
+        // 尝试其他标记
+        extEndIdx = existingPrompt.indexOf('\n\n---\n\n', extStartIdx);
+      }
+      if (extEndIdx === -1) {
+        extEndIdx = existingPrompt.length;
+      } else {
+        extEndIdx += extEndMarker.length;
+      }
+      const newPrompt = existingPrompt.substring(0, extStartIdx) + extDesc + existingPrompt.substring(extEndIdx);
+      agent.setSystemPrompt(newPrompt);
+    } else {
+      agent.setSystemPrompt(existingPrompt + '\n\n' + extDesc);
+    }
+  }
+
+  _buildExtensionsDescription() {
+    if (this._extensions.size === 0) {
+      return '';
+    }
+
+    let desc = '[Extensions]\n\n';
+    desc += '你可以通过 `ext_call` 工具调用以下扩展插件的功能。\n\n';
+
+    for (const [name, ext] of this._extensions) {
+      desc += `### ${ext.name || name}\n\n`;
+      desc += `${ext.description || ''}\n\n`;
+      for (const tool of ext.tools) {
+        desc += `- **${tool.name}**: ${tool.description || '无描述'}\n`;
+        // 添加参数描述
+        if (tool.inputSchema) {
+          try {
+            if (typeof tool.inputSchema.toJSON === 'function') {
+              desc += `**参数:**\n\n`;
+              desc += zodSchemaToMarkdown(tool.inputSchema) + '\n\n';
+            } else if (tool.inputSchema.properties) {
+              // JSON Schema 格式
+              desc += `**参数:**\n\n`;
+              desc += this._schemaToMarkdown(tool.inputSchema) + '\n\n';
+            }
+          } catch (e) {
+            // 忽略转换错误
+          }
+        }
+      }
+      desc += '\n';
+    }
+
+    desc += '**调用格式:**\n';
+    desc += '```\next_call({ plugin: "插件名", tool: "工具名", args: {参数} })\n';
+    desc += '```\n';
+    return desc.trim();
+  }
+
+  /**
+   * 将 JSON Schema 转换为 Markdown
+   */
+  _schemaToMarkdown(schema) {
+    if (!schema || !schema.properties) return '';
+
+    const props = schema.properties || {};
+    const required = schema.required || [];
+    let md = '';
+
+    for (const [key, prop] of Object.entries(props)) {
+      const isRequired = required.includes(key);
+      const type = prop.type || 'any';
+      const descText = prop.description || '';
+      md += `- \`${key}\`${isRequired ? ' (必填)' : ''}: ${type} ${descText}\n`;
+    }
+    return md;
+  }
+
+  getExtensions() {
+    return Array.from(this._extensions.entries()).map(([name, ext]) => ({
+      name,
+      description: ext.description,
+      version: ext.version,
+      tools: ext.tools,
+    }));
+  }
+
+  reload(framework) {
+    this._framework = framework;
+  }
+
+  async uninstall(framework) {
+    this._extensions.clear();
+    this._framework = null;
+  }
+}
+
+module.exports = ExtensionExecutorPlugin