diff --git a/skyvern/forge/app.py b/skyvern/forge/app.py index 16f50a73..d48403fc 100644 --- a/skyvern/forge/app.py +++ b/skyvern/forge/app.py @@ -103,8 +103,10 @@ AUTO_COMPLETION_LLM_API_HANDLER = ( else SECONDARY_LLM_API_HANDLER ) SVG_CSS_CONVERTER_LLM_API_HANDLER = SECONDARY_LLM_API_HANDLER if SETTINGS_MANAGER.SECONDARY_LLM_KEY else None -SCRIPT_GENERATION_LLM_API_HANDLER = LLMAPIHandlerFactory.get_llm_api_handler( - SETTINGS_MANAGER.SCRIPT_GENERATION_LLM_KEY or SETTINGS_MANAGER.SECONDARY_LLM_KEY or SETTINGS_MANAGER.LLM_KEY +SCRIPT_GENERATION_LLM_API_HANDLER = ( + LLMAPIHandlerFactory.get_llm_api_handler(SETTINGS_MANAGER.SCRIPT_GENERATION_LLM_KEY) + if SETTINGS_MANAGER.SCRIPT_GENERATION_LLM_KEY + else SECONDARY_LLM_API_HANDLER ) WORKFLOW_CONTEXT_MANAGER = WorkflowContextManager()