shumengya/sproutclaw
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ai): switch xiaomi default to api billing, add per-region token plan providers (#4112)

Browse Source 
Built-in `xiaomi` provider now targets the API billing endpoint (https://api.xiaomimimo.com/anthropic) — a single stable URL for keys issued at platform.xiaomimimo.com. The Token Plan endpoints are exposed as three sibling providers, each with its own env var:

- xiaomi-token-plan-cn: XIAOMI_TOKEN_PLAN_CN_API_KEY
- xiaomi-token-plan-ams: XIAOMI_TOKEN_PLAN_AMS_API_KEY
- xiaomi-token-plan-sgp: XIAOMI_TOKEN_PLAN_SGP_API_KEY

BREAKING CHANGE: users who previously set XIAOMI_API_KEY against the Token Plan AMS endpoint must move to xiaomi-token-plan-ams and set XIAOMI_TOKEN_PLAN_AMS_API_KEY. This also resolves the 401 reported by on #4005, where a platform.xiaomimimo.com key fails against the Token Plan endpoint.

closes #4082
This commit is contained in:
Jake Jia
2026-05-03 18:57:11 +08:00
committed by Mario Zechner
parent 924a1395bc
commit b326bf1a3a
22 changed files with 859 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
packages/ai/test/total-tokens.test.ts
View File

@@ -447,7 +447,7 @@ describe("totalTokens field", () => {
// Xiaomi MiMo
// =========================================================================
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo", () => {
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo (API billing)", () => {
it(
"mimo-v2.5-pro - should return totalTokens equal to sum of components",
{ retry: 3, timeout: 60000 },
@@ -466,6 +466,81 @@ describe("totalTokens field", () => {
);
});
// =========================================================================
// Xiaomi MiMo Token Plan CN
// =========================================================================
describe.skipIf(!process.env.XIAOMI_TOKEN_PLAN_CN_API_KEY)("Xiaomi MiMo Token Plan (CN)", () => {
it(
"mimo-v2.5-pro - should return totalTokens equal to sum of components",
{ retry: 3, timeout: 60000 },
async () => {
const llm = getModel("xiaomi-token-plan-cn", "mimo-v2.5-pro");
console.log(`\nXiaomi MiMo Token Plan CN / ${llm.id}:`);
const { first, second } = await testTotalTokensWithCache(llm, {
apiKey: process.env.XIAOMI_TOKEN_PLAN_CN_API_KEY,
});
logUsage("First request", first);
logUsage("Second request", second);
assertTotalTokensEqualsComponents(first);
assertTotalTokensEqualsComponents(second);
},
);
});
// =========================================================================
// Xiaomi MiMo Token Plan AMS
// =========================================================================
describe.skipIf(!process.env.XIAOMI_TOKEN_PLAN_AMS_API_KEY)("Xiaomi MiMo Token Plan (AMS)", () => {
it(
"mimo-v2.5-pro - should return totalTokens equal to sum of components",
{ retry: 3, timeout: 60000 },
async () => {
const llm = getModel("xiaomi-token-plan-ams", "mimo-v2.5-pro");
console.log(`\nXiaomi MiMo Token Plan AMS / ${llm.id}:`);
const { first, second } = await testTotalTokensWithCache(llm, {
apiKey: process.env.XIAOMI_TOKEN_PLAN_AMS_API_KEY,
});
logUsage("First request", first);
logUsage("Second request", second);
assertTotalTokensEqualsComponents(first);
assertTotalTokensEqualsComponents(second);
},
);
});
// =========================================================================
// Xiaomi MiMo Token Plan SGP
// =========================================================================
describe.skipIf(!process.env.XIAOMI_TOKEN_PLAN_SGP_API_KEY)("Xiaomi MiMo Token Plan (SGP)", () => {
it(
"mimo-v2.5-pro - should return totalTokens equal to sum of components",
{ retry: 3, timeout: 60000 },
async () => {
const llm = getModel("xiaomi-token-plan-sgp", "mimo-v2.5-pro");
console.log(`\nXiaomi MiMo Token Plan SGP / ${llm.id}:`);
const { first, second } = await testTotalTokensWithCache(llm, {
apiKey: process.env.XIAOMI_TOKEN_PLAN_SGP_API_KEY,
});
logUsage("First request", first);
logUsage("Second request", second);
assertTotalTokensEqualsComponents(first);
assertTotalTokensEqualsComponents(second);
},
);
});
// =========================================================================
// Kimi For Coding
// =========================================================================