Часто помогает разбивать код реально мелко, чтобы AI успевал "впитывать" контекст по шагам. Еще лучше — сразу класть комментарии, что именно должен делать каждый блок, иначе модель быстро теряется. Иногда сохраняю промежуточные результаты в текст, чтобы потом подсовывать вместе с кодом — вроде как память подгоняет. Но идеального способа пока не нашел, приходится на глаз балансировать.