diff --git a/packages/capabilities/image-generation/pyproject.toml b/packages/capabilities/image-generation/pyproject.toml index e91680d5..4583f72c 100644 --- a/packages/capabilities/image-generation/pyproject.toml +++ b/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" diff --git a/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/models.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/models.py index 4f7d80f5..ba0cbdde 100644 --- a/packages/capabilities/image-generation/src/celeste_image_generation/providers/bfl/models.py +++ b/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, diff --git a/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/models.py b/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/models.py index 8183a18a..c2b54a04 100644 --- a/packages/capabilities/image-generation/src/celeste_image_generation/providers/openai/models.py +++ b/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"] + ), + }, + ), ] diff --git a/pyproject.toml b/pyproject.toml index ec2609ee..1270fe70 100644 --- a/pyproject.toml +++ b/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"