Save context after generating groundtruth in generateGroundtruth function

pelikhan · pelikhan · commit 39249c372a1e · 2025-07-24T16:05:13.000Z
diff --git a/cmd/generate/pipeline.go b/cmd/generate/pipeline.go
@@ -385,6 +385,8 @@ func (h *generateCommandHandler) generateGroundtruth(context *PromptPexContext)
 			}
 			test.Groundtruth = &output
 			test.GroundtruthModel = groundtruthModel
+
+			h.SaveContext(context) // Save context after generating groundtruth
 		}
 
 	}

Original file line number	Diff line number	Diff line change
`@@ -385,6 +385,8 @@ func (h generateCommandHandler) generateGroundtruth(context PromptPexContext)`
`385`	`385`	`}`
`386`	`386`	`test.Groundtruth = &output`
`387`	`387`	`test.GroundtruthModel = groundtruthModel`
	`388`	`+`
	`389`	`+ h.SaveContext(context) // Save context after generating groundtruth`
`388`	`390`	`}`
`389`	`391`
`390`	`392`	`}`