plugins: keep google provider policy lightweight

src/agents/models-config.providers.policy.lookup.test.ts

@@ -0,0 +1,52 @@
+import { describe, expect, it } from "vitest";
+import { resolveProviderPluginLookupKey } from "./models-config.providers.policy.lookup.js";
+
+describe("resolveProviderPluginLookupKey", () => {
+  it("routes Google Generative AI custom providers to the google policy artifact", () => {
+    expect(
+      resolveProviderPluginLookupKey("google-paid", {
+        baseUrl: "https://generativelanguage.googleapis.com/v1beta",
+        api: "google-generative-ai",
+        models: [],
+      }),
+    ).toBe("google");
+  });
+
+  it("routes model-level Google Generative AI providers to the google policy artifact", () => {
+    expect(
+      resolveProviderPluginLookupKey("custom-google", {
+        baseUrl: "https://generativelanguage.googleapis.com/v1beta",
+        models: [
+          {
+            id: "gemini-3-pro",
+            name: "Gemini 3 Pro",
+            api: "google-generative-ai",
+            reasoning: true,
+            input: ["text", "image"],
+            cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+            contextWindow: 1_048_576,
+            maxTokens: 65_536,
+          },
+        ],
+      }),
+    ).toBe("google");
+  });
+
+  it("routes google-antigravity to the google policy artifact", () => {
+    expect(
+      resolveProviderPluginLookupKey("google-antigravity", {
+        baseUrl: "https://generativelanguage.googleapis.com/v1beta",
+        models: [],
+      }),
+    ).toBe("google");
+  });
+
+  it("routes google-vertex to the google policy artifact", () => {
+    expect(
+      resolveProviderPluginLookupKey("google-vertex", {
+        baseUrl: "https://aiplatform.googleapis.com",
+        models: [],
+      }),
+    ).toBe("google");
+  });
+});

src/agents/models-config.providers.policy.lookup.ts

@@ -14,6 +14,18 @@ export function resolveProviderPluginLookupKey(
   provider?: ProviderConfig,
 ): string {
   const api = normalizeOptionalString(provider?.api) ?? "";
+  if (
+    providerKey === "google-antigravity" ||
+    providerKey === "google-vertex" ||
+    api === "google-generative-ai"
+  ) {
+    return "google";
+  }
+  if (
+    provider?.models?.some((model) => normalizeOptionalString(model.api) === "google-generative-ai")
+  ) {
+    return "google";
+  }
   if (
     api &&
     MODEL_APIS.includes(api as (typeof MODEL_APIS)[number]) &&