diff --git a/skyvern/config.py b/skyvern/config.py index 79c248f2..8f7c34f8 100644 --- a/skyvern/config.py +++ b/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 diff --git a/skyvern/forge/app.py b/skyvern/forge/app.py index e56f9920..9165455f 100644 --- a/skyvern/forge/app.py +++ b/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() diff --git a/skyvern/webeye/actions/handler.py b/skyvern/webeye/actions/handler.py index 9fd816b5..4ab1725e 100644 --- a/skyvern/webeye/actions/handler.py +++ b/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",