Compare commits

...

2 Commits

Author SHA1 Message Date
Probe Test
006c2ca6fb test(model-alias-seed): expect canonical antigravity provider for the agy alias
Same #8013 realignment as this PR: `getModelInfo` resolves the stored `agy/…` alias
target to its canonical provider id `antigravity` (ALIAS_TO_PROVIDER_ID). The stored
alias STRING stays `agy/gemini-pro-agent`; only the resolved `provider` is now
`antigravity`, not `agy`. Test updated to match. 6/0.
2026-07-23 04:02:18 -03:00
Probe Test
59152050e3 test(antigravity): align catalog tests with the #8013/#8123 model realignment
Base-red slice 3. Two test files referenced antigravity model IDs that
#8013 (align official clients / callable catalog) and #8123 (discover live AGY
models) intentionally renamed/retired — the candidate builder and static catalog
are correct; the tests were stale.

- auto-combo-credentialed-model-pool: claude-sonnet-5 -> claude-sonnet-4-6 and the
  gemini-3.5-flash-{low,medium,high} tier -> gemini-3.6-flash-{low,medium,high}
  (verified against the live createVirtualAutoCombo candidate set); the exclusion
  wildcard and per-account transparency assertions are unchanged in intent.
- T31 static-catalog: gemini-3-pro-preview was retired by #8013, so assert the
  current client-visible top flash tier (gemini-3.6-flash-high) plus its absence.

Test-only. Validated: auto-combo-credentialed-model-pool 4/4, the T31/T33/T34/T38
model-specs file 8/8.
2026-07-23 03:41:23 -03:00
3 changed files with 24 additions and 17 deletions

View File

@@ -86,10 +86,10 @@ test("credentialed providers expose one logical candidate per visible registry m
"provider/model candidates must not be duplicated per connection"
);
for (const model of [
"antigravity/claude-sonnet-5",
"antigravity/gemini-3.5-flash-low",
"antigravity/gemini-3.5-flash-medium",
"antigravity/gemini-3.5-flash-high",
"antigravity/claude-sonnet-4-6",
"antigravity/gemini-3.6-flash-low",
"antigravity/gemini-3.6-flash-medium",
"antigravity/gemini-3.6-flash-high",
]) {
assert.ok(modelStrings.includes(model), `${model} should be eligible for auto routing`);
}
@@ -110,7 +110,7 @@ test("candidate transparency expands a logical model into per-account rows", asy
const result = await candidateHandler.getAutoComboCandidates("auto", null);
const sonnetRows = result.candidates.filter(
(candidate) =>
candidate.provider === "antigravity" && candidate.model === "antigravity/claude-sonnet-5"
candidate.provider === "antigravity" && candidate.model === "antigravity/claude-sonnet-4-6"
);
assert.deepEqual(
@@ -121,12 +121,12 @@ test("candidate transparency expands a logical model into per-account rows", asy
});
test("connection model exclusions narrow only that model's account allowlist", async () => {
const { first, second } = await seedConnections(["gemini-3.5-*"]);
const { first, second } = await seedConnections(["gemini-3.6-*"]);
const combo = await virtualFactory.createVirtualAutoCombo(undefined);
const candidates = antigravityCandidates(combo);
const geminiCandidates = candidates.filter((candidate) =>
candidate.model.startsWith("antigravity/gemini-3.5-")
candidate.model.startsWith("antigravity/gemini-3.6-")
);
assert.ok(geminiCandidates.length >= 3, "Gemini 3.5 candidates should remain available");
@@ -134,23 +134,25 @@ test("connection model exclusions narrow only that model's account allowlist", a
assert.deepEqual(candidate.allowedConnectionIds, [second.id]);
}
const sonnet = candidates.find((candidate) => candidate.model === "antigravity/claude-sonnet-5");
const sonnet = candidates.find(
(candidate) => candidate.model === "antigravity/claude-sonnet-4-6"
);
assert.ok(sonnet, "Claude Sonnet 5 should remain in the candidate pool");
assert.deepEqual([...(sonnet.allowedConnectionIds ?? [])].sort(), [first.id, second.id].sort());
});
test("hiding the first registry model does not drop the credentialed provider", async () => {
await seedConnections();
modelsDb.setModelIsHidden("antigravity", "claude-sonnet-5", true);
modelsDb.setModelIsHidden("antigravity", "claude-sonnet-4-6", true);
const combo = await virtualFactory.createVirtualAutoCombo(undefined);
const modelStrings = antigravityCandidates(combo).map((candidate) => candidate.model);
assert.equal(modelStrings.includes("antigravity/claude-sonnet-5"), false);
assert.equal(modelStrings.includes("antigravity/claude-sonnet-4-6"), false);
for (const model of [
"antigravity/gemini-3.5-flash-low",
"antigravity/gemini-3.5-flash-medium",
"antigravity/gemini-3.5-flash-high",
"antigravity/gemini-3.6-flash-low",
"antigravity/gemini-3.6-flash-medium",
"antigravity/gemini-3.6-flash-high",
]) {
assert.ok(modelStrings.includes(model), `${model} should remain after Sonnet is hidden`);
}

View File

@@ -42,8 +42,10 @@ test("default model alias seed writes missing aliases and is idempotent", async
assert.equal(aliases["gemini-3-flash-preview"], undefined);
const routed = await sseModelService.getModelInfo("gemini-3.1-pro");
// The stored alias target is "agy/gemini-pro-agent", but getModelInfo canonicalizes
// the "agy" alias to its provider id "antigravity" (ALIAS_TO_PROVIDER_ID, #8013).
assert.deepEqual(routed, {
provider: "agy",
provider: "antigravity",
model: "gemini-pro-agent",
extendedContext: false,
});

View File

@@ -16,10 +16,13 @@ const {
} = await import("../../src/shared/constants/modelSpecs.ts");
test("T31: antigravity static catalog exposes client-visible Gemini preview IDs", () => {
// Antigravity exposes preview aliases to clients even though the upstream
// still accepts its internal model identifiers.
// Antigravity exposes client-visible Gemini IDs even though the upstream still
// accepts its internal model identifiers. #8013 (align official clients / callable
// catalog) retired the `gemini-3-pro-preview` alias, so assert the current
// client-visible top flash tier instead.
const staticIds = (getStaticModelsForProvider("antigravity") || []).map((m) => m.id);
assert.ok(staticIds.includes("gemini-3-pro-preview"));
assert.ok(staticIds.includes("gemini-3.6-flash-high"));
assert.ok(!staticIds.includes("gemini-3-pro-preview"));
// #3303 (agy parity, discussion #3184): the Gemini + Claude budget tiers ARE
// client-visible on the Antigravity OAuth backend (Claude was never removed).
assert.ok(staticIds.includes("gemini-3.1-pro-low"));