foliko 1.1.93 → 2.0.0

package/src/core/subagent.js

@@ -1,442 +0,0 @@
-/**
- * Subagent 类
- * 轻量级子 Agent，用于完成单次任务
- * 不需要 session、上下文压缩等复杂功能
- */
-
-const { EventEmitter } = require('../utils/event-emitter');
-const { cleanResponse } = require('../utils');
-const { generateText, tool, stepCountIs, isLoopFinished, RetryError, ToolLoopAgent } = require('ai');
-const { z } = require('zod');
-const { logger } = require('../utils/logger');
-const { validateAll } = require('../utils/message-validator');
-const { isNetworkError, calculateDelay } = require('../utils/retry');
-
-class Subagent extends EventEmitter {
-  /**
-   * @param {Object} config - 配置
-   * @param {string} config.name - Subagent 名称
-   * @param {string} [config.role] - 角色描述
-   * @param {string} [config.description] - 描述
-   * @param {string} [config.systemPrompt] - 系统提示词（可选，如果不提供则在 chat 时动态构建）
-   * @param {string} [config.model] - 模型名称
-   * @param {string} [config.provider] - 提供者
-   * @param {string} [config.apiKey] - API 密钥
-   * @param {string} [config.baseURL] - base URL
-   * @param {Object} [config.providerOptions] - 提供者选项
-   * @param {Array|Object} [config.tools] - 工具定义
-   * @param {Object} [config.framework] - Framework 引用（用于动态构建系统提示词）
-   */
-  constructor(config = {}) {
-    super();
-
-    this.name = config.name || 'Subagent';
-    this.role = config.role || '助手';
-    this.description = config.description || '';
-    this.model = config.model || 'deepseek-chat';
-    this.provider = config.provider || 'deepseek';
-    this.apiKey = config.apiKey;
-    this.baseURL = config.baseURL;
-    this.providerOptions = config.providerOptions || {};
-    this.framework = config.framework || null;
-
-    this.defaulTools = ['ext_skill','loadSkill','ext_call','subagent_call','mcp_call','mcp_tool_schema'];
-
-    this.bindTools = {
-      read: 'read_file',
-      write: 'write_file',
-      edit: 'modify_file',
-      glob: 'read_directory',
-      grep: 'search_file',
-      bash: 'bash',
-    };
-    // 如果提供了 systemPrompt 则使用，否则标记为需要动态构建
-    this._customSystemPrompt = config.systemPrompt || null;
-    this.parentTools = config?.parentTools;
-    // 工具管理
-    this.tools=config.tools||{}
-
-    // 重试配置
-    this.maxRetries = config.maxRetries ?? 2;
-    this.retryDelay = config.retryDelay ?? 5000;
-    // 禁用工具
-    this.disableTools = config.disableTools ?? false;
-
-    // 提示词部分注册（支持插件注册提示词）
-    this._promptParts = [];
-
-    // 标记为子Agent（用于区分主Agent）
-    this._isSubagent = true;
-
-    this.providerOptions.deepseek = this.providerOptions.deepseek || {};
-    this.providerOptions.deepseek.thinking={ type: 'disabled' }
-
-    this.framework.once('framework:ready', () => {
-      const extExecutor = this.framework?.pluginManager?.get('extension-executor');
-      extExecutor._refreshAllAgentsExtPrompt(this.framework);
-    });
-  }
-
-  /**
-   * 注册系统提示词部分
-   * @param {string} name - 部分名称
-   * @param {number} [priority=100] - 优先级
-   * @param {Function} provider - 返回提示词内容的函数
-   */
-  registerPromptPart(name, priority = 100, provider) {
-    this._promptParts.push({ name, priority, provider });
-    // 按优先级排序
-    this._promptParts.sort((a, b) => a.priority - b.priority);
-    return this;
-  }
-
-  /**
-   * 验证消息配对（工具调用和结果配对）
-   * @private
-   */
-  _validateMessagesPairing(messages) {
-    return validateAll(messages);
-  }
-
-  /**
-   * 验证工具调用
-   * @private
-   */
-  _validateToolCalls(messages) {
-    // validateAll 内部已包含 validateMessagesPairing + validateToolCalls
-    // Subagent 的 chat() 中 _validateMessagesPairing 已调用 validateAll，所以这里不再重复
-  }
-
-  /**
-   * 注册工具
-   * @param {Array|Object} tools - 工具定义
-   */
-  _registerTools(tools) {
-    if (Array.isArray(tools)) {
-      for (const t of tools) {
-        if (t && t.name) {
-          this.tools[t.name]=t
-        }
-      }
-    } else if (tools && typeof tools === 'object') {
-      for (const [name, t] of Object.entries(tools)) {
-        this.tools[name]=t
-      }
-    }
-  }
-
-  /**
-   * 添加工具
-   * @param {Object} toolDef - 工具定义 { name, description, inputSchema, execute }
-   */
-  addTool(toolDef) {
-    this.tools[toolDef.name]=toolDef
-    return this;
-  }
-
-  /**
-   * 获取所有工具
-   */
-  getTools() {
-    return Object.values(this.tools);
-  }
-
-  /**
-   * 获取 AI 提供者
-   */
-  _getAIProvider() {
-    const { createAI } = require('./provider');
-    return createAI({
-      provider: this.provider,
-      model: this.model,
-      apiKey: this.apiKey,
-      baseURL: this.baseURL,
-    });
-  }
-
-  /**
-   * 构建 AI 工具格式
-   */
-  _buildAITools() {
-    // 禁用工具时返回空
-    const tools = {};
-    // 从父Agent继承工具
-    const all_tools = this.framework.getTools();
-    let parentTools = [];
-    if (Array.isArray(this.parentTools)) {
-      parentTools = this.parentTools.map((key) => {
-        return this.bindTools[key.toLocaleLowerCase()] || key;
-      });
-      for (const toolName of parentTools) {
-        const toolDef = all_tools.find((t) => t.name === toolName);
-        if (toolDef) {
-          tools[toolDef.name] = toolDef;
-        }
-      }
-      const defaulTools = all_tools.filter((a) => this.defaulTools.includes(a.name));
-
-      defaulTools.map((tool) => {
-        tools[tool.name] = tool;
-      });
-    } else {
-      
-      all_tools.forEach(item => {
-        tools[item.name]=item
-      });
-    } 
-    return { ...tools, ...this.tools };
-  }
-
-  /**
-   * 动态构建系统提示词
-   * @private
-   */
-  _buildSystemPrompt() {
-    const lines = [];
-    const roleName = this.role || '助手';
-    const descText = this.description || '';
-
-    // 1. 角色定义
-    lines.push(`你是 ${roleName}。`);
-    if (descText) {
-      lines.push(descText);
-    }
-    lines.push('');
-    const now = new Date();
-    const timeStr = now.toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai' });
-    const dateStr = now.toLocaleDateString('zh-CN', {
-      timeZone: 'Asia/Shanghai',
-      weekday: 'long',
-    });
-    lines.push(`【当前时间】${timeStr}（${dateStr}）`)
-
-    if (this._customSystemPrompt) {
-      lines.push(this._customSystemPrompt);
-    }
-
-    // 2. 主Agent的系统提示词
-    const mainAgent = this.framework?.getMainAgent?.();
-    if (mainAgent) {
-      const mainPrompt = mainAgent.getOriginalPrompt();
-      if (mainPrompt) {
-        lines.push('## 主Agent的系统提示词');
-        lines.push(mainPrompt);
-        lines.push('');
-      }
-    }
-
-    // // 3. 工具描述
-    // const tools = this._buildAITools();
-
-    // const toolKeys = Object.keys(tools);
-    // if (toolKeys.length > 0) {
-    //   lines.push('## 可用工具');
-    //   lines.push('');
-    //   for (const key of toolKeys) {
-    //     const toolDef = tools[key];
-    //     const desc = toolDef.description || '无描述';
-    //     lines.push(`- **${toolDef.name}**: ${desc}`);
-    //   }
-    //   lines.push('');
-    // }
-    const subagentManager = this.framework?.pluginManager?.get('subagent-manager');
-    lines.push(subagentManager._buildDescription());
-    lines.push('');
-    // 3.5 扩展插件工具描述
-    const extExecutor = this.framework?.pluginManager?.get('extension-executor');
-    const extDesc = extExecutor?._buildExtensionsDescription();
-    if (extDesc) {
-      lines.push(extDesc);
-      lines.push('');
-    }
-
-    // 5. 注册的提示词部分（来自插件）
-    if (this._promptParts && this._promptParts.length > 0) {
-      for (const part of this._promptParts) {
-        try {
-          const content = typeof part.provider === 'function' ? part.provider() : '';
-          if (content && content.trim()) {
-            lines.push('');
-            lines.push(content.trim());
-          }
-        } catch (err) {
-          logger.warn(`[Subagent:${this.name}] Failed to get prompt part '${part.name}':`, err.message);
-        }
-      }
-    }
-
-    // 6. SubAgentPlugin 的核心职责和规则
-    lines.push('');
-    lines.push(`## 核心职责`);
-    lines.push(`- **专注本职**：只完成与 ${this.role} 相关的任务，不要处理其他领域的请求`);
-    lines.push(`- **不越界**：不要代替其他子Agent完成任务，不要主动介入其他专业领域`);
-    lines.push(`- **返回结果**：返回简洁明确的操作结果，不需要解释过程`);
-    lines.push('');
-    lines.push(`当你被调用时，你应该：`);
-    lines.push(`1. 仔细理解任务要求，确认是否属于你的职责范围`);
-    lines.push(`2. 如果任务不属于你的职责范围，返回"此任务不在我的职责范围内"`);
-    lines.push(`3. 使用你的工具集完成任务`);
-    lines.push(`4. 返回完整的操作结果`);
-    lines.push('');
-    lines.push(`## 禁止事项`);
-    lines.push(`- 处理本职外的任务`);
-    lines.push(`- 代替其他子Agent执行操作`);
-    lines.push(`- 在回复中透露你的内部实现细节`);
-    lines.push(`- 不先调用 ext_skill/loadSkill 就直接使用 ext_call`);
-
-    return lines.join('\n');
-  }
-
-  /**
-   * 执行任务
-   * @param {string|Array} taskOrMessages - 任务描述或消息数组
-   * @param {Object} [options] - 选项
-   * @param {number} [options.maxSteps] - 最大步数
-   * @param {AbortSignal} [options.signal] - 中止信号
-   * @param {number} [options.maxRetries] - 最大重试次数
-   * @returns {Promise<{success: boolean, message: string, steps: number}>}
-   */
-  async chat(taskOrMessages, options = {}) {
-    const maxSteps = options?.maxSteps || 30;
-    const maxRetries = options?.maxRetries ?? this.maxRetries;
-    const retryDelay = options?.retryDelay ?? this.retryDelay;
-    const aiProvider = this._getAIProvider();
-
-    // 支持传入消息数组（用于持续对话上下文）或单个任务字符串
-    let messages;
-    if (Array.isArray(taskOrMessages)) {
-      messages = [...taskOrMessages];
-    } else {
-      messages = [{ role: 'user', content: taskOrMessages }];
-    }
-
-    // logger.info(
-    //   `[Subagent:${this.name}] chat called, maxRetries=${maxRetries}, disableTools=${this.disableTools}`
-    // );
-
-    let lastError;
-    for (let attempt = 0; attempt <= maxRetries; attempt++) {
-      // logger.info(
-      //   `[Subagent:${this.name}] attempt ${attempt + 1}, model=${this.model}, provider=${this.provider}`
-      // );
-      try {
-        const tools = this.disableTools ? {} : this._buildAITools();
-        const systemPrompt = this.disableTools
-          ? this._customSystemPrompt
-          : this._buildSystemPrompt();
-        // console.log(Object.keys(tools))
-        //await fs.writeFile('subagent.md',systemPrompt)
-        // 验证消息配对，清理无效的 tool call/result
-        const validated = this._validateMessagesPairing(messages);
-        if (validated.length !== messages.length) {
-          messages.length = 0;
-          messages.push(...validated);
-        }
-        this._validateToolCalls(messages);
-
-        // 使用 ToolLoopAgent 实现自动工具执行循环
-        const agent = new ToolLoopAgent({
-          model: aiProvider(this.model),
-          instructions: systemPrompt,
-          tools,
-          stopWhen: stepCountIs(maxSteps),
-        });
-
-        const result = await agent.generate({
-          messages,
-          ...this.providerOptions,
-          abortSignal: options.signal,
-        });
-        // console.log('[子Agent] ' ,result.text)
-        // ToolLoopAgent 会自动执行工具并收集结果，messages 已经被修改
-        messages.push(...result.response.messages);
-        const full_text = cleanResponse(result.text);
-        this.emit('complete', { message: full_text, steps: result.steps?.length || 0 });
-        return {
-          success: true,
-          message: full_text,
-          steps: result.steps?.length || 0,
-          messages,  // 返回完整消息历史，用于持续对话上下文
-        };
-      } catch (err) {
-        
-        lastError = err;
-          // 统一使用 retry.js 的错误判断
-        if (isNetworkError(err) && attempt < maxRetries) {
-          logger.warn(
-            `[Subagent:${this.name}] AI 服务暂时不可用，正在重试 (${attempt + 1}/${maxRetries})`
-          );
-          await new Promise((resolve) => setTimeout(resolve, retryDelay * Math.pow(2, attempt)));
-          continue;
-        }
-        logger.warn(`[Subagent:${this.name}] generateText error: ${err.message}`);
-        // 最终失败，转换为友好消息
-        const friendlyMessage = this._getFriendlyError(err);
-        this.emit('error', { error: friendlyMessage });
-        return {
-          success: false,
-          message: '',
-          error: friendlyMessage,
-          steps: 0,
-          messages,  // 即使失败也返回消息历史
-        };
-      }
-    }
-  }
-
-  /**
-   * 获取友好错误消息
-   * @private
-   */
-  _getFriendlyError(err) {
-    const errName = err?.name || '';
-
-    // 先用 isNetworkError 统一判断
-    if (isNetworkError(err)) {
-      return 'AI 服务暂时不可用，请稍后重试';
-    }
-
-    const errorMessages = {
-      AI_NoContentGeneratedError: 'AI 未生成有效内容，请重试',
-      AI_NoSuchModelError: '指定的 AI 模型不存在',
-      AI_NoSuchProviderError: 'AI 提供商配置错误',
-      AI_LoadAPIKeyError: 'AI API 密钥配置错误',
-    };
-    return errorMessages[errName] || err.message?.split('\n')[0] || '未知错误';
-  }
-
-  /**
-   * 设置角色（会清除自定义 systemPrompt）
-   * @param {string} role - 角色描述
-   */
-  setRole(role) {
-    this.role = role;
-    this._customSystemPrompt = null;
-  }
-
-  /**
-   * 设置描述
-   * @param {string} description - 描述
-   */
-  setDescription(description) {
-    this.description = description;
-  }
-
-  /**
-   * 设置系统提示词（会覆盖动态构建）
-   * @param {string} prompt - 系统提示词
-   */
-  setSystemPrompt(prompt) {
-    this._customSystemPrompt = prompt;
-  }
-
-  /**
-   * 销毁
-   */
-  destroy() {
-    this.removeAllListeners();
-    this.tools={}
-  }
-}
-
-module.exports = { Subagent };

package/src/core/system-prompt-builder.js

@@ -1,120 +0,0 @@
-/**
- * 系统提示词构建器
- * 负责从多个来源组合系统提示词各部分
- * 支持插件注册自己的提示词部分，支持重载后更新
- */
-
-class SystemPromptPart {
-  constructor(name, priority, provider) {
-    this.name = name;
-    this.priority = priority;
-    this._provider = provider;
-    this._dirty = true;
-    this._cached = null;
-  }
-
-  get() {
-    if (this._dirty || this._cached === null) {
-      try {
-        this._cached = this._provider();
-        this._dirty = false;
-      } catch (e) {
-        console.warn(`[SystemPromptPart:${this.name}] Provider failed:`, e.message);
-        this._cached = null;
-      }
-    }
-    return this._cached;
-  }
-
-  invalidate() {
-    this._dirty = true;
-  }
-}
-
-class SystemPromptBuilder {
-  constructor() {
-    this._parts = new Map(); // name -> SystemPromptPart
-    this._orderedParts = []; // sorted by priority
-  }
-
-  /**
-   * 注册一个提示词部分
-   * @param {string} name - 部分名称（唯一标识）
-   * @param {number} priority - 优先级（数字越小越靠前）
-   * @param {Function} provider - 返回提示词内容的函数
-   */
-  register(name, priority, provider) {
-    const part = new SystemPromptPart(name, priority, provider);
-    this._parts.set(name, part);
-    this._reorder();
-    return this;
-  }
-
-  /**
-   * 注销一个提示词部分
-   */
-  unregister(name) {
-    this._parts.delete(name);
-    this._reorder();
-    return this;
-  }
-
-  /**
-   * 使某个部分失效（下次 get 时会重新计算）
-   */
-  invalidate(name) {
-    const part = this._parts.get(name);
-    if (part) {
-      part.invalidate();
-    }
-    return this;
-  }
-
-  /**
-   * 使所有部分失效
-   */
-  invalidateAll() {
-    for (const part of this._parts.values()) {
-      part.invalidate();
-    }
-    return this;
-  }
-
-  /**
-   * 获取某个部分的内容
-   */
-  getPart(name) {
-    const part = this._parts.get(name);
-    return part ? part.get() : null;
-  }
-
-  /**
-   * 构建完整的系统提示词
-   * @returns {string}
-   */
-  build() {
-    const parts = [];
-    for (const part of this._orderedParts) {
-      const content = part.get();
-      if (content && content.trim()) {
-        parts.push(content.trim());
-      }
-    }
-    return parts.join('\n\n');
-  }
-
-  /**
-   * 获取所有部分名称
-   */
-  getPartNames() {
-    return [...this._parts.keys()];
-  }
-
-  _reorder() {
-    this._orderedParts = Array.from(this._parts.values()).sort(
-      (a, b) => a.priority - b.priority
-    );
-  }
-}
-
-module.exports = { SystemPromptBuilder, SystemPromptPart };