foliko 1.1.93 → 2.0.0

package/plugins/{tools-plugin.js → tools/index.js}

@@ -1,34 +1,31 @@
 /**
- * 工具插件
- * 提供内置工具和热重载功能
+ * ToolsPlugin - 内置工具插件
+ * 提供热重载、插件管理、工具列表等内置工具
  */
 
-const { Plugin } = require('../src/core/plugin-base')
-const { logger } = require('../src/utils/logger')
-const log = logger.child('Tools')
-const { z } = require('zod')
+const { Plugin } = require('../../src/plugin/base');
+const { logger } = require('../../src/common/logger');
+const { z } = require('zod');
+const log = logger.child('Tools');
 
 class ToolsPlugin extends Plugin {
   constructor(config = {}) {
-    super()
-    this.name = 'tools'
-    this.version = '1.0.0'
-    this.description = '内置工具插件，提供热重载和工具管理功能'
-    this.priority = 10
-    
-    this.system = true 
-
-    this._framework = null
+    super();
+    this.name = 'tools';
+    this.version = '1.0.0';
+    this.description = '内置工具插件，提供热重载和工具管理功能';
+    this.priority = 10;
+    this.system = true;
+    this._framework = null;
   }
 
   install(framework) {
-    this._framework = framework
-    this._registerBuiltinTools()
-    return this
+    this._framework = framework;
+    this._registerBuiltinTools();
+    return this;
   }
 
   start(framework) {
-    // 注册工具调用核心规则提示词
     this.registerPromptPart('tool-core-rules', 95, () => this._getToolCoreRules());
   }
 
@@ -62,165 +59,120 @@ class ToolsPlugin extends Plugin {
   }
 
   _registerBuiltinTools() {
-    const framework = this._framework
+    const framework = this._framework;
 
-    // 热重载插件工具
     framework.registerTool({
       name: 'reload_plugins',
       description: '热重载插件。可以重载单个指定插件，或重载所有插件。',
       inputSchema: z.object({
-        pluginName: z.string().optional().describe('可选：指定要重载的插件名称，不指定则重载所有插件')
+        pluginName: z.string().optional().describe('可选：指定要重载的插件名称，不指定则重载所有插件'),
       }),
       execute: async (args) => {
         try {
           if (args.pluginName) {
-            await framework.reloadPlugin(args.pluginName)
-            return {
-              success: true,
-              data: `Plugin '${args.pluginName}' reloaded successfully`
-            }
-          } else {
-            await framework.reloadAllPlugins()
-            return {
-              success: true,
-              data: 'All plugins reloaded successfully'
-            }
+            await framework.reloadPlugin(args.pluginName);
+            return { success: true, data: `Plugin '${args.pluginName}' reloaded successfully` };
           }
+          await framework.reloadAllPlugins();
+          return { success: true, data: 'All plugins reloaded successfully' };
         } catch (err) {
-          return {
-            success: false,
-            error: err.message
-          }
+          return { success: false, error: err.message };
         }
-      }
-    })
+      },
+    });
 
-    // 列出所有插件工具（包括未加载的）
     framework.registerTool({
       name: 'list_plugins',
       description: '列出所有插件（包括未加载的）',
       inputSchema: z.object({}),
       execute: async () => {
-        const plugins = framework.pluginManager.getAllKnown()
-        return {
-          success: true,
-          data: plugins.map(p => ({
-            name: p.name,
-            status: p.status,
-            enabled: p.enabled,
-            version: p.version,
-            system: p.system
-          }))
-        }
-      }
-    })
+        const plugins = framework.pluginManager.getAllKnown();
+        return { success: true, data: plugins };
+      },
+    });
 
-    // 启用插件工具
     framework.registerTool({
       name: 'enable_plugin',
       description: '启用指定插件',
-      inputSchema: z.object({
-        name: z.string().describe('插件名称')
-      }),
+      inputSchema: z.object({ name: z.string().describe('插件名称') }),
       execute: async (args) => {
         try {
-          await framework.enablePlugin(args.name)
-          return { success: true, data: `插件 '${args.name}' 已启用` }
+          await framework.enablePlugin(args.name);
+          return { success: true, data: `插件 '${args.name}' 已启用` };
         } catch (err) {
-          return { success: false, error: err.message }
+          return { success: false, error: err.message };
         }
-      }
-    })
+      },
+    });
 
-    // 禁用插件工具
     framework.registerTool({
       name: 'disable_plugin',
       description: '禁用指定插件',
-      inputSchema: z.object({
-        name: z.string().describe('插件名称')
-      }),
+      inputSchema: z.object({ name: z.string().describe('插件名称') }),
       execute: async (args) => {
         try {
-          await framework.disablePlugin(args.name)
-          return { success: true, data: `插件 '${args.name}' 已禁用` }
+          await framework.disablePlugin(args.name);
+          return { success: true, data: `插件 '${args.name}' 已禁用` };
         } catch (err) {
-          return { success: false, error: err.message }
+          return { success: false, error: err.message };
         }
-      }
-    })
+      },
+    });
 
-    // 获取工具列表
     framework.registerTool({
       name: 'list_tools',
       description: '列出所有已注册的工具',
       inputSchema: z.object({}),
       execute: async () => {
-        const tools = framework.getTools()
-        return {
-          success: true,
-          data: tools.map(t => ({
-            name: t.name,
-            description: t.description
-          }))
-        }
-      }
-    })
+        const tools = framework.getTools();
+        return { success: true, data: tools.map(t => ({ name: t.name, description: t.description })) };
+      },
+    });
 
-    // 获取插件配置
     framework.registerTool({
       name: 'get_plugin_config',
       description: '获取指定插件的当前配置',
-      inputSchema: z.object({
-        name: z.string().describe('插件名称')
-      }),
+      inputSchema: z.object({ name: z.string().describe('插件名称') }),
       execute: async (args) => {
-        const plugin = framework.pluginManager.get(args.name)
-        if (!plugin) {
-          return { success: false, error: `Plugin '${args.name}' not found` }
-        }
-        return {
-          success: true,
-          data: plugin.config || {}
-        }
-      }
-    })
+        const plugin = framework.pluginManager.get(args.name);
+        if (!plugin) return { success: false, error: `Plugin '${args.name}' not found` };
+        return { success: true, data: plugin.config || {} };
+      },
+    });
 
-    // 更新插件配置
     framework.registerTool({
       name: 'update_plugin_config',
       description: '更新指定插件的配置，会合并到现有配置并持久化保存',
       inputSchema: z.object({
         name: z.string().describe('插件名称'),
-        config: z.record(z.any()).describe('要更新的配置（会合并到现有配置）')
+        config: z.record(z.any()).describe('要更新的配置（会合并到现有配置）'),
       }),
       execute: async (args) => {
         try {
-          const newConfig = framework.updatePluginConfig(args.name, args.config)
-          return { success: true, data: `插件 '${args.name}' 配置已更新`, metadata: { config: newConfig } }
+          const newConfig = framework.updatePluginConfig(args.name, args.config);
+          return { success: true, data: `插件 '${args.name}' 配置已更新`, metadata: { config: newConfig } };
         } catch (err) {
-          return { success: false, error: err.message }
+          return { success: false, error: err.message };
         }
-      }
-    })
+      },
+    });
   }
 
   reload(framework) {
-    // log.info(' Reloading...')
-    this._framework = framework
-    this._registerBuiltinTools()
+    this._framework = framework;
+    this._registerBuiltinTools();
   }
 
   uninstall(framework) {
-    // 注销内置工具
-    framework.toolRegistry.unregister('reload_plugins')
-    framework.toolRegistry.unregister('list_plugins')
-    framework.toolRegistry.unregister('list_tools')
-    framework.toolRegistry.unregister('enable_plugin')
-    framework.toolRegistry.unregister('disable_plugin')
-    framework.toolRegistry.unregister('get_plugin_config')
-    framework.toolRegistry.unregister('update_plugin_config')
-    this._framework = null
+    framework.toolRegistry.unregister('reload_plugins');
+    framework.toolRegistry.unregister('list_plugins');
+    framework.toolRegistry.unregister('list_tools');
+    framework.toolRegistry.unregister('enable_plugin');
+    framework.toolRegistry.unregister('disable_plugin');
+    framework.toolRegistry.unregister('get_plugin_config');
+    framework.toolRegistry.unregister('update_plugin_config');
+    this._framework = null;
   }
 }
 
-module.exports = ToolsPlugin
+module.exports = ToolsPlugin;

package/plugins/trading/README.md

@@ -0,0 +1,15 @@
+# Gate Trading Plugin
+
+Gate.io API 交易集成。
+
+## 配置
+
+通过 `config.gate` 在 plugins.json 或环境变量中配置：
+- `GATE_API_KEY` — Gate.io API Key
+- `GATE_API_SECRET` — Gate.io API Secret
+
+## 工具
+
+- 行情查询
+- 账户资产查询
+- 订单操作

package/plugins/{gate-trading.js → trading/index.js}

@@ -1,4 +1,4 @@
-const { Plugin } = require('../src/core/plugin-base')
+const { Plugin } = require('../../src/plugin/base')
 class GateTradingPlugin extends Plugin {
   constructor(config = {}) {
     super();
@@ -6,7 +6,7 @@ class GateTradingPlugin extends Plugin {
     this.version = '1.1.0';
     this.description = 'Gate.io 加密货币交易插件（支持现货+合约）';
     this.priority = 100;
-    
+
     // 从 .env 文件读取密钥
     const fs = require('fs');
     const path = require('path');
@@ -423,7 +423,7 @@ class GateTradingPlugin extends Plugin {
           this._checkAuth();
           const futuresApi = this._createFuturesApi();
           const settle = args.settle.toLowerCase();
-          
+
           // 构造订单对象 - settle 不能在订单内部
           const order = {
             contract: args.contract,
@@ -433,7 +433,7 @@ class GateTradingPlugin extends Plugin {
           };
           if (args.price) order.price = args.price;
           if (args.reduce_only !== undefined) order.reduce_only = args.reduce_only;
-          
+
           // SDK签名: createFuturesOrder(settle, futuresOrder, opts)
           const result = await futuresApi.createFuturesOrder(settle, order);
           const r = result.body;
@@ -449,7 +449,7 @@ class GateTradingPlugin extends Plugin {
             fill_price: r.fill_price,
             create_time: r.create_time,
           };
-        } catch (error) { 
+        } catch (error) {
           return { success: false, error: error.message }; }
       },
     });
@@ -522,13 +522,13 @@ class GateTradingPlugin extends Plugin {
           this._checkAuth();
           const futuresApi = this._createFuturesApi();
           const settle = (args.settle || 'USDT').toLowerCase();
-          
+
           // SDK签名: listFuturesOrders(settle, status, opts)
           // status: open = 未完成, finished = 已完成
           const status = 'open';
           const opts = {};
           if (args.contract) opts.contract = args.contract;
-          
+
           const result = await futuresApi.listFuturesOrders(settle, status, opts);
           const orders = (Array.isArray(result.body) ? result.body : []);
           return {
@@ -565,7 +565,7 @@ class GateTradingPlugin extends Plugin {
           const futuresApi = this._createFuturesApi();
           const settle = (args.settle || 'USDT').toLowerCase();
           const { info_type, contract } = args;
-          
+
           if (info_type === 'book' || !info_type) {
             // SDK签名: listFuturesOrderBook(settle, contract, opts)
             const book = await futuresApi.listFuturesOrderBook(settle, contract || 'BTC_USDT');

package/{examples → sandbox}/test-concurrent-chat.js

@@ -3,8 +3,8 @@
  */
 
 const { Framework } = require('../src');
-const AIPlugin = require('../plugins/ai-plugin');
-const SessionPlugin = require('../plugins/session-plugin');
+const AIPlugin = require('../plugins/core/ai');
+const SessionPlugin = require('../plugins/core/session');
 require('dotenv').config();
 
 async function main() {

package/{examples → sandbox}/test-long-chat.js

@@ -3,8 +3,8 @@
  */
 
 const { Framework } = require('../src');
-const AIPlugin = require('../plugins/ai-plugin');
-const SessionPlugin = require('../plugins/session-plugin');
+const AIPlugin = require('../plugins/core/ai');
+const SessionPlugin = require('../plugins/core/session');
 require('dotenv').config();
 
 async function main() {

package/{examples → sandbox}/test-session-chat.js

@@ -4,8 +4,8 @@
  */
 
 const { Framework } = require('../src');
-const AIPlugin = require('../plugins/ai-plugin');
-const SessionPlugin = require('../plugins/session-plugin');
+const AIPlugin = require('../plugins/core/ai');
+const SessionPlugin = require('../plugins/core/session');
 require('dotenv').config();
 
 async function main() {

package/{examples → sandbox}/test-web-plugin.js

@@ -3,7 +3,7 @@
  * 测试路由注册和请求响应
  */
 
-const { WebPlugin } = require('../plugins/web-plugin');
+const { WebPlugin } = require('../plugins/io/web');
 
 async function test() {
   console.log('=== Web 插件测试 ===\n');

package/{examples → sandbox}/test-weixin-feishu.js

@@ -21,14 +21,14 @@ async function main() {
   });
 
   // 加载微信插件
-  const { WeixinPlugin } = require('../plugins/weixin-plugin');
+  const { WeixinPlugin } = require('../plugins/messaging/weixin');
   const weixinPlugin = new WeixinPlugin({
     forceLogin: false,
     allowedUsers: [], // 空数组表示允许所有人
   });
 
   // 加载飞书插件
-  const { FeishuPlugin } = require('../plugins/feishu-plugin');
+  const { FeishuPlugin } = require('../plugins/messaging/feishu');
   const feishuPlugin = new FeishuPlugin({
     // appId: 'your-app-id',
     // appSecret: 'your-app-secret',

package/src/agent/base.js

@@ -0,0 +1,56 @@
+'use strict';
+
+/**
+ * BaseAgent - 所有 Agent 类型的公共基类
+ * 提供事件、状态、prompt 部分注册等公共功能
+ */
+
+const { EventEmitter } = require('../common/events');
+const { SystemPromptBuilder } = require('./prompt');
+
+class BaseAgent extends EventEmitter {
+  constructor(framework, config = {}) {
+    super();
+    this.framework = framework;
+    this.config = config;
+    this.name = config.name || 'Agent';
+    this._status = 'idle';
+    this._promptBuilder = new SystemPromptBuilder();
+  }
+
+  // === Status ===
+
+  getStatus() { return this._status; }
+
+  resetStatus() {
+    this._status = 'idle';
+    this.emit('status', { status: 'idle' });
+    return this;
+  }
+
+  // === Prompt Parts ===
+
+  registerPromptPart(name, priority, provider) {
+    this._promptBuilder.register(name, priority, provider);
+    return this;
+  }
+
+  unregisterPromptPart(name) {
+    this._promptBuilder.unregister(name);
+    return this;
+  }
+
+  invalidatePromptPart(name) {
+    this._promptBuilder.invalidate(name);
+    return this;
+  }
+
+  // === Destroy ===
+
+  destroy() {
+    this.removeAllListeners();
+    this.emit('destroyed');
+  }
+}
+
+module.exports = { BaseAgent };

package/src/{core/agent-chat.js → agent/chat.js}

@@ -8,23 +8,23 @@
  * - ContextCompressor: 上下文压缩
  */
 
-const { EventEmitter } = require('../utils/event-emitter');
-const { logger } = require('../utils/logger');
+const { EventEmitter } = require('../common/events');
+const { logger } = require('../common/logger');
 const fs = require('fs');
 const {
   tool: aiTool,
   ToolLoopAgent,
   isLoopFinished,
 } = require('ai');
-const { autoSplitToolResult } = require('../../plugins/data-splitter-plugin');
+const { autoSplitToolResult } = require('../../plugins/executors/data-splitter');
 const { cleanResponse } = require('../utils');
-const { ChatQueueManager } = require('../utils/chat-queue');
-const { TokenCounter } = require('./token-counter');
-const { isThinkingModel } = require('./provider');
+const { ChatQueueManager } = require('../common/queue');
+const { TokenCounter } = require('../llm/tokens');
+const { isThinkingModel } = require('../llm/provider');
 // 新模块
-const { ChatSession } = require('./chat-session');
-const { ToolExecutor } = require('./tool-executor');
-const { ContextCompressor } = require('./context-compressor');
+const { ChatSession } = require('../session/chat');
+const { ToolExecutor } = require('../tool/executor');
+const { ContextCompressor } = require('../context/compressor');
 const {
   DEFAULT_MAX_OUTPUT_TOKENS,
   DEFAULT_TEMPERATURE,
@@ -32,7 +32,7 @@ const {
   DEFAULT_MAX_CONCURRENT,
   DEFAULT_RETRY_ATTEMPTS,
   DEFAULT_RETRY_DELAY_MS,
-} = require('./constants');
+} = require('../common/constants');
 
 class AgentChatHandler extends EventEmitter {
   /**
@@ -254,7 +254,7 @@ class AgentChatHandler extends EventEmitter {
    * @private
    */
   _createAIProvider() {
-    const { createAI, createModel } = require('./provider');
+    const { createAI, createModel } = require('../llm/provider');
     const aiProvider = createAI({
       provider: this.provider,
       model: this.model,

package/src/{core/coordinator-manager.js → agent/coordinator.js}

@@ -3,9 +3,9 @@
  * 管理Worker的创建、消息传递、生命周期
  */
 
-const { EventEmitter } = require('../utils/event-emitter');
-const { WorkerAgent } = require('./worker-agent');
-const { logger } = require('../utils/logger');
+const { EventEmitter } = require('../common/events');
+const { WorkerAgent } = require('../agent/worker');
+const { logger } = require('../common/logger');
 
 const log = logger.child('CoordinatorManager');
 

package/src/agent/index.js

@@ -0,0 +1,111 @@
+'use strict';
+
+const { BaseAgent } = require('./base');
+const { MainAgent } = require('./main');
+const { SubAgent } = require('./sub');
+const { WorkerAgent } = require('./worker');
+const { SystemPromptBuilder } = require('./prompt');
+
+/**
+ * Create a MainAgent instance
+ * @param {Framework} framework
+ * @param {Object} config
+ * @returns {MainAgent}
+ */
+/**
+ * Create a MainAgent with AI config merge
+ * @param {Framework} framework
+ * @param {Object} config
+ * @returns {MainAgent}
+ */
+function createAgent(framework, config = {}) {
+  const merged = _mergeAIConfig(framework, config);
+  const agent = new MainAgent(framework, merged);
+  framework._agents.push(agent);
+  if (!framework._mainAgent) {
+    framework._mainAgent = agent;
+  }
+  framework.emit('agent:created', agent);
+  return agent;
+}
+
+/**
+ * Create a SubAgent
+ */
+function createSubAgent(framework, config = {}) {
+  const aiPlugin = framework.pluginManager?.get('ai');
+  const aiConfig = aiPlugin ? aiPlugin.getConfig() : {};
+  const mergedConfig = { ...aiConfig, ...(config.llmConfig || {}) };
+  const subagent = new SubAgent({
+    name: `subagent_${config.name}`,
+    role: config.role || config.name,
+    description: config.description || '',
+    systemPrompt: config.systemPrompt,
+    model: mergedConfig.model || 'deepseek-chat',
+    provider: mergedConfig.provider || 'deepseek',
+    apiKey: mergedConfig.apiKey,
+    baseURL: mergedConfig.baseURL,
+    providerOptions: { maxOutputTokens: mergedConfig.maxOutputTokens || 8192, temperature: mergedConfig.temperature || 0.3 },
+    tools: config.tools || {},
+    parentTools: config.parentTools,
+    framework,
+    maxRetries: config.maxRetries,
+    disableTools: config.disableTools,
+  });
+  _syncPromptParts(framework, subagent);
+  framework._agents.push(subagent);
+  framework.emit('agent:created', subagent);
+  return subagent;
+}
+
+/**
+ * Create a Session-bound Agent
+ */
+function createSessionAgent(framework, sessionId, config = {}) {
+  return createAgent(framework, { name: `session_${sessionId}`, ...config, sessionId });
+}
+
+/**
+ * Merge AI plugin config into agent config
+ */
+function _mergeAIConfig(framework, target) {
+  if (target.apiKey) return target;
+  const aiPlugin = framework.pluginManager?.get('ai');
+  if (!aiPlugin) return target;
+  return {
+    ...target,
+    apiKey: target.apiKey || aiPlugin.config?.apiKey,
+    provider: target.provider || aiPlugin.config?.provider,
+    model: target.model || aiPlugin.config?.model,
+    baseURL: target.baseURL || aiPlugin.config?.baseURL,
+    providerOptions: target.providerOptions || aiPlugin.config?.providerOptions || {},
+  };
+}
+
+/**
+ * Sync main agent prompt parts to subagent
+ */
+function _syncPromptParts(framework, subagent) {
+  if (!framework._mainAgent?._systemPromptBuilder?._parts) return;
+  const mainParts = framework._mainAgent._systemPromptBuilder._parts;
+  const registeredNames = new Set(subagent._promptParts?.map(p => p.name) || []);
+  for (const [name, part] of mainParts) {
+    if (['original-prompt', 'shared-prompt', 'datetime', 'metadata', 'capabilities'].includes(name)) continue;
+    if (!registeredNames.has(name)) {
+      subagent.registerPromptPart(name, part.priority, part.provider);
+      registeredNames.add(name);
+    }
+  }
+}
+
+module.exports = {
+  BaseAgent,
+  MainAgent,
+  SubAgent,
+  WorkerAgent,
+  SystemPromptBuilder,
+  createAgent,
+  createSubAgent,
+  createSessionAgent,
+  _syncPromptParts,
+};