From f01a7aa3572d81340fa13429f760563c6342f058 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 27 Nov 2024 22:03:53 -0800 Subject: [PATCH] fix empty llm_key promblem for the TextPromptBlock (#1282) --- skyvern/forge/sdk/workflow/models/block.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/workflow/models/block.py b/skyvern/forge/sdk/workflow/models/block.py index eec6a97d..0cb2d85a 100644 --- a/skyvern/forge/sdk/workflow/models/block.py +++ b/skyvern/forge/sdk/workflow/models/block.py @@ -783,7 +783,8 @@ class TextPromptBlock(Block): return self.parameters async def send_prompt(self, prompt: str, parameter_values: dict[str, Any]) -> dict[str, Any]: - llm_api_handler = LLMAPIHandlerFactory.get_llm_api_handler(self.llm_key) + llm_key = self.llm_key or DEFAULT_TEXT_PROMPT_LLM_KEY + llm_api_handler = LLMAPIHandlerFactory.get_llm_api_handler(llm_key) if not self.json_schema: self.json_schema = { "type": "object",