kc-beta 0.1.2 → 0.2.1

package/src/providers.js

@@ -0,0 +1,370 @@
+/**
+ * Provider registry for Multi-LLM support.
+ * Centralizes provider metadata, default models, and model classification.
+ *
+ * Aligned with kc_reborn/platform/backend/app/providers.py
+ */
+
+const PROVIDERS = [
+  {
+    id: "siliconflow",
+    name: "SiliconFlow",
+    baseUrl: "https://api.siliconflow.cn/v1",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: "/models",
+    defaultModel: "glm-5",
+    defaultTiers: {
+      tier1: "Pro/zai-org/GLM-5, Pro/moonshotai/Kimi-K2.5",
+      tier2: "Pro/deepseek-ai/DeepSeek-V3.2, Pro/MiniMaxAI/MiniMax-M2.5",
+      tier3: "Qwen/Qwen3.5-122B-A10B",
+      tier4: "Qwen/Qwen3.5-35B-A3B",
+    },
+    labels: {
+      en: "SiliconFlow (recommended for China)",
+      zh: "SiliconFlow（国内推荐）",
+    },
+  },
+  {
+    id: "aliyun",
+    name: "Aliyun",
+    // Coding plan URL — regular API uses dashscope.aliyuncs.com/compatible-mode/v1
+    baseUrl: "https://dashscope.aliyuncs.com/compatible-mode/v1",
+    codingPlanUrl: "https://coding.dashscope.aliyuncs.com/v1",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: null, // Aliyun coding plan doesn't support /models
+    supportsCodingPlanKey: true,
+    defaultModel: "glm-5",
+    defaultTiers: {
+      tier1: "qwen3.6-plus",
+      tier2: "",
+      tier3: "",
+      tier4: "",
+    },
+    // Curated model list (coding plan doesn't have /models endpoint)
+    curatedModels: [
+      { id: "qwen3.6-plus", ownedBy: "qwen" },
+      { id: "qwen3.5-plus", ownedBy: "qwen" },
+      { id: "qwen3-max-2026-01-23", ownedBy: "qwen" },
+      { id: "qwen3-coder-next", ownedBy: "qwen" },
+      { id: "qwen3-coder-plus", ownedBy: "qwen" },
+      { id: "glm-5", ownedBy: "zhipu" },
+      { id: "glm-4.7", ownedBy: "zhipu" },
+      { id: "kimi-k2.5", ownedBy: "kimi" },
+      { id: "MiniMax-M2.5", ownedBy: "minimax" },
+    ],
+    labels: {
+      en: "Aliyun Bailian",
+      zh: "阿里云百炼",
+    },
+  },
+  {
+    id: "volcanocloud",
+    name: "VolcanoCloud",
+    baseUrl: "https://ark.cn-beijing.volces.com/api/v3",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: null, // VolcanoCloud coding plan — use curated list
+    defaultModel: "glm-5",
+    defaultTiers: {
+      tier1: "doubao-seed-2-0-pro-260215, deepseek-v3-2-251201",
+      tier2: "glm-4-7-251222, doubao-1-5-pro-32k-250115",
+      tier3: "doubao-seed-2-0-mini-260215",
+      tier4: "doubao-seed-2-0-lite-260215, doubao-1-5-lite-32k-250115",
+    },
+    curatedModels: [
+      { id: "doubao-seed-2-0-pro-260215", ownedBy: "bytedance" },
+      { id: "deepseek-v3-2-251201", ownedBy: "deepseek" },
+      { id: "glm-4-7-251222", ownedBy: "zhipu" },
+      { id: "doubao-1-5-pro-32k-250115", ownedBy: "bytedance" },
+      { id: "doubao-seed-2-0-mini-260215", ownedBy: "bytedance" },
+      { id: "doubao-seed-2-0-lite-260215", ownedBy: "bytedance" },
+      { id: "doubao-1-5-lite-32k-250115", ownedBy: "bytedance" },
+    ],
+    labels: {
+      en: "VolcanoCloud (ByteDance)",
+      zh: "火山云（字节跳动）",
+    },
+  },
+  {
+    id: "anthropic",
+    name: "Anthropic",
+    baseUrl: "https://api.anthropic.com",
+    authType: "x-api-key",
+    apiFormat: "anthropic",
+    modelsEndpoint: null, // Use curated list
+    defaultModel: "claude-sonnet-4-20250514",
+    defaultTiers: {
+      tier1: "claude-sonnet-4-20250514",
+      tier2: "claude-sonnet-4-20250514",
+      tier3: "claude-haiku-4-5-20251001",
+      tier4: "claude-haiku-4-5-20251001",
+    },
+    curatedModels: [
+      { id: "claude-opus-4-20250514", ownedBy: "anthropic" },
+      { id: "claude-sonnet-4-20250514", ownedBy: "anthropic" },
+      { id: "claude-haiku-4-5-20251001", ownedBy: "anthropic" },
+    ],
+    labels: {
+      en: "Anthropic",
+      zh: "Anthropic",
+    },
+  },
+  {
+    id: "openai",
+    name: "OpenAI",
+    baseUrl: "https://api.openai.com/v1",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: "/models",
+    defaultModel: "gpt-4o",
+    defaultTiers: {
+      tier1: "gpt-4o",
+      tier2: "gpt-4o-mini",
+      tier3: "gpt-4o-mini",
+      tier4: "gpt-4o-mini",
+    },
+    labels: {
+      en: "OpenAI",
+      zh: "OpenAI",
+    },
+  },
+  {
+    id: "zhipu",
+    name: "Zhipu/GLM",
+    baseUrl: "https://open.bigmodel.cn/api/paas/v4",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: "/models",
+    defaultModel: "glm-4-plus",
+    defaultTiers: {
+      tier1: "glm-4-plus",
+      tier2: "glm-4-air",
+      tier3: "glm-4-flash",
+      tier4: "glm-4-flash",
+    },
+    labels: {
+      en: "Zhipu GLM",
+      zh: "智谱 GLM",
+    },
+  },
+  {
+    id: "minimax",
+    name: "MiniMax",
+    baseUrl: "https://api.minimax.chat/v1",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: "/models",
+    defaultModel: "MiniMax-M2.5",
+    defaultTiers: {
+      tier1: "MiniMax-M2.5",
+      tier2: "MiniMax-M2.5",
+      tier3: "MiniMax-M1",
+      tier4: "MiniMax-M1",
+    },
+    labels: {
+      en: "MiniMax",
+      zh: "MiniMax",
+    },
+  },
+  {
+    id: "openrouter",
+    name: "OpenRouter",
+    baseUrl: "https://openrouter.ai/api/v1",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: "/models",
+    defaultModel: "anthropic/claude-sonnet-4-20250514",
+    defaultTiers: {
+      tier1: "anthropic/claude-sonnet-4-20250514",
+      tier2: "google/gemini-2.5-flash",
+      tier3: "google/gemini-2.5-flash",
+      tier4: "google/gemini-2.5-flash",
+    },
+    labels: {
+      en: "OpenRouter",
+      zh: "OpenRouter",
+    },
+  },
+  {
+    id: "bedrock",
+    name: "Bedrock",
+    baseUrl: "",
+    authType: "aws-sigv4",
+    apiFormat: "anthropic",
+    modelsEndpoint: null,
+    defaultModel: "anthropic.claude-sonnet-4-20250514-v1:0",
+    defaultTiers: {
+      tier1: "anthropic.claude-sonnet-4-20250514-v1:0",
+      tier2: "anthropic.claude-sonnet-4-20250514-v1:0",
+      tier3: "anthropic.claude-haiku-4-5-20251001-v1:0",
+      tier4: "anthropic.claude-haiku-4-5-20251001-v1:0",
+    },
+    labels: {
+      en: "AWS Bedrock (not yet supported)",
+      zh: "AWS Bedrock（暂未支持）",
+    },
+  },
+  {
+    id: "custom",
+    name: "Custom",
+    baseUrl: "",
+    authType: "bearer",
+    apiFormat: "openai",
+    modelsEndpoint: "/models",
+    defaultModel: "",
+    defaultTiers: { tier1: "", tier2: "", tier3: "", tier4: "" },
+    labels: {
+      en: "Custom (enter base URL)",
+      zh: "自定义（输入接口地址）",
+    },
+  },
+];
+
+/**
+ * Known model capability rankings (partial — used to sort discovered models).
+ * Pattern-matched against lowercase model ID. Higher = more capable.
+ * Aligned with kc_reborn providers.py _MODEL_RANKING.
+ */
+const MODEL_RANKING = {
+  // Anthropic
+  "claude-opus-4": 100,
+  "claude-sonnet-4": 90,
+  "claude-haiku-4": 70,
+  // OpenAI
+  "gpt-4o": 90,
+  "gpt-4o-mini": 70,
+  "gpt-4-turbo": 85,
+  "o1": 95,
+  "o3": 95,
+  // Qwen (Aliyun Bailian)
+  "qwen3.6-plus": 90,
+  "qwen3.5-plus": 85,
+  "qwen3-max": 88,
+  "qwen3-coder-next": 85,
+  "qwen3-coder-plus": 80,
+  "qwen-plus": 75,
+  "qwen-turbo": 60,
+  "qwen3.5-397b": 85,
+  "qwen3.5-122b": 75,
+  "qwen3.5-35b": 65,
+  // Zhipu
+  "glm-5": 90,
+  "glm-4.7": 80,
+  "glm-4": 75,
+  // Others
+  "kimi-k2.5": 85,
+  "kimi-k2": 80,
+  "minimax-m2": 80,
+  "deepseek-v3": 85,
+  "deepseek-r1": 90,
+  // VolcanoCloud (ByteDance Doubao)
+  "doubao-seed-2-0-pro": 90,
+  "doubao-seed-2-0-code": 88,
+  "doubao-seed-2-0-mini": 75,
+  "doubao-seed-2-0-lite": 65,
+  "doubao-seed-1-8": 85,
+  "doubao-seed-1-6": 80,
+  "doubao-1-5-pro": 80,
+  "doubao-1-5-lite": 60,
+};
+
+/**
+ * Estimate model capability rank (0-100) based on known patterns.
+ * @param {string} modelId
+ * @returns {number}
+ */
+function rankModel(modelId) {
+  const lower = modelId.toLowerCase();
+  for (const [pattern, rank] of Object.entries(MODEL_RANKING)) {
+    if (lower.includes(pattern)) return rank;
+  }
+  return 50; // Unknown model: assume mid-tier
+}
+
+/**
+ * Patterns to filter out non-chat models from discovery results.
+ */
+const EXCLUDE_PATTERNS = [
+  /embed/i, /tts/i, /whisper/i, /dall-e/i, /audio/i, /image/i,
+  /moderation/i, /rerank/i,
+];
+
+/** @returns {Array} All provider definitions */
+export function getProviders() {
+  return PROVIDERS;
+}
+
+/**
+ * @param {string} id - Provider ID
+ * @returns {object|undefined}
+ */
+export function getProviderById(id) {
+  return PROVIDERS.find((p) => p.id === id);
+}
+
+/**
+ * Get display labels for the onboard menu.
+ * @param {string} lang - "en" or "zh"
+ * @returns {Array<{id: string, label: string}>}
+ */
+export function getProviderLabels(lang) {
+  return PROVIDERS.map((p) => ({
+    id: p.id,
+    label: p.labels[lang] || p.labels.en || p.name,
+  }));
+}
+
+/**
+ * Classify a list of discovered models into tier assignments.
+ * Uses the same ranking-based approach as kc_reborn's propose_tiers().
+ *
+ * @param {Array<{id: string, name?: string}>} models - Models from /models endpoint or curated list
+ * @returns {{ conductor: string, tiers: {tier1: string, tier2: string, tier3: string, tier4: string}, unclassified: string[] }}
+ */
+export function classifyModels(models) {
+  // Filter out non-chat models
+  const chatModels = models.filter((m) => {
+    const name = m.id || m.name || "";
+    return !EXCLUDE_PATTERNS.some((re) => re.test(name));
+  });
+
+  // Rank and sort by capability
+  const ranked = [...chatModels].sort((a, b) => rankModel(b.id) - rankModel(a.id));
+
+  // Select conductor (highest ranked)
+  const conductor = ranked[0]?.id || "";
+
+  // Distribute across tiers by rank
+  const tierBuckets = { tier1: [], tier2: [], tier3: [], tier4: [] };
+
+  for (const m of ranked) {
+    const rank = rankModel(m.id);
+    if (rank >= 85) tierBuckets.tier1.push(m.id);
+    else if (rank >= 70) tierBuckets.tier2.push(m.id);
+    else if (rank >= 55) tierBuckets.tier3.push(m.id);
+    else tierBuckets.tier4.push(m.id);
+  }
+
+  const tiers = {
+    tier1: tierBuckets.tier1.slice(0, 3).join(", "),
+    tier2: tierBuckets.tier2.slice(0, 3).join(", "),
+    tier3: tierBuckets.tier3.slice(0, 2).join(", "),
+    tier4: tierBuckets.tier4.slice(0, 2).join(", "),
+  };
+
+  const unclassified = ranked.filter((m) => rankModel(m.id) === 50).map((m) => m.id);
+
+  return { conductor, tiers, unclassified };
+}
+
+/**
+ * Get curated models for providers that don't support /models endpoint.
+ * @param {string} providerId
+ * @returns {Array<{id: string, ownedBy: string}>|null}
+ */
+export function getCuratedModels(providerId) {
+  const provider = getProviderById(providerId);
+  return provider?.curatedModels || null;
+}