From 7b9fab05b8c7a8112e60a74c11e224ca8f39b666 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 5 Mar 2025 22:12:23 -0800 Subject: [PATCH] make text promtp llm pickable in settings (#1894) --- skyvern/config.py | 1 + skyvern/forge/sdk/workflow/models/block.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/skyvern/config.py b/skyvern/config.py index 909a2582..a4d60470 100644 --- a/skyvern/config.py +++ b/skyvern/config.py @@ -98,6 +98,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 # 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 86ebd648..3639bec6 100644 --- a/skyvern/forge/sdk/workflow/models/block.py +++ b/skyvern/forge/sdk/workflow/models/block.py @@ -1186,7 +1186,7 @@ async def user_code(): ) -DEFAULT_TEXT_PROMPT_LLM_KEY = settings.SECONDARY_LLM_KEY or settings.LLM_KEY +DEFAULT_TEXT_PROMPT_LLM_KEY = settings.TEXT_ONLY_AGENT_LLM_KEY or settings.LLM_KEY class TextPromptBlock(Block):