From a2d53dcedd3234fee5ff50139f770d2a39e124c6 Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Thu, 11 Dec 2025 20:55:30 +0100 Subject: [PATCH 1/3] feat: add Devstral 2 and Devstral Small 2 models --- .../providers/mistral/models.py | 52 +++++++++++++++++-- 1 file changed, 48 insertions(+), 4 deletions(-) diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py b/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py index 57ad1447..5b358f5a 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py @@ -9,7 +9,7 @@ Model( id="mistral-large-latest", provider=Provider.MISTRAL, - display_name="Mistral Large", + display_name="Mistral Large 3", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -20,7 +20,7 @@ Model( id="mistral-medium-latest", provider=Provider.MISTRAL, - display_name="Mistral Medium", + display_name="Mistral Medium 3.1", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -31,7 +31,7 @@ Model( id="mistral-small-latest", provider=Provider.MISTRAL, - display_name="Mistral Small", + display_name="Mistral Small 3.2", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -105,6 +105,28 @@ TextGenerationParameter.OUTPUT_SCHEMA: Schema(), }, ), + Model( + id="devstral-2512", + provider=Provider.MISTRAL, + display_name="Devstral 2", + streaming=True, + parameter_constraints={ + Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), + Parameter.MAX_TOKENS: Range(min=1, max=32768, step=1), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), + Model( + id="labs-devstral-small-2512", + provider=Provider.MISTRAL, + display_name="Devstral 2 Small", + streaming=True, + parameter_constraints={ + Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), + Parameter.MAX_TOKENS: Range(min=1, max=32768, step=1), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), Model( id="pixtral-12b-2409", provider=Provider.MISTRAL, @@ -116,10 +138,32 @@ TextGenerationParameter.OUTPUT_SCHEMA: Schema(), }, ), + Model( + id="ministral-3b-latest", + provider=Provider.MISTRAL, + display_name="Ministral 3 3B", + streaming=True, + parameter_constraints={ + Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), + Parameter.MAX_TOKENS: Range(min=1, max=32768, step=1), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), Model( id="ministral-8b-latest", provider=Provider.MISTRAL, - display_name="Ministral 8B", + display_name="Ministral 3 8B", + streaming=True, + parameter_constraints={ + Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), + Parameter.MAX_TOKENS: Range(min=1, max=32768, step=1), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), + Model( + id="ministral-14b-latest", + provider=Provider.MISTRAL, + display_name="Ministral 3 14B", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), From 226caed382e2ad4f79e5fece5b2cdcf3da65803b Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Thu, 11 Dec 2025 20:58:00 +0100 Subject: [PATCH 2/3] fix: remove version numbers from display names for -latest models --- .../providers/mistral/models.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py b/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py index 5b358f5a..dd9c30ea 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/mistral/models.py @@ -9,7 +9,7 @@ Model( id="mistral-large-latest", provider=Provider.MISTRAL, - display_name="Mistral Large 3", + display_name="Mistral Large", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -20,7 +20,7 @@ Model( id="mistral-medium-latest", provider=Provider.MISTRAL, - display_name="Mistral Medium 3.1", + display_name="Mistral Medium", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -31,7 +31,7 @@ Model( id="mistral-small-latest", provider=Provider.MISTRAL, - display_name="Mistral Small 3.2", + display_name="Mistral Small", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -141,7 +141,7 @@ Model( id="ministral-3b-latest", provider=Provider.MISTRAL, - display_name="Ministral 3 3B", + display_name="Ministral 3B", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -152,7 +152,7 @@ Model( id="ministral-8b-latest", provider=Provider.MISTRAL, - display_name="Ministral 3 8B", + display_name="Ministral 8B", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), @@ -163,7 +163,7 @@ Model( id="ministral-14b-latest", provider=Provider.MISTRAL, - display_name="Ministral 3 14B", + display_name="Ministral 14B", streaming=True, parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0, step=0.01), From 52d1e0c4ff4e65a693126133587d72ce96dfe81c Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Thu, 11 Dec 2025 21:02:30 +0100 Subject: [PATCH 3/3] feat: add seedream-4.5 model for ByteDance image generation --- .../providers/bytedance/models.py | 25 +++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py b/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py index 6273ccba..0cde1a95 100644 --- a/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py +++ b/packages/image-generation/src/celeste_image_generation/providers/bytedance/models.py @@ -31,4 +31,29 @@ ImageGenerationParameter.WATERMARK: Bool(), }, ), + Model( + id="seedream-4-5-251128", + provider=Provider.BYTEDANCE, + display_name="Seedream 4.5", + parameter_constraints={ + ImageGenerationParameter.ASPECT_RATIO: Dimensions( + min_pixels=2560 * 1440, # 3,686,400 + max_pixels=4096 * 4096, # 16,777,216 + min_aspect_ratio=1 / 16, # 0.0625 + max_aspect_ratio=16, + presets={ + "Square 2K": "2048x2048", + "Square 4K": "4096x4096", + "HD 16:9": "1920x1080", + "2K 16:9": "2560x1440", + "4K 16:9": "3840x2160", + "Portrait HD": "1080x1920", + "Portrait 2K": "1440x2560", + "Ultra-wide 21:9": "3024x1296", + }, + ), + ImageGenerationParameter.QUALITY: Choice(options=["2K", "4K"]), + ImageGenerationParameter.WATERMARK: Bool(), + }, + ), ]