Vendor OpenClaw source as Adolf fork baseline
Some checks failed
ClawSweeper Dispatch / dispatch (push) Has been cancelled
CodeQL / Security High (actions) (push) Has been cancelled
CodeQL / Security High (channel-runtime-boundary) (push) Has been cancelled
CodeQL / Security High (core-auth-secrets) (push) Has been cancelled
CodeQL / Security High (mcp-process-tool-boundary) (push) Has been cancelled
CodeQL / Security High (network-ssrf-boundary) (push) Has been cancelled
CodeQL / Security High (plugin-trust-boundary) (push) Has been cancelled
CodeQL / Security High (process-exec-boundary) (push) Has been cancelled
Docs Sync Publish Repo / sync-publish-repo (push) Has been cancelled
Docs / docs (push) Has been cancelled
OpenClaw Stable Main Closeout / Resolve stable release closeout inputs (push) Has been cancelled
OpenClaw Stable Main Closeout / Verify stable main closeout (push) Has been cancelled
Workflow Sanity / no-tabs (push) Has been cancelled
Workflow Sanity / actionlint (push) Has been cancelled
Workflow Sanity / generated-doc-baselines (push) Has been cancelled
CI / runner-admission (push) Has been cancelled
CI / preflight (push) Has been cancelled
CI / security-fast (push) Has been cancelled
CI / pnpm-store-warmup (push) Has been cancelled
CI / build-artifacts (push) Has been cancelled
CI / native-i18n (push) Has been cancelled
CI / ${{ matrix.check_name }} (push) Has been cancelled
CI / ${{ matrix.checkName }} (push) Has been cancelled
CI / checks-node-compat-node22 (push) Has been cancelled
CI / check-bundled-channel-config-metadata (push) Has been cancelled
CI / check-dependencies (push) Has been cancelled
CI / check-guards (push) Has been cancelled
CI / check-lint (push) Has been cancelled
CI / check-prod-types (push) Has been cancelled
CI / check-shrinkwrap (push) Has been cancelled
CI / check-test-types (push) Has been cancelled
CI / check-additional-boundaries-a (push) Has been cancelled
CI / check-additional-boundaries-bcd (push) Has been cancelled
CI / check-additional-extension-bundled (push) Has been cancelled
CI / check-additional-extension-channels (push) Has been cancelled
CI / check-additional-extension-package-boundary (push) Has been cancelled
CI / check-additional-runtime-topology-architecture (push) Has been cancelled
CI / check-session-accessor-boundary (push) Has been cancelled
CI / check-session-transcript-reader-boundary (push) Has been cancelled
CI / check-docs (push) Has been cancelled
CI / skills-python (push) Has been cancelled
CI / macos-swift (push) Has been cancelled
CI / ios-build (push) Has been cancelled
CI / ci-timings-summary (push) Has been cancelled
Native App Locale Refresh / Refresh native fa (push) Has been cancelled
Native App Locale Refresh / Refresh native fr (push) Has been cancelled
Native App Locale Refresh / Refresh native hi (push) Has been cancelled
Native App Locale Refresh / Refresh native id (push) Has been cancelled
Native App Locale Refresh / Refresh native it (push) Has been cancelled
Native App Locale Refresh / Refresh native ja-JP (push) Has been cancelled
Control UI Locale Refresh / plan (push) Has been cancelled
Control UI Locale Refresh / Refresh ${{ matrix.locale }} (push) Has been cancelled
Control UI Locale Refresh / Commit control UI locale refresh (push) Has been cancelled
Live Media Runner Image / Build live media runner image (push) Has been cancelled
Native App Locale Refresh / Refresh native ar (push) Has been cancelled
Native App Locale Refresh / Refresh native de (push) Has been cancelled
Native App Locale Refresh / Refresh native es (push) Has been cancelled
Native App Locale Refresh / Refresh native ko (push) Has been cancelled
Native App Locale Refresh / Refresh native nl (push) Has been cancelled
Native App Locale Refresh / Refresh native pl (push) Has been cancelled
Native App Locale Refresh / Refresh native pt-BR (push) Has been cancelled
Native App Locale Refresh / Refresh native ru (push) Has been cancelled
Native App Locale Refresh / Refresh native sv (push) Has been cancelled
Native App Locale Refresh / Refresh native th (push) Has been cancelled
Native App Locale Refresh / Refresh native tr (push) Has been cancelled
Native App Locale Refresh / Refresh native uk (push) Has been cancelled
Native App Locale Refresh / Refresh native vi (push) Has been cancelled
Native App Locale Refresh / Refresh native zh-CN (push) Has been cancelled
Native App Locale Refresh / Refresh native zh-TW (push) Has been cancelled
Native App Locale Refresh / Commit native locale refresh (push) Has been cancelled
Plugin Init Scaffold Validation / Validate provider scaffold (push) Has been cancelled
Plugin NPM Release / preview_plugins_npm (push) Has been cancelled
Plugin NPM Release / Validate release publish approval (push) Has been cancelled
Plugin NPM Release / preview_plugin_pack (push) Has been cancelled
Plugin NPM Release / publish_plugins_npm (push) Has been cancelled
Sandbox Common Smoke / sandbox-common-smoke (push) Has been cancelled
Website Installer Sync / static (push) Has been cancelled
Website Installer Sync / linux-docker (push) Has been cancelled
Website Installer Sync / macos-installer (push) Has been cancelled
Website Installer Sync / windows-installer (push) Has been cancelled
Website Installer Sync / sync-website (push) Has been cancelled
Some checks failed
ClawSweeper Dispatch / dispatch (push) Has been cancelled
CodeQL / Security High (actions) (push) Has been cancelled
CodeQL / Security High (channel-runtime-boundary) (push) Has been cancelled
CodeQL / Security High (core-auth-secrets) (push) Has been cancelled
CodeQL / Security High (mcp-process-tool-boundary) (push) Has been cancelled
CodeQL / Security High (network-ssrf-boundary) (push) Has been cancelled
CodeQL / Security High (plugin-trust-boundary) (push) Has been cancelled
CodeQL / Security High (process-exec-boundary) (push) Has been cancelled
Docs Sync Publish Repo / sync-publish-repo (push) Has been cancelled
Docs / docs (push) Has been cancelled
OpenClaw Stable Main Closeout / Resolve stable release closeout inputs (push) Has been cancelled
OpenClaw Stable Main Closeout / Verify stable main closeout (push) Has been cancelled
Workflow Sanity / no-tabs (push) Has been cancelled
Workflow Sanity / actionlint (push) Has been cancelled
Workflow Sanity / generated-doc-baselines (push) Has been cancelled
CI / runner-admission (push) Has been cancelled
CI / preflight (push) Has been cancelled
CI / security-fast (push) Has been cancelled
CI / pnpm-store-warmup (push) Has been cancelled
CI / build-artifacts (push) Has been cancelled
CI / native-i18n (push) Has been cancelled
CI / ${{ matrix.check_name }} (push) Has been cancelled
CI / ${{ matrix.checkName }} (push) Has been cancelled
CI / checks-node-compat-node22 (push) Has been cancelled
CI / check-bundled-channel-config-metadata (push) Has been cancelled
CI / check-dependencies (push) Has been cancelled
CI / check-guards (push) Has been cancelled
CI / check-lint (push) Has been cancelled
CI / check-prod-types (push) Has been cancelled
CI / check-shrinkwrap (push) Has been cancelled
CI / check-test-types (push) Has been cancelled
CI / check-additional-boundaries-a (push) Has been cancelled
CI / check-additional-boundaries-bcd (push) Has been cancelled
CI / check-additional-extension-bundled (push) Has been cancelled
CI / check-additional-extension-channels (push) Has been cancelled
CI / check-additional-extension-package-boundary (push) Has been cancelled
CI / check-additional-runtime-topology-architecture (push) Has been cancelled
CI / check-session-accessor-boundary (push) Has been cancelled
CI / check-session-transcript-reader-boundary (push) Has been cancelled
CI / check-docs (push) Has been cancelled
CI / skills-python (push) Has been cancelled
CI / macos-swift (push) Has been cancelled
CI / ios-build (push) Has been cancelled
CI / ci-timings-summary (push) Has been cancelled
Native App Locale Refresh / Refresh native fa (push) Has been cancelled
Native App Locale Refresh / Refresh native fr (push) Has been cancelled
Native App Locale Refresh / Refresh native hi (push) Has been cancelled
Native App Locale Refresh / Refresh native id (push) Has been cancelled
Native App Locale Refresh / Refresh native it (push) Has been cancelled
Native App Locale Refresh / Refresh native ja-JP (push) Has been cancelled
Control UI Locale Refresh / plan (push) Has been cancelled
Control UI Locale Refresh / Refresh ${{ matrix.locale }} (push) Has been cancelled
Control UI Locale Refresh / Commit control UI locale refresh (push) Has been cancelled
Live Media Runner Image / Build live media runner image (push) Has been cancelled
Native App Locale Refresh / Refresh native ar (push) Has been cancelled
Native App Locale Refresh / Refresh native de (push) Has been cancelled
Native App Locale Refresh / Refresh native es (push) Has been cancelled
Native App Locale Refresh / Refresh native ko (push) Has been cancelled
Native App Locale Refresh / Refresh native nl (push) Has been cancelled
Native App Locale Refresh / Refresh native pl (push) Has been cancelled
Native App Locale Refresh / Refresh native pt-BR (push) Has been cancelled
Native App Locale Refresh / Refresh native ru (push) Has been cancelled
Native App Locale Refresh / Refresh native sv (push) Has been cancelled
Native App Locale Refresh / Refresh native th (push) Has been cancelled
Native App Locale Refresh / Refresh native tr (push) Has been cancelled
Native App Locale Refresh / Refresh native uk (push) Has been cancelled
Native App Locale Refresh / Refresh native vi (push) Has been cancelled
Native App Locale Refresh / Refresh native zh-CN (push) Has been cancelled
Native App Locale Refresh / Refresh native zh-TW (push) Has been cancelled
Native App Locale Refresh / Commit native locale refresh (push) Has been cancelled
Plugin Init Scaffold Validation / Validate provider scaffold (push) Has been cancelled
Plugin NPM Release / preview_plugins_npm (push) Has been cancelled
Plugin NPM Release / Validate release publish approval (push) Has been cancelled
Plugin NPM Release / preview_plugin_pack (push) Has been cancelled
Plugin NPM Release / publish_plugins_npm (push) Has been cancelled
Sandbox Common Smoke / sandbox-common-smoke (push) Has been cancelled
Website Installer Sync / static (push) Has been cancelled
Website Installer Sync / linux-docker (push) Has been cancelled
Website Installer Sync / macos-installer (push) Has been cancelled
Website Installer Sync / windows-installer (push) Has been cancelled
Website Installer Sync / sync-website (push) Has been cancelled
Adolf is a fork/vendored clone of github.com/openclaw/openclaw (v2026.6.11), free to diverge. Tree copied sans upstream .git; upstream remote added for future syncs. Node pinned to 24 (.nvmrc); engines already require >=22.19. Preserves docs/ARCHITECTURE.md. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01LeqyaxJF2nbRXJtae2kNB2
This commit is contained in:
683
extensions/xiaomi/index.test.ts
Normal file
683
extensions/xiaomi/index.test.ts
Normal file
@@ -0,0 +1,683 @@
|
||||
// Xiaomi tests cover index plugin behavior.
|
||||
import type { StreamFn } from "openclaw/plugin-sdk/agent-core";
|
||||
import type { Context, Model } from "openclaw/plugin-sdk/llm";
|
||||
import { createAssistantMessageEventStream } from "openclaw/plugin-sdk/llm";
|
||||
import {
|
||||
registerProviderPlugin,
|
||||
requireRegisteredProvider,
|
||||
resolveProviderPluginChoice,
|
||||
type RegisteredProviderCollections,
|
||||
} from "openclaw/plugin-sdk/plugin-test-runtime";
|
||||
import { buildOpenAICompletionsParams } from "openclaw/plugin-sdk/provider-transport-runtime";
|
||||
import { describe, expect, it, vi } from "vitest";
|
||||
import { runSingleProviderCatalog } from "../test-support/provider-model-test-helpers.js";
|
||||
import xiaomiPlugin from "./index.js";
|
||||
import { createMiMoThinkingWrapper } from "./stream.js";
|
||||
|
||||
type OpenAICompletionsModel = Model<"openai-completions">;
|
||||
|
||||
type PayloadCapture = {
|
||||
payload?: Record<string, unknown>;
|
||||
};
|
||||
|
||||
type ThinkingPayload = {
|
||||
type?: unknown;
|
||||
};
|
||||
|
||||
type ReplayToolCall = {
|
||||
id?: unknown;
|
||||
type?: unknown;
|
||||
function?: {
|
||||
name?: unknown;
|
||||
arguments?: unknown;
|
||||
};
|
||||
};
|
||||
|
||||
type RegisteredProvider = RegisteredProviderCollections["providers"][number];
|
||||
type FakeStream = {
|
||||
result: () => Promise<unknown>;
|
||||
[Symbol.asyncIterator]: () => AsyncIterator<unknown>;
|
||||
};
|
||||
|
||||
const emptyUsage = {
|
||||
input: 0,
|
||||
output: 0,
|
||||
cacheRead: 0,
|
||||
cacheWrite: 0,
|
||||
totalTokens: 0,
|
||||
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
||||
};
|
||||
|
||||
function requireThinkingProfileResolver(
|
||||
provider: RegisteredProvider,
|
||||
): NonNullable<RegisteredProvider["resolveThinkingProfile"]> {
|
||||
if (!provider.resolveThinkingProfile) {
|
||||
throw new Error("Xiaomi provider did not register a thinking profile resolver");
|
||||
}
|
||||
return provider.resolveThinkingProfile;
|
||||
}
|
||||
|
||||
const readToolCall = { type: "toolCall", id: "call_1", name: "read", arguments: {} };
|
||||
const readToolResult = {
|
||||
role: "toolResult",
|
||||
toolCallId: "call_1",
|
||||
toolName: "read",
|
||||
content: [{ type: "text", text: "ok" }],
|
||||
isError: false,
|
||||
timestamp: 3,
|
||||
};
|
||||
const readTool = {
|
||||
name: "read",
|
||||
description: "Read data",
|
||||
parameters: { type: "object", properties: {}, required: [], additionalProperties: false },
|
||||
};
|
||||
|
||||
const registerXiaomiPlugin = () =>
|
||||
registerProviderPlugin({
|
||||
plugin: xiaomiPlugin,
|
||||
id: "xiaomi",
|
||||
name: "Xiaomi Provider",
|
||||
});
|
||||
|
||||
async function getXiaomiProvider() {
|
||||
const { providers } = await registerXiaomiPlugin();
|
||||
return requireRegisteredProvider(providers, "xiaomi");
|
||||
}
|
||||
|
||||
async function getXiaomiTokenPlanProvider() {
|
||||
const { providers } = await registerXiaomiPlugin();
|
||||
return requireRegisteredProvider(providers, "xiaomi-token-plan");
|
||||
}
|
||||
|
||||
function mimoReasoningModel(
|
||||
id: "mimo-v2-pro" | "mimo-v2-omni" | "mimo-v2.5" | "mimo-v2.5-pro" | "mimo-v2.6-pro",
|
||||
provider: "xiaomi" | "xiaomi-token-plan" = "xiaomi",
|
||||
): OpenAICompletionsModel {
|
||||
return {
|
||||
provider,
|
||||
id,
|
||||
name: id,
|
||||
api: "openai-completions",
|
||||
baseUrl: "https://api.xiaomimimo.com/v1",
|
||||
reasoning: true,
|
||||
input: ["text"],
|
||||
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
|
||||
contextWindow: 1_048_576,
|
||||
maxTokens: 32_000,
|
||||
compat: {},
|
||||
} as OpenAICompletionsModel;
|
||||
}
|
||||
|
||||
function replayAssistantMessage(params: {
|
||||
provider: string;
|
||||
model: string;
|
||||
content: Array<Record<string, unknown>>;
|
||||
stopReason: "stop" | "toolUse";
|
||||
}) {
|
||||
return {
|
||||
role: "assistant",
|
||||
api: "openai-completions",
|
||||
provider: params.provider,
|
||||
model: params.model,
|
||||
content: params.content,
|
||||
usage: emptyUsage,
|
||||
stopReason: params.stopReason,
|
||||
timestamp: 2,
|
||||
};
|
||||
}
|
||||
|
||||
function readToolReplayContext(assistantMessage: ReturnType<typeof replayAssistantMessage>) {
|
||||
return {
|
||||
messages: [{ role: "user", content: "hi", timestamp: 1 }, assistantMessage, readToolResult],
|
||||
tools: [readTool],
|
||||
} as Context;
|
||||
}
|
||||
|
||||
function mimoReasoningToolReplayContext(provider = "xiaomi") {
|
||||
return readToolReplayContext(
|
||||
replayAssistantMessage({
|
||||
provider,
|
||||
model: "mimo-v2.5-pro",
|
||||
content: [
|
||||
{
|
||||
type: "thinking",
|
||||
thinking: "call reasoning",
|
||||
thinkingSignature: "reasoning_content",
|
||||
},
|
||||
readToolCall,
|
||||
],
|
||||
stopReason: "toolUse",
|
||||
}),
|
||||
);
|
||||
}
|
||||
|
||||
function createPayloadCapturingStream(capture: PayloadCapture, model: OpenAICompletionsModel) {
|
||||
return (
|
||||
_streamModel: OpenAICompletionsModel,
|
||||
streamContext: Context,
|
||||
options?: { onPayload?: (payload: unknown, m: unknown) => unknown },
|
||||
) => {
|
||||
capture.payload = buildOpenAICompletionsParams(model, streamContext, {
|
||||
reasoning: "high",
|
||||
} as never);
|
||||
options?.onPayload?.(capture.payload, model);
|
||||
const stream = createAssistantMessageEventStream();
|
||||
queueMicrotask(() => stream.end());
|
||||
return stream;
|
||||
};
|
||||
}
|
||||
|
||||
function createFakeStream(params: { events: unknown[]; resultMessage: unknown }): FakeStream {
|
||||
return {
|
||||
async result() {
|
||||
return params.resultMessage;
|
||||
},
|
||||
[Symbol.asyncIterator]() {
|
||||
return (async function* () {
|
||||
for (const event of params.events) {
|
||||
yield event;
|
||||
}
|
||||
})();
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
function createResultStreamFn(params: { events?: unknown[]; resultMessage: unknown }): StreamFn {
|
||||
return () =>
|
||||
createFakeStream({
|
||||
events: params.events ?? [],
|
||||
resultMessage: params.resultMessage,
|
||||
}) as ReturnType<StreamFn>;
|
||||
}
|
||||
|
||||
function requireThinkingWrapper(
|
||||
wrapper: ReturnType<typeof createMiMoThinkingWrapper>,
|
||||
label: string,
|
||||
): NonNullable<ReturnType<typeof createMiMoThinkingWrapper>> {
|
||||
if (!wrapper) {
|
||||
throw new Error(`expected MiMo thinking wrapper for ${label}`);
|
||||
}
|
||||
return wrapper;
|
||||
}
|
||||
|
||||
function readThinking(payload: Record<string, unknown> | undefined): ThinkingPayload | undefined {
|
||||
return payload?.thinking as ThinkingPayload | undefined;
|
||||
}
|
||||
|
||||
function readPayloadMessage(
|
||||
capture: PayloadCapture,
|
||||
index: number,
|
||||
): Record<string, unknown> | undefined {
|
||||
return (capture.payload?.messages as Array<Record<string, unknown>> | undefined)?.[index];
|
||||
}
|
||||
|
||||
function readFirstToolCall(
|
||||
message: Record<string, unknown> | undefined,
|
||||
): ReplayToolCall | undefined {
|
||||
return (message?.tool_calls as ReplayToolCall[] | undefined)?.[0];
|
||||
}
|
||||
|
||||
describe("xiaomi provider plugin", () => {
|
||||
it("registers Xiaomi pay-as-you-go auth metadata", async () => {
|
||||
const { providers } = await registerXiaomiPlugin();
|
||||
const provider = requireRegisteredProvider(providers, "xiaomi");
|
||||
const resolved = resolveProviderPluginChoice({
|
||||
providers,
|
||||
choice: "xiaomi-api-key",
|
||||
});
|
||||
|
||||
expect(provider.id).toBe("xiaomi");
|
||||
expect(provider.label).toBe("Xiaomi");
|
||||
expect(provider.envVars).toEqual(["XIAOMI_API_KEY"]);
|
||||
expect(provider.auth).toHaveLength(1);
|
||||
expect(provider.auth[0]?.label).toBe("Xiaomi API key (Pay-as-you-go)");
|
||||
if (!resolved) {
|
||||
throw new Error("expected Xiaomi api-key auth choice");
|
||||
}
|
||||
expect(resolved.provider.id).toBe("xiaomi");
|
||||
expect(resolved.method.id).toBe("api-key");
|
||||
});
|
||||
|
||||
it("registers Xiaomi Token Plan regional auth metadata", async () => {
|
||||
const { providers } = await registerXiaomiPlugin();
|
||||
const provider = requireRegisteredProvider(providers, "xiaomi-token-plan");
|
||||
const resolved = resolveProviderPluginChoice({
|
||||
providers,
|
||||
choice: "xiaomi-token-plan-sgp",
|
||||
});
|
||||
|
||||
expect(provider.id).toBe("xiaomi-token-plan");
|
||||
expect(provider.label).toBe("Xiaomi Token Plan");
|
||||
expect(provider.envVars).toEqual(["XIAOMI_TOKEN_PLAN_API_KEY"]);
|
||||
expect(
|
||||
provider.auth.map((method) => ({
|
||||
id: method.id,
|
||||
label: method.label,
|
||||
hint: method.hint,
|
||||
choiceId: method.wizard?.choiceId,
|
||||
})),
|
||||
).toEqual([
|
||||
{
|
||||
id: "token-plan-ams",
|
||||
label: "Xiaomi Token Plan (Europe)",
|
||||
hint: "Endpoint preset: token-plan-ams.xiaomimimo.com/v1",
|
||||
choiceId: "xiaomi-token-plan-ams",
|
||||
},
|
||||
{
|
||||
id: "token-plan-cn",
|
||||
label: "Xiaomi Token Plan (China)",
|
||||
hint: "Endpoint preset: token-plan-cn.xiaomimimo.com/v1",
|
||||
choiceId: "xiaomi-token-plan-cn",
|
||||
},
|
||||
{
|
||||
id: "token-plan-sgp",
|
||||
label: "Xiaomi Token Plan (Singapore)",
|
||||
hint: "Endpoint preset: token-plan-sgp.xiaomimimo.com/v1",
|
||||
choiceId: "xiaomi-token-plan-sgp",
|
||||
},
|
||||
]);
|
||||
if (!resolved) {
|
||||
throw new Error("expected Xiaomi token-plan auth choice");
|
||||
}
|
||||
expect(resolved.provider.id).toBe("xiaomi-token-plan");
|
||||
expect(resolved.method.id).toBe("token-plan-sgp");
|
||||
});
|
||||
|
||||
it("builds the static Xiaomi model catalog with reasoning flags", async () => {
|
||||
const provider = await getXiaomiProvider();
|
||||
const catalogProvider = await runSingleProviderCatalog(provider);
|
||||
|
||||
expect(catalogProvider.api).toBe("openai-completions");
|
||||
expect(catalogProvider.baseUrl).toBe("https://api.xiaomimimo.com/v1");
|
||||
|
||||
const modelIds = catalogProvider.models?.map((m) => m.id);
|
||||
expect(modelIds).toContain("mimo-v2-pro");
|
||||
expect(modelIds).toContain("mimo-v2-omni");
|
||||
expect(modelIds).toContain("mimo-v2-flash");
|
||||
|
||||
expect(catalogProvider.models?.find((m) => m.id === "mimo-v2-pro")?.reasoning).toBe(true);
|
||||
expect(catalogProvider.models?.find((m) => m.id === "mimo-v2-omni")?.reasoning).toBe(true);
|
||||
expect(catalogProvider.models?.find((m) => m.id === "mimo-v2-flash")?.reasoning).toBeFalsy();
|
||||
});
|
||||
|
||||
it("exposes Token Plan v2.5 catalog rows only after a provider config selects a region", async () => {
|
||||
const provider = await getXiaomiTokenPlanProvider();
|
||||
|
||||
const missingConfig = await provider.catalog?.run({
|
||||
config: {},
|
||||
env: {},
|
||||
resolveProviderApiKey: () => ({ apiKey: "tp-test" }),
|
||||
resolveProviderAuth: () => ({
|
||||
apiKey: "tp-test",
|
||||
mode: "api_key",
|
||||
source: "env",
|
||||
}),
|
||||
} as never);
|
||||
expect(missingConfig).toBeNull();
|
||||
|
||||
const configured = await provider.catalog?.run({
|
||||
config: {
|
||||
models: {
|
||||
providers: {
|
||||
"xiaomi-token-plan": {
|
||||
baseUrl: "https://token-plan-cn.xiaomimimo.com/v1",
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
env: {},
|
||||
resolveProviderApiKey: () => ({ apiKey: "tp-test" }),
|
||||
resolveProviderAuth: () => ({
|
||||
apiKey: "tp-test",
|
||||
mode: "api_key",
|
||||
source: "profile",
|
||||
profileId: "xiaomi-token-plan:default",
|
||||
}),
|
||||
} as never);
|
||||
if (!configured || !("provider" in configured)) {
|
||||
throw new Error("expected configured Xiaomi Token Plan catalog");
|
||||
}
|
||||
expect(configured.provider.baseUrl).toBe("https://token-plan-cn.xiaomimimo.com/v1");
|
||||
expect(configured.provider.api).toBe("openai-completions");
|
||||
expect(configured.provider.models?.map((model) => model.id)).toEqual([
|
||||
"mimo-v2.5-pro",
|
||||
"mimo-v2.5",
|
||||
]);
|
||||
expect(configured.provider.models?.find((model) => model.id === "mimo-v2.5")?.input).toEqual([
|
||||
"text",
|
||||
"image",
|
||||
]);
|
||||
});
|
||||
|
||||
it("rejects token-plan keys on the pay-as-you-go auth choice", async () => {
|
||||
const provider = await getXiaomiProvider();
|
||||
const method = provider.auth[0];
|
||||
if (!method?.runNonInteractive) {
|
||||
throw new Error("expected Xiaomi pay-as-you-go non-interactive auth");
|
||||
}
|
||||
|
||||
await expect(
|
||||
method.runNonInteractive({
|
||||
authChoice: "xiaomi-api-key",
|
||||
config: {},
|
||||
baseConfig: {},
|
||||
opts: { xiaomiApiKey: "tp-test" },
|
||||
runtime: {} as never,
|
||||
resolveApiKey: async () => ({
|
||||
key: "tp-test",
|
||||
source: "flag",
|
||||
}),
|
||||
toApiKeyCredential: vi.fn(),
|
||||
} as never),
|
||||
).rejects.toThrow(
|
||||
"This looks like a Xiaomi MiMo Token Plan key (tp-...). " +
|
||||
"Re-run onboarding with one of: --auth-choice xiaomi-token-plan-cn, " +
|
||||
"--auth-choice xiaomi-token-plan-sgp, or --auth-choice xiaomi-token-plan-ams.",
|
||||
);
|
||||
});
|
||||
|
||||
it("rejects pay-as-you-go keys on Token Plan auth choices", async () => {
|
||||
const provider = await getXiaomiTokenPlanProvider();
|
||||
const method = provider.auth.find((entry) => entry.id === "token-plan-ams");
|
||||
if (!method?.runNonInteractive) {
|
||||
throw new Error("expected Xiaomi Token Plan non-interactive auth");
|
||||
}
|
||||
|
||||
await expect(
|
||||
method.runNonInteractive({
|
||||
authChoice: "xiaomi-token-plan-ams",
|
||||
config: {},
|
||||
baseConfig: {},
|
||||
opts: { xiaomiTokenPlanApiKey: "sk-test" },
|
||||
runtime: {} as never,
|
||||
resolveApiKey: async () => ({
|
||||
key: "sk-test",
|
||||
source: "flag",
|
||||
}),
|
||||
toApiKeyCredential: vi.fn(),
|
||||
} as never),
|
||||
).rejects.toThrow(
|
||||
"This looks like a Xiaomi MiMo pay-as-you-go key (sk-...). " +
|
||||
`Re-run onboarding with --auth-choice xiaomi-api-key or pass --xiaomi-api-key.`,
|
||||
);
|
||||
});
|
||||
|
||||
it("rejects keys that do not start with sk- on the pay-as-you-go auth choice", async () => {
|
||||
const provider = await getXiaomiProvider();
|
||||
const method = provider.auth[0];
|
||||
if (!method?.runNonInteractive) {
|
||||
throw new Error("expected Xiaomi pay-as-you-go non-interactive auth");
|
||||
}
|
||||
|
||||
await expect(
|
||||
method.runNonInteractive({
|
||||
authChoice: "xiaomi-api-key",
|
||||
config: {},
|
||||
baseConfig: {},
|
||||
opts: { xiaomiApiKey: "bad-key" },
|
||||
runtime: {} as never,
|
||||
resolveApiKey: async () => ({
|
||||
key: "bad-key",
|
||||
source: "flag",
|
||||
}),
|
||||
toApiKeyCredential: vi.fn(),
|
||||
} as never),
|
||||
).rejects.toThrow(
|
||||
'Xiaomi MiMo pay-as-you-go keys must start with "sk-". The entered key does not match the expected format.',
|
||||
);
|
||||
});
|
||||
|
||||
it("rejects keys that do not start with tp- on Token Plan auth choices", async () => {
|
||||
const provider = await getXiaomiTokenPlanProvider();
|
||||
const method = provider.auth.find((entry) => entry.id === "token-plan-ams");
|
||||
if (!method?.runNonInteractive) {
|
||||
throw new Error("expected Xiaomi Token Plan non-interactive auth");
|
||||
}
|
||||
|
||||
await expect(
|
||||
method.runNonInteractive({
|
||||
authChoice: "xiaomi-token-plan-ams",
|
||||
config: {},
|
||||
baseConfig: {},
|
||||
opts: { xiaomiTokenPlanApiKey: "bad-key" },
|
||||
runtime: {} as never,
|
||||
resolveApiKey: async () => ({
|
||||
key: "bad-key",
|
||||
source: "flag",
|
||||
}),
|
||||
toApiKeyCredential: vi.fn(),
|
||||
} as never),
|
||||
).rejects.toThrow(
|
||||
'Xiaomi MiMo Token Plan keys must start with "tp-". The entered key does not match the expected format.',
|
||||
);
|
||||
});
|
||||
|
||||
it("owns OpenAI-compatible replay policy", async () => {
|
||||
const provider = await getXiaomiProvider();
|
||||
|
||||
const replayPolicy = provider.buildReplayPolicy?.({ modelApi: "openai-completions" } as never);
|
||||
expect(replayPolicy?.sanitizeToolCallIds).toBe(true);
|
||||
expect(replayPolicy?.toolCallIdMode).toBe("strict");
|
||||
expect(replayPolicy?.validateGeminiTurns).toBe(true);
|
||||
expect(replayPolicy?.validateAnthropicTurns).toBe(true);
|
||||
});
|
||||
|
||||
it("marks resolved MiMo models for empty array items omission", async () => {
|
||||
const provider = await getXiaomiProvider();
|
||||
const model = mimoReasoningModel("mimo-v2.5");
|
||||
|
||||
const normalized = provider.normalizeResolvedModel?.({
|
||||
provider: "xiaomi",
|
||||
modelId: model.id,
|
||||
modelApi: model.api,
|
||||
model,
|
||||
} as never);
|
||||
|
||||
expect(
|
||||
(normalized?.compat as { omitEmptyArrayItems?: unknown } | undefined)?.omitEmptyArrayItems,
|
||||
).toBe(true);
|
||||
});
|
||||
|
||||
it("advertises thinking profiles for MiMo reasoning models only", async () => {
|
||||
const provider = await getXiaomiProvider();
|
||||
const resolveThinkingProfile = requireThinkingProfileResolver(provider);
|
||||
const expectedLevels = ["off", "minimal", "low", "medium", "high", "xhigh", "max"];
|
||||
|
||||
for (const modelId of [
|
||||
"mimo-v2-pro",
|
||||
"mimo-v2-omni",
|
||||
"mimo-v2.5",
|
||||
"mimo-v2.5-pro",
|
||||
"mimo-v2.6-pro",
|
||||
]) {
|
||||
const profile = resolveThinkingProfile({ provider: "xiaomi", modelId } as never);
|
||||
expect(profile?.levels.map((l) => l.id)).toEqual(expectedLevels);
|
||||
expect(profile?.defaultLevel).toBe("high");
|
||||
}
|
||||
|
||||
expect(resolveThinkingProfile({ provider: "xiaomi", modelId: "mimo-v2-flash" } as never)).toBe(
|
||||
undefined,
|
||||
);
|
||||
});
|
||||
|
||||
it("isModernModelRef returns true only for MiMo reasoning models", async () => {
|
||||
const provider = await getXiaomiProvider();
|
||||
|
||||
expect(
|
||||
provider.isModernModelRef?.({ provider: "xiaomi", modelId: "mimo-v2.5-pro" } as never),
|
||||
).toBe(true);
|
||||
expect(
|
||||
provider.isModernModelRef?.({ provider: "xiaomi", modelId: "mimo-v2.6-pro" } as never),
|
||||
).toBe(true);
|
||||
expect(
|
||||
provider.isModernModelRef?.({ provider: "xiaomi", modelId: "mimo-v2-pro" } as never),
|
||||
).toBe(true);
|
||||
expect(
|
||||
provider.isModernModelRef?.({ provider: "xiaomi", modelId: "mimo-v2-flash" } as never),
|
||||
).toBe(false);
|
||||
});
|
||||
|
||||
it("adds blank reasoning_content for replayed tool calls from non-xiaomi turns", async () => {
|
||||
const capture: PayloadCapture = {};
|
||||
const model = mimoReasoningModel("mimo-v2.5-pro");
|
||||
const context = readToolReplayContext(
|
||||
replayAssistantMessage({
|
||||
provider: "openai",
|
||||
model: "gpt-5.5",
|
||||
content: [readToolCall],
|
||||
stopReason: "toolUse",
|
||||
}),
|
||||
);
|
||||
const baseStreamFn = createPayloadCapturingStream(capture, model);
|
||||
|
||||
const wrapThinkingHigh = requireThinkingWrapper(
|
||||
createMiMoThinkingWrapper(baseStreamFn as never, "high"),
|
||||
"high",
|
||||
);
|
||||
await wrapThinkingHigh(model, context, {});
|
||||
|
||||
const assistantMessage = readPayloadMessage(capture, 1);
|
||||
expect(assistantMessage?.role).toBe("assistant");
|
||||
expect(assistantMessage?.reasoning_content).toBe("");
|
||||
const toolCall = readFirstToolCall(assistantMessage);
|
||||
expect(toolCall?.id).toBe("call_1");
|
||||
expect(toolCall?.type).toBe("function");
|
||||
expect(toolCall?.function?.name).toBe("read");
|
||||
expect(toolCall?.function?.arguments).toBe("{}");
|
||||
});
|
||||
|
||||
it("preserves replayed reasoning_content when MiMo thinking is enabled", async () => {
|
||||
const capture: PayloadCapture = {};
|
||||
const model = mimoReasoningModel("mimo-v2.5-pro", "xiaomi-token-plan");
|
||||
const context = mimoReasoningToolReplayContext("xiaomi-token-plan");
|
||||
const baseStreamFn = createPayloadCapturingStream(capture, model);
|
||||
|
||||
const wrapThinkingHigh = requireThinkingWrapper(
|
||||
createMiMoThinkingWrapper(baseStreamFn as never, "high"),
|
||||
"high",
|
||||
);
|
||||
await wrapThinkingHigh(model, context, {});
|
||||
|
||||
expect(readThinking(capture.payload)?.type).toBe("enabled");
|
||||
const assistantMessage = readPayloadMessage(capture, 1);
|
||||
expect(assistantMessage?.role).toBe("assistant");
|
||||
expect(assistantMessage?.reasoning_content).toBe("call reasoning");
|
||||
const toolCall = readFirstToolCall(assistantMessage);
|
||||
expect(toolCall?.id).toBe("call_1");
|
||||
expect(toolCall?.type).toBe("function");
|
||||
expect(toolCall?.function?.name).toBe("read");
|
||||
});
|
||||
|
||||
it("strips reasoning_content when MiMo thinking is disabled", async () => {
|
||||
const capture: PayloadCapture = {};
|
||||
const model = mimoReasoningModel("mimo-v2-pro");
|
||||
const context = mimoReasoningToolReplayContext();
|
||||
const baseStreamFn = createPayloadCapturingStream(capture, model);
|
||||
|
||||
const wrapThinkingNone = requireThinkingWrapper(
|
||||
createMiMoThinkingWrapper(baseStreamFn as never, "none" as never),
|
||||
"none",
|
||||
);
|
||||
await wrapThinkingNone(model, context, {});
|
||||
|
||||
expect(readThinking(capture.payload)?.type).toBe("disabled");
|
||||
expect((capture.payload!.messages as Array<Record<string, unknown>>)[1]).not.toHaveProperty(
|
||||
"reasoning_content",
|
||||
);
|
||||
});
|
||||
|
||||
it.each(["mimo-v2-pro", "mimo-v2-omni"] as const)(
|
||||
"promotes reasoning-only terminal output to visible text for %s",
|
||||
async (modelId) => {
|
||||
const model = mimoReasoningModel(modelId);
|
||||
const wrapped = requireThinkingWrapper(
|
||||
createMiMoThinkingWrapper(
|
||||
createResultStreamFn({
|
||||
events: [
|
||||
{
|
||||
type: "message_end",
|
||||
message: {
|
||||
role: "assistant",
|
||||
content: [{ type: "thinking", thinking: "MiMo final answer" }],
|
||||
stopReason: "stop",
|
||||
},
|
||||
},
|
||||
],
|
||||
resultMessage: {
|
||||
role: "assistant",
|
||||
content: [{ type: "thinking", thinking: "MiMo final answer" }],
|
||||
stopReason: "stop",
|
||||
},
|
||||
}),
|
||||
"high",
|
||||
),
|
||||
modelId,
|
||||
);
|
||||
|
||||
const stream = (await wrapped(model, { messages: [] } as Context, {})) as FakeStream;
|
||||
const events: unknown[] = [];
|
||||
for await (const event of stream) {
|
||||
events.push(event);
|
||||
}
|
||||
|
||||
expect(events).toEqual([
|
||||
{
|
||||
type: "message_end",
|
||||
message: {
|
||||
role: "assistant",
|
||||
content: [{ type: "text", text: "MiMo final answer" }],
|
||||
stopReason: "stop",
|
||||
},
|
||||
},
|
||||
]);
|
||||
await expect(stream.result()).resolves.toEqual({
|
||||
role: "assistant",
|
||||
content: [{ type: "text", text: "MiMo final answer" }],
|
||||
stopReason: "stop",
|
||||
});
|
||||
},
|
||||
);
|
||||
|
||||
it("does not promote reasoning when the MiMo assistant turn also has text or tool calls", async () => {
|
||||
const model = mimoReasoningModel("mimo-v2-pro");
|
||||
const textMessage = {
|
||||
role: "assistant",
|
||||
content: [
|
||||
{ type: "thinking", thinking: "internal" },
|
||||
{ type: "text", text: "already visible" },
|
||||
],
|
||||
stopReason: "stop",
|
||||
};
|
||||
const toolMessage = {
|
||||
role: "assistant",
|
||||
content: [{ type: "thinking", thinking: "call reasoning" }, readToolCall],
|
||||
stopReason: "toolUse",
|
||||
};
|
||||
|
||||
for (const resultMessage of [textMessage, toolMessage]) {
|
||||
const wrapped = requireThinkingWrapper(
|
||||
createMiMoThinkingWrapper(createResultStreamFn({ resultMessage }), "high"),
|
||||
"mixed-content",
|
||||
);
|
||||
const stream = (await wrapped(model, { messages: [] } as Context, {})) as FakeStream;
|
||||
|
||||
await expect(stream.result()).resolves.toEqual(resultMessage);
|
||||
}
|
||||
});
|
||||
|
||||
it("does not promote reasoning-only output for newer MiMo replay models", async () => {
|
||||
const model = mimoReasoningModel("mimo-v2.5-pro");
|
||||
const resultMessage = {
|
||||
role: "assistant",
|
||||
content: [{ type: "thinking", thinking: "actual reasoning" }],
|
||||
stopReason: "stop",
|
||||
};
|
||||
const wrapped = requireThinkingWrapper(
|
||||
createMiMoThinkingWrapper(createResultStreamFn({ resultMessage }), "high"),
|
||||
"mimo-v2.5-pro",
|
||||
);
|
||||
const stream = (await wrapped(model, { messages: [] } as Context, {})) as FakeStream;
|
||||
|
||||
await expect(stream.result()).resolves.toEqual(resultMessage);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user