Use gemini 2_5 flash lite for create_extract_action (#3429)

2025-09-13 16:22:57 +08:00
parent 031d9083a6
commit 66b2004b70
3 changed files with 24 additions and 2 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -3020,7 +3020,7 @@ class ForgeAgent:
            local_datetime=datetime.now(context.tz_info).isoformat(),
        )

-        data_extraction_summary_resp = await app.SECONDARY_LLM_API_HANDLER(
+        data_extraction_summary_resp = await app.EXTRACTION_LLM_API_HANDLER(
            prompt=prompt, step=step, prompt_name="data-extraction-summary"
        )
        return ExtractAction(