add generate_script to the task v2 table (#3332)

2025-08-31 15:08:01 +08:00
parent 494b009ef9
commit 687bf1e2c7
6 changed files with 51 additions and 0 deletions
--- a/skyvern/forge/sdk/db/client.py
+++ b/skyvern/forge/sdk/db/client.py
@@ -2639,6 +2639,7 @@ class AgentDB:
        max_screenshot_scrolling_times: int | None = None,
        extra_http_headers: dict[str, str] | None = None,
        browser_address: str | None = None,
+        generate_script: bool = False,
    ) -> TaskV2:
        async with self.Session() as session:
            new_task_v2 = TaskV2Model(
@@ -2658,6 +2659,7 @@ class AgentDB:
                max_screenshot_scrolling_times=max_screenshot_scrolling_times,
                extra_http_headers=extra_http_headers,
                browser_address=browser_address,
+                generate_script=generate_script,
            )
            session.add(new_task_v2)
            await session.commit()