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

WYV-1186 Create API to replay webhook callbacks (#4)

Browse Source
This commit is contained in:
Suchintan
2024-03-02 00:38:32 -05:00
committed by GitHub
parent 9eddb3d812
commit 861114794a
3 changed files with 88 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
skyvern/forge/agent.py
View File

@@ -692,16 +692,19 @@ class ForgeAgent(Agent):
# Wait for all tasks to complete before generating the links for the artifacts
await app.ARTIFACT_MANAGER.wait_for_upload_aiotasks_for_task(task.task_id)
if not task.webhook_callback_url:
await self.execute_task_webhook(task=task, last_step=last_step, api_key=api_key)
async def execute_task_webhook(self, task: Task, last_step: Step, api_key: str | None) -> None:
if not api_key:
LOG.warning(
"Task has no webhook callback url. Not sending task response",
"Request has no api key. Not sending task response",
task_id=task.task_id,
)
return
if not api_key:
if not task.webhook_callback_url:
LOG.warning(
"Request has no api key. Not sending task response",
"Task has no webhook callback url. Not sending task response",
task_id=task.task_id,
)
return
@@ -732,6 +735,7 @@ class ForgeAgent(Agent):
if not task_from_db:
LOG.error("Failed to get task from db when sending task response")
raise TaskNotFound(task_id=task.task_id)
task = task_from_db
if not task.webhook_callback_url:
LOG.info("Task has no webhook callback url. Not sending task response")