Python: support checkpoints for workflow orchestrations and sub-workflows (#863)

* Magentic checkpoint wip

* Magentic checkpoint updates

* Support checkpointing for magentic orchestration.

* Checkpointing for sub-workflows

* Use _execute_contexts instead of _pending_requests

* Remove unnecessary type ignores

* Support checkpoints for other orchestrations, refactor some code.

* Regenerate uv.lock

python/packages/main/tests/workflow/test_concurrent.py

@@ -18,6 +18,7 @@ from agent_framework import (
     WorkflowStatusEvent,
     handler,
 )
+from agent_framework._workflow._checkpoint import InMemoryCheckpointStorage
 
 
 class _FakeAgentExec(Executor):
@@ -156,3 +157,54 @@ def test_concurrent_custom_aggregator_uses_callback_name_for_id() -> None:
     assert "summarize" in wf.executors
     aggregator = wf.executors["summarize"]
     assert aggregator.id == "summarize"
+
+
+@pytest.mark.asyncio
+async def test_concurrent_checkpoint_resume_round_trip() -> None:
+    storage = InMemoryCheckpointStorage()
+
+    participants = (
+        _FakeAgentExec("agentA", "Alpha"),
+        _FakeAgentExec("agentB", "Beta"),
+        _FakeAgentExec("agentC", "Gamma"),
+    )
+
+    wf = ConcurrentBuilder().participants(list(participants)).with_checkpointing(storage).build()
+
+    baseline_output: list[ChatMessage] | None = None
+    async for ev in wf.run_stream("checkpoint concurrent"):
+        if isinstance(ev, WorkflowOutputEvent):
+            baseline_output = ev.data  # type: ignore[assignment]
+        if isinstance(ev, WorkflowStatusEvent) and ev.state == WorkflowRunState.IDLE:
+            break
+
+    assert baseline_output is not None
+
+    checkpoints = await storage.list_checkpoints()
+    assert checkpoints
+    checkpoints.sort(key=lambda cp: cp.timestamp)
+    resume_checkpoint = next(
+        (cp for cp in checkpoints if (cp.metadata or {}).get("checkpoint_type") == "superstep"),
+        checkpoints[-1],
+    )
+
+    resumed_participants = (
+        _FakeAgentExec("agentA", "Alpha"),
+        _FakeAgentExec("agentB", "Beta"),
+        _FakeAgentExec("agentC", "Gamma"),
+    )
+    wf_resume = ConcurrentBuilder().participants(list(resumed_participants)).with_checkpointing(storage).build()
+
+    resumed_output: list[ChatMessage] | None = None
+    async for ev in wf_resume.run_stream_from_checkpoint(resume_checkpoint.checkpoint_id):
+        if isinstance(ev, WorkflowOutputEvent):
+            resumed_output = ev.data  # type: ignore[assignment]
+        if isinstance(ev, WorkflowStatusEvent) and ev.state in (
+            WorkflowRunState.IDLE,
+            WorkflowRunState.IDLE_WITH_PENDING_REQUESTS,
+        ):
+            break
+
+    assert resumed_output is not None
+    assert [m.role for m in resumed_output] == [m.role for m in baseline_output]
+    assert [m.text for m in resumed_output] == [m.text for m in baseline_output]