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

add support for gpt5 and azure gpt5 series (#3136)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-08-07 15:12:47 -07:00
committed by GitHub
parent 149cea6893
commit e356d9fea0
2 changed files with 125 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
skyvern/config.py
View File

@@ -215,6 +215,27 @@ class Settings(BaseSettings):
AZURE_O3_API_BASE: str | None = None
AZURE_O3_API_VERSION: str = "2025-01-01-preview"
# AZURE gpt-5
ENABLE_AZURE_GPT5: bool = False
AZURE_GPT5_DEPLOYMENT: str = "gpt-5"
AZURE_GPT5_API_KEY: str | None = None
AZURE_GPT5_API_BASE: str | None = None
AZURE_GPT5_API_VERSION: str = "2025-04-01-preview"
# AZURE gpt-5 mini
ENABLE_AZURE_GPT5_MINI: bool = False
AZURE_GPT5_MINI_DEPLOYMENT: str = "gpt-5-mini"
AZURE_GPT5_MINI_API_KEY: str | None = None
AZURE_GPT5_MINI_API_BASE: str | None = None
AZURE_GPT5_MINI_API_VERSION: str = "2025-04-01-preview"
# AZURE gpt-5 nano
ENABLE_AZURE_GPT5_NANO: bool = False
AZURE_GPT5_NANO_DEPLOYMENT: str = "gpt-5-nano"
AZURE_GPT5_NANO_API_KEY: str | None = None
AZURE_GPT5_NANO_API_BASE: str | None = None
AZURE_GPT5_NANO_API_VERSION: str = "2025-04-01-preview"
# GEMINI
GEMINI_API_KEY: str | None = None

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

@@ -72,6 +72,39 @@ class LLMConfigRegistry:
if settings.ENABLE_OPENAI:
LLMConfigRegistry.register_config(
"OPENAI_GPT5",
LLMConfig(
"gpt-5-2025-08-07",
["OPENAI_API_KEY"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=128000,
temperature=None,
),
)
LLMConfigRegistry.register_config(
"OPENAI_GPT5_MINI",
LLMConfig(
"gpt-5-mini-2025-08-07",
["OPENAI_API_KEY"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=128000,
temperature=None,
),
)
LLMConfigRegistry.register_config(
"OPENAI_GPT5_NANO",
LLMConfig(
"gpt-5-nano-2025-08-07",
["OPENAI_API_KEY"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=128000,
temperature=None,
),
)
LLMConfigRegistry.register_config(
"OPENAI_GPT4_TURBO",
LLMConfig(
@@ -537,6 +570,77 @@ if settings.ENABLE_AZURE_GPT4_1_NANO:
),
)
if settings.ENABLE_AZURE_GPT5:
LLMConfigRegistry.register_config(
"AZURE_OPENAI_GPT5",
LLMConfig(
f"azure/{settings.AZURE_GPT5_DEPLOYMENT}",
[
"AZURE_GPT5_DEPLOYMENT",
"AZURE_GPT5_API_KEY",
"AZURE_GPT5_API_BASE",
"AZURE_GPT5_API_VERSION",
],
litellm_params=LiteLLMParams(
api_base=settings.AZURE_GPT5_API_BASE,
api_key=settings.AZURE_GPT5_API_KEY,
api_version=settings.AZURE_GPT5_API_VERSION,
model_info={"model_name": "azure/gpt-5"},
),
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=128000,
temperature=None,
),
)
if settings.ENABLE_AZURE_GPT5_MINI:
LLMConfigRegistry.register_config(
"AZURE_OPENAI_GPT5_MINI",
LLMConfig(
f"azure/{settings.AZURE_GPT5_MINI_DEPLOYMENT}",
[
"AZURE_GPT5_MINI_DEPLOYMENT",
"AZURE_GPT5_MINI_API_KEY",
"AZURE_GPT5_MINI_API_BASE",
"AZURE_GPT5_MINI_API_VERSION",
],
litellm_params=LiteLLMParams(
api_base=settings.AZURE_GPT5_MINI_API_BASE,
api_key=settings.AZURE_GPT5_MINI_API_KEY,
api_version=settings.AZURE_GPT5_MINI_API_VERSION,
model_info={"model_name": "azure/gpt-5-mini"},
),
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=128000,
temperature=None,
),
)
if settings.ENABLE_AZURE_GPT5_NANO:
LLMConfigRegistry.register_config(
"AZURE_OPENAI_GPT5_NANO",
LLMConfig(
f"azure/{settings.AZURE_GPT5_NANO_DEPLOYMENT}",
[
"AZURE_GPT5_NANO_DEPLOYMENT",
"AZURE_GPT5_NANO_API_KEY",
"AZURE_GPT5_NANO_API_BASE",
"AZURE_GPT5_NANO_API_VERSION",
],
litellm_params=LiteLLMParams(
api_base=settings.AZURE_GPT5_NANO_API_BASE,
api_key=settings.AZURE_GPT5_NANO_API_KEY,
api_version=settings.AZURE_GPT5_NANO_API_VERSION,
model_info={"model_name": "azure/gpt-5-nano"},
),
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=128000,
temperature=None,
),
)
if settings.ENABLE_AZURE_O4_MINI:
LLMConfigRegistry.register_config(