feat: Model to track node states and has_ended flag

[yuechao-qin]

TL;DR

Added an ExecutionSummary class to track execution state with comprehensive test coverage.

What changed?

• Created a new ExecutionSummary dataclass in api_server_sql.py that tracks:
  • Total number of nodes
  • Number of ended nodes
  • Whether the entire execution has ended
• Added a count_node_status method to update counters based on node status
• Created a new test file test_api_server_sql.py.

How to test?

Run the new test suite:

pytest tests/test_api_server_sql.py

Why make this change?

This will be used in upstream PRs to calculate execution summary for runs and root nodes.

[yuechao-qin]

• feat: GetExecutionNodes API returns number of total and ended nodes #99
• test: Add tests for GetGraphExecution API #98
• feat: API pipeline run get has execution summary #97
• feat: API pipeline run list has execution summary #96
• refactor: Function to get execution status stats #95
• feat: Model to track node states and has_ended flag #94 👈 (View in Graphite)
• master

This stack of pull requests is managed by Graphite. Learn more about stacking.

[yuechao-qin]

Merge activity

• Feb 21, 2:40 AM UTC: @yuechao-qin merged this pull request with Graphite.