singularity/singularity-forge
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix: constrain mimo proxy fallbacks

Browse source
This commit is contained in:
Mikael Hugo 2026-05-05 17:18:56 +02:00
parent c4ee341852
commit 1e8a05dc70
2 changed files with 37 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
packages/pi-coding-agent/src/core/model-registry-proxy-routing.test.ts
View file

@ -253,6 +253,22 @@ describe("ModelRegistry.getModelsForProxy — basic", () => {
);
});
it("routes MiMo proxy candidates through direct Xiaomi and subscribed/free relays, never OpenRouter", () => {
const registry = createRegistry(() => true);
registerNone(registry, "openrouter", "mimo-v2-pro");
registerNone(registry, "ollama-cloud", "mimo-v2-pro");
const result = registry.getModelsForProxy("mimo-v2-pro");
assert.deepEqual(
result.map((m) => `${m.provider}/${m.id}`),
[
"xiaomi/mimo-v2-pro",
"opencode-go/mimo-v2-pro",
"ollama-cloud/mimo-v2-pro",
],
);
});
it("hides Grok models even when they arrive through aggregators", () => {
const registry = createRegistry(() => true);
const available = registry.getAvailable();
@ -439,9 +455,9 @@ describe("ModelRegistry.getModelsForProxy — family priority ordering", () => {
it("provider not in any family rule falls back to end of list", () => {
const registry = createRegistry();
registerNone(registry, "zai", "glm-4-air");
registerNone(registry, "unknown-aggregator", "glm-4-air");
const result = registry.getModelsForProxy("glm-4-air");
registerNone(registry, "zai", "unknown-model");
registerNone(registry, "unknown-aggregator", "unknown-model");
const result = registry.getModelsForProxy("unknown-model");
const providers = result.map((m) => m.provider);
assert.equal(providers[0], "zai");
assert.equal(providers[providers.length - 1], "unknown-aggregator");

21
packages/pi-coding-agent/src/core/model-registry.ts
View file

@ -87,7 +87,13 @@ export const PROXY_FAMILY_PRIORITY: ReadonlyArray<{
// MiMo/Xiaomi — direct API via Xiaomi MiMo Open Platform (api.xiaomimimo.com)
// or the Token Plan endpoint (token-plan-sgp.xiaomimimo.com). Both served
// under the `xiaomi` provider namespace.
{ match: /^mimo-|^XiaomiMiMo\//i, prefix: "mimo-", providers: ["xiaomi"] },
{
match: /^mimo-|^xiaomi\/mimo-|^XiaomiMiMo\//i,
prefix: "mimo-",
providers: ["xiaomi"],
family_failover: ["opencode-go", "ollama-cloud"],
global_fallback: false,
},
// Gemini/Gemma: route bare model IDs through google-gemini-cli only.
// Direct GenAI and Vertex providers stay explicit provider-qualified routes,
// but they are hidden from normal SF/TUI selection and default fallback.
@ -1183,13 +1189,22 @@ export class ModelRegistry {
overrides: Record<string, string[]> = {},
providerModelAllow?: ProviderModelAllowList,
): Model<Api>[] {
const order = this.buildCandidateOrder(modelId, overrides);
const hasScopedProviderOrder =
Object.entries(overrides).some(([k]) => modelId.startsWith(k)) ||
PROXY_FAMILY_PRIORITY.some((r) => r.match.test(modelId));
const allowedProviders =
hasScopedProviderOrder && order.length > 0 ? new Set(order) : undefined;
const candidates = this.filterProviderModelAllow(
this.models.filter((m) => m.id === modelId),
this.getAllWithDiscovered().filter(
(m) =>
m.id === modelId &&
(!allowedProviders || allowedProviders.has(m.provider)),
),
providerModelAllow,
);
if (candidates.length === 0) return [];
const order = this.buildCandidateOrder(modelId, overrides);
const sorted = [...candidates].sort((a, b) => {
const pa = order.indexOf(a.provider);
const pb = order.indexOf(b.provider);