feat: add flux-2-max and gpt-image-1.5 models

packages/capabilities/image-generation/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "celeste-image-generation"
-version = "0.3.3"
+version = "0.3.5"
 description = "Image generation package for Celeste AI. Unified interface for all providers"
 authors = [{name = "Kamilbenkirane", email = "kamil@withceleste.ai"}]
 readme = "README.md"

packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/models.py

@@ -6,6 +6,34 @@
 from celeste_image_generation.parameters import ImageGenerationParameter
 
 MODELS: list[Model] = [
+    Model(
+        id="flux-2-max",
+        provider=Provider.BFL,
+        display_name="FLUX.2 [max]",
+        parameter_constraints={
+            ImageGenerationParameter.ASPECT_RATIO: Dimensions(
+                min_pixels=64 * 64,  # 4,096
+                max_pixels=2048 * 2048,  # 4,194,304 (4MP)
+                min_aspect_ratio=9 / 21,  # ~0.429
+                max_aspect_ratio=21 / 9,  # ~2.333
+                presets={
+                    "Square 1K": "1024x1024",
+                    "Square 2K": "2048x2048",
+                    "HD 16:9": "1920x1080",
+                    "Portrait HD": "1080x1920",
+                    "4:3": "1280x960",
+                    "3:4": "960x1280",
+                    "Ultra-wide 21:9": "1920x832",
+                    "Portrait 9:21": "832x1920",
+                },
+            ),
+            # Note: flux-2-max always upsamples prompts (no prompt_upsampling parameter)
+            # Includes grounding search for real-time information integration
+            ImageGenerationParameter.SEED: Int(),
+            ImageGenerationParameter.SAFETY_TOLERANCE: Range(min=0, max=5),
+            ImageGenerationParameter.OUTPUT_FORMAT: Choice(options=["jpeg", "png"]),
+        },
+    ),
     Model(
         id="flux-2-pro",
         provider=Provider.BFL,

packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/models.py

@@ -56,4 +56,18 @@
             ),
         },
     ),
+    Model(
+        id="gpt-image-1.5",
+        provider=Provider.OPENAI,
+        display_name="GPT Image 1.5",
+        streaming=False,
+        parameter_constraints={
+            ImageGenerationParameter.ASPECT_RATIO: Choice(
+                options=["1024x1024", "1536x1024", "1024x1536", "auto"]
+            ),
+            ImageGenerationParameter.QUALITY: Choice(
+                options=["low", "medium", "high", "auto"]
+            ),
+        },
+    ),
 ]

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "celeste-ai"
-version = "0.3.4"
+version = "0.3.5"
 description = "Open source, type-safe primitives for multi-modal AI. All capabilities, all providers, one interface"
 authors = [{name = "Kamilbenkirane", email = "kamil@withceleste.ai"}]
 readme = "README.md"