giers10/Heimgeist
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add explicit 'run_interrupted' event when streaming workflow events

Browse Source
This commit is contained in:
Victor Giers
2026-06-15 21:52:17 +02:00
parent d415dd7ead
commit 117a16aca5
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
backend/agent/api.py
View File

@@ -536,6 +536,7 @@ def get_workflow_events(run_id: str, after_sequence: int = Query(0, ge=0), follo
async def stream():
sequence = after_sequence
idle_after_terminal = 0
emitted_interrupted = False
while True:
db = SessionLocal()
try:
@@ -555,6 +556,17 @@ def get_workflow_events(run_id: str, after_sequence: int = Query(0, ge=0), follo
"payload": _json(item.payload_json, {}),
}, ensure_ascii=False) + "\n"
terminal = run.status in {"completed", "failed", "cancelled", "interrupted"}
if run.status == "interrupted" and not emitted_interrupted:
emitted_interrupted = True
yield json.dumps({
"run_id": run.id,
"workflow_id": run.workflow_id,
"node_id": None,
"sequence": sequence + 1,
"timestamp": (run.finished_at or datetime.utcnow()).isoformat(),
"type": "run_interrupted",
"payload": _json(run.error_json, {"message": "Workflow run was interrupted."}),
}, ensure_ascii=False) + "\n"
finally:
db.close()
if not follow: