GPT-o4-mini and GPT-o3 support is here (#2170)

This commit is contained in:
Suchintan 2025-04-16 21:34:00 -04:00 committed by GitHub
parent a9c8a1d889
commit eefddb4105
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 35 additions and 1 deletions

View file

@ -148,6 +148,36 @@ if settings.ENABLE_OPENAI:
max_completion_tokens=16384,
),
)
LLMConfigRegistry.register_config(
"OPENAI_O4_MINI",
LLMConfig(
"o4-mini",
["OPENAI_API_KEY"],
supports_vision=False,
add_assistant_prefix=False,
max_completion_tokens=16384,
temperature=None, # Temperature isn't supported in the O-model series
reasoning_effort="high",
litellm_params=LiteLLMParams(
drop_params=True, # type: ignore
),
),
)
LLMConfigRegistry.register_config(
"OPENAI_O3",
LLMConfig(
"o3",
["OPENAI_API_KEY"],
supports_vision=False,
add_assistant_prefix=False,
max_completion_tokens=16384,
temperature=None, # Temperature isn't supported in the O-model series
reasoning_effort="high",
litellm_params=LiteLLMParams(
drop_params=True, # type: ignore
),
),
)
if settings.ENABLE_ANTHROPIC:
@ -343,7 +373,7 @@ if settings.ENABLE_AZURE_O3_MINI:
add_assistant_prefix=False,
max_completion_tokens=16384,
temperature=None, # Temperature isn't supported in the O-model series
reasoning_effort="low",
reasoning_effort="high",
),
)