claude-code-model-router 1.2.2 → 1.3.0

package/dist/config.js

@@ -16,100 +16,177 @@ const dotenv_1 = require("dotenv");
 // Load .env file
 (0, dotenv_1.config)();
 const DEFAULT_CONFIG = {
-    default_model: 'deepseek',
-    models: {
+    default_model: 'deepseek-v4-pro',
+    providers: {
         deepseek: {
-            display_name: 'DeepSeek V3.2',
+            display_name: 'DeepSeek',
             provider: 'deepseek',
-            model_id: 'deepseek-chat',
             base_url: 'https://api.deepseek.com/anthropic',
             api_key_env: 'DEEPSEEK_API_KEY',
             auth_header: 'x-api-key',
+            auth_type: 'api_key',
             supports_streaming: true,
             supports_tools: true,
-            max_tokens: 128000,
-            context_window: 128000,
+            default_variant: 'v4-pro',
+            variants: {
+                'v4-pro': {
+                    display_name: 'DeepSeek V4 Pro',
+                    model_id: 'deepseek-v4-pro',
+                    max_tokens: 393216,
+                    context_window: 1048576,
+                },
+                'v4-flash': {
+                    display_name: 'DeepSeek V4 Flash',
+                    model_id: 'deepseek-v4-flash',
+                    max_tokens: 393216,
+                    context_window: 1048576,
+                },
+            },
         },
         kimi: {
-            display_name: 'Kimi K2.5',
+            display_name: 'Kimi',
             provider: 'moonshot',
-            model_id: 'kimi-for-coding',
-            base_url: 'https://api.kimi.com/coding/',
+            base_url: 'https://api.moonshot.ai/anthropic',
             api_key_env: 'KIMI_API_KEY',
-            auth_header: 'x-api-key',
+            auth_header: 'Authorization',
+            auth_type: 'bearer',
             supports_streaming: true,
             supports_tools: true,
-            max_tokens: 32768,
-            context_window: 256000,
+            default_variant: 'k2.6',
+            variants: {
+                'k2.6': {
+                    display_name: 'Kimi K2.6',
+                    model_id: 'kimi-k2.6',
+                    max_tokens: 32768,
+                    context_window: 262144,
+                },
+            },
         },
         minimax: {
-            display_name: 'MiniMax M2.7',
+            display_name: 'MiniMax',
             provider: 'minimax',
-            model_id: 'MiniMax-M2.7',
-            base_url: 'https://api.minimaxi.com/anthropic',
+            base_url: 'https://api.minimax.io/anthropic',
             api_key_env: 'MINIMAX_API_KEY',
-            auth_header: 'x-api-key',
+            auth_header: 'Authorization',
+            auth_type: 'bearer',
             supports_streaming: true,
             supports_tools: true,
-            max_tokens: 196608,
-            context_window: 204800,
+            default_variant: 'm2.7',
+            variants: {
+                'm2.7': {
+                    display_name: 'MiniMax M2.7',
+                    model_id: 'MiniMax-M2.7',
+                    max_tokens: 196608,
+                    context_window: 204800,
+                },
+                'm2.7-highspeed': {
+                    display_name: 'MiniMax M2.7 Highspeed',
+                    model_id: 'MiniMax-M2.7-highspeed',
+                    max_tokens: 196608,
+                    context_window: 204800,
+                },
+            },
         },
         qwen: {
-            display_name: 'Qwen3.5 Plus',
+            display_name: 'Qwen',
             provider: 'alibaba',
-            model_id: 'qwen3.5-plus',
             base_url: 'https://dashscope.aliyuncs.com/apps/anthropic',
             api_key_env: 'QWEN_API_KEY',
             auth_header: 'x-api-key',
+            auth_type: 'api_key',
             supports_streaming: true,
             supports_tools: true,
-            max_tokens: 65536,
-            context_window: 1000000,
+            default_variant: '3.5-plus',
+            variants: {
+                '3.5-plus': {
+                    model_key: 'qwen3.5-plus',
+                    display_name: 'Qwen3.5 Plus',
+                    model_id: 'qwen3.5-plus',
+                    max_tokens: 65536,
+                    context_window: 1000000,
+                },
+                '3.5-flash': {
+                    model_key: 'qwen3.5-flash',
+                    display_name: 'Qwen3.5 Flash',
+                    model_id: 'qwen3.5-flash',
+                    max_tokens: 65536,
+                    context_window: 1000000,
+                },
+                '3-max': {
+                    model_key: 'qwen3-max',
+                    display_name: 'Qwen3 Max',
+                    model_id: 'qwen3-max',
+                    max_tokens: 65536,
+                    context_window: 1000000,
+                },
+            },
         },
         glm: {
-            display_name: 'GLM-5',
+            display_name: 'GLM',
             provider: 'zhipu',
-            model_id: 'glm-5',
             base_url: 'https://open.bigmodel.cn/api/anthropic',
             api_key_env: 'GLM_API_KEY',
             auth_header: 'x-api-key',
+            auth_type: 'api_key',
             supports_streaming: true,
             supports_tools: true,
-            max_tokens: 128000,
-            context_window: 200000,
+            default_variant: '5.1',
+            variants: {
+                '5.1': {
+                    display_name: 'GLM-5.1',
+                    model_id: 'glm-5.1',
+                    max_tokens: 131072,
+                    context_window: 204800,
+                },
+            },
         },
-        kat: {
-            display_name: 'KAT-Coder-Pro V2',
-            provider: 'kuaishou',
-            model_id: 'kat-coder-pro-v2',
-            base_url: 'https://wanqing.streamlakeapi.com/api/gateway/coding/kat-coder-pro-v2/claude-code-proxy',
-            api_key_env: 'KAT_API_KEY',
-            auth_header: 'x-api-key',
+        mimo: {
+            display_name: 'MiMo',
+            provider: 'xiaomi',
+            base_url: 'https://api.xiaomimimo.com/anthropic',
+            api_key_env: 'MIMO_API_KEY',
+            auth_header: 'api-key',
+            auth_type: 'api_key',
             supports_streaming: true,
             supports_tools: true,
-            max_tokens: 81920,
-            context_window: 262144,
+            default_variant: 'v2.5',
+            variants: {
+                'v2.5': {
+                    display_name: 'MiMo V2.5',
+                    model_id: 'mimo-v2.5',
+                    max_tokens: 131072,
+                    context_window: 1048576,
+                },
+            },
         },
     },
+    models: {},
     aliases: {
-        'deepseek-v3': 'deepseek',
-        'deepseek-chat': 'deepseek',
-        'ds': 'deepseek',
-        'kimi-k2.5': 'kimi',
-        'kimi-k2': 'kimi',
-        'moonshot': 'kimi',
-        'minimax-m2.7': 'minimax',
-        'minimax-m2': 'minimax',
-        'mm': 'minimax',
-        'qwen3.5': 'qwen',
-        'qwen3.5-plus': 'qwen',
-        'tongyi': 'qwen',
-        'glm-5': 'glm',
-        'zhipu': 'glm',
-        'chatglm': 'glm',
-        'kat-coder': 'kat',
-        'kat-coder-pro-v2': 'kat',
-        'kuaishou': 'kat',
+        deepseek: 'deepseek-v4-pro',
+        'deepseek-v4': 'deepseek-v4-pro',
+        'deepseek-pro': 'deepseek-v4-pro',
+        'deepseek-flash': 'deepseek-v4-flash',
+        'deepseek-chat': 'deepseek-v4-flash',
+        ds: 'deepseek-v4-pro',
+        kimi: 'kimi-k2.6',
+        'kimi-k2': 'kimi-k2.6',
+        'kimi-k2.6': 'kimi-k2.6',
+        moonshot: 'kimi-k2.6',
+        minimax: 'minimax-m2.7',
+        'minimax-m2': 'minimax-m2.7',
+        'minimax-highspeed': 'minimax-m2.7-highspeed',
+        mm: 'minimax-m2.7',
+        qwen: 'qwen3.5-plus',
+        'qwen3.5': 'qwen3.5-plus',
+        tongyi: 'qwen3.5-plus',
+        glm: 'glm-5.1',
+        'glm-5': 'glm-5.1',
+        zhipu: 'glm-5.1',
+        chatglm: 'glm-5.1',
+        mimo: 'mimo-v2.5',
+        'mimo-v2': 'mimo-v2.5',
+        'mimo-v2.5': 'mimo-v2.5',
+        xiaomi: 'mimo-v2.5',
     },
     gateway: {
         host: '0.0.0.0',
@@ -139,7 +216,7 @@ class ConfigManager {
                 try {
                     const content = node_fs_1.default.readFileSync(p, 'utf-8');
                     const parsed = js_yaml_1.default.load(content);
-                    return this.mergeConfig(parsed);
+                    return this.mergeConfig(parsed ?? {});
                 }
                 catch (e) {
                     console.warn(`Warning: Failed to parse config file ${p}:`, e);
@@ -147,17 +224,80 @@ class ConfigManager {
             }
         }
         // Return default config if no file found
-        return DEFAULT_CONFIG;
+        return this.normalizeConfig(DEFAULT_CONFIG);
     }
     mergeConfig(parsed) {
-        return {
+        return this.normalizeConfig({
             default_model: parsed.default_model ?? DEFAULT_CONFIG.default_model,
+            providers: this.mergeProviders(DEFAULT_CONFIG.providers ?? {}, parsed.providers ?? {}),
             models: { ...DEFAULT_CONFIG.models, ...parsed.models },
             aliases: { ...DEFAULT_CONFIG.aliases, ...parsed.aliases },
             gateway: { ...DEFAULT_CONFIG.gateway, ...parsed.gateway },
+        });
+    }
+    mergeProviders(defaults, overrides) {
+        const providers = { ...defaults };
+        for (const [providerKey, override] of Object.entries(overrides)) {
+            const base = providers[providerKey];
+            providers[providerKey] = base
+                ? {
+                    ...base,
+                    ...override,
+                    variants: { ...base.variants, ...override.variants },
+                }
+                : override;
+        }
+        return providers;
+    }
+    normalizeConfig(config) {
+        const models = {};
+        for (const [providerKey, provider] of Object.entries(config.providers ?? {})) {
+            for (const [variantKey, variant] of Object.entries(provider.variants)) {
+                const modelKey = variant.model_key ?? `${providerKey}-${variantKey}`;
+                models[modelKey] = this.buildModelConfig(providerKey, variantKey, provider, variant);
+            }
+            if (provider.default_variant) {
+                const defaultVariant = provider.variants[provider.default_variant];
+                const defaultKey = defaultVariant?.model_key ?? `${providerKey}-${provider.default_variant}`;
+                if (models[defaultKey]) {
+                    models[providerKey] = models[defaultKey];
+                }
+            }
+        }
+        for (const [modelKey, model] of Object.entries(config.models ?? {})) {
+            models[modelKey] = model;
+        }
+        return {
+            ...config,
+            default_model: this.resolveAlias(config.default_model, config.aliases ?? {}),
+            models,
+            aliases: config.aliases ?? {},
+            gateway: config.gateway,
+        };
+    }
+    buildModelConfig(providerKey, variantKey, provider, variant) {
+        return {
+            display_name: variant.display_name,
+            provider: provider.provider,
+            model_id: variant.model_id,
+            base_url: provider.base_url,
+            api_key_env: provider.api_key_env,
+            auth_header: provider.auth_header,
+            auth_type: provider.auth_type,
+            supports_streaming: variant.supports_streaming ?? provider.supports_streaming,
+            supports_tools: variant.supports_tools ?? provider.supports_tools,
+            max_tokens: variant.max_tokens,
+            context_window: variant.context_window,
+            provider_key: providerKey,
+            variant_key: variantKey,
+            provider_display_name: provider.display_name,
         };
     }
+    resolveAlias(name, aliases) {
+        return aliases[name] ?? name;
+    }
     loadApiKeys() {
+        this.apiKeys.clear();
         for (const [name, model] of Object.entries(this.config.models)) {
             const key = process.env[model.api_key_env];
             if (key) {
@@ -169,25 +309,11 @@ class ConfigManager {
         return this.config;
     }
     getModel(name) {
-        // Check direct model name
-        if (this.config.models[name]) {
-            return this.config.models[name];
-        }
-        // Check aliases
-        const aliasTarget = this.config.aliases[name];
-        if (aliasTarget && this.config.models[aliasTarget]) {
-            return this.config.models[aliasTarget];
-        }
-        return undefined;
+        const resolved = this.resolveModelName(name);
+        return this.config.models[resolved];
     }
     resolveModelName(name) {
-        if (this.config.aliases[name]) {
-            return this.config.aliases[name];
-        }
-        if (this.config.models[name]) {
-            return name;
-        }
-        return name;
+        return this.resolveAlias(name, this.config.aliases);
     }
     getApiKey(modelName) {
         const resolved = this.resolveModelName(modelName);
@@ -196,9 +322,13 @@ class ConfigManager {
     listModels() {
         const result = {};
         for (const [name, model] of Object.entries(this.config.models)) {
+            if (model.provider_key && name === model.provider_key) {
+                continue;
+            }
             result[name] = {
                 displayName: model.display_name,
                 provider: model.provider,
+                variant: model.variant_key,
                 available: this.apiKeys.has(name),
             };
         }
@@ -208,110 +338,177 @@ class ConfigManager {
 exports.ConfigManager = ConfigManager;
 // Generate default config file content
 function generateConfigFile() {
-    return `# Claude Code Model Router Configuration
-# Place this file as models.yaml or .claude-router.yaml in your project root
-
-default_model: deepseek
-
-models:
-  deepseek:
-    display_name: "DeepSeek V3.2"
-    provider: deepseek
-    model_id: deepseek-chat
-    base_url: https://api.deepseek.com/anthropic
-    api_key_env: DEEPSEEK_API_KEY
-    auth_header: x-api-key
-    max_tokens: 128000
-    context_window: 128000
-
-  kimi:
-    display_name: "Kimi K2.5"
-    provider: moonshot
-    model_id: kimi-for-coding
-    base_url: https://api.kimi.com/coding/
-    api_key_env: KIMI_API_KEY
-    auth_header: x-api-key
-    max_tokens: 32768
-    context_window: 256000
-
-  minimax:
-    display_name: "MiniMax M2.7"
-    provider: minimax
-    model_id: MiniMax-M2.7
-    base_url: https://api.minimaxi.com/anthropic
-    api_key_env: MINIMAX_API_KEY
-    auth_header: x-api-key
-    max_tokens: 196608
-    context_window: 204800
-
-  qwen:
-    display_name: "Qwen3.5 Plus"
-    provider: alibaba
-    model_id: qwen3.5-plus
-    base_url: https://dashscope.aliyuncs.com/apps/anthropic
-    api_key_env: QWEN_API_KEY
-    auth_header: x-api-key
-    max_tokens: 65536
-    context_window: 1000000
-
-  glm:
-    display_name: "GLM-5"
-    provider: zhipu
-    model_id: glm-5
-    base_url: https://open.bigmodel.cn/api/anthropic
-    api_key_env: GLM_API_KEY
-    auth_header: x-api-key
-    max_tokens: 128000
-    context_window: 200000
-
-  kat:
-    display_name: "KAT-Coder-Pro V2"
-    provider: kuaishou
-    model_id: kat-coder-pro-v2
-    base_url: https://wanqing.streamlakeapi.com/api/gateway/coding/kat-coder-pro-v2/claude-code-proxy
-    api_key_env: KAT_API_KEY
-    auth_header: x-api-key
-    max_tokens: 81920
-    context_window: 262144
-
-aliases:
-  ds: deepseek
-  deepseek-v3.2: deepseek
-  mm: minimax
-  minimax-m2.7: minimax
-  kimi-k2.5: kimi
-  qwen3.5-plus: qwen
-  glm-5: glm
-  kat-coder: kat
-  kat-coder-pro-v2: kat
-  kuaishou: kat
-
-gateway:
-  port: 8080
-  timeout: 300
+    return `# Claude Code Model Router Configuration
+# Place this file as models.yaml or .claude-router.yaml in your project root
+
+default_model: deepseek-v4-pro
+
+providers:
+  deepseek:
+    display_name: DeepSeek
+    provider: deepseek
+    base_url: https://api.deepseek.com/anthropic
+    api_key_env: DEEPSEEK_API_KEY
+    auth_header: x-api-key
+    auth_type: api_key
+    default_variant: v4-pro
+    variants:
+      v4-pro:
+        display_name: "DeepSeek V4 Pro"
+        model_id: deepseek-v4-pro
+        max_tokens: 393216
+        context_window: 1048576
+      v4-flash:
+        display_name: "DeepSeek V4 Flash"
+        model_id: deepseek-v4-flash
+        max_tokens: 393216
+        context_window: 1048576
+
+  kimi:
+    display_name: Kimi
+    provider: moonshot
+    base_url: https://api.moonshot.ai/anthropic
+    api_key_env: KIMI_API_KEY
+    auth_header: Authorization
+    auth_type: bearer
+    default_variant: k2.6
+    variants:
+      k2.6:
+        display_name: "Kimi K2.6"
+        model_id: kimi-k2.6
+        max_tokens: 32768
+        context_window: 262144
+
+  minimax:
+    display_name: MiniMax
+    provider: minimax
+    base_url: https://api.minimax.io/anthropic
+    api_key_env: MINIMAX_API_KEY
+    auth_header: Authorization
+    auth_type: bearer
+    default_variant: m2.7
+    variants:
+      m2.7:
+        display_name: "MiniMax M2.7"
+        model_id: MiniMax-M2.7
+        max_tokens: 196608
+        context_window: 204800
+      m2.7-highspeed:
+        display_name: "MiniMax M2.7 Highspeed"
+        model_id: MiniMax-M2.7-highspeed
+        max_tokens: 196608
+        context_window: 204800
+
+  qwen:
+    display_name: Qwen
+    provider: alibaba
+    base_url: https://dashscope.aliyuncs.com/apps/anthropic
+    api_key_env: QWEN_API_KEY
+    auth_header: x-api-key
+    auth_type: api_key
+    default_variant: 3.5-plus
+    variants:
+      3.5-plus:
+        model_key: qwen3.5-plus
+        display_name: "Qwen3.5 Plus"
+        model_id: qwen3.5-plus
+        max_tokens: 65536
+        context_window: 1000000
+      3.5-flash:
+        model_key: qwen3.5-flash
+        display_name: "Qwen3.5 Flash"
+        model_id: qwen3.5-flash
+        max_tokens: 65536
+        context_window: 1000000
+      3-max:
+        model_key: qwen3-max
+        display_name: "Qwen3 Max"
+        model_id: qwen3-max
+        max_tokens: 65536
+        context_window: 1000000
+
+  glm:
+    display_name: GLM
+    provider: zhipu
+    base_url: https://open.bigmodel.cn/api/anthropic
+    api_key_env: GLM_API_KEY
+    auth_header: x-api-key
+    auth_type: api_key
+    default_variant: 5.1
+    variants:
+      5.1:
+        display_name: "GLM-5.1"
+        model_id: glm-5.1
+        max_tokens: 131072
+        context_window: 204800
+
+  mimo:
+    display_name: MiMo
+    provider: xiaomi
+    base_url: https://api.xiaomimimo.com/anthropic
+    api_key_env: MIMO_API_KEY
+    auth_header: api-key
+    auth_type: api_key
+    default_variant: v2.5
+    variants:
+      v2.5:
+        display_name: "MiMo V2.5"
+        model_id: mimo-v2.5
+        max_tokens: 131072
+        context_window: 1048576
+
+aliases:
+  deepseek: deepseek-v4-pro
+  deepseek-v4: deepseek-v4-pro
+  deepseek-pro: deepseek-v4-pro
+  deepseek-flash: deepseek-v4-flash
+  deepseek-chat: deepseek-v4-flash
+  ds: deepseek-v4-pro
+  kimi: kimi-k2.6
+  kimi-k2: kimi-k2.6
+  moonshot: kimi-k2.6
+  minimax: minimax-m2.7
+  minimax-m2: minimax-m2.7
+  minimax-highspeed: minimax-m2.7-highspeed
+  mm: minimax-m2.7
+  qwen: qwen3.5-plus
+  qwen3.5: qwen3.5-plus
+  tongyi: qwen3.5-plus
+  glm: glm-5.1
+  glm-5: glm-5.1
+  zhipu: glm-5.1
+  chatglm: glm-5.1
+  mimo: mimo-v2.5
+  mimo-v2: mimo-v2.5
+  mimo-v2.5: mimo-v2.5
+  xiaomi: mimo-v2.5
+
+gateway:
+  port: 8080
+  timeout: 300
 `;
 }
 function generateEnvFile() {
-    return `# Claude Code Model Router - API Keys
-# Fill in your API keys below
-
-# DeepSeek - https://platform.deepseek.com/
-DEEPSEEK_API_KEY=
-
-# Kimi - https://www.kimi.com/
-KIMI_API_KEY=
-
-# MiniMax - https://platform.minimax.io/
-MINIMAX_API_KEY=
-
-# Qwen - https://dashscope.console.aliyun.com/
-QWEN_API_KEY=
-
-# GLM - https://open.bigmodel.cn/
-GLM_API_KEY=
-
-# KAT-Coder (快手) - https://wanqing.streamlakeapi.com/
-KAT_API_KEY=
+    return `# Claude Code Model Router - API Keys
+# Fill in your API keys below
+
+# DeepSeek - https://platform.deepseek.com/
+DEEPSEEK_API_KEY=
+
+# Kimi / Moonshot - https://platform.kimi.ai/
+KIMI_API_KEY=
+
+# MiniMax - https://platform.minimax.io/
+MINIMAX_API_KEY=
+
+# Qwen - https://dashscope.console.aliyun.com/
+QWEN_API_KEY=
+
+# GLM - https://open.bigmodel.cn/
+GLM_API_KEY=
+
+# MiMo - https://platform.xiaomimimo.com/
+MIMO_API_KEY=
 `;
 }
 //# sourceMappingURL=config.js.map