Update cmd/cli/commands/run.go

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

Author: ericcurtin

cmd/cli/commands/run.go

@@ -206,7 +206,7 @@ func generateInteractiveWithReadline(cmd *cobra.Command, desktopClient *desktop.
 
 			// Create a cancellable context for the chat request
 			// This allows us to cancel the request if the user presses Ctrl+C during response generation
-			chatCtx, cancelChat := context.WithCancel(context.Background())
+			chatCtx, cancelChat := context.WithCancel(cmd.Context())
 
 			// Set up signal handler to cancel the context on Ctrl+C
 			sigChan := make(chan os.Signal, 1)