Ask each last workflow stage to merge as soon as it finishes

2026-10-21 08:46:51 -07:00
parent b34f9c1511
commit 279919c6cd
6 changed files with 20 additions and 27 deletions
@@ -171,19 +171,22 @@ func (m *Manager) tryPromptReady(ctx context.Context, taskID string, fromReview
 	}
 	stagePrompt := ""
 	if task.WorkflowStage != nil {
+		frozen, err := m.store.TaskWorkflow(ctx, taskID)
+		if err != nil {
+			return err
+		}
+		output, _, err := m.store.StageResult(ctx, taskID)
+		if err != nil {
+			return err
+		}
+		if output != "" && len(workflow.Next(frozen.Definition, *task.WorkflowStage)) > 0 {
+			return nil
+		}
 		finished, err := m.store.WorkflowFinished(ctx, *task.WorkflowRun)
 		if err != nil {
 			return err
 		}
-		if !finished {
-			output, _, err := m.store.StageResult(ctx, taskID)
-			if err != nil || output != "" {
-				return err
-			}
-			frozen, err := m.store.TaskWorkflow(ctx, taskID)
-			if err != nil {
-				return err
-			}
+		if output == "" && !finished {
 			stagePrompt = finishStagePrompt(frozen, *task.WorkflowStage)
 		}
 	}