script generation improvement (#3216)

2025-08-16 17:48:10 -07:00
parent 82f0b98fca
commit b9b7591079
14 changed files with 841 additions and 279 deletions
--- a/skyvern/forge/app.py
+++ b/skyvern/forge/app.py
@@ -69,6 +69,11 @@ SINGLE_CLICK_AGENT_LLM_API_HANDLER = (
    if SETTINGS_MANAGER.SINGLE_CLICK_AGENT_LLM_KEY
    else SECONDARY_LLM_API_HANDLER
 )
+SINGLE_INPUT_AGENT_LLM_API_HANDLER = (
+    LLMAPIHandlerFactory.get_llm_api_handler(SETTINGS_MANAGER.SINGLE_INPUT_AGENT_LLM_KEY)
+    if SETTINGS_MANAGER.SINGLE_INPUT_AGENT_LLM_KEY
+    else SECONDARY_LLM_API_HANDLER
+)
 WORKFLOW_CONTEXT_MANAGER = WorkflowContextManager()
 WORKFLOW_SERVICE = WorkflowService()
 AGENT_FUNCTION = AgentFunction()