foliko 1.0.74 → 1.0.76

package/src/core/provider.js

@@ -1,111 +1,114 @@
-/**
- * AI Provider 工厂
- * 支持多种 AI 提供商
- */
-
-const { createOpenAICompatible } = require('@ai-sdk/openai-compatible')
-const { createOpenAI } = require('@ai-sdk/openai')
-const { createAnthropic } = require('@ai-sdk/anthropic')
-
-/**
- * 默认提供商配置
- */
-const DEFAULT_PROVIDERS = {
-  openai: {
-    name: 'OpenAI',
-    baseURL: 'https://api.openai.com/v1'
-  },
-  ollama: {
-    name: 'Ollama',
-    baseURL: 'http://localhost:11434/v1'
-  },
-  lmstudio: {
-    name: 'LM Studio',
-    baseURL: 'http://localhost:1234/v1'
-  },
-  deepseek: {
-    name: 'DeepSeek',
-    baseURL: 'https://api.deepseek.com/v1'
-  },
-  anthropic: {
-    name: 'Anthropic',
-    baseURL: 'https://api.anthropic.com/v1'
-  },
-  minimax: {
-    name: 'MiniMax',
-    baseURL: 'https://api.minimaxi.com/v1'
-  }
-}
-
-/**
- * 创建 AI 客户端
- * @param {Object} config - 配置
- * @param {string} config.provider - 提供商名称
- * @param {string} config.model - 模型名称
- * @param {string} config.apiKey - API 密钥
- * @param {string} [config.baseURL] - 自定义 API 地址
- */
-function createAI(config) {
-  const { provider, model, apiKey, baseURL } = config
-  const providerName = (provider || 'deepseek').toLowerCase()
-  // 检查是否是预定义提供商
-  if (DEFAULT_PROVIDERS[providerName]) {
-    const providerConfig = DEFAULT_PROVIDERS[providerName]
-    return createOpenAICompatible({
-      name: providerConfig.name,
-      baseURL: baseURL || providerConfig.baseURL,
-      apiKey: apiKey || 'dummy-key',
-      headers: provider === 'anthropic' ? {
-        'x-api-key': apiKey,
-        'anthropic-version': '2023-06-01'
-      } : undefined,
-      models: {
-        default: {
-          id: model || 'deepseek-chat',
-          streamOptions: {
-            includeUsage: true
-          }
-        }
-      }
-    })
-  }
-
-  // 自定义提供商
-  return createOpenAICompatible({
-    name: providerName || 'Custom',
-    baseURL: baseURL,
-    apiKey: apiKey || 'dummy-key',
-    models: {
-      default: {
-        id: model || 'gpt-4'
-      }
-    }
-  })
-}
-
-/**
- * 获取可用提供商列表
- */
-function getAvailableProviders() {
-  return Object.entries(DEFAULT_PROVIDERS).map(([key, value]) => ({
-    id: key,
-    name: value.name,
-    baseURL: value.baseURL
-  }))
-}
-
-/**
- * 创建模型实例
- * @param {string} model - 模型名称
- * @param {Object} provider - Provider 实例
- */
-function createModel(model, provider) {
-  return provider(model)
-}
-
-module.exports = {
-  createAI,
-  createModel,
-  getAvailableProviders,
-  DEFAULT_PROVIDERS
-}
+/**
+ * AI Provider 工厂
+ * 支持多种 AI 提供商
+ */
+
+const { createOpenAICompatible } = require('@ai-sdk/openai-compatible');
+const { createOpenAI } = require('@ai-sdk/openai');
+const { createAnthropic } = require('@ai-sdk/anthropic');
+
+/**
+ * 默认提供商配置
+ */
+const DEFAULT_PROVIDERS = {
+  openai: {
+    name: 'OpenAI',
+    baseURL: 'https://api.openai.com/v1',
+  },
+  ollama: {
+    name: 'Ollama',
+    baseURL: 'http://localhost:11434/v1',
+  },
+  lmstudio: {
+    name: 'LM Studio',
+    baseURL: 'http://localhost:1234/v1',
+  },
+  deepseek: {
+    name: 'DeepSeek',
+    baseURL: 'https://api.deepseek.com/v1',
+  },
+  anthropic: {
+    name: 'Anthropic',
+    baseURL: 'https://api.anthropic.com/v1',
+  },
+  minimax: {
+    name: 'MiniMax',
+    baseURL: 'https://api.minimaxi.com/v1',
+  },
+};
+
+/**
+ * 创建 AI 客户端
+ * @param {Object} config - 配置
+ * @param {string} config.provider - 提供商名称
+ * @param {string} config.model - 模型名称
+ * @param {string} config.apiKey - API 密钥
+ * @param {string} [config.baseURL] - 自定义 API 地址
+ */
+function createAI(config) {
+  const { provider, model, apiKey, baseURL } = config;
+  const providerName = (provider || 'deepseek').toLowerCase();
+  // 检查是否是预定义提供商
+  if (DEFAULT_PROVIDERS[providerName]) {
+    const providerConfig = DEFAULT_PROVIDERS[providerName];
+    return createOpenAICompatible({
+      name: providerConfig.name,
+      baseURL: baseURL || providerConfig.baseURL,
+      apiKey: apiKey || 'dummy-key',
+      headers:
+        provider === 'anthropic'
+          ? {
+              'x-api-key': apiKey,
+              'anthropic-version': '2023-06-01',
+            }
+          : undefined,
+      models: {
+        default: {
+          id: model || 'deepseek-chat',
+          streamOptions: {
+            includeUsage: true,
+          },
+        },
+      },
+    });
+  }
+
+  // 自定义提供商
+  return createOpenAICompatible({
+    name: providerName || 'Custom',
+    baseURL: baseURL,
+    apiKey: apiKey || 'dummy-key',
+    models: {
+      default: {
+        id: model || 'gpt-4',
+      },
+    },
+  });
+}
+
+/**
+ * 获取可用提供商列表
+ */
+function getAvailableProviders() {
+  return Object.entries(DEFAULT_PROVIDERS).map(([key, value]) => ({
+    id: key,
+    name: value.name,
+    baseURL: value.baseURL,
+  }));
+}
+
+/**
+ * 创建模型实例
+ * @param {string} model - 模型名称
+ * @param {Object} provider - Provider 实例
+ */
+function createModel(model, provider) {
+  return provider(model);
+}
+
+module.exports = {
+  createAI,
+  createModel,
+  getAvailableProviders,
+  DEFAULT_PROVIDERS,
+};

package/src/core/sub-agent-config.js

@@ -0,0 +1,264 @@
+/**
+ * SubAgent 配置管理
+ * 负责读取、解析和缓存 .agent/agents/ 目录下的 md 文件
+ * 类似 SkillManager 的工作方式
+ */
+
+const fs = require('fs');
+const path = require('path');
+
+/**
+ * 解析 YAML frontmatter
+ */
+function parseFrontmatter(content) {
+  const match = content.match(/^---\r?\n([\s\S]*?)\r?\n---/);
+  if (!match) return null;
+
+  const frontmatter = {};
+  const lines = match[1].split('\n');
+  let currentKey = null;
+
+  for (const line of lines) {
+    // 处理 metadata 嵌套
+    if (currentKey === 'metadata') {
+      if (line.match(/^ {2}[a-zA-Z]/)) {
+        const colonIndex = line.indexOf(':');
+        if (colonIndex > 0) {
+          const key = line.substring(2, colonIndex).trim();
+          const value = line
+            .substring(colonIndex + 1)
+            .trim()
+            .replace(/^["']|["']$/g, '');
+          frontmatter[currentKey][key] = value;
+          continue;
+        }
+      } else if (line.match(/^[a-zA-Z]/)) {
+        currentKey = null;
+        continue;
+      }
+    }
+
+    const colonIndex = line.indexOf(':');
+    if (colonIndex === -1) continue;
+
+    const key = line.substring(0, colonIndex).trim();
+    let value = line.substring(colonIndex + 1).trim();
+
+    // 移除引号
+    if (
+      (value.startsWith('"') && value.endsWith('"')) ||
+      (value.startsWith("'") && value.endsWith("'"))
+    ) {
+      value = value.slice(1, -1);
+    }
+
+    if (key === 'metadata') {
+      currentKey = 'metadata';
+      frontmatter.metadata = {};
+    } else if (key === 'allowed-tools') {
+      frontmatter[key] = value
+        .split(',')
+        .map((v) => v.trim())
+        .filter(Boolean);
+    } else if (key === 'skills') {
+      frontmatter[key] = value
+        .split(',')
+        .map((v) => v.trim())
+        .filter(Boolean);
+    } else if (key === 'tools') {
+      frontmatter[key] = value
+        .split(',')
+        .map((v) => v.trim())
+        .filter(Boolean);
+    } else {
+      frontmatter[key] = value;
+    }
+  }
+
+  return frontmatter;
+}
+
+/**
+ * 移除 frontmatter，只保留正文内容
+ */
+function stripFrontmatter(content) {
+  return content.replace(/^---\r?\n[\s\S]*?\r?\n---\r?\n/, '');
+}
+
+class SubAgentConfig {
+  constructor(name, filePath) {
+    this.name = name;
+    this.filePath = filePath;
+    this._frontmatter = null;
+    this._content = null;
+    this._loaded = false;
+  }
+
+  /**
+   * 加载并解析 md 文件
+   */
+  load() {
+    if (this._loaded) return;
+
+    try {
+      const content = fs.readFileSync(this.filePath, 'utf-8');
+      this._frontmatter = parseFrontmatter(content);
+      this._content = stripFrontmatter(content);
+      this._loaded = true;
+    } catch (e) {
+      console.warn(`[SubAgentConfig] Failed to load ${this.filePath}:`, e.message);
+      this._frontmatter = {};
+      this._content = '';
+      this._loaded = true;
+    }
+  }
+
+  /**
+   * 获取 frontmatter 中的值
+   */
+  get(key, defaultValue = null) {
+    this.load();
+    return this._frontmatter?.[key] ?? defaultValue;
+  }
+
+  /**
+   * 获取名称
+   */
+  getName() {
+    return this.get('name', this.name);
+  }
+
+  /**
+   * 获取描述
+   */
+  getDescription() {
+    return this.get('description', '');
+  }
+
+  /**
+   * 获取技能列表
+   */
+  getSkills() {
+    return this.get('skills', []);
+  }
+
+  /**
+   * 获取工具列表
+   */
+  getTools() {
+    return this.get('tools', []);
+  }
+
+  /**
+   * 获取模型配置
+   */
+  getModel() {
+    return this.get('model', 'inherit');
+  }
+
+  /**
+   * 获取完整内容（用于构建系统提示词）
+   */
+  getContent() {
+    this.load();
+    return this._content || '';
+  }
+
+  /**
+   * 获取完整的系统提示词（frontmatter + 内容）
+   */
+  getSystemPrompt() {
+    this.load();
+    const desc = this.getDescription();
+    const content = this._content || '';
+
+    // 构建系统提示词
+    let prompt = `# ${this.getName()}`;
+    if (desc) {
+      prompt += `\n\n${desc}`;
+    }
+    if (content.trim()) {
+      prompt += `\n\n${content.trim()}`;
+    }
+    return prompt;
+  }
+
+  /**
+   * 检查是否有效
+   */
+  isValid() {
+    this.load();
+    return this._loaded && this._frontmatter !== null;
+  }
+}
+
+/**
+ * SubAgent 配置管理器
+ * 管理目录下所有子代理配置的加载和缓存
+ */
+class SubAgentConfigManager {
+  constructor(agentsDir) {
+    this.agentsDir = agentsDir;
+    this._configs = new Map(); // name -> SubAgentConfig
+  }
+
+  /**
+   * 加载目录下所有子代理配置
+   */
+  loadAll() {
+    if (!fs.existsSync(this.agentsDir)) {
+      return;
+    }
+
+    const entries = fs.readdirSync(this.agentsDir, { withFileTypes: true });
+    for (const entry of entries) {
+      if (entry.isFile() && entry.name.endsWith('.md')) {
+        const name = entry.name.replace('.md', '');
+        const filePath = path.join(this.agentsDir, entry.name);
+
+        if (!this._configs.has(name)) {
+          this._configs.set(name, new SubAgentConfig(name, filePath));
+        }
+      }
+    }
+  }
+
+  /**
+   * 刷新配置（清除后重新加载）
+   */
+  refresh() {
+    this._configs.clear();
+    this.loadAll();
+  }
+
+  /**
+   * 获取某个子代理配置
+   */
+  get(name) {
+    return this._configs.get(name) || null;
+  }
+
+  /**
+   * 获取所有配置
+   */
+  getAll() {
+    return Array.from(this._configs.values());
+  }
+
+  /**
+   * 获取所有有效配置
+   */
+  getValidConfigs() {
+    return this.getAll().filter((c) => c.isValid());
+  }
+
+  /**
+   * 重新加载
+   */
+  reload() {
+    this._configs.clear();
+    this.loadAll();
+  }
+}
+
+module.exports = { SubAgentConfig, SubAgentConfigManager, parseFrontmatter, stripFrontmatter };

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

@@ -0,0 +1,120 @@
+/**
+ * 系统提示词构建器
+ * 负责从多个来源组合系统提示词各部分
+ * 支持插件注册自己的提示词部分，支持重载后更新
+ */
+
+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 };