Update cmd/cli/commands/run.go

Co-authored-by: Copilot <[email protected]>

Author: ericcurtin

cmd/cli/commands/run.go

@@ -266,7 +266,7 @@ func generateInteractiveBasic(cmd *cobra.Command, desktopClient *desktop.Client,
 
 		// 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)