add step prepare and browser clean up (#554)

2024-07-05 02:17:34 +08:00
parent 7cdd986bde
commit be97d0bdb1
4 changed files with 40 additions and 7 deletions
--- a/skyvern/forge/agent.py
+++ b/skyvern/forge/agent.py
@@ -503,6 +503,10 @@ class ForgeAgent:
                step_retry=step.retry_index,
            )
            step = await self.update_step(step=step, status=StepStatus.running)
+            await app.AGENT_FUNCTION.prepare_step_execution(
+                organization=organization, task=task, step=step, browser_state=browser_state
+            )
+
            (
                scraped_page,
                extract_action_prompt,