ModelSelector: add 'Skyvern Optimized' as hard choice (#2558)

2025-05-31 18:42:57 -07:00
parent 800a26d323
commit b4d5837196
6 changed files with 59 additions and 21 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -867,13 +867,12 @@ 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=task.llm_key,
+                        llm_key_override=llm_caller.llm_key if llm_caller else None,
                    )
                    try:
                        json_response = await self.handle_potential_verification_code(