From 97f67feaa66b2a6fcfc5bbbb22a546b44723aa66 Mon Sep 17 00:00:00 2001 From: Suchintan Date: Thu, 7 Mar 2024 00:19:02 -0500 Subject: [PATCH] Fix agent webhook replay bug (#39) --- skyvern/forge/sdk/routes/agent_protocol.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index cd2e19f9..7d8d1d84 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -236,7 +236,7 @@ async def retry_webhook( ) -> TaskResponse: analytics.capture("skyvern-oss-agent-task-retry-webhook") agent = request["agent"] - task_obj = await agent.db.get_task(task_id, organization_id=current_org.organization_id) + task_obj = await app.DATABASE.get_task(task_id, organization_id=current_org.organization_id) if not task_obj: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, @@ -244,7 +244,7 @@ async def retry_webhook( ) # get latest step - latest_step = await agent.db.get_latest_step(task_id, organization_id=current_org.organization_id) + latest_step = await app.DATABASE.get_latest_step(task_id, organization_id=current_org.organization_id) if not latest_step: return task_obj.to_task_response()