weave LLMCaller.llm_key through to api handler/agent (#2524)

2025-05-29 13:49:59 -07:00
parent 7f6b65ba61
commit ea5620acd2
3 changed files with 34 additions and 18 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -864,6 +864,7 @@ class ForgeAgent:
                        prompt_name="extract-actions",
                        step=step,
                        screenshots=scraped_page.screenshots,
+                        llm_key_override=llm_caller.llm_key if llm_caller else None,
                    )
                    try:
                        json_response = await self.handle_potential_verification_code(