introduce llm key for script gen (#3736)

2025-10-16 02:50:16 -07:00
parent b93154e7b7
commit c186418d41
3 changed files with 5 additions and 1 deletions
--- a/skyvern/forge/app.py
+++ b/skyvern/forge/app.py
@@ -103,6 +103,9 @@ 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
+)

 WORKFLOW_CONTEXT_MANAGER = WorkflowContextManager()
 WORKFLOW_SERVICE = WorkflowService()