vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 20:40:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: support Gemini latest thinking config

Browse Source
This commit is contained in:
Peter Steinberger
2026-04-18 18:18:22 +01:00
parent 00e613f12d
commit 28fe0296c4
9 changed files with 341 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
extensions/google/provider-models.test.ts
View File

@@ -186,6 +186,48 @@ describe("resolveGoogleGeminiForwardCompatModel", () => {
});
});
it("resolves Gemini latest aliases from current Google templates", () => {
const models = [
createTemplateModel("google", "gemini-3-pro-preview", { reasoning: true }),
createTemplateModel("google", "gemini-3-flash-preview", { reasoning: true }),
createTemplateModel("google", "gemini-3.1-flash-lite-preview", { reasoning: true }),
];
expect(
resolveGoogleGeminiForwardCompatModel({
providerId: "google",
ctx: createContext({ provider: "google", modelId: "gemini-pro-latest", models }),
}),
).toMatchObject({
provider: "google",
id: "gemini-pro-latest",
api: "google-generative-ai",
reasoning: true,
});
expect(
resolveGoogleGeminiForwardCompatModel({
providerId: "google",
ctx: createContext({ provider: "google", modelId: "gemini-flash-latest", models }),
}),
).toMatchObject({
provider: "google",
id: "gemini-flash-latest",
api: "google-generative-ai",
reasoning: true,
});
expect(
resolveGoogleGeminiForwardCompatModel({
providerId: "google",
ctx: createContext({ provider: "google", modelId: "gemini-flash-lite-latest", models }),
}),
).toMatchObject({
provider: "google",
id: "gemini-flash-lite-latest",
api: "google-generative-ai",
reasoning: true,
});
});
it("prefers the flash-lite template before the broader flash prefix", () => {
const model = resolveGoogleGeminiForwardCompatModel({
providerId: "google-vertex",
@@ -217,6 +259,12 @@ describe("resolveGoogleGeminiForwardCompatModel", () => {
expect(isModernGoogleModel("gemini-1.5-pro")).toBe(false);
});
it("treats Gemini latest aliases as modern google models", () => {
expect(isModernGoogleModel("gemini-pro-latest")).toBe(true);
expect(isModernGoogleModel("gemini-flash-latest")).toBe(true);
expect(isModernGoogleModel("gemini-flash-lite-latest")).toBe(true);
});
it("treats gemma models as modern google models", () => {
expect(isModernGoogleModel("gemma-4-26b-a4b-it")).toBe(true);
expect(isModernGoogleModel("gemma-3-4b-it")).toBe(true);

19
extensions/google/provider-models.ts
View File

@@ -12,6 +12,9 @@ const GEMINI_2_5_FLASH_PREFIX = "gemini-2.5-flash";
const GEMINI_3_1_PRO_PREFIX = "gemini-3.1-pro";
const GEMINI_3_1_FLASH_LITE_PREFIX = "gemini-3.1-flash-lite";
const GEMINI_3_1_FLASH_PREFIX = "gemini-3.1-flash";
const GEMINI_PRO_LATEST_ID = "gemini-pro-latest";
const GEMINI_FLASH_LATEST_ID = "gemini-flash-latest";
const GEMINI_FLASH_LITE_LATEST_ID = "gemini-flash-lite-latest";
const GEMMA_PREFIX = "gemma-";
const GEMINI_2_5_PRO_TEMPLATE_IDS = ["gemini-2.5-pro"] as const;
const GEMINI_2_5_FLASH_LITE_TEMPLATE_IDS = ["gemini-2.5-flash-lite"] as const;
@@ -128,7 +131,7 @@ export function resolveGoogleGeminiForwardCompatModel(params: {
cliTemplateIds: GEMINI_3_1_FLASH_TEMPLATE_IDS,
preferExternalFirstForCli: true,
};
} else if (lower.startsWith(GEMINI_3_1_PRO_PREFIX)) {
} else if (lower.startsWith(GEMINI_3_1_PRO_PREFIX) || lower === GEMINI_PRO_LATEST_ID) {
family = {
googleTemplateIds: GEMINI_3_1_PRO_TEMPLATE_IDS,
cliTemplateIds: GEMINI_3_1_PRO_TEMPLATE_IDS,
@@ -136,12 +139,15 @@ export function resolveGoogleGeminiForwardCompatModel(params: {
if (params.providerId === "google" || params.providerId === GOOGLE_GEMINI_CLI_PROVIDER_ID) {
patch = { reasoning: true };
}
} else if (lower.startsWith(GEMINI_3_1_FLASH_LITE_PREFIX)) {
} else if (
lower.startsWith(GEMINI_3_1_FLASH_LITE_PREFIX) ||
lower === GEMINI_FLASH_LITE_LATEST_ID
) {
family = {
googleTemplateIds: GEMINI_3_1_FLASH_LITE_TEMPLATE_IDS,
cliTemplateIds: GEMINI_3_1_FLASH_LITE_TEMPLATE_IDS,
};
} else if (lower.startsWith(GEMINI_3_1_FLASH_PREFIX)) {
} else if (lower.startsWith(GEMINI_3_1_FLASH_PREFIX) || lower === GEMINI_FLASH_LATEST_ID) {
family = {
googleTemplateIds: GEMINI_3_1_FLASH_TEMPLATE_IDS,
cliTemplateIds: GEMINI_3_1_FLASH_TEMPLATE_IDS,
@@ -182,6 +188,11 @@ export function resolveGoogleGeminiForwardCompatModel(params: {
export function isModernGoogleModel(modelId: string): boolean {
const lower = normalizeOptionalLowercaseString(modelId) ?? "";
return (
lower.startsWith("gemini-2.5") || lower.startsWith("gemini-3") || lower.startsWith(GEMMA_PREFIX)
lower.startsWith("gemini-2.5") ||
lower.startsWith("gemini-3") ||
lower === GEMINI_PRO_LATEST_ID ||
lower === GEMINI_FLASH_LATEST_ID ||
lower === GEMINI_FLASH_LITE_LATEST_ID ||
lower.startsWith(GEMMA_PREFIX)
);
}