Setting for check-user-goal in sequential click for dropdown (flash_lite xp) (#3451)

skyvern/config.py

@@ -123,6 +123,7 @@ class Settings(BaseSettings):
     PROMPT_BLOCK_LLM_KEY: str | None = None
     PARSE_SELECT_LLM_KEY: str | None = None
     EXTRACTION_LLM_KEY: str | None = None
+    CHECK_USER_GOAL_LLM_KEY: str | None = None
     # COMMON
     LLM_CONFIG_TIMEOUT: int = 300
     LLM_CONFIG_MAX_TOKENS: int = 4096

skyvern/forge/app.py

@@ -78,6 +78,11 @@ EXTRACTION_LLM_API_HANDLER = (
     if SETTINGS_MANAGER.EXTRACTION_LLM_KEY
     else LLM_API_HANDLER
 )
+CHECK_USER_GOAL_LLM_API_HANDLER = (
+    LLMAPIHandlerFactory.get_llm_api_handler(SETTINGS_MANAGER.CHECK_USER_GOAL_LLM_KEY)
+    if SETTINGS_MANAGER.CHECK_USER_GOAL_LLM_KEY
+    else SECONDARY_LLM_API_HANDLER
+)
 SVG_CSS_CONVERTER_LLM_API_HANDLER = SECONDARY_LLM_API_HANDLER if SETTINGS_MANAGER.SECONDARY_LLM_KEY else None
 
 WORKFLOW_CONTEXT_MANAGER = WorkflowContextManager()

skyvern/webeye/actions/handler.py

@@ -753,7 +753,7 @@ async def handle_sequential_click_for_dropdown(
         action_history=action_history_str,
         local_datetime=datetime.now(skyvern_context.ensure_context().tz_info).isoformat(),
     )
-    response = await app.SECONDARY_LLM_API_HANDLER(
+    response = await app.CHECK_USER_GOAL_LLM_API_HANDLER(
         prompt=prompt,
         step=step,
         prompt_name="check-user-goal",