From df57f982f8983b0c2334fbc368c7fb89e95141ba Mon Sep 17 00:00:00 2001
From: ekko <152005280+EKKOLearnAI@users.noreply.github.com>
Date: Fri, 1 May 2026 15:57:41 +0800
Subject: [PATCH] feat: add Xiaomi Token Plan provider and update MiMo models
 (#376)

Add new Xiaomi Token Plan provider with updated model catalog and refresh existing Xiaomi MiMo provider models to match actual API response.

Changes:
- Add new Xiaomi Token Plan provider (base_url: https://token-plan-sgp.xiaomimimo.com/v1)
- Update Xiaomi MiMo provider models: remove mimo-v2-flash, add v2.5 series and TTS variants
- Add environment variable mapping for xiaomi-token-plan provider

Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 .../server/src/services/config-helpers.ts     |  1 +
 packages/server/src/shared/providers.ts       | 27 ++++++++++++++++++-
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/packages/server/src/services/config-helpers.ts b/packages/server/src/services/config-helpers.ts
index 87a303b..2d79502 100644
--- a/packages/server/src/services/config-helpers.ts
+++ b/packages/server/src/services/config-helpers.ts
@@ -21,6 +21,7 @@ export const PROVIDER_ENV_MAP: Record<string, { api_key_env: string; base_url_en
   anthropic: { api_key_env: 'ANTHROPIC_API_KEY', base_url_env: '' },
   xai: { api_key_env: 'XAI_API_KEY', base_url_env: '' },
   xiaomi: { api_key_env: 'XIAOMI_API_KEY', base_url_env: '' },
+  'xiaomi-token-plan': { api_key_env: '', base_url_env: '' },
   gemini: { api_key_env: 'GEMINI_API_KEY', base_url_env: '' },
   kilocode: { api_key_env: 'KILO_API_KEY', base_url_env: '' },
   'ai-gateway': { api_key_env: 'AI_GATEWAY_API_KEY', base_url_env: '' },
diff --git a/packages/server/src/shared/providers.ts b/packages/server/src/shared/providers.ts
index 7691679..5fd99c5 100644
--- a/packages/server/src/shared/providers.ts
+++ b/packages/server/src/shared/providers.ts
@@ -165,12 +165,37 @@ export const PROVIDER_PRESETS: ProviderPreset[] = [
       'moonshotai/Kimi-K2-Thinking',
     ],
   },
+  {
+    label: 'Xiaomi Token Plan',
+    value: 'xiaomi-token-plan',
+    builtin: true,
+    base_url: 'https://token-plan-sgp.xiaomimimo.com/v1',
+    models: [
+      'mimo-v2-omni',
+      'mimo-v2-pro',
+      'mimo-v2-tts',
+      'mimo-v2.5',
+      'mimo-v2.5-pro',
+      'mimo-v2.5-tts',
+      'mimo-v2.5-tts-voiceclone',
+      'mimo-v2.5-tts-voicedesign',
+    ],
+  },
   {
     label: 'Xiaomi MiMo',
     value: 'xiaomi',
     builtin: true,
     base_url: 'https://api.xiaomimimo.com/v1',
-    models: ['mimo-v2-pro', 'mimo-v2-omni', 'mimo-v2-flash'],
+    models: [
+      'mimo-v2-omni',
+      'mimo-v2-pro',
+      'mimo-v2-tts',
+      'mimo-v2.5',
+      'mimo-v2.5-pro',
+      'mimo-v2.5-tts',
+      'mimo-v2.5-tts-voiceclone',
+      'mimo-v2.5-tts-voicedesign',
+    ],
   },
   {
     label: 'Kilo Code',