diff --git a/extensions/google/provider-catalog.test.ts b/extensions/google/provider-catalog.test.ts index a94652fbc6f..875e56fde34 100644 --- a/extensions/google/provider-catalog.test.ts +++ b/extensions/google/provider-catalog.test.ts @@ -11,8 +11,13 @@ describe("google provider catalog", () => { expect(provider.api).toBe("google-vertex"); expect(provider.baseUrl).toBe("https://{location}-aiplatform.googleapis.com"); expect(provider.models.map((model) => model.id)).toEqual( - expect.arrayContaining(["gemini-2.5-pro", "gemini-3.1-pro-preview"]), + expect.arrayContaining(["gemini-2.5-pro", "gemini-3.1-pro-preview", "gemini-3.1-flash-lite"]), ); + expect(provider.models.find((model) => model.id === "gemini-3.1-flash-lite")).toMatchObject({ + contextWindow: 1_048_576, + maxTokens: 65_536, + reasoning: true, + }); }); it("keeps Google AI Studio and Vertex model ids aligned", () => { diff --git a/extensions/google/provider-catalog.ts b/extensions/google/provider-catalog.ts index 1c982c8e570..923a8b0a997 100644 --- a/extensions/google/provider-catalog.ts +++ b/extensions/google/provider-catalog.ts @@ -43,6 +43,15 @@ const GOOGLE_GEMINI_TEXT_MODELS: ModelDefinitionConfig[] = [ contextWindow: 1_048_576, maxTokens: 65_536, }, + { + id: "gemini-3.1-flash-lite", + name: "Gemini 3.1 Flash Lite", + reasoning: true, + input: ["text", "image"], + cost: GOOGLE_GEMINI_COST, + contextWindow: 1_048_576, + maxTokens: 65_536, + }, { id: "gemini-3-flash-preview", name: "Gemini 3 Flash Preview",