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

feat(ai): add Xiaomi MiMo provider (#4005)

Browse Source 
* fix(ai): include minimax-cn in cross-provider-handoff matrix

* feat(ai): add Xiaomi MiMo provider

Adds Xiaomi MiMo as an openai-completions-compatible provider.

- packages/ai: register provider in types/KnownProvider, env-api-keys (XIAOMI_API_KEY), generate-models, models.generated.ts, overflow util, README, CHANGELOG
- packages/ai/test: extend stream, tokens, abort, empty, context-overflow, overflow, image-tool-result, tool-call-without-result, total-tokens, unicode-surrogate, cross-provider-handoff matrices with Xiaomi
- packages/coding-agent: default model (mimo-v2.5-pro), display name (Xiaomi MiMo), CLI env var docs, README, docs/providers.md

closes #3912

---------

Co-authored-by: Mario Zechner <badlogicgames@gmail.com>
This commit is contained in:
Jake Jia
2026-05-02 06:46:05 +08:00
committed by GitHub
parent ddb8ed0c73
commit a44622670f
23 changed files with 319 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
packages/ai/test/abort.test.ts
View File

@@ -190,6 +190,18 @@ describe("AI Providers Abort Tests", () => {
});
});
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo Provider Abort", () => {
const llm = getModel("xiaomi", "mimo-v2.5-pro");
it("should abort mid-stream", { retry: 3 }, async () => {
await testAbortSignal(llm);
});
it("should handle immediate abort", { retry: 3 }, async () => {
await testImmediateAbort(llm);
});
});
describe.skipIf(!process.env.KIMI_API_KEY)("Kimi For Coding Provider Abort", () => {
const llm = getModel("kimi-coding", "kimi-k2-thinking");

19
packages/ai/test/context-overflow.test.ts
View File

@@ -389,6 +389,25 @@ describe("Context overflow error handling", () => {
}, 120000);
});
// =============================================================================
// Xiaomi MiMo
// =============================================================================
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo", () => {
// Xiaomi silently truncates oversized input to fill the context window exactly,
// then returns finish_reason "length" with output=0 (no room left to generate).
// This is a detectable overflow signal but uses stopReason "length" rather than "error".
it("mimo-v2.5-pro - should detect overflow via isContextOverflow", async () => {
const model = getModel("xiaomi", "mimo-v2.5-pro");
const result = await testContextOverflow(model, process.env.XIAOMI_API_KEY!);
logResult(result);
expect(result.stopReason).toBe("length");
expect(result.usage.output).toBe(0);
expect(isContextOverflow(result.response, model.contextWindow)).toBe(true);
}, 120000);
});
// =============================================================================
// Kimi For Coding
// =============================================================================

3
packages/ai/test/cross-provider-handoff.test.ts
View File

@@ -113,6 +113,7 @@ const PROVIDER_MODEL_PAIRS: ProviderModelPair[] = [
{ provider: "mistral", model: "devstral-medium-latest", label: "mistral-devstral-medium" },
// MiniMax
{ provider: "minimax", model: "MiniMax-M2.7", label: "minimax-m2.7" },
{ provider: "minimax-cn", model: "MiniMax-M2.7", label: "minimax-m2.7" },
// OpenCode Zen
{ provider: "opencode", model: "big-pickle", label: "zen-big-pickle" },
{ provider: "opencode", model: "claude-sonnet-4-5", label: "zen-claude-sonnet-4-5" },
@@ -123,6 +124,8 @@ const PROVIDER_MODEL_PAIRS: ProviderModelPair[] = [
// OpenCode Go
{ provider: "opencode-go", model: "kimi-k2.5", label: "go-kimi-k2.5" },
{ provider: "opencode-go", model: "minimax-m2.5", label: "go-minimax-m2.5" },
// Xiaomi MiMo
{ provider: "xiaomi", model: "mimo-v2.5-pro", label: "xiaomi-mimo-v2.5-pro" },
];
// Cached context structure

20
packages/ai/test/empty.test.ts
View File

@@ -427,6 +427,26 @@ describe("AI Providers Empty Message Tests", () => {
});
});
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo Provider Empty Messages", () => {
const llm = getModel("xiaomi", "mimo-v2.5-pro");
it("should handle empty content array", { retry: 3, timeout: 30000 }, async () => {
await testEmptyMessage(llm);
});
it("should handle empty string content", { retry: 3, timeout: 30000 }, async () => {
await testEmptyStringMessage(llm);
});
it("should handle whitespace-only content", { retry: 3, timeout: 30000 }, async () => {
await testWhitespaceOnlyMessage(llm);
});
it("should handle empty assistant message in conversation", { retry: 3, timeout: 30000 }, async () => {
await testEmptyAssistantMessage(llm);
});
});
describe.skipIf(!process.env.KIMI_API_KEY)("Kimi For Coding Provider Empty Messages", () => {
const llm = getModel("kimi-coding", "kimi-k2-thinking");

12
packages/ai/test/image-tool-result.test.ts
View File

@@ -298,6 +298,18 @@ describe("Tool Results with Images", () => {
});
});
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo Provider (mimo-v2.5-pro)", () => {
const llm = getModel("xiaomi", "mimo-v2.5-pro");
it("should handle tool result with only image", { retry: 3, timeout: 30000 }, async () => {
await handleToolWithImageResult(llm);
});
it("should handle tool result with text and image", { retry: 3, timeout: 30000 }, async () => {
await handleToolWithTextAndImageResult(llm);
});
});
describe.skipIf(!process.env.KIMI_API_KEY)("Kimi For Coding Provider (kimi-for-coding)", () => {
const llm = getModel("kimi-coding", "kimi-for-coding");

35
packages/ai/test/overflow.test.ts
View File

@@ -61,4 +61,39 @@ describe("isContextOverflow", () => {
const message = createErrorMessage("Too many requests. Please slow down.");
expect(isContextOverflow(message, 200000)).toBe(false);
});
function createLengthStopMessage(input: number, cacheRead: number, output: number): AssistantMessage {
return {
role: "assistant",
content: [],
api: "openai-completions",
provider: "xiaomi",
model: "mimo-v2.5-pro",
usage: {
input,
output,
cacheRead,
cacheWrite: 0,
totalTokens: input + cacheRead + output,
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
},
stopReason: "length",
timestamp: Date.now(),
};
}
it("detects Xiaomi-style overflow (length stop with zero output and filled context)", () => {
const message = createLengthStopMessage(58, 1048512, 0);
expect(isContextOverflow(message, 1048576)).toBe(true);
});
it("does not treat normal length stops with output as overflow", () => {
const message = createLengthStopMessage(1000, 0, 4096);
expect(isContextOverflow(message, 200000)).toBe(false);
});
it("does not treat length stops far below context as overflow", () => {
const message = createLengthStopMessage(100, 0, 0);
expect(isContextOverflow(message, 200000)).toBe(false);
});
});

18
packages/ai/test/stream.test.ts
View File

@@ -993,6 +993,24 @@ describe("Generate E2E Tests", () => {
},
);
describe.skipIf(!process.env.XIAOMI_API_KEY)(
"Xiaomi MiMo Provider (Xiaomi MiMo-V2.5-Pro via OpenAI Completions)",
() => {
const llm = getModel("xiaomi", "mimo-v2.5-pro");
it("should complete basic text generation", { retry: 3 }, async () => {
await basicTextGeneration(llm);
});
it("should handle tool calling", { retry: 3 }, async () => {
await handleToolCall(llm);
});
it("should handle streaming", { retry: 3 }, async () => {
await handleStreaming(llm);
});
},
);
// =========================================================================
// OAuth-based providers (credentials from ~/.pi/agent/oauth.json)
// Tokens are resolved at module level (see oauthTokens above)

8
packages/ai/test/tokens.test.ts
View File

@@ -220,6 +220,14 @@ describe("Token Statistics on Abort", () => {
});
});
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo Provider", () => {
const llm = getModel("xiaomi", "mimo-v2.5-pro");
it("should include token stats when aborted mid-stream", { retry: 3, timeout: 30000 }, async () => {
await testTokensOnAbort(llm);
});
});
// =========================================================================
// OAuth-based providers (credentials from ~/.pi/agent/oauth.json)
// =========================================================================

8
packages/ai/test/tool-call-without-result.test.ts
View File

@@ -215,6 +215,14 @@ describe("Tool Call Without Result Tests", () => {
});
});
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo Provider", () => {
const model = getModel("xiaomi", "mimo-v2.5-pro");
it("should filter out tool calls without corresponding tool results", { retry: 3, timeout: 30000 }, async () => {
await testToolCallWithoutResult(model);
});
});
describe.skipIf(!process.env.KIMI_API_KEY)("Kimi For Coding Provider", () => {
const model = getModel("kimi-coding", "kimi-k2-thinking");

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

@@ -443,6 +443,29 @@ describe("totalTokens field", () => {
);
});
// =========================================================================
// Xiaomi MiMo
// =========================================================================
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo", () => {
it(
"mimo-v2.5-pro - should return totalTokens equal to sum of components",
{ retry: 3, timeout: 60000 },
async () => {
const llm = getModel("xiaomi", "mimo-v2.5-pro");
console.log(`\nXiaomi MiMo / ${llm.id}:`);
const { first, second } = await testTotalTokensWithCache(llm, { apiKey: process.env.XIAOMI_API_KEY });
logUsage("First request", first);
logUsage("Second request", second);
assertTotalTokensEqualsComponents(first);
assertTotalTokensEqualsComponents(second);
},
);
});
// =========================================================================
// Kimi For Coding
// =========================================================================

16
packages/ai/test/unicode-surrogate.test.ts
View File

@@ -594,6 +594,22 @@ describe("AI Providers Unicode Surrogate Pair Tests", () => {
});
});
describe.skipIf(!process.env.XIAOMI_API_KEY)("Xiaomi MiMo Provider Unicode Handling", () => {
const llm = getModel("xiaomi", "mimo-v2.5-pro");
it("should handle emoji in tool results", { retry: 3, timeout: 30000 }, async () => {
await testEmojiInToolResults(llm);
});
it("should handle real-world LinkedIn comment data with emoji", { retry: 3, timeout: 30000 }, async () => {
await testRealWorldLinkedInData(llm);
});
it("should handle unpaired high surrogate (0xD83D) in tool results", { retry: 3, timeout: 30000 }, async () => {
await testUnpairedHighSurrogate(llm);
});
});
describe.skipIf(!process.env.KIMI_API_KEY)("Kimi For Coding Provider Unicode Handling", () => {
const llm = getModel("kimi-coding", "kimi-k2-thinking");