add cleanup function (#631)

2024-07-23 14:23:15 +08:00
parent 3e8ac05f56
commit 9245a329ee
3 changed files with 39 additions and 32 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -899,6 +899,7 @@ class ForgeAgent:
        return await scrape_website(
            browser_state,
            task.url,
+            app.AGENT_FUNCTION.cleanup_element_tree,
            scrape_exclude=app.scrape_exclude,
        )