From 511360273b12dbec0d84f1f7ccaa8f55534bfc8e Mon Sep 17 00:00:00 2001
From: kamilbenkirane <benkirane.kamil@gmail.com>
Date: Wed, 1 Jul 2026 15:40:44 +0200
Subject: [PATCH 1/2] feat(google): add gemini-3.1-flash-lite-image to catalog

Adds Nano Banana 2 Lite, the low-latency/low-cost variant of the Gemini
3.1 image family: 1K-only resolution (2K/4K unsupported), 14 object
reference images, and the same minimal/high thinking levels as its
siblings.
---
 .../images/providers/google/models.py         | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/celeste/modalities/images/providers/google/models.py b/src/celeste/modalities/images/providers/google/models.py
index 5e725e9..0a7fe6a 100644
--- a/src/celeste/modalities/images/providers/google/models.py
+++ b/src/celeste/modalities/images/providers/google/models.py
@@ -125,6 +125,31 @@
             ImageParameter.THINKING_LEVEL: Choice(options=["minimal", "High"]),
         },
     ),
+    Model(
+        id="gemini-3.1-flash-lite-image",
+        provider=Provider.GOOGLE,
+        display_name="Nano Banana 2 Lite",
+        operations={Modality.IMAGES: {Operation.GENERATE, Operation.EDIT}},
+        parameter_constraints={
+            ImageParameter.ASPECT_RATIO: Choice(
+                options=[
+                    "1:1",
+                    "2:3",
+                    "3:2",
+                    "3:4",
+                    "4:3",
+                    "4:5",
+                    "5:4",
+                    "9:16",
+                    "16:9",
+                    "21:9",
+                ]
+            ),
+            ImageParameter.QUALITY: Choice(options=["1K"]),
+            ImageParameter.REFERENCE_IMAGES: ImagesConstraint(max_count=14),
+            ImageParameter.THINKING_LEVEL: Choice(options=["minimal", "high"]),
+        },
+    ),
 ]
 
 # Unified model list for registration

From 07992985da310d8979e0e2e174fdadc58fda9055 Mon Sep 17 00:00:00 2001
From: kamilbenkirane <benkirane.kamil@gmail.com>
Date: Wed, 1 Jul 2026 15:46:27 +0200
Subject: [PATCH 2/2] fix(google): correct Gemini image catalog errors found
 during audit

- gemini-3-pro-image-preview and gemini-3.1-flash-image-preview are
  deprecated (shut down June 25, 2026); rename to their GA replacement
  IDs gemini-3-pro-image and gemini-3.1-flash-image.
- Fix THINKING_LEVEL casing across the Gemini 3.x image family: the API
  expects lowercase "high", not "High".
- Add the 512px quality tier to gemini-3.1-flash-image (GA gained it
  over the preview).
- Remove QUALITY from imagen-4.0-fast-generate-001: the imageSize
  parameter is only supported on the Standard and Ultra Imagen 4
  variants, not Fast.
---
 .../modalities/images/providers/google/models.py    | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/celeste/modalities/images/providers/google/models.py b/src/celeste/modalities/images/providers/google/models.py
index 0a7fe6a..971ce2f 100644
--- a/src/celeste/modalities/images/providers/google/models.py
+++ b/src/celeste/modalities/images/providers/google/models.py
@@ -32,7 +32,6 @@
             ImageParameter.ASPECT_RATIO: Choice(
                 options=["1:1", "3:4", "4:3", "9:16", "16:9"]
             ),
-            ImageParameter.QUALITY: Choice(options=["1K"]),
         },
     ),
     Model(
@@ -76,9 +75,9 @@
         },
     ),
     Model(
-        id="gemini-3-pro-image-preview",
+        id="gemini-3-pro-image",
         provider=Provider.GOOGLE,
-        display_name="Gemini 3 Pro Image (Preview)",
+        display_name="Nano Banana Pro",
         operations={Modality.IMAGES: {Operation.GENERATE, Operation.EDIT}},
         parameter_constraints={
             ImageParameter.ASPECT_RATIO: Choice(
@@ -97,11 +96,11 @@
             ),
             ImageParameter.QUALITY: Choice(options=["1K", "2K", "4K"]),
             ImageParameter.REFERENCE_IMAGES: ImagesConstraint(max_count=14),
-            ImageParameter.THINKING_LEVEL: Choice(options=["minimal", "High"]),
+            ImageParameter.THINKING_LEVEL: Choice(options=["minimal", "high"]),
         },
     ),
     Model(
-        id="gemini-3.1-flash-image-preview",
+        id="gemini-3.1-flash-image",
         provider=Provider.GOOGLE,
         display_name="Nano Banana 2",
         operations={Modality.IMAGES: {Operation.GENERATE, Operation.EDIT}},
@@ -120,9 +119,9 @@
                     "21:9",
                 ]
             ),
-            ImageParameter.QUALITY: Choice(options=["1K", "2K", "4K"]),
+            ImageParameter.QUALITY: Choice(options=["512px", "1K", "2K", "4K"]),
             ImageParameter.REFERENCE_IMAGES: ImagesConstraint(max_count=14),
-            ImageParameter.THINKING_LEVEL: Choice(options=["minimal", "High"]),
+            ImageParameter.THINKING_LEVEL: Choice(options=["minimal", "high"]),
         },
     ),
     Model(