From 6a5fd7aaaad072a006b458635898a91968311679 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Thu, 13 Feb 2025 16:32:21 +0800 Subject: [PATCH] post step execution (#1765) --- skyvern/forge/agent.py | 1 + skyvern/forge/agent_functions.py | 3 +++ 2 files changed, 4 insertions(+) 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,