diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index 6dd0a3b1..f62ee6cb 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -378,6 +378,7 @@ class ForgeAgent: step, detailed_output = await self.agent_step( task, step, browser_state, organization=organization, task_block=task_block ) + await app.AGENT_FUNCTION.post_step_execution(task, step) task = await self.update_task_errors_from_detailed_output(task, detailed_output) retry = False diff --git a/skyvern/forge/agent_functions.py b/skyvern/forge/agent_functions.py index dc528ce3..9bca9a2c 100644 --- a/skyvern/forge/agent_functions.py +++ b/skyvern/forge/agent_functions.py @@ -436,6 +436,9 @@ class AgentFunction: """ return + async def post_step_execution(self, task: Task, step: Step) -> None: + return + async def generate_async_operations( self, organization: Organization,