From a2d53dcedd3234fee5ff50139f770d2a39e124c6 Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Thu, 11 Dec 2025 20:55:30 +0100 Subject: [PATCH 1/2] 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 57ad144..5b358f5 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/2] 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 5b358f5..dd9c30e 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),