extend select agent to support date picker (#2849)

2025-07-01 14:12:39 +09:00
parent 7a96642c12
commit cb17dbbb6f
6 changed files with 140 additions and 60 deletions
--- a/skyvern/services/task_v2_service.py
+++ b/skyvern/services/task_v2_service.py
@@ -1251,7 +1251,7 @@ async def _generate_extraction_task(
    # extract the data
    context = skyvern_context.ensure_context()
    generate_extraction_task_prompt = load_prompt_with_elements(
-        scraped_page=scraped_page,
+        element_tree_builder=scraped_page,
        prompt_engine=prompt_engine,
        template_name="task_v2_generate_extraction_task",
        current_url=current_url,