fix terminate action in caching - do not replay terminate + introduce SINGLE_CLICK_AGENT_LLM_API_HANDLER (#1760)

2025-02-12 17:10:24 +08:00
parent fbf1473884
commit f16367bf96
3 changed files with 17 additions and 14 deletions
--- a/skyvern/forge/app.py
+++ b/skyvern/forge/app.py
@@ -40,6 +40,11 @@ SELECT_AGENT_LLM_API_HANDLER = (
    if SETTINGS_MANAGER.SELECT_AGENT_LLM_KEY
    else SECONDARY_LLM_API_HANDLER
 )
+SINGLE_CLICK_AGENT_LLM_API_HANDLER = (
+    LLMAPIHandlerFactory.get_llm_api_handler(SETTINGS_MANAGER.SINGLE_CLICK_AGENT_LLM_KEY)
+    if SETTINGS_MANAGER.SINGLE_CLICK_AGENT_LLM_KEY
+    else SECONDARY_LLM_API_HANDLER
+)
 WORKFLOW_CONTEXT_MANAGER = WorkflowContextManager()
 WORKFLOW_SERVICE = WorkflowService()
 AGENT_FUNCTION = AgentFunction()