fix(provider): normalize bare gemini-3 Pro model IDs for google-antigravity (#24145)

* fix(provider): normalize bare gemini-3 Pro model IDs for google-antigravity The Antigravity Cloud Code Assist API requires a thinking-tier suffix (-low or -high) for all Gemini 3 Pro variants. When a user configures a bare model ID like `gemini-3.1-pro`, the API returns a 404 because it only recognises `gemini-3.1-pro-low` or `gemini-3.1-pro-high`. Add `normalizeAntigravityModelId()` that appends `-low` (the default tier) to bare Pro model IDs, and apply it during provider normalisation for `google-antigravity`. Also refactor the per-provider model normalisation into a shared `normalizeProviderModels()` helper. Closes #24071 Co-authored-by: Cursor <cursoragent@cursor.com> * Tests: cover antigravity model ID normalization * Changelog: note antigravity pro tier normalization * Tests: type antigravity model helper inputs --------- Co-authored-by: Cursor <cursoragent@cursor.com> Co-authored-by: Vincent Koc <vincentkoc@ieee.org>
2026-05-10 21:34:32 +00:00 · 2026-02-27 07:53:46 +08:00
parent 17578d77e1
commit e6be26ef1c
3 changed files with 118 additions and 2 deletions
--- a/src/agents/models-config.providers.google-antigravity.test.ts
+++ b/src/agents/models-config.providers.google-antigravity.test.ts
@@ -0,0 +1,87 @@
+import { mkdtempSync } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { describe, expect, it } from "vitest";
+import {
+  normalizeAntigravityModelId,
+  normalizeProviders,
+  type ProviderConfig,
+} from "./models-config.providers.js";
+
+function buildModel(id: string): NonNullable<ProviderConfig["models"]>[number] {
+  return {
+    id,
+    name: id,
+    reasoning: true,
+    input: ["text"],
+    cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+    contextWindow: 1,
+    maxTokens: 1,
+  };
+}
+
+function buildProvider(modelIds: string[]): ProviderConfig {
+  return {
+    baseUrl: "https://example.invalid/v1",
+    api: "openai-completions",
+    apiKey: "EXAMPLE_KEY",
+    models: modelIds.map((id) => buildModel(id)),
+  };
+}
+
+describe("normalizeAntigravityModelId", () => {
+  it.each(["gemini-3-pro", "gemini-3.1-pro", "gemini-3-1-pro"])(
+    "adds default -low suffix to bare pro id: %s",
+    (id) => {
+      expect(normalizeAntigravityModelId(id)).toBe(`${id}-low`);
+    },
+  );
+
+  it.each([
+    "gemini-3-pro-low",
+    "gemini-3-pro-high",
+    "gemini-3.1-flash",
+    "claude-opus-4-6-thinking",
+  ])("keeps already-tiered and non-pro ids unchanged: %s", (id) => {
+    expect(normalizeAntigravityModelId(id)).toBe(id);
+  });
+});
+
+describe("google-antigravity provider normalization", () => {
+  it("normalizes bare gemini pro IDs only for google-antigravity providers", () => {
+    const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-"));
+    const providers = {
+      "google-antigravity": buildProvider([
+        "gemini-3-pro",
+        "gemini-3.1-pro",
+        "gemini-3-1-pro",
+        "gemini-3-pro-high",
+        "claude-opus-4-6-thinking",
+      ]),
+      openai: buildProvider(["gpt-5"]),
+    };
+
+    const normalized = normalizeProviders({ providers, agentDir });
+
+    expect(normalized).not.toBe(providers);
+    expect(normalized?.["google-antigravity"]?.models.map((model) => model.id)).toEqual([
+      "gemini-3-pro-low",
+      "gemini-3.1-pro-low",
+      "gemini-3-1-pro-low",
+      "gemini-3-pro-high",
+      "claude-opus-4-6-thinking",
+    ]);
+    expect(normalized?.openai).toBe(providers.openai);
+  });
+
+  it("returns original providers object when no antigravity IDs need normalization", () => {
+    const agentDir = mkdtempSync(join(tmpdir(), "openclaw-test-"));
+    const providers = {
+      "google-antigravity": buildProvider(["gemini-3-pro-low", "claude-opus-4-6-thinking"]),
+    };
+
+    const normalized = normalizeProviders({ providers, agentDir });
+
+    expect(normalized).toBe(providers);
+  });
+});