@huyooo/ai-chat-core 0.2.45 → 0.3.3

package/src/providers/model-registry.ts

@@ -1,466 +0,0 @@
-/**
- * 模型注册表
- * 
- * 核心设计：Protocol + Family 分离
- * - Protocol：负责 API 通信协议（HTTP/SSE/认证）
- * - Family：负责模型行为差异（thinking格式、搜索方式、工具格式）
- * 
- * 这样设计的好处：
- * 1. 同协议不同行为的模型可以复用协议层（如 ARK 上的豆包和 DeepSeek）
- * 2. 新增模型只需配置 family，无需修改 adapter 代码
- * 3. 行为差异集中管理，易于维护
- */
-
-// ==================== 模型家族定义 ====================
-
-/** 模型家族 ID */
-export type ModelFamilyId =
-  | 'doubao'      // 豆包系列
-  | 'deepseek'    // DeepSeek 系列
-  | 'qwen'        // 通义千问系列
-  | 'gemini'      // Gemini 系列
-  | 'gpt'         // GPT 系列（OpenAI）
-  | 'claude';     // Claude 系列（Anthropic）
-
-/** 协议类型 */
-export type ProtocolId =
-  | 'ark'         // 火山引擎 Responses API（豆包）
-  | 'deepseek'    // DeepSeek（通过火山引擎 ARK）
-  | 'qwen'        // 通义千问 DashScope API
-  | 'gemini'      // Google Gemini API
-  | 'openai'      // OpenAI API（通过 OpenRouter）
-  | 'anthropic';  // Anthropic API（通过 OpenRouter）
-
-/** Thinking 输出格式 */
-export type ThinkingFormat =
-  | 'reasoning'          // ARK/豆包/DeepSeek: reasoning_summary_text
-  | 'thinking_enabled'   // Qwen: thinking 参数
-  | 'thought_signature'  // Gemini: 需要 thought_signature 循环
-  | 'none';              // 不支持 thinking
-
-/**
- * 搜索实现方式（谁提供搜索、行为是否由我们统一）
- * - provider_native：厂商原生，协议层各自实现（ARK type:web_search / Gemini googleSearch 等），行为各异
- * - tavily：我们注入 Tavily web_search 工具，行为一致（GPT/Claude/Qwen/Gemini 有工具时）
- */
-export type SearchStrategy =
-  | 'provider_native'
-  | 'tavily';
-
-/** 工具调用格式 */
-export type ToolCallFormat =
-  | 'responses'    // ARK Responses API: function_call/function_call_output
-  | 'openai'       // OpenAI 兼容: tool_calls/tool
-  | 'gemini';      // Gemini: functionCall/functionResponse
-
-/**
- * 模型家族配置
- * 定义同一家族模型的共同行为特征
- */
-export interface ModelFamilyConfig {
-  /** 家族 ID */
-  id: ModelFamilyId;
-  /** 显示名称 */
-  displayName: string;
-
-  // === 多模态配置 ===
-  /** 是否支持图片理解 */
-  supportsVision: boolean;
-
-  // === Thinking 配置 ===
-  /** 是否支持 thinking */
-  supportsThinking: boolean;
-  /** Thinking 输出格式 */
-  thinkingFormat: ThinkingFormat;
-
-  // === 搜索配置 ===
-  /** 是否支持原生搜索 */
-  supportsNativeSearch: boolean;
-  /** 搜索实现方式 */
-  searchStrategy: SearchStrategy;
-
-  // === 工具调用配置 ===
-  /** 工具调用格式 */
-  toolCallFormat: ToolCallFormat;
-
-  // === 默认参数 ===
-  /** 默认最大输出 token */
-  defaultMaxTokens?: number;
-  /** 是否需要特殊处理（如 Gemini 的 thought_signature） */
-  requiresSpecialHandling?: string[];
-}
-
-// ==================== 家族配置定义 ====================
-
-/** 豆包家族（联网搜索统一走 web_search_ai/Tavily，与其它模型事件与数据格式一致） */
-export const DOUBAO_FAMILY: ModelFamilyConfig = {
-  id: 'doubao',
-  displayName: '豆包',
-  supportsVision: true,
-  supportsThinking: true,
-  thinkingFormat: 'reasoning',
-  supportsNativeSearch: false,
-  searchStrategy: 'tavily',
-  toolCallFormat: 'responses',
-  // Seed 1.6 最大输出 16K（默认 4K，需手动设置）
-  defaultMaxTokens: 16384,
-};
-
-/** DeepSeek 家族（联网搜索统一走 web_search_ai/Tavily，与其它模型事件与数据格式一致） */
-export const DEEPSEEK_FAMILY: ModelFamilyConfig = {
-  id: 'deepseek',
-  displayName: 'DeepSeek',
-  supportsVision: false,
-  supportsThinking: true,
-  thinkingFormat: 'reasoning',
-  supportsNativeSearch: false,
-  searchStrategy: 'tavily',
-  toolCallFormat: 'responses',
-  defaultMaxTokens: 32768,
-};
-
-/** 通义千问家族 */
-/** Qwen 家族（使用 Tavily 统一搜索） */
-export const QWEN_FAMILY: ModelFamilyConfig = {
-  id: 'qwen',
-  displayName: '通义千问',
-  supportsVision: false,
-  supportsThinking: true,
-  thinkingFormat: 'thinking_enabled',
-  supportsNativeSearch: false, // 关闭原生搜索，使用 Tavily
-  searchStrategy: 'tavily',
-  toolCallFormat: 'openai',
-  defaultMaxTokens: 32768,
-};
-
-/** Gemini 家族（注意：googleSearch 不能与其他工具同时使用，有工具时走 Tavily） */
-export const GEMINI_FAMILY: ModelFamilyConfig = {
-  id: 'gemini',
-  displayName: 'Gemini',
-  supportsVision: true,
-  supportsThinking: true,
-  thinkingFormat: 'thought_signature',
-  supportsNativeSearch: false, // googleSearch 与其他工具冲突，统一使用 Tavily
-  searchStrategy: 'tavily',
-  toolCallFormat: 'gemini',
-  defaultMaxTokens: 65536,
-  requiresSpecialHandling: ['thought_signature'],
-};
-
-/** GPT 家族 */
-export const GPT_FAMILY: ModelFamilyConfig = {
-  id: 'gpt',
-  displayName: 'GPT',
-  supportsVision: true,
-  supportsThinking: true, // GPT-5.x 支持 reasoning
-  thinkingFormat: 'reasoning',
-  supportsNativeSearch: false,
-  searchStrategy: 'tavily',
-  toolCallFormat: 'openai',
-  defaultMaxTokens: 128000,
-};
-
-/** Claude 家族（使用 Vercel AI SDK / @ai-sdk/anthropic） */
-export const CLAUDE_FAMILY: ModelFamilyConfig = {
-  id: 'claude',
-  displayName: 'Claude',
-  supportsVision: true,
-  supportsThinking: true,
-  thinkingFormat: 'reasoning',
-  supportsNativeSearch: false,
-  searchStrategy: 'tavily',
-  toolCallFormat: 'openai',
-  // Bedrock fallback 限制 64K output，取 64000 保证各 provider 兼容
-  defaultMaxTokens: 64000,
-};
-
-/** 家族配置映射 */
-export const MODEL_FAMILIES: Record<ModelFamilyId, ModelFamilyConfig> = {
-  doubao: DOUBAO_FAMILY,
-  deepseek: DEEPSEEK_FAMILY,
-  qwen: QWEN_FAMILY,
-  gemini: GEMINI_FAMILY,
-  gpt: GPT_FAMILY,
-  claude: CLAUDE_FAMILY,
-};
-
-// ==================== 模型注册项 ====================
-
-/**
- * 模型注册项
- */
-export interface ModelRegistryEntry {
-  /** 模型 ID（API 调用用） */
-  id: string;
-  /** 显示名称 */
-  displayName: string;
-  /** 所属家族 */
-  family: ModelFamilyId;
-  /** 使用的协议 */
-  protocol: ProtocolId;
-  /** 是否在前端显示 */
-  visible?: boolean;
-  /** 是否支持图片理解（优先级高于 family.supportsVision） */
-  supportsVision?: boolean;
-  /** 上下文窗口大小（如 "256K"），用于展示 */
-  contextWindow?: string;
-  /** 上下文窗口精确 token 数（用于压缩器计算） */
-  contextWindowTokens?: number;
-  /** 价格信息（数组，分行显示） */
-  pricing?: string[];
-}
-
-// ==================== 模型注册表 ====================
-
-/**
- * 全局模型注册表
- * 
- * 每个模型只有一家供应商，无需分组
- */
-export const MODEL_REGISTRY: ModelRegistryEntry[] = [
-  // 豆包（价格为输入<=32k档，输出价格取决于输出长度）
-  { id: 'doubao-seed-1-6-250615', displayName: '豆包 Seed 1.6', family: 'doubao', protocol: 'ark', visible: true, supportsVision: true, contextWindow: '256K', contextWindowTokens: 256_000, pricing: ['输入 0.8 元/百万tokens', '输出 2-8 元/百万tokens'] },
-  { id: 'doubao-seed-1-8-251215', displayName: '豆包 Seed 1.8', family: 'doubao', protocol: 'ark', contextWindow: '256K', contextWindowTokens: 256_000, pricing: ['输入 0.8 元/百万tokens', '输出 2-8 元/百万tokens'] },
-
-  // DeepSeek（价格为输入<=32k档）
-  { id: 'deepseek-v3-2-251201', displayName: 'DeepSeek V3.2', family: 'deepseek', protocol: 'deepseek', visible: true, supportsVision: false, contextWindow: '128K', contextWindowTokens: 128_000, pricing: ['输入 2 元/百万tokens', '输出 3 元/百万tokens'] },
-
-  // 通义千问
-  { id: 'qwen3-vl-plus', displayName: '通义千问 3 VL', family: 'qwen', protocol: 'qwen', visible: true, supportsVision: true, contextWindow: '262K', contextWindowTokens: 262_144, pricing: ['输入 1 元/百万tokens', '输出 10 元/百万tokens'] },
-  { id: 'qwen3.5-plus', displayName: '通义千问 3.5 Plus', family: 'qwen', protocol: 'qwen', contextWindow: '1M', contextWindowTokens: 1_000_000, pricing: ['输入 0.8 元/百万tokens', '输出 4.8 元/百万tokens'] },
-  { id: 'qwen-long', displayName: '通义千问 Long', family: 'qwen', protocol: 'qwen', contextWindow: '10M', contextWindowTokens: 10_000_000, pricing: ['输入 0.5 元/百万tokens', '输出 2 元/百万tokens'] },
-
-  // Gemini
-  { id: 'gemini-3-pro-preview', displayName: 'Gemini 3 Pro', family: 'gemini', protocol: 'gemini', visible: true, supportsVision: true, contextWindow: '1M', contextWindowTokens: 1_000_000, pricing: ['输入 1.25 元/百万tokens', '输出 10 元/百万tokens'] },
-  { id: 'gemini-2.5-flash-preview-05-20', displayName: 'Gemini 2.5 Flash', family: 'gemini', protocol: 'gemini', contextWindow: '1M', contextWindowTokens: 1_000_000, pricing: ['输入 0.15 元/百万tokens', '输出 0.6 元/百万tokens'] },
-  { id: 'gemini-2.5-pro-preview-05-06', displayName: 'Gemini 2.5 Pro', family: 'gemini', protocol: 'gemini', contextWindow: '1M', contextWindowTokens: 1_000_000, pricing: ['输入 1.25 元/百万tokens', '输出 10 元/百万tokens'] },
-
-  // GPT（OpenRouter，美元价格按约7.2汇率换算）
-  { id: 'openai/gpt-5.2', displayName: 'GPT-5.2', family: 'gpt', protocol: 'openai', visible: true, supportsVision: true, contextWindow: '400K', contextWindowTokens: 400_000, pricing: ['输入 12.6 元/百万tokens', '输出 100.8 元/百万tokens'] },
-
-  // Claude（Vercel AI SDK，美元价格按约7.2汇率换算）
-  { id: 'anthropic/claude-opus-4.5', displayName: 'Claude Opus 4.5', family: 'claude', protocol: 'anthropic', visible: true, supportsVision: true, contextWindow: '200K', contextWindowTokens: 200_000, pricing: ['输入 36 元/百万tokens', '输出 180 元/百万tokens'] },
-];
-
-// ==================== 查询辅助函数 ====================
-
-/**
- * 根据模型 ID 获取注册信息
- */
-export function getModelEntry(modelId: string): ModelRegistryEntry | undefined {
-  return MODEL_REGISTRY.find(m => m.id === modelId || modelId.includes(m.id));
-}
-
-/**
- * 根据模型 ID 获取家族配置
- */
-export function getModelFamily(modelId: string): ModelFamilyConfig | undefined {
-  const entry = getModelEntry(modelId);
-  if (!entry) return undefined;
-  return MODEL_FAMILIES[entry.family];
-}
-
-/**
- * 根据模型 ID 获取协议类型
- */
-export function getModelProtocol(modelId: string): ProtocolId | undefined {
-  const entry = getModelEntry(modelId);
-  return entry?.protocol;
-}
-
-/**
- * 获取所有可见模型（用于前端显示）
- */
-export function getVisibleModels(): ModelRegistryEntry[] {
-  return MODEL_REGISTRY.filter(m => m.visible);
-}
-
-/**
- * 根据家族 ID 获取所有模型
- */
-export function getModelsByFamily(familyId: ModelFamilyId): ModelRegistryEntry[] {
-  return MODEL_REGISTRY.filter(m => m.family === familyId);
-}
-
-/**
- * 根据协议获取所有模型
- */
-export function getModelsByProtocol(protocol: ProtocolId): ModelRegistryEntry[] {
-  return MODEL_REGISTRY.filter(m => m.protocol === protocol);
-}
-
-/**
- * 检查模型是否支持 thinking
- */
-export function modelSupportsThinking(modelId: string): boolean {
-  const family = getModelFamily(modelId);
-  return family?.supportsThinking ?? false;
-}
-
-/**
- * 检查模型是否支持原生搜索
- */
-export function modelSupportsNativeSearch(modelId: string): boolean {
-  const family = getModelFamily(modelId);
-  return family?.supportsNativeSearch ?? false;
-}
-
-/**
- * 检查模型是否支持图片理解
- */
-export function modelSupportsVision(modelId: string): boolean {
-  const entry = getModelEntry(modelId);
-  if (!entry) return false;
-  if (typeof entry.supportsVision === 'boolean') return entry.supportsVision;
-  const family = getModelFamily(modelId);
-  return family?.supportsVision ?? false;
-}
-
-/**
- * 获取模型的搜索策略
- */
-export function getModelSearchStrategy(modelId: string): SearchStrategy {
-  const family = getModelFamily(modelId);
-  return family?.searchStrategy ?? 'tavily';
-}
-
-// ==================== Context 配置 ====================
-
-export interface ModelContextConfig {
-  contextWindowTokens: number;
-  maxOutputTokens: number;
-}
-
-/**
- * 获取模型的 context 配置（供压缩器使用）
- *
- * 所有模型必须注册且声明 contextWindowTokens，否则抛错。
- */
-export function getModelContextConfig(modelId: string): ModelContextConfig {
-  const entry = getModelEntry(modelId);
-  if (!entry) throw new Error(`模型 ${modelId} 未在 MODEL_REGISTRY 中注册`);
-
-  const family = MODEL_FAMILIES[entry.family];
-  if (!family) throw new Error(`模型 ${modelId} 的家族 ${entry.family} 未定义`);
-
-  if (!entry.contextWindowTokens) {
-    throw new Error(`模型 ${modelId} 缺少 contextWindowTokens 配置`);
-  }
-  if (!family.defaultMaxTokens) {
-    throw new Error(`模型家族 ${entry.family} 缺少 defaultMaxTokens 配置`);
-  }
-
-  return {
-    contextWindowTokens: entry.contextWindowTokens,
-    maxOutputTokens: family.defaultMaxTokens,
-  };
-}
-
-/** 解析 "256K" / "1M" 格式为 token 数 */
-function parseContextWindowString(s: string): number | undefined {
-  const match = s.match(/^([\d.]+)\s*(K|M)$/i);
-  if (!match) return undefined;
-  const num = parseFloat(match[1]);
-  const unit = match[2].toUpperCase();
-  return unit === 'M' ? num * 1_000_000 : num * 1_000;
-}
-
-// ==================== 配置校验 ====================
-
-export interface ConfigValidationError {
-  modelId: string;
-  field: string;
-  message: string;
-  severity: 'error' | 'warning';
-}
-
-/**
- * 校验所有模型配置的合法性
- *
- * 检查项：
- * 1. defaultMaxTokens 不能超过 contextWindow
- * 2. contextWindowTokens 与 contextWindow 字符串必须一致
- * 3. 必须声明 contextWindowTokens（否则压缩器无法工作）
- * 4. defaultMaxTokens 必须大于 0
- * 5. 各 provider 的已知硬限制（如 Bedrock 64K output）
- */
-export function validateModelConfigs(): ConfigValidationError[] {
-  const errors: ConfigValidationError[] = [];
-
-  // 各 provider 已知 maxOutputTokens 硬限制
-  const providerOutputLimits: Record<string, number> = {
-    anthropic: 64_000,  // Bedrock fallback 限制
-    gemini: 65_536,
-    ark: 16_384,        // 豆包 Seed 系列最大输出 16K
-    deepseek: 64_000,   // V3.2 reasoner 最大 64K
-    qwen: 32_768,
-  };
-
-  for (const entry of MODEL_REGISTRY) {
-    const family = MODEL_FAMILIES[entry.family];
-    if (!family) {
-      errors.push({ modelId: entry.id, field: 'family', message: `未找到家族配置: ${entry.family}`, severity: 'error' });
-      continue;
-    }
-
-    // 检查 contextWindowTokens 声明
-    if (!entry.contextWindowTokens) {
-      errors.push({
-        modelId: entry.id,
-        field: 'contextWindowTokens',
-        message: '缺少 contextWindowTokens，压缩器将回退到字符数阈值',
-        severity: 'warning',
-      });
-    }
-
-    // 检查 contextWindow 字符串与数值一致性
-    if (entry.contextWindow && entry.contextWindowTokens) {
-      const parsed = parseContextWindowString(entry.contextWindow);
-      if (parsed && parsed !== entry.contextWindowTokens) {
-        errors.push({
-          modelId: entry.id,
-          field: 'contextWindow',
-          message: `contextWindow "${entry.contextWindow}" (${parsed}) 与 contextWindowTokens (${entry.contextWindowTokens}) 不一致`,
-          severity: 'error',
-        });
-      }
-    }
-
-    // 检查 defaultMaxTokens
-    const maxTokens = family.defaultMaxTokens;
-    if (maxTokens !== undefined) {
-      if (maxTokens <= 0) {
-        errors.push({
-          modelId: entry.id,
-          field: 'defaultMaxTokens',
-          message: `defaultMaxTokens (${maxTokens}) 必须大于 0`,
-          severity: 'error',
-        });
-      }
-
-      // 不能超过 contextWindow
-      if (entry.contextWindowTokens && maxTokens > entry.contextWindowTokens) {
-        errors.push({
-          modelId: entry.id,
-          field: 'defaultMaxTokens',
-          message: `defaultMaxTokens (${maxTokens}) 超过 contextWindow (${entry.contextWindowTokens})`,
-          severity: 'error',
-        });
-      }
-
-      // 检查 provider 硬限制
-      const providerLimit = providerOutputLimits[entry.protocol];
-      if (providerLimit && maxTokens > providerLimit) {
-        errors.push({
-          modelId: entry.id,
-          field: 'defaultMaxTokens',
-          message: `defaultMaxTokens (${maxTokens}) 超过 ${entry.protocol} 的硬限制 (${providerLimit})`,
-          severity: 'error',
-        });
-      }
-    }
-  }
-
-  return errors;
-}
-