vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix agent webhook replay bug (#39)

Browse Source
This commit is contained in:
Suchintan
2024-03-07 00:19:02 -05:00
committed by GitHub
parent 7085334607
commit 97f67feaa6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -236,7 +236,7 @@ async def retry_webhook(
) -> TaskResponse: ) -> TaskResponse:
analytics.capture("skyvern-oss-agent-task-retry-webhook") analytics.capture("skyvern-oss-agent-task-retry-webhook")
agent = request["agent"] 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: if not task_obj:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_404_NOT_FOUND, status_code=status.HTTP_404_NOT_FOUND,
@@ -244,7 +244,7 @@ async def retry_webhook(
) )
# get latest step # 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: if not latest_step:
return task_obj.to_task_response() return task_obj.to_task_response()