Fix extraction prompt templating (#3335)

2025-09-01 15:41:40 +08:00
parent ae8fda4ed5
commit 9a699e70f8
5 changed files with 65 additions and 3 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -3010,6 +3010,7 @@ class ForgeAgent:
        return ExtractAction(
            reasoning=data_extraction_summary_resp.get("summary", "Extracting information from the page"),
            data_extraction_goal=task.data_extraction_goal,
+            data_extraction_schema=task.extracted_information_schema,
            organization_id=task.organization_id,
            task_id=task.task_id,
            workflow_run_id=task.workflow_run_id,