fix: pass DagStatusData model instead of dict to append_event_and_broadcast

add_event() calls .model_dump() on data, so it needs a Pydantic model not a dict.
This commit is contained in:
Igor Loskutov
2026-02-09 14:00:43 -05:00
parent 6dd96bfa5e
commit 2410688559
2 changed files with 4 additions and 3 deletions

View File

@@ -216,7 +216,7 @@ async def broadcast_dag_status(transcript_id: str, workflow_run_id: str) -> None
transcript_id,
transcript,
"DAG_STATUS",
dag_status.model_dump(mode="json"),
dag_status,
logger,
)
except Exception:

View File

@@ -676,8 +676,9 @@ class TestBroadcastDagStatus:
assert call_args[0][1] is mock_transcript # transcript
assert call_args[0][2] == "DAG_STATUS" # event_name
data = call_args[0][3]
assert data["workflow_run_id"] == "wf-abc"
assert len(data["tasks"]) == 1
assert isinstance(data, DagStatusData)
assert data.workflow_run_id == "wf-abc"
assert len(data.tasks) == 1
@pytest.mark.asyncio
async def test_swallows_exceptions(self):