mirror of
https://github.com/openclaw/openclaw.git
synced 2026-06-06 05:51:15 +08:00
50 lines
2.1 KiB
TypeScript
50 lines
2.1 KiB
TypeScript
// Google tests cover model id plugin behavior.
|
|
import { describe, expect, it } from "vitest";
|
|
import { normalizeAntigravityModelId, normalizeGoogleModelId } from "./api.js";
|
|
|
|
describe("google model id helpers", () => {
|
|
it.each(["gemini-3-pro", "gemini-3.1-pro", "gemini-3-1-pro"])(
|
|
"adds default -low suffix to bare antigravity 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);
|
|
});
|
|
|
|
it("maps the deprecated 3.1 flash alias to the real preview model", () => {
|
|
expect(normalizeGoogleModelId("gemini-3.1-flash")).toBe("gemini-3-flash-preview");
|
|
expect(normalizeGoogleModelId("gemini-3.1-flash-preview")).toBe("gemini-3-flash-preview");
|
|
});
|
|
|
|
it("keeps bare Gemini 3.1 Pro as an alias for Google's preview-suffixed API id", () => {
|
|
expect(normalizeGoogleModelId("gemini-3-pro")).toBe("gemini-3.1-pro-preview");
|
|
expect(normalizeGoogleModelId("gemini-3-pro-preview")).toBe("gemini-3.1-pro-preview");
|
|
expect(normalizeGoogleModelId("gemini-3.1-pro")).toBe("gemini-3.1-pro-preview");
|
|
expect(normalizeGoogleModelId("gemini-3.1-pro-preview")).toBe("gemini-3.1-pro-preview");
|
|
});
|
|
|
|
it("normalizes provider-prefixed Gemini 3 Pro config ids", () => {
|
|
expect(normalizeGoogleModelId("google/gemini-3-pro-preview")).toBe(
|
|
"google/gemini-3.1-pro-preview",
|
|
);
|
|
});
|
|
|
|
it("keeps GA gemini-3.1-flash-lite unchanged and maps old preview name to GA", () => {
|
|
expect(normalizeGoogleModelId("gemini-3.1-flash-lite")).toBe("gemini-3.1-flash-lite");
|
|
expect(normalizeGoogleModelId("gemini-3.1-flash-lite-preview")).toBe("gemini-3.1-flash-lite");
|
|
});
|
|
|
|
it("maps the old Gemma 4 26B shorthand to Google's canonical API id", () => {
|
|
expect(normalizeGoogleModelId("gemma-4-26b")).toBe("gemma-4-26b-a4b-it");
|
|
expect(normalizeGoogleModelId("google/gemma-4-26b")).toBe("google/gemma-4-26b-a4b-it");
|
|
});
|
|
});
|