diff --git a/skills/baoyu-image-gen/scripts/providers/google.test.ts b/skills/baoyu-image-gen/scripts/providers/google.test.ts index 88d4e00..0c2e7fd 100644 --- a/skills/baoyu-image-gen/scripts/providers/google.test.ts +++ b/skills/baoyu-image-gen/scripts/providers/google.test.ts @@ -67,6 +67,9 @@ test("Google provider helpers normalize model IDs and select image size defaults "gemini-3.1-flash-image-preview", ); assert.equal(isGoogleMultimodal("models/gemini-3-pro-image-preview"), true); + assert.equal(isGoogleMultimodal("gemini-3-pro-image"), true); + assert.equal(isGoogleMultimodal("gemini-3.1-flash-image"), true); + assert.equal(isGoogleMultimodal("models/gemini-3-pro-image"), true); assert.equal(isGoogleImagen("imagen-3.0-generate-002"), true); assert.equal(getGoogleImageSize(makeArgs({ imageSize: null, quality: "2k" })), "2K"); assert.equal(getGoogleImageSize(makeArgs({ imageSize: "4K", quality: "normal" })), "4K"); diff --git a/skills/baoyu-image-gen/scripts/providers/google.ts b/skills/baoyu-image-gen/scripts/providers/google.ts index 09985d4..bed9a47 100644 --- a/skills/baoyu-image-gen/scripts/providers/google.ts +++ b/skills/baoyu-image-gen/scripts/providers/google.ts @@ -4,6 +4,8 @@ import { execFileSync } from "node:child_process"; import type { CliArgs } from "../types"; const GOOGLE_MULTIMODAL_MODELS = [ + "gemini-3-pro-image", + "gemini-3.1-flash-image", "gemini-3-pro-image-preview", "gemini-3-flash-preview", "gemini-3.1-flash-image-preview", @@ -333,7 +335,7 @@ export async function generateImage( if (isGoogleImagen(model)) { if (args.referenceImages.length > 0) { throw new Error( - "Reference images are not supported with Imagen models. Use gemini-3-pro-image-preview, gemini-3-flash-preview, or gemini-3.1-flash-image-preview.", + "Reference images are not supported with Imagen models. Use a Gemini multimodal model such as gemini-3-pro-image, gemini-3.1-flash-image, gemini-3-pro-image-preview, gemini-3-flash-preview, or gemini-3.1-flash-image-preview.", ); } return generateWithImagen(prompt, model, args); @@ -341,7 +343,7 @@ export async function generateImage( if (!isGoogleMultimodal(model) && args.referenceImages.length > 0) { throw new Error( - "Reference images are only supported with Gemini multimodal models. Use gemini-3-pro-image-preview, gemini-3-flash-preview, or gemini-3.1-flash-image-preview.", + "Reference images are only supported with Gemini multimodal models such as gemini-3-pro-image, gemini-3.1-flash-image, gemini-3-pro-image-preview, gemini-3-flash-preview, or gemini-3.1-flash-image-preview.", ); }