.NET: Add checkpoint on super step started event: issue #4280 by elgold92 · Pull Request #4604 · microsoft/agent-framework

elgold92 · 2026-03-10T20:11:02Z

Motivation and Context

Address issue #4280, allowing workflows to resume from checkpoints saved from SuperStepStarted events.

Description

Adds CheckpointInfo? field to the SuperStepStartInfo class, populating this information in the InProcessRunner and InProcStepTracer. Also updates associated unit tests to expect more checkpoints to be created on checkpointed workflows.

…ely b/c the test definitions are now wrong

…c definition. I think this is correct, but the test now hangs, likely because the first checkpoint (or it's related data structures) we're now creating aren't initialized correctly.

…ShouldPersist_CheckpointedAsync test definition

Copilot

Pull request overview

Adds support for creating/checkpointing workflow state at the SuperStepStarted boundary so runs can resume from “pre-delivery” checkpoints (addressing #4280), and updates tests accordingly.

Changes:

Add a CheckpointInfo? field to SuperStepStartInfo and populate it on SuperStepStartedEvent.
Create a checkpoint at the start of each superstep (capturing pre-delivery queued messages) by extending runner state export to accept an override StepContext.
Update and expand unit tests to account for the additional checkpoints and validate parent chaining/resume behavior.

Reviewed changes

Copilot reviewed 6 out of 6 changed files in this pull request and generated 6 comments.

Show a summary per file

File	Description
dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/InProcessStateTests.cs	Updates expected checkpoint count due to start+end checkpointing per superstep.
dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/CheckpointParentTests.cs	Extends tests to include checkpoints emitted on `SuperStepStartedEvent` and adds new resume/count assertions.
dotnet/src/Microsoft.Agents.AI.Workflows/SuperStepStartInfo.cs	Adds `Checkpoint` property to expose the checkpoint emitted at superstep start.
dotnet/src/Microsoft.Agents.AI.Workflows/InProc/InProcessRunnerContext.cs	Allows exporting runner state from an alternate `StepContext` (pre-delivery snapshot).
dotnet/src/Microsoft.Agents.AI.Workflows/InProc/InProcessRunner.cs	Saves a checkpoint before superstep execution and wires it into the started event.
dotnet/src/Microsoft.Agents.AI.Workflows/InProc/InProcStepTracer.cs	Plumbs the start-checkpoint into `SuperStepStartedEvent` payload.

dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/CheckpointParentTests.cs

dotnet/src/Microsoft.Agents.AI.Workflows/SuperStepStartInfo.cs

dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/CheckpointParentTests.cs

…tParentTests.cs rename local variable Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

elgold92 · 2026-03-10T21:10:39Z

@copilot open a new pull request to apply changes based on the comments in this thread. Changes look to be generally minor improvements.

Copilot

Pull request overview

Copilot reviewed 6 out of 6 changed files in this pull request and generated 4 comments.

dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/CheckpointParentTests.cs

…r changes

Copilot

Pull request overview

Copilot reviewed 6 out of 6 changed files in this pull request and generated 2 comments.

dotnet/src/Microsoft.Agents.AI.Workflows/InProc/InProcessRunner.cs

dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/CheckpointParentTests.cs

…b.com/elgold92/agent-framework into ericgold/CheckpointOnSuperStepStarted

Copilot

Pull request overview

Copilot reviewed 6 out of 6 changed files in this pull request and generated 5 comments.

dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/CheckpointParentTests.cs

…tParentTests.cs Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Copilot

Pull request overview

Copilot reviewed 6 out of 6 changed files in this pull request and generated no new comments.

Eric Gold added 7 commits March 9, 2026 16:53

Add Checkpoint member to SuperStepStartInfo

a5cf61c

initial proposed solution from Claude. Causes unit tests to fail, lik…

a81c3e2

…ely b/c the test definitions are now wrong

Fix Checkpoint_FirstCheckpoint_ShouldHaveNullParentAsync definition

d1b4449

fix Checkpoint_SubsequentCheckpoints_ShouldChainParentsAsync definition

151f7db

update Checkpoint_AfterResume_ShouldHaveResumedCheckpointAsParentAsyn…

30380d6

…c definition. I think this is correct, but the test now hangs, likely because the first checkpoint (or it's related data structures) we're now creating aren't initialized correctly.

All unit tests now passing.

61fb686

Add a couple more unit tests as well as fix failed InProcessRun_State…

4c13736

…ShouldPersist_CheckpointedAsync test definition

Copilot AI review requested due to automatic review settings March 10, 2026 20:11

markwallace-microsoft added .NET workflows Related to Workflows in agent-framework labels Mar 10, 2026

github-actions bot changed the title ~~Add checkpoint on super step started event: issue #4280~~ .NET: Add checkpoint on super step started event: issue #4280 Mar 10, 2026

Copilot started reviewing on behalf of elgold92 March 10, 2026 20:12 View session

Copilot AI reviewed Mar 10, 2026

View reviewed changes

Update dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/Checkpoin…

1f859cc

…tParentTests.cs rename local variable Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Copilot AI review requested due to automatic review settings March 10, 2026 21:07

Copilot started reviewing on behalf of elgold92 March 10, 2026 21:08 View session

Copilot AI reviewed Mar 10, 2026

View reviewed changes

Eric Gold and others added 2 commits March 10, 2026 16:35

Address GitHubCopilot's comments on CheckpointParentTest.cs. All mino…

4c44228

…r changes

Merge branch 'main' into ericgold/CheckpointOnSuperStepStarted

71f50f0

Copilot AI review requested due to automatic review settings March 10, 2026 23:50

Copilot started reviewing on behalf of elgold92 March 10, 2026 23:51 View session

Copilot AI reviewed Mar 10, 2026

View reviewed changes

dotnet/src/Microsoft.Agents.AI.Workflows/InProc/InProcessRunner.cs Show resolved Hide resolved

dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/CheckpointParentTests.cs Outdated Show resolved Hide resolved

Eric Gold and others added 3 commits March 10, 2026 17:29

minor suggestions from Copilot

9f4de94

Merge branch 'ericgold/CheckpointOnSuperStepStarted' of https://githu…

90d7b08

…b.com/elgold92/agent-framework into ericgold/CheckpointOnSuperStepStarted

Merge branch 'main' into ericgold/CheckpointOnSuperStepStarted

3bf2b1c

Copilot AI review requested due to automatic review settings March 11, 2026 00:32

Copilot started reviewing on behalf of elgold92 March 11, 2026 00:33 View session

Copilot AI reviewed Mar 11, 2026

View reviewed changes

resolve 1 more of Copilot's test comments

386a5df

Update dotnet/tests/Microsoft.Agents.AI.Workflows.UnitTests/Checkpoin…

2fcf732

…tParentTests.cs Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Copilot AI review requested due to automatic review settings March 11, 2026 04:00

Copilot started reviewing on behalf of elgold92 March 11, 2026 04:01 View session

Copilot AI reviewed Mar 11, 2026

View reviewed changes

elgold92 added 2 commits March 10, 2026 21:25

Merge branch 'main' into ericgold/CheckpointOnSuperStepStarted

f797794

Merge branch 'main' into ericgold/CheckpointOnSuperStepStarted

3b8cb64

Conversation

elgold92 commented Mar 10, 2026

Motivation and Context

Description

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

elgold92 commented Mar 10, 2026

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants