Jon/model name massage (#2559)

2025-05-31 19:34:30 -07:00
parent b4d5837196
commit 2167d88c20
11 changed files with 67 additions and 60 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -867,12 +867,13 @@ class ForgeAgent:
                else:
                    if engine in CUA_ENGINES:
                        self.async_operation_pool.run_operation(task.task_id, AgentPhase.llm)
+
                    json_response = await app.LLM_API_HANDLER(
                        prompt=extract_action_prompt,
                        prompt_name="extract-actions",
                        step=step,
                        screenshots=scraped_page.screenshots,
-                        llm_key_override=llm_caller.llm_key if llm_caller else None,
+                        llm_key_override=task.llm_key,
                    )
                    try:
                        json_response = await self.handle_potential_verification_code(