86694ed8d0
Add gemini-3.7-flash and its tiered high/medium/low variants to the Antigravity and Gemini registries, with matching capabilities, pricing and Antigravity quota tracking. extractModel now recognises gemini-3.7-flash-tiered alongside 3.6 and derives the version from the request, so thinkingLevel still maps to the right tiered alias. Closes #3286 Closes #3281
61 lines
2 KiB
JavaScript
61 lines
2 KiB
JavaScript
import { describe, expect, it, vi, beforeEach } from "vitest";
|
|
|
|
const proxyAwareFetch = vi.fn(async (url) => ({
|
|
ok: true,
|
|
status: 200,
|
|
json: async () => url.includes(":loadCodeAssist")
|
|
? { cloudaicompanionProject: "project-1", currentTier: { name: "Pro" } }
|
|
: {
|
|
models: {
|
|
"gemini-3.7-flash-high": {
|
|
displayName: "Gemini 3.7 Flash (High)",
|
|
quotaInfo: { remainingFraction: 0.85, resetTime: "2026-08-25T12:00:00Z" },
|
|
},
|
|
"gemini-3.7-flash-medium": {
|
|
displayName: "Gemini 3.7 Flash (Medium)",
|
|
quotaInfo: { remainingFraction: 0.6, resetTime: "2026-08-25T12:00:00Z" },
|
|
},
|
|
"gemini-3.7-flash-low": {
|
|
displayName: "Gemini 3.7 Flash (Low)",
|
|
quotaInfo: { remainingFraction: 0.35, resetTime: "2026-08-25T12:00:00Z" },
|
|
},
|
|
"internal-model": {
|
|
displayName: "Internal",
|
|
isInternal: true,
|
|
quotaInfo: { remainingFraction: 0.5 },
|
|
},
|
|
},
|
|
},
|
|
text: async () => "{}",
|
|
}));
|
|
|
|
vi.mock("../../open-sse/utils/proxyFetch.js", () => ({
|
|
proxyAwareFetch,
|
|
}));
|
|
|
|
describe("Antigravity quota tracker: Gemini 3.7 Flash usage bars", () => {
|
|
beforeEach(() => proxyAwareFetch.mockClear());
|
|
|
|
it("returns Gemini 3.7 Flash tier quotas so the dashboard can render usage bars", async () => {
|
|
const { getAntigravityUsage } = await import("../../open-sse/services/usage/google.js");
|
|
|
|
const usage = await getAntigravityUsage("access-token", {});
|
|
|
|
expect(usage.quotas["gemini-3.7-flash-high"]).toMatchObject({
|
|
used: 150,
|
|
total: 1000,
|
|
remainingPercentage: 85,
|
|
displayName: "Gemini 3.7 Flash (High)",
|
|
});
|
|
expect(usage.quotas["gemini-3.7-flash-medium"]).toMatchObject({
|
|
used: 400,
|
|
total: 1000,
|
|
remainingPercentage: 60,
|
|
});
|
|
expect(usage.quotas["gemini-3.7-flash-low"]).toMatchObject({
|
|
used: 650,
|
|
total: 1000,
|
|
remainingPercentage: 35,
|
|
});
|
|
});
|
|
});
|