Skip to content

Add CheckPrerequisite to WorkflowStep to validate state machine edges#374

Open
Zoe Zhao (zoez7) wants to merge 2 commits into
agent-substrate:mainfrom
zoez7:check-prerequisite
Open

Add CheckPrerequisite to WorkflowStep to validate state machine edges#374
Zoe Zhao (zoez7) wants to merge 2 commits into
agent-substrate:mainfrom
zoez7:check-prerequisite

Conversation

@zoez7

@zoez7 Zoe Zhao (zoez7) commented Jul 2, 2026

Copy link
Copy Markdown
Collaborator

Fixes #369

Add a CheckPrerequisite method to the WorkflowStep interface, called by RunWorkflow after IsComplete returns false and before Execute, so that each workflow validates its actor state-machine edge up front while retried (reentrant) workflows still fast-forward past completed steps.

@zoez7 Zoe Zhao (zoez7) force-pushed the check-prerequisite branch 4 times, most recently from 9e8acdf to 3921ac8 Compare July 8, 2026 23:55
}
func (s *AssignWorkerStep) CheckPrerequisite(ctx context.Context, input *ResumeInput, state *ResumeState) error {
// The resume edge exists from SUSPENDED and PAUSED.
// RESUMING is allowed for retrying this step.

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is because of line 168 below.

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

could you please clarify what is special at line 168? Is it for cases it was an error in resuming process and we are OK to try resuming one more time?

Same case might happening with Pausing and the CheckPrerequisite for CallAteletPauseStep does not accept pausing state.

func (s *FinalizeSuspendedStep) Name() string { return "FinalizeSuspended" }
func (s *FinalizeSuspendedStep) IsComplete(ctx context.Context, input *SuspendInput, state *SuspendState) (bool, error) {
// The workflow is completely done ONLY if the status is SUSPENDED *and* we've successfully freed the worker.
return state.Actor.GetStatus() == ateapipb.Actor_STATUS_SUSPENDED && state.Actor.GetAteomPodNamespace() == "", nil

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I removed this check, because we actually update the Actor's status and worker assignment in the same UpdateActor request in Execute() below, so we only need to check if status==suspended.

@zoez7 Zoe Zhao (zoez7) marked this pull request as ready for review July 8, 2026 23:58
return nil
}
func (s *CallAteletPauseStep) Execute(ctx context.Context, input *PauseInput, state *PauseState) error {
if state.Actor.GetAteomPodNamespace() == "" || state.Actor.GetAteomPodName() == "" {

@dberkov Dmitry Berkovich (dberkov) Jul 10, 2026

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this is a pre-requisite too, however function in current implementation does not support transition to crashActor.

Might be rename the "CheckPrerequisite" to different name? might be allow transition to crash? or might be from the beginning we were not supposed to be in this state, that actor does not have pod or namespace?

// seeded actor statuses, covering both the rejected and the idempotent-success
// paths. The atelet dialer is nil, so any step that unexpectedly reaches it
// panics.
func TestPauseActorWorkflow(t *testing.T) {

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TestPauseSteps_CheckPrerequisite validates Prerequisite that matches name of the test.
In this case, the name is generic for entire Step, however it validates CheckPrerequisite transition only.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Actor state machine] Add CheckPrerequisite steps to WorkflowStep definition

3 participants