my-ai-chat-framework 2.7.0 → 3.0.0

package/src/plugins/model-registry.js

@@ -1,187 +1,223 @@
-/**
- * model-registry 插件
- * 
- * 职责：
- * 1. 内置常用模型的能力标签表（model → capabilities 映射）
- * 2. 安装时自动根据 config.model 查表，写入 config.capabilities
- * 3. 监听 config-updated 事件，切换模型时自动同步能力
- * 4. 提供 chat.registerModel()，允许用户追加/覆盖自定义模型
- * 
- * 能力标签字段（全部可选，缺失视为 false / 未知）：
- * {
- *   reasoning:    boolean — 是否返回 reasoning_content（思维链）
- *   vision:       boolean — 是否支持图片输入
- *   toolCalling:  boolean — 是否支持 function calling
- *   streaming:    boolean — 是否支持流式传输
- *   maxInputTokens:  number — 最大输入 token 数
- *   maxOutputTokens: number — 最大输出 token 数
- * }
- * 
- * 使用：
- *   chat.use(modelRegistryPlugin);
- *   // config.capabilities 现在已自动填充
- *   chat.registerModel('my-custom-model', { toolCalling: true });
- */
-
-// ====== 内置模型能力表 ======
-// 数据来源：各厂商官方文档（2026-05 快照），用户可通过 registerModel() 覆盖
-const BUILTIN_MODELS = {
-  // ---- DeepSeek ----
-  'deepseek-chat': {
-    reasoning: false,
-    vision: false,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 128000,
-    maxOutputTokens: 8192
-  },
-  'deepseek-reasoner': {
-    reasoning: true,
-    vision: false,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 128000,
-    maxOutputTokens: 8192
-  },
-
-  // ---- OpenAI ----
-  'gpt-4o': {
-    reasoning: false,
-    vision: true,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 128000,
-    maxOutputTokens: 16384
-  },
-  'gpt-4o-mini': {
-    reasoning: false,
-    vision: true,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 128000,
-    maxOutputTokens: 16384
-  },
-  'gpt-3.5-turbo': {
-    reasoning: false,
-    vision: false,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 16385,
-    maxOutputTokens: 4096
-  },
-  'o1': {
-    reasoning: true,
-    vision: false,
-    toolCalling: false,
-    streaming: false,
-    maxInputTokens: 200000,
-    maxOutputTokens: 100000
-  },
-  'o3-mini': {
-    reasoning: true,
-    vision: false,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 200000,
-    maxOutputTokens: 100000
-  },
-
-  // ---- Anthropic ----
-  'claude-3.5-sonnet': {
-    reasoning: false,
-    vision: true,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 200000,
-    maxOutputTokens: 8192
-  },
-  'claude-3.5-haiku': {
-    reasoning: false,
-    vision: true,
-    toolCalling: true,
-    streaming: true,
-    maxInputTokens: 200000,
-    maxOutputTokens: 8192
-  }
-};
-
-const DEFAULT_CAPABILITIES = {
-  streaming: true,
-  toolCalling: false,
-  reasoning: false,
-  vision: false
-};
-
-export const modelRegistryPlugin = {
-  name: 'model-registry',
-
-  install(chatService) {
-    this.chat = chatService;
-    // 合并内置表 + 用户通过 registerModel 追加的
-    this._registry = new Map(Object.entries(BUILTIN_MODELS));
-
-    /**
-     * 注册/覆盖一个模型的能力标签
-     * @param {string} name — 模型名称
-     * @param {Object} capabilities — 能力标签对象（部分字段即可，未提供的取默认值）
-     * @returns {ChatService}
-     */
-    chatService.registerModel = (name, capabilities = {}) => {
-      if (!name || typeof name !== 'string' || !name.trim()) {
-        throw new Error('[model-registry] 模型名称必须是非空字符串');
-      }
-      const merged = { ...DEFAULT_CAPABILITIES, ...capabilities };
-      this._registry.set(name.trim(), merged);
-      // 如果注册的模型与当前使用的模型同名，立即同步
-      if (chatService.config.model === name.trim()) {
-        this._syncCapabilities();
-      }
-      return chatService;
-    };
-
-    /**
-     * 列出所有已注册的模型名称
-     * @returns {Array<string>}
-     */
-    chatService.listModels = () => {
-      return [...this._registry.keys()];
-    };
-
-    // 安装时立即同步一次
-    this._syncCapabilities();
-
-    // 监听配置变更：model 变了就重新同步
-    chatService.on('config-updated', ({ changes }) => {
-      if ('model' in changes) {
-        this._syncCapabilities();
-      }
-      // 如果用户手动改了 capabilities，用 merge 保留未指定的字段
-      if ('capabilities' in changes && !('model' in changes)) {
-        const current = this._lookupCapabilities();
-        if (current) {
-          chatService.config.capabilities = { ...current, ...changes.capabilities };
-        }
-      }
-    });
-  },
-
-  // ---- 内部方法 ----
-
-  /** 查表获取模型能力，未注册返回 null */
-  _lookupCapabilities() {
-    const model = this.chat.config.model;
-    if (!model) return null;
-    return this._registry.get(model) || null;
-  },
-
-  /** 将查到的能力写入 config.capabilities */
-  _syncCapabilities() {
-    const caps = this._lookupCapabilities();
-    if (caps) {
-      // 若用户已手动设了 capabilities，用模型的合并（用户值优先？还是模型优先？）
-      // 策略：模型优先，因为换了模型能力应该重置
-      this.chat.config.capabilities = { ...caps };
-    }
-    // 未注册的模型：不清除已有 capabilities，保留用户手动设的值
-  }
-};
+/**
+ * model-registry 插件
+ * 
+ * 职责：
+ * 1. 内置常用模型的能力标签表（model → capabilities 映射）
+ * 2. 安装时自动根据 config.model 查表，写入 config.capabilities
+ * 3. 监听 config-updated 事件，切换模型时自动同步能力
+ * 4. 提供 chat.registerModel()，允许用户追加/覆盖自定义模型
+ * 
+ * 能力标签字段（全部可选，缺失视为 false / 未知）：
+ * {
+ *   reasoning:    boolean — 是否返回 reasoning_content（思维链）
+ *   vision:       boolean — 是否支持图片输入
+ *   toolCalling:  boolean — 是否支持 function calling
+ *   streaming:    boolean — 是否支持流式传输
+ *   maxInputTokens:  number — 最大输入 token 数
+ *   maxOutputTokens: number — 最大输出 token 数
+ * }
+ * 
+ * 使用：
+ *   chat.use(modelRegistryPlugin);
+ *   // config.capabilities 现在已自动填充
+ *   chat.registerModel('my-custom-model', { toolCalling: true });
+ */
+
+// ====== 内置模型能力表 ======
+// 数据来源：各厂商官方文档（2026-05 快照），用户可通过 registerModel() 覆盖
+const BUILTIN_MODELS = {
+  // ---- DeepSeek ----
+  'deepseek-chat': {
+    reasoning: false,
+    vision: false,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 128000,
+    maxOutputTokens: 8192
+  },
+  'deepseek-reasoner': {
+    reasoning: true,
+    vision: false,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 128000,
+    maxOutputTokens: 8192
+  },
+
+  // ---- OpenAI ----
+  'gpt-4o': {
+    reasoning: false,
+    vision: true,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 128000,
+    maxOutputTokens: 16384
+  },
+  'gpt-4o-mini': {
+    reasoning: false,
+    vision: true,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 128000,
+    maxOutputTokens: 16384
+  },
+  'gpt-3.5-turbo': {
+    reasoning: false,
+    vision: false,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 16385,
+    maxOutputTokens: 4096
+  },
+  'o1': {
+    reasoning: true,
+    vision: false,
+    toolCalling: false,
+    streaming: false,
+    maxInputTokens: 200000,
+    maxOutputTokens: 100000
+  },
+  'o3-mini': {
+    reasoning: true,
+    vision: false,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 200000,
+    maxOutputTokens: 100000
+  },
+
+  // ---- Anthropic ----
+  'claude-3.5-sonnet': {
+    reasoning: false,
+    vision: true,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 200000,
+    maxOutputTokens: 8192
+  },
+  'claude-3.5-haiku': {
+    reasoning: false,
+    vision: true,
+    toolCalling: true,
+    streaming: true,
+    maxInputTokens: 200000,
+    maxOutputTokens: 8192
+  }
+};
+
+const DEFAULT_CAPABILITIES = {
+  streaming: true,
+  toolCalling: false,
+  reasoning: false,
+  vision: false
+};
+
+/**
+ * 注意：请使用 createModelRegistryPlugin() 工厂创建实例。
+ * 默认导出的 modelRegistryPlugin 是兼容用的模块级单例，装到多个 ChatService
+ * 实例会互相覆盖（注册表、实例引用），新代码不要直接用单例。
+ */
+export const modelRegistryPlugin = createModelRegistryPlugin();
+
+/**
+ * 创建模型能力注册表插件实例（工厂）。
+ * @param {Object} [options] — { models: { '模型名': capabilities } }（合并进内置表）
+ */
+export function createModelRegistryPlugin(options = {}) {
+  return {
+    name: 'model-registry',
+
+    _options: { ...options },
+    _registry: new Map([
+      ...Object.entries(BUILTIN_MODELS),
+      ...Object.entries(options.models || {})
+    ]),
+    chat: null,
+
+    /**
+     * 安装插件
+     * @param {ChatService} chatService - ChatService 实例
+     * @param {Object} [options] - 安装时配置（合并进 _options）
+     */
+    install(chatService, options = {}) {
+      this._options = { ...this._options, ...options };
+      this.chat = chatService;
+
+      /**
+       * 注册/覆盖一个模型的能力标签
+       * @param {string} name — 模型名称
+       * @param {Object} capabilities — 能力标签对象（部分字段即可，未提供的取默认值）
+       * @returns {ChatService}
+       */
+      chatService.registerModel = (name, capabilities = {}) => {
+        if (!name || typeof name !== 'string' || !name.trim()) {
+          throw new Error('[model-registry] 模型名称必须是非空字符串');
+        }
+        const merged = { ...DEFAULT_CAPABILITIES, ...capabilities };
+        this._registry.set(name.trim(), merged);
+        // 如果注册的模型与当前使用的模型同名，立即同步
+        if (chatService.config.model === name.trim()) {
+          this._syncCapabilities();
+        }
+        return chatService;
+      };
+
+      /**
+       * 列出所有已注册的模型名称
+       * @returns {Array<string>}
+       */
+      chatService.listModels = () => {
+        return [...this._registry.keys()];
+      };
+
+      // 安装时立即同步一次
+      this._syncCapabilities();
+
+      // 每次请求时按"本次生效的 model"（含请求级覆盖）解析 capabilities
+      // v3.0：beforeSend 车间，不再用 _hooks 钩子（时机与之前一致）
+      chatService.pipe({
+        name: 'model-registry-caps',
+        phase: 'beforeSend',
+        run: ({ config }) => {
+          if (!config.model) return;
+          const caps = this._registry.get(config.model);
+          if (caps) config.capabilities = { ...caps };
+        }
+      });
+
+      // 监听配置变更：model 变了就重新同步（兼容路径；请求时按生效 model 实时查表）
+      chatService.on('config-updated', ({ changes }) => {
+        if ('model' in changes) {
+          this._syncCapabilities();
+        }
+        // 如果用户手动改了 capabilities，用 merge 保留未指定的字段
+        if ('capabilities' in changes && !('model' in changes)) {
+          const current = this._lookupCapabilities();
+          if (current) {
+            chatService.config.capabilities = { ...current, ...changes.capabilities };
+          }
+        }
+      });
+    },
+
+    // ---- 内部方法 ----
+
+    /** 查表获取模型能力，未注册返回 null */
+    _lookupCapabilities() {
+      const model = this.chat.config.model;
+      if (!model) return null;
+      return this._registry.get(model) || null;
+    },
+
+    /** 将查到的能力写入 config.capabilities */
+    _syncCapabilities() {
+      const caps = this._lookupCapabilities();
+      if (caps) {
+        // 若用户已手动设了 capabilities，用模型的合并（用户值优先？还是模型优先？）
+        // 策略：模型优先，因为换了模型能力应该重置
+        this.chat.config.capabilities = { ...caps };
+      }
+      // 未注册的模型：不清除已有 capabilities，保留用户手动设的值
+    }
+  };
+}