ZacharyZhang-NY/Kigi-CLI
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

feat(providers): add Xiaomi MiMo (global + China token plan)

Browse Source 
Providers 19-20 (22nd & 23rd registry variants), sourced from Pi
(earendil-works/pi). Pi's xiaomi.ts / xiaomi-token-plan-cn.ts use plain
openAICompletionsApi (no thinking dialect) → Kigi Passthrough. Global: MiMo
at api.xiaomimimo.com/v1, XIAOMI_API_KEY, models.dev xiaomi. China token plan:
token-plan-cn.xiaomimimo.com/v1, XIAOMI_TOKEN_PLAN_CN_API_KEY, models.dev
xiaomi-token-plan-cn. Both Bearer + OpenAI listing + ChatCompletions +
restrict_to_enriched; /models auth-gated → validator.

id-match proven vs Pi's static ids (mimo-v2.5-pro, mimo-v2-flash, ...). The
CN token plan lists 4 mimo TTS models (tool_call=false); restrict's tool_call
cut drops them, keeping only the 3 chat models — the e2e proves this
non-vacuously with the real mimo-v2-tts. Review found no defects.

Registry at 23; picker 24 rows (welcome picker verified rendering all rows at
a taller viewport).
This commit is contained in:
ZacharyZhang-NY
2026-07-21 19:55:51 -04:00
parent 010f6c3be3
commit 347311bfe5
4 changed files with 237 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/codegen/kigi-shell/src/agent/auth_method.rs
+62 -4
View File
@@ -625,7 +625,9 @@ mod tests {
"qwen-token-plan-cn",
"kimi-coding",
"zai",
"zai-coding-cn"
"zai-coding-cn",
"xiaomi",
"xiaomi-token-plan-cn"
]
);
assert_eq!(default_id(&built), Some(XAI_API_KEY_METHOD_ID));
@@ -669,7 +671,9 @@ mod tests {
"qwen-token-plan-cn",
"kimi-coding",
"zai",
"zai-coding-cn"
"zai-coding-cn",
"xiaomi",
"xiaomi-token-plan-cn"
]
);
assert_eq!(default_id(&built), Some(CACHED_TOKEN_AUTH_METHOD_ID));
@@ -706,7 +710,9 @@ mod tests {
"qwen-token-plan-cn",
"kimi-coding",
"zai",
"zai-coding-cn"
"zai-coding-cn",
"xiaomi",
"xiaomi-token-plan-cn"
]
);
assert_eq!(default_id(&built), Some(CACHED_TOKEN_AUTH_METHOD_ID));
@@ -746,7 +752,9 @@ mod tests {
"qwen-token-plan-cn",
"kimi-coding",
"zai",
"zai-coding-cn"
"zai-coding-cn",
"xiaomi",
"xiaomi-token-plan-cn"
]
);
assert_eq!(default_id(&built), None);
@@ -1147,4 +1155,54 @@ mod tests {
"Invalid API key for zai-coding-cn \u{2014} check your key on open.bigmodel.cn"
);
}
/// Xiaomi MiMo (global): /models is auth-gated (401 for a bad key).
#[tokio::test]
#[serial]
async fn xiaomi_validates_against_models_and_rejects_bad_key() {
use wiremock::matchers::{method, path};
let server = wiremock::MockServer::start().await;
wiremock::Mock::given(method("GET"))
.and(path("/models"))
.respond_with(wiremock::ResponseTemplate::new(401))
.expect(1)
.mount(&server)
.await;
let _base = EnvGuard::set(kigi_models::XIAOMI_BASE_URL_ENV, &server.uri());
let err = authenticate_platform_api_key(kigi_models::PlatformId::Xiaomi, Some("xm-bad"))
.await
.expect_err("a 401 from /models must reject the key");
assert_eq!(
err.message,
"Invalid API key for xiaomi \u{2014} check your key on xiaomimimo.com"
);
}
/// Xiaomi Token Plan CN: distinct base URL; same auth-gated validation.
#[tokio::test]
#[serial]
async fn xiaomi_token_plan_cn_validates_against_models_and_rejects_bad_key() {
use wiremock::matchers::{method, path};
let server = wiremock::MockServer::start().await;
wiremock::Mock::given(method("GET"))
.and(path("/models"))
.respond_with(wiremock::ResponseTemplate::new(401))
.expect(1)
.mount(&server)
.await;
let _base = EnvGuard::set(
kigi_models::XIAOMI_TOKEN_PLAN_CN_BASE_URL_ENV,
&server.uri(),
);
let err = authenticate_platform_api_key(
kigi_models::PlatformId::XiaomiTokenPlanCn,
Some("xm-cn-bad"),
)
.await
.expect_err("a 401 from /models must reject the key");
assert_eq!(
err.message,
"Invalid API key for xiaomi-token-plan-cn \u{2014} check your key on xiaomimimo.com"
);
}
}