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

@ -57,6 +57,8 @@ setup_llm_providers() {
"OPENAI_GPT4_1_MINI" "OPENAI_GPT4_1_MINI"
"OPENAI_GPT4_1_NANO" "OPENAI_GPT4_1_NANO"
"OPENAI_GPT4O" "OPENAI_GPT4O"
"OPENAI_O4_MINI"
"OPENAI_O3"
) )
fi fi
else else

View file

@ -227,6 +227,8 @@ def setup_llm_providers() -> None:
"OPENAI_GPT4_1_MINI", "OPENAI_GPT4_1_MINI",
"OPENAI_GPT4_1_NANO", "OPENAI_GPT4_1_NANO",
"OPENAI_GPT4O", "OPENAI_GPT4O",
"OPENAI_O4_MINI",
"OPENAI_O3",
] ]
) )
else: else:

View file

@ -148,6 +148,36 @@ if settings.ENABLE_OPENAI:
max_completion_tokens=16384, 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: if settings.ENABLE_ANTHROPIC:
@ -343,7 +373,7 @@ if settings.ENABLE_AZURE_O3_MINI:
add_assistant_prefix=False, add_assistant_prefix=False,
max_completion_tokens=16384, max_completion_tokens=16384,
temperature=None, # Temperature isn't supported in the O-model series temperature=None, # Temperature isn't supported in the O-model series
reasoning_effort="low", reasoning_effort="high",
), ),
) )