foliko 1.0.77 → 1.0.78

package/plugins/extension-executor-plugin.js

@@ -0,0 +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

package/plugins/feishu-plugin.js

@@ -478,4 +478,4 @@ class FeishuPlugin extends Plugin {
   }
 }
 
-module.exports = { FeishuPlugin }
+module.exports = FeishuPlugin

package/plugins/file-system-plugin.js

@@ -4,7 +4,7 @@
  */
 
 const { Plugin } = require('../src/core/plugin-base')
-
+const { NodeHtmlMarkdown } = require('node-html-markdown')
 class FileSystemPlugin extends Plugin {
   constructor(config = {}) {
     super()
@@ -743,10 +743,12 @@ class FileSystemPlugin extends Plugin {
         headers: z.record(z.string()).optional().describe('请求头'),
         body: z.string().optional().describe('请求体（POST/PUT/PATCH 时使用）'),
         timeout: z.number().optional().describe('超时时间(ms)，默认 30000'),
-        proxy: z.boolean().optional().describe('是否使用代理，默认 false。访问失败时可自动设为 true 重试')
+        proxy: z.boolean().optional().describe('是否使用代理，默认 false。访问失败时可自动设为 true 重试'),
+        toMarkdown: z.boolean().default(true).describe('是否将 HTML 响应转换为 Markdown 格式，默认 true'),
+        maxLength: z.number().optional().describe('最大返回长度（字符数），超过则截断')
       }),
       execute: async (args, framework) => {
-        const { url, method = 'GET', headers = {}, body, timeout = 30000, proxy = false } = args
+        const { url, method = 'GET', headers = {}, body, timeout = 30000, proxy = false, toMarkdown = true, maxLength } = args
         try {
           const controller = new AbortController()
           const timeoutId = setTimeout(() => controller.abort(), timeout)
@@ -769,13 +771,62 @@ class FileSystemPlugin extends Plugin {
             data = text
           }
 
+          // 如果 toMarkdown 为 true 且响应是 HTML，则转换为 Markdown
+          if (toMarkdown && typeof data === 'string') {
+            // 检测是否为 HTML 内容
+            const isHtml = /<[a-z][\s\S]*>/i.test(data) || data.trim().startsWith('<')
+            if (isHtml) {
+              try {
+                
+                const markdown = NodeHtmlMarkdown.translate(data)
+                // 限制返回长度
+                const truncatedMarkdown = maxLength && markdown.length > maxLength 
+                  ? markdown.substring(0, maxLength) + '\n\n...(truncated)' 
+                  : markdown
+                return {
+                  success: true,
+                  status: response.status,
+                  statusText: response.statusText,
+                  headers: Object.fromEntries(response.headers.entries()),
+                  usedProxy: proxy,
+                  markdown: truncatedMarkdown,
+                  originalLength: markdown.length,
+                  truncated: maxLength && markdown.length > maxLength
+                }
+              } catch (e) {
+                // 转换失败时返回原始 body
+                const truncatedBody = maxLength && data.length > maxLength
+                  ? data.substring(0, maxLength) + '\n\n...(truncated)'
+                  : data
+                return {
+                  success: true,
+                  status: response.status,
+                  statusText: response.statusText,
+                  headers: Object.fromEntries(response.headers.entries()),
+                  usedProxy: proxy,
+                  markdown: null,
+                  body: truncatedBody,
+                  originalLength: data.length,
+                  truncated: maxLength && data.length > maxLength,
+                  markdownError: e.message
+                }
+              }
+            }
+          }
+
+          // toMarkdown 为 false 或非 HTML 内容时，返回原始内容
+          const truncatedData = maxLength && typeof data === 'string' && data.length > maxLength
+            ? data.substring(0, maxLength) + '\n\n...(truncated)'
+            : data
           return {
             success: true,
             status: response.status,
             statusText: response.statusText,
             headers: Object.fromEntries(response.headers.entries()),
-            body: data,
-            usedProxy: proxy
+            usedProxy: proxy,
+            body: truncatedData,
+            originalLength: typeof data === 'string' ? data.length : null,
+            truncated: maxLength && typeof data === 'string' && data.length > maxLength
           }
         } catch (error) {
           return {
@@ -823,4 +874,4 @@ class FileSystemPlugin extends Plugin {
   }
 }
 
-module.exports = { FileSystemPlugin }
+module.exports = FileSystemPlugin