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 4ec67fb3..ed0b5247 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,38 @@ TextGenerationParameter.OUTPUT_SCHEMA: Schema(), }, ), + Model( + id="gpt-5.1", + provider=Provider.OPENAI, + display_name="GPT-5.1", + streaming=True, + parameter_constraints={ + Parameter.MAX_TOKENS: Range(min=1, max=128000), + TextGenerationParameter.THINKING_BUDGET: Choice( + options=["minimal", "low", "medium", "high"] + ), + TextGenerationParameter.VERBOSITY: Choice( + options=["low", "medium", "high"] + ), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), + Model( + id="gpt-5.1-codex", + provider=Provider.OPENAI, + display_name="GPT-5.1 Codex", + streaming=True, + parameter_constraints={ + Parameter.MAX_TOKENS: Range(min=1, max=128000), + TextGenerationParameter.THINKING_BUDGET: Choice( + options=["minimal", "low", "medium", "high"] + ), + TextGenerationParameter.VERBOSITY: Choice( + options=["low", "medium", "high"] + ), + TextGenerationParameter.OUTPUT_SCHEMA: Schema(), + }, + ), Model( id="gpt-5-mini", provider=Provider.OPENAI,