1 files changed, 33 insertions, 35 deletions
diff --git a/pkg/aflow/flow/patching/patching.go b/pkg/aflow/flow/patching/patching.go
index a0b8d4a6c..9959e0bdb 100644
--- a/pkg/aflow/flow/patching/patching.go
+++ b/pkg/aflow/flow/patching/patching.go
@@ -45,42 +45,40 @@ func init() {
 		ai.WorkflowPatching,
 		"generate a kernel patch fixing a provided bug reproducer",
 		&aflow.Flow{
-			Root: &aflow.Pipeline{
-				Actions: []aflow.Action{
-					baseCommitPicker,
-					kernel.Checkout,
-					kernel.Build,
-					// Ensure we can reproduce the crash (and the build boots).
-					crash.Reproduce,
-					codesearcher.PrepareIndex,
-					&aflow.LLMAgent{
-						Name:        "debugger",
-						Model:       aflow.BestExpensiveModel,
-						Reply:       "BugExplanation",
-						Temperature: 1,
-						Instruction: debuggingInstruction,
-						Prompt:      debuggingPrompt,
-						Tools:       tools,
-					},
-					&aflow.LLMAgent{
-						Name:        "diff-generator",
-						Model:       aflow.BestExpensiveModel,
-						Reply:       "PatchDiff",
-						Temperature: 1,
-						Instruction: diffInstruction,
-						Prompt:      diffPrompt,
-						Tools:       tools,
-					},
-					&aflow.LLMAgent{
-						Name:        "description-generator",
-						Model:       aflow.BestExpensiveModel,
-						Reply:       "PatchDescription",
-						Temperature: 1,
-						Instruction: descriptionInstruction,
-						Prompt:      descriptionPrompt,
-					},
+			Root: aflow.Pipeline(
+				baseCommitPicker,
+				kernel.Checkout,
+				kernel.Build,
+				// Ensure we can reproduce the crash (and the build boots).
+				crash.Reproduce,
+				codesearcher.PrepareIndex,
+				&aflow.LLMAgent{
+					Name:        "debugger",
+					Model:       aflow.BestExpensiveModel,
+					Reply:       "BugExplanation",
+					Temperature: 1,
+					Instruction: debuggingInstruction,
+					Prompt:      debuggingPrompt,
+					Tools:       tools,
 				},
-			},
+				&aflow.LLMAgent{
+					Name:        "diff-generator",
+					Model:       aflow.BestExpensiveModel,
+					Reply:       "PatchDiff",
+					Temperature: 1,
+					Instruction: diffInstruction,
+					Prompt:      diffPrompt,
+					Tools:       tools,
+				},
+				&aflow.LLMAgent{
+					Name:        "description-generator",
+					Model:       aflow.BestExpensiveModel,
+					Reply:       "PatchDescription",
+					Temperature: 1,
+					Instruction: descriptionInstruction,
+					Prompt:      descriptionPrompt,
+				},
+			),
 		},
 	)
 }