Skip to content

Commit fe03daf

Browse files
empezarceroKludex
andauthored
Add service_tier to OpenAIModelSettings (#1923)
Co-authored-by: Marcelo Trylesinski <marcelotryle@gmail.com>
1 parent 4734135 commit fe03daf

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

pydantic_ai_slim/pydantic_ai/models/openai.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,13 @@ class OpenAIModelSettings(ModelSettings, total=False):
116116
See [OpenAI's safety best practices](https://platform.openai.com/docs/guides/safety-best-practices#end-user-ids) for more details.
117117
"""
118118

119+
openai_service_tier: Literal['auto', 'default', 'flex']
120+
"""The service tier to use for the model request.
121+
122+
Currently supported values are `auto`, `default`, and `flex`.
123+
For more information, see [OpenAI's service tiers documentation](https://platform.openai.com/docs/api-reference/chat/object#chat/object-service_tier).
124+
"""
125+
119126

120127
class OpenAIResponsesModelSettings(OpenAIModelSettings, total=False):
121128
"""Settings used for an OpenAI Responses model request.
@@ -298,6 +305,7 @@ async def _completions_create(
298305
logprobs=model_settings.get('openai_logprobs', NOT_GIVEN),
299306
top_logprobs=model_settings.get('openai_top_logprobs', NOT_GIVEN),
300307
user=model_settings.get('openai_user', NOT_GIVEN),
308+
service_tier=model_settings.get('openai_service_tier', NOT_GIVEN),
301309
extra_headers=extra_headers,
302310
extra_body=model_settings.get('extra_body'),
303311
)

0 commit comments

Comments
 (0)