Allow tracking function environments for partially processed ASTs