Update cmd/cli/commands/run.go

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: ericcurtin

cmd/cli/commands/run.go

@@ -272,15 +272,17 @@ func generateInteractiveBasic(cmd *cobra.Command, desktopClient *desktop.Client,
 		sigChan := make(chan os.Signal, 1)
 		signal.Notify(sigChan, syscall.SIGINT)
 		go func() {
-			<-sigChan
-			cancelChat()
+			select {
+			case <-sigChan:
+				cancelChat()
+			case <-chatCtx.Done():
+			}
 		}()
 
 		err = chatWithMarkdownContext(chatCtx, cmd, desktopClient, backend, model, userInput, apiKey)
 
 		// Clean up signal handler
 		signal.Stop(sigChan)
-		close(sigChan)
 		cancelChat()
 
 		if err != nil {