Files
OmniRoute/tests/unit/antigravity-image-catalog-separation.test.ts
backryun 888c872459 refactor(antigravity): align official clients and callable catalog (#8013)
* fix(antigravity): preserve protocol fidelity and fail closed

* chore: add PR-numbered changelog fragment

* test: split oversized Antigravity suites

* refactor(antigravity): align official IDE and CLI identities

* fix(antigravity): align catalog with callable models

* test(antigravity): update 2 test files to renamed version-cache API (#8013 fix)

Co-authored-by: diegosouzapw <8016841+diegosouzapw@users.noreply.github.com>

---------

Co-authored-by: nguyenha935 <208228297+nguyenha935@users.noreply.github.com>
Co-authored-by: backryun <backryun@users.noreply.github.com>
Co-authored-by: diegosouzapw <8016841+diegosouzapw@users.noreply.github.com>
Co-authored-by: nguyenha935 <nguyenha935@users.noreply.github.com>
Co-authored-by: Probe Test <probe@example.com>
2026-07-22 20:41:19 -03:00

36 lines
1.5 KiB
TypeScript

import assert from "node:assert/strict";
import test from "node:test";
import {
ANTIGRAVITY_PUBLIC_MODELS,
isUserCallableAntigravityModelId,
} from "../../open-sse/config/antigravityModelAliases.ts";
import { AGY_PUBLIC_MODELS, isUserCallableAgyModelId } from "../../open-sse/config/agyModels.ts";
import { FREE_MODEL_BUDGETS } from "../../open-sse/config/freeModelCatalog.data.ts";
import { getImageProvider } from "../../open-sse/config/imageRegistry.ts";
test("Antigravity and AGY chat catalogs exclude image-generation-only models", () => {
const chatModelIds = new Set(ANTIGRAVITY_PUBLIC_MODELS.map((model) => model.id));
const agyChatModelIds = new Set(AGY_PUBLIC_MODELS.map((model) => model.id));
assert.equal(chatModelIds.has("gemini-3.1-flash-image"), false);
assert.equal(chatModelIds.has("gemini-3-pro-image-preview"), false);
assert.equal(isUserCallableAntigravityModelId("gemini-3.1-flash-image"), false);
assert.equal(isUserCallableAntigravityModelId("gemini-3-pro-image-preview"), false);
assert.equal(agyChatModelIds.has("gemini-3.1-flash-image"), false);
assert.equal(isUserCallableAgyModelId("gemini-3.1-flash-image"), false);
assert.equal(
FREE_MODEL_BUDGETS.some(
(model) => model.provider === "agy" && model.modelId === "gemini-3.1-flash-image"
),
false
);
const imageProvider = getImageProvider("antigravity");
assert.ok(imageProvider);
assert.equal(
imageProvider.models.some((model) => model.id === "gemini-3.1-flash-image"),
true
);
});