Update cmd/generate/parser.go

pelikhan · Copilot · web-flow · commit 8f7da6c80a78 · 2025-07-25T03:13:05.000-07:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/cmd/generate/parser.go b/cmd/generate/parser.go
@@ -54,7 +54,7 @@ func (h *generateCommandHandler) ParseTestsFromLLMResponse(content string) ([]Pr
 	for _, rawTest := range rawTests {
 		test := PromptPexTest{}
 
-		for _, key := range []string{"testInput", "testinput", "testInput"} {
+		for _, key := range []string{"testInput", "testinput", "input"} {
 			if input, ok := rawTest[key].(string); ok {
 				test.Input = input
 				break
