diff --git a/packages/text-generation/src/celeste_text_generation/providers/openai/models.py b/packages/text-generation/src/celeste_text_generation/providers/openai/models.py index 64a7afa6..4ec67fb3 100644 --- a/packages/text-generation/src/celeste_text_generation/providers/openai/models.py +++ b/packages/text-generation/src/celeste_text_generation/providers/openai/models.py @@ -71,6 +71,32 @@ TextGenerationParameter.OUTPUT_SCHEMA: Schema(), }, ), + Model( + id="gpt-5-mini", + provider=Provider.OPENAI, + display_name="GPT-5 Mini", + streaming=True, + parameter_constraints={ + Parameter.MAX_TOKENS: Range(min=1, max=128000), + TextGenerationParameter.THINKING_BUDGET: Choice( + options=["minimal", "low", "medium", "high"] + ), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), + Model( + id="gpt-5-nano", + provider=Provider.OPENAI, + display_name="GPT-5 Nano", + streaming=True, + parameter_constraints={ + Parameter.MAX_TOKENS: Range(min=1, max=128000), + TextGenerationParameter.THINKING_BUDGET: Choice( + options=["minimal", "low", "medium", "high"] + ), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), Model( id="gpt-4.1", provider=Provider.OPENAI,