vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add support for gpt 5.1 (#4197)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-12-04 17:08:16 -05:00
committed by GitHub
parent 944c95f456
commit 43e89d2991
2 changed files with 44 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
skyvern/config.py
View File

@@ -281,6 +281,13 @@ class Settings(BaseSettings):
AZURE_GPT5_NANO_API_BASE: str | None = None
AZURE_GPT5_NANO_API_VERSION: str = "2025-04-01-preview"
# AZURE gpt-5.1
ENABLE_AZURE_GPT5_1: bool = False
AZURE_GPT5_1_DEPLOYMENT: str = "gpt-5.1"
AZURE_GPT5_1_API_KEY: str | None = None
AZURE_GPT5_1_API_BASE: str | None = None
AZURE_GPT5_1_API_VERSION: str = "2025-04-01-preview"
# GEMINI
GEMINI_API_KEY: str | None = None
GEMINI_INCLUDE_THOUGHT: bool = False

37
skyvern/forge/sdk/api/llm/config_registry.py
View File

@@ -108,6 +108,18 @@ if settings.ENABLE_OPENAI:
reasoning_effort=settings.GPT5_REASONING_EFFORT,
),
)
LLMConfigRegistry.register_config(
"OPENAI_GPT5_1",
LLMConfig(
"gpt-5.1",
["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(
@@ -661,6 +673,31 @@ if settings.ENABLE_AZURE_GPT5_NANO:
),
)
if settings.ENABLE_AZURE_GPT5_1:
LLMConfigRegistry.register_config(
"AZURE_OPENAI_GPT5_1",
LLMConfig(
f"azure/{settings.AZURE_GPT5_1_DEPLOYMENT}",
[
"AZURE_GPT5_1_DEPLOYMENT",
"AZURE_GPT5_1_API_KEY",
"AZURE_GPT5_1_API_BASE",
"AZURE_GPT5_1_API_VERSION",
],
litellm_params=LiteLLMParams(
api_base=settings.AZURE_GPT5_1_API_BASE,
api_key=settings.AZURE_GPT5_1_API_KEY,
api_version=settings.AZURE_GPT5_1_API_VERSION,
model_info={"model_name": "azure/gpt-5.1"},
),
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",