handle workflow run cancel for child workflow runs / task v2 + observer cancel handling (#1776)

skyvern/forge/sdk/routes/agent_protocol.py

@@ -339,6 +339,19 @@ async def cancel_workflow_run(
             status_code=status.HTTP_404_NOT_FOUND,
             detail=f"Workflow run not found {workflow_run_id}",
         )
+    # get all the child workflow runs and cancel them
+    child_workflow_runs = await app.DATABASE.get_workflow_runs_by_parent_workflow_run_id(
+        organization_id=current_org.organization_id,
+        parent_workflow_run_id=workflow_run_id,
+    )
+    for child_workflow_run in child_workflow_runs:
+        if child_workflow_run.status not in [
+            WorkflowRunStatus.running,
+            WorkflowRunStatus.created,
+            WorkflowRunStatus.queued,
+        ]:
+            continue
+        await app.WORKFLOW_SERVICE.mark_workflow_run_as_canceled(child_workflow_run.workflow_run_id)
     await app.WORKFLOW_SERVICE.mark_workflow_run_as_canceled(workflow_run_id)
     await app.WORKFLOW_SERVICE.execute_workflow_webhook(workflow_run, api_key=x_api_key)