Improving context retention for long-form AI contracts