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:
parent
a7941ddab4
commit
42c691b3ea
2 changed files with 82 additions and 0 deletions
|
|
@ -161,6 +161,9 @@ export async function getAntigravityUsage(accessToken, providerSpecificData, pro
|
||||||
if (data.models) {
|
if (data.models) {
|
||||||
// Filter only recommended/important models (must match PROVIDER_MODELS ag ids)
|
// Filter only recommended/important models (must match PROVIDER_MODELS ag ids)
|
||||||
const importantModels = [
|
const importantModels = [
|
||||||
|
'gemini-3.6-flash-high',
|
||||||
|
'gemini-3.6-flash-medium',
|
||||||
|
'gemini-3.6-flash-low',
|
||||||
'gemini-3-flash-agent',
|
'gemini-3-flash-agent',
|
||||||
'gemini-3.5-flash-low',
|
'gemini-3.5-flash-low',
|
||||||
'gemini-3.5-flash-extra-low',
|
'gemini-3.5-flash-extra-low',
|
||||||
|
|
|
||||||
79
tests/unit/antigravity-quota-gemini-3.6.test.js
Normal file
79
tests/unit/antigravity-quota-gemini-3.6.test.js
Normal 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",
|
||||||
|
]);
|
||||||
|
});
|
||||||
|
});
|
||||||
Loading…
Reference in a new issue