wehub-resource-sync
112 lines
3.9 KiB
TypeScript
112 lines
3.9 KiB
TypeScript
import { getModelSpec } from "../../src/shared/constants/modelSpecs.ts";
|
|
import test from "node:test";
|
|
import assert from "node:assert/strict";
|
|
|
|
import { REGISTRY } from "../../open-sse/config/providerRegistry.ts";
|
|
import { getAllAudioModels, getSpeechProvider } from "../../open-sse/config/audioRegistry.ts";
|
|
import { DefaultExecutor } from "../../open-sse/executors/default.ts";
|
|
import {
|
|
createProviderSchema,
|
|
updateProviderConnectionSchema,
|
|
} from "../../src/shared/validation/schemas.ts";
|
|
import { validateBody } from "../../src/shared/validation/helpers.ts";
|
|
|
|
const DEPRECATED_MIMO_V2_MODELS = ["mimo-v2-pro", "mimo-v2-omni", "mimo-v2-flash", "mimo-v2-tts"];
|
|
|
|
test("xiaomi-mimo registry uses the current default base URL and MiMo V2.5 models", () => {
|
|
const entry = REGISTRY["xiaomi-mimo"];
|
|
|
|
assert.ok(entry, "xiaomi-mimo should exist in registry");
|
|
assert.equal(entry.baseUrl, "https://api.xiaomimimo.com/v1");
|
|
for (const modelId of DEPRECATED_MIMO_V2_MODELS) {
|
|
assert.ok(!entry.models.some((model) => model.id === modelId), `${modelId} is deprecated`);
|
|
}
|
|
assert.deepEqual(
|
|
entry.models.map((model) => model.id),
|
|
["mimo-v2.5-pro", "mimo-v2.5"]
|
|
);
|
|
});
|
|
|
|
test("xiaomi-mimo TTS models are registered in the audio speech registry", () => {
|
|
const provider = getSpeechProvider("xiaomi-mimo");
|
|
|
|
assert.ok(provider, "xiaomi-mimo should exist in speech registry");
|
|
assert.equal(provider.id, "xiaomi-mimo");
|
|
assert.equal(provider.authType, "apikey");
|
|
assert.equal(provider.authHeader, "bearer");
|
|
assert.equal(provider.format, "xiaomi-mimo-tts");
|
|
assert.deepEqual(
|
|
provider.models.map((model) => model.id),
|
|
["mimo-v2.5-tts", "mimo-v2.5-tts-voicedesign", "mimo-v2.5-tts-voiceclone"]
|
|
);
|
|
assert.ok(
|
|
getAllAudioModels().some(
|
|
(model) => model.id === "xiaomi-mimo/mimo-v2.5-tts" && model.subtype === "speech"
|
|
)
|
|
);
|
|
});
|
|
|
|
test("xiaomi-mimo executor appends /chat/completions for regional base URLs", () => {
|
|
const executor = new DefaultExecutor("xiaomi-mimo");
|
|
|
|
assert.equal(
|
|
executor.buildUrl("mimo-v2.5-pro", true, 0, {
|
|
providerSpecificData: {
|
|
baseUrl: "https://token-plan-sgp.xiaomimimo.com/v1",
|
|
},
|
|
}),
|
|
"https://token-plan-sgp.xiaomimimo.com/v1/chat/completions"
|
|
);
|
|
|
|
assert.equal(
|
|
executor.buildUrl("mimo-v2.5-pro", true, 0, {
|
|
providerSpecificData: {
|
|
baseUrl: "https://token-plan-cn.xiaomimimo.com/v1/chat/completions",
|
|
},
|
|
}),
|
|
"https://token-plan-cn.xiaomimimo.com/v1/chat/completions"
|
|
);
|
|
});
|
|
|
|
test("xiaomi-mimo create schema accepts custom regional baseUrl", () => {
|
|
const validation = validateBody(createProviderSchema, {
|
|
provider: "xiaomi-mimo",
|
|
apiKey: "xm-placeholder-key",
|
|
name: "Xiaomi MiMo SGP",
|
|
providerSpecificData: {
|
|
baseUrl: "https://token-plan-sgp.xiaomimimo.com/v1",
|
|
},
|
|
});
|
|
|
|
assert.equal(validation.success, true, "create schema should accept Xiaomi regional baseUrl");
|
|
if (validation.success) {
|
|
assert.equal(
|
|
validation.data.providerSpecificData?.baseUrl,
|
|
"https://token-plan-sgp.xiaomimimo.com/v1"
|
|
);
|
|
}
|
|
});
|
|
|
|
test("xiaomi-mimo update schema accepts custom regional baseUrl", () => {
|
|
const validation = validateBody(updateProviderConnectionSchema, {
|
|
providerSpecificData: {
|
|
baseUrl: "https://token-plan-cn.xiaomimimo.com/v1",
|
|
},
|
|
});
|
|
|
|
assert.equal(validation.success, true, "update schema should accept Xiaomi regional baseUrl");
|
|
if (validation.success) {
|
|
assert.equal(
|
|
validation.data.providerSpecificData?.baseUrl,
|
|
"https://token-plan-cn.xiaomimimo.com/v1"
|
|
);
|
|
}
|
|
});
|
|
|
|
test("registered Xiaomi MiMo V2.5 chat models keep capability overrides", () => {
|
|
// models.dev mislabels the *-pro models (hermes-agent#18884); see the hard override in
|
|
// src/lib/modelCapabilities.ts.
|
|
assert.equal(getModelSpec("mimo-v2.5")?.supportsVision, true);
|
|
assert.equal(getModelSpec("mimo-v2.5-pro")?.supportsVision, false);
|
|
});
|