From 1d38c7bfe831fc8af654a97d12edc8d1f1be70e2 Mon Sep 17 00:00:00 2001 From: Stanislav Novosad Date: Mon, 12 Jan 2026 14:09:19 -0700 Subject: [PATCH] Workflow Copilot: Use get_llm_handler_for_prompt_type (#4436) --- skyvern/forge/sdk/routes/workflow_copilot.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/routes/workflow_copilot.py b/skyvern/forge/sdk/routes/workflow_copilot.py index ee792068..ea6b7a26 100644 --- a/skyvern/forge/sdk/routes/workflow_copilot.py +++ b/skyvern/forge/sdk/routes/workflow_copilot.py @@ -12,6 +12,7 @@ from skyvern.forge import app from skyvern.forge.prompts import prompt_engine from skyvern.forge.sdk.api.llm.exceptions import LLMProviderError from skyvern.forge.sdk.artifact.models import Artifact, ArtifactType +from skyvern.forge.sdk.experimentation.llm_prompt_config import get_llm_handler_for_prompt_type from skyvern.forge.sdk.routes.routers import base_router from skyvern.forge.sdk.routes.run_blocks import DEFAULT_LOGIN_PROMPT from skyvern.forge.sdk.schemas.organizations import Organization @@ -115,8 +116,12 @@ async def copilot_call_llm( llm_prompt_len=len(llm_prompt), llm_prompt=llm_prompt, ) + llm_api_handler = ( + await get_llm_handler_for_prompt_type("workflow-copilot", chat_request.workflow_permanent_id, organization_id) + or app.LLM_API_HANDLER + ) llm_start_time = time.monotonic() - llm_response = await app.LLM_API_HANDLER( + llm_response = await llm_api_handler( prompt=llm_prompt, prompt_name="workflow-copilot", organization_id=organization_id,