oh man i totally get this, it's so annoying when they just completely zone out. what usually helps me is making sure my initial system prompt is really detailed about key lore/char facts. also, if your platform has a 'remember' or 'key memories' feature, load it up. sometimes i just manually summarize the last few important turns and stick that at the start of my next message too. keeps them on track usually. good luck!