There is a program called CT-Art[1], that uses "motifs" to train tactical sight for these sorts of things. Instead of next-move type puzzles where the moves are obvious, it gives you a game position several moves out from the targeted tactic, so you learn to recognize the conditions to be able to steer the game toward the tactical position. I think in it's current iteration it's broken out into separate courses or something but the older programs (v2 or v3 that I can vouch for) were really great for improving in these kinds of areas.

[1]: https://chesskingtraining.com/ct-art/