From c5dbe403017a2079e79a679ddeb1349a3472e590 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 12 Mar 2025 02:10:47 -0700 Subject: [PATCH] PROMPT_BLOCK_LLM_KEY renaming (#1929) --- skyvern/config.py | 2 +- skyvern/forge/sdk/workflow/models/block.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/skyvern/config.py b/skyvern/config.py index a229a80a..d89e7f19 100644 --- a/skyvern/config.py +++ b/skyvern/config.py @@ -99,7 +99,7 @@ class Settings(BaseSettings): SECONDARY_LLM_KEY: str | None = None SELECT_AGENT_LLM_KEY: str | None = None SINGLE_CLICK_AGENT_LLM_KEY: str | None = None - TEXT_ONLY_AGENT_LLM_KEY: str | None = None + PROMPT_BLOCK_LLM_KEY: str | None = None # COMMON LLM_CONFIG_TIMEOUT: int = 300 LLM_CONFIG_MAX_TOKENS: int = 4096 diff --git a/skyvern/forge/sdk/workflow/models/block.py b/skyvern/forge/sdk/workflow/models/block.py index 19387b92..75fff2c3 100644 --- a/skyvern/forge/sdk/workflow/models/block.py +++ b/skyvern/forge/sdk/workflow/models/block.py @@ -1280,7 +1280,7 @@ async def wrapper(): ) -DEFAULT_TEXT_PROMPT_LLM_KEY = settings.TEXT_ONLY_AGENT_LLM_KEY or settings.LLM_KEY +DEFAULT_TEXT_PROMPT_LLM_KEY = settings.PROMPT_BLOCK_LLM_KEY or settings.LLM_KEY class TextPromptBlock(Block):