fix compute exec stream close handling

cursoragent · cursoragent · commit ec9182b51565 · 2026-03-10T09:19:09.000Z
diff --git a/pkg/cmd/compute/exec.go b/pkg/cmd/compute/exec.go
@@ -68,6 +68,7 @@ func newComputeExec() *cobra.Command {
 				// nolint:wrapcheck
 				return err
 			}
+			defer stream.Close()
 
 			for stream.Receive() {
 				msg := stream.Msg()
@@ -78,6 +79,7 @@ func newComputeExec() *cobra.Command {
 					fmt.Fprint(os.Stderr, v.Stderr)
 				case *civ1.ExecuteCommandResponse_ExitCode:
 					if v.ExitCode != 0 {
+						stream.Close()
 						os.Exit(int(v.ExitCode))
 					}
 				}

Original file line number	Diff line number	Diff line change
`@@ -68,6 +68,7 @@ func newComputeExec() *cobra.Command {`
`68`	`68`	`// nolint:wrapcheck`
`69`	`69`	`return err`
`70`	`70`	`}`
	`71`	`+ defer stream.Close()`
`71`	`72`
`72`	`73`	`for stream.Receive() {`
`73`	`74`	`msg := stream.Msg()`
`@@ -78,6 +79,7 @@ func newComputeExec() *cobra.Command {`
`78`	`79`	`fmt.Fprint(os.Stderr, v.Stderr)`
`79`	`80`	`case *civ1.ExecuteCommandResponse_ExitCode:`
`80`	`81`	`if v.ExitCode != 0 {`
	`82`	`+ stream.Close()`
`81`	`83`	`os.Exit(int(v.ExitCode))`
`82`	`84`	`}`
`83`	`85`	`}`