Regarding Kylo watching Ochi’s ship take off. I like the idea in theory, it’s an “Oh shit, they’re getting away!” kind of moment, but… why not just pull them towards himself?
He already accomplished what he set out to do there: make Rey lose control. Now he wants Rey to be alone long enough to wallow in that misery, so that she accepts him later on or straight up goes to him herself.
And yeah, the Knights were also in charge of tracking them.