GPT 5.2 Support, update models (#4287)
Co-authored-by: Suchintan Singh <suchintan@skyvern.com>
This commit is contained in:
@@ -120,6 +120,18 @@ if settings.ENABLE_OPENAI:
|
||||
reasoning_effort=settings.GPT5_REASONING_EFFORT,
|
||||
),
|
||||
)
|
||||
LLMConfigRegistry.register_config(
|
||||
"OPENAI_GPT5_2",
|
||||
LLMConfig(
|
||||
"gpt-5.2",
|
||||
["OPENAI_API_KEY"],
|
||||
supports_vision=True,
|
||||
add_assistant_prefix=False,
|
||||
max_completion_tokens=128000,
|
||||
temperature=1, # GPT-5 only supports temperature=1
|
||||
reasoning_effort=settings.GPT5_REASONING_EFFORT,
|
||||
),
|
||||
)
|
||||
LLMConfigRegistry.register_config(
|
||||
"OPENAI_GPT4_TURBO",
|
||||
LLMConfig(
|
||||
@@ -698,6 +710,31 @@ if settings.ENABLE_AZURE_GPT5_1:
|
||||
),
|
||||
)
|
||||
|
||||
if settings.ENABLE_AZURE_GPT5_2:
|
||||
LLMConfigRegistry.register_config(
|
||||
"AZURE_OPENAI_GPT5_2",
|
||||
LLMConfig(
|
||||
f"azure/{settings.AZURE_GPT5_2_DEPLOYMENT}",
|
||||
[
|
||||
"AZURE_GPT5_2_DEPLOYMENT",
|
||||
"AZURE_GPT5_2_API_KEY",
|
||||
"AZURE_GPT5_2_API_BASE",
|
||||
"AZURE_GPT5_2_API_VERSION",
|
||||
],
|
||||
litellm_params=LiteLLMParams(
|
||||
api_base=settings.AZURE_GPT5_2_API_BASE,
|
||||
api_key=settings.AZURE_GPT5_2_API_KEY,
|
||||
api_version=settings.AZURE_GPT5_2_API_VERSION,
|
||||
model_info={"model_name": "azure/gpt-5.2"},
|
||||
),
|
||||
supports_vision=True,
|
||||
add_assistant_prefix=False,
|
||||
max_completion_tokens=128000,
|
||||
temperature=1, # GPT-5 only supports temperature=1
|
||||
reasoning_effort=settings.GPT5_REASONING_EFFORT,
|
||||
),
|
||||
)
|
||||
|
||||
if settings.ENABLE_AZURE_O4_MINI:
|
||||
LLMConfigRegistry.register_config(
|
||||
"AZURE_OPENAI_O4_MINI",
|
||||
|
||||
Reference in New Issue
Block a user