feat(antigravity): show Gemini 3.6 Flash usage bars in quota tracker

getAntigravityUsage filtered the fetchAvailableModels response through a
hardcoded importantModels list that only contained 3.5 Flash, silently
dropping the 3.6 Flash quota buckets so no usage bar rendered.
This commit is contained in:
whale9820
2026-08-05 13:19:49 +07:00
parent a7941ddab4
commit 42c691b3ea
2 changed files with 82 additions and 0 deletions

View File

@@ -161,6 +161,9 @@ export async function getAntigravityUsage(accessToken, providerSpecificData, pro
if (data.models) {
// Filter only recommended/important models (must match PROVIDER_MODELS ag ids)
const importantModels = [
'gemini-3.6-flash-high',
'gemini-3.6-flash-medium',
'gemini-3.6-flash-low',
'gemini-3-flash-agent',
'gemini-3.5-flash-low',
'gemini-3.5-flash-extra-low',

View File

@@ -0,0 +1,79 @@
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.6-flash-high": {
displayName: "Gemini 3.6 Flash (High)",
quotaInfo: { remainingFraction: 0.8, resetTime: "2026-07-25T12:00:00Z" },
},
"gemini-3.6-flash-medium": {
displayName: "Gemini 3.6 Flash (Medium)",
quotaInfo: { remainingFraction: 0.5, resetTime: "2026-07-25T12:00:00Z" },
},
"gemini-3.6-flash-low": {
displayName: "Gemini 3.6 Flash (Low)",
quotaInfo: { remainingFraction: 0.2, resetTime: "2026-07-25T12:00:00Z" },
},
"gemini-3.5-flash-low": {
displayName: "Gemini 3.5 Flash (Medium)",
quotaInfo: { remainingFraction: 0.9, resetTime: "2026-07-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.6 Flash usage bars", () => {
beforeEach(() => proxyAwareFetch.mockClear());
it("returns Gemini 3.6 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.6-flash-high"]).toMatchObject({
used: 200,
total: 1000,
remainingPercentage: 80,
displayName: "Gemini 3.6 Flash (High)",
});
expect(usage.quotas["gemini-3.6-flash-medium"]).toMatchObject({
used: 500,
total: 1000,
remainingPercentage: 50,
});
expect(usage.quotas["gemini-3.6-flash-low"]).toMatchObject({
used: 800,
total: 1000,
remainingPercentage: 20,
});
});
it("still filters out internal and non-important models", async () => {
const { getAntigravityUsage } = await import("../../open-sse/services/usage/google.js");
const usage = await getAntigravityUsage("access-token", {});
expect(usage.quotas).not.toHaveProperty("internal-model");
expect(Object.keys(usage.quotas)).toEqual([
"gemini-3.6-flash-high",
"gemini-3.6-flash-medium",
"gemini-3.6-flash-low",
"gemini-3.5-flash-low",
]);
});
});