Do you wait for the AI while it works if you are a lawyer?

I think it's a good approach. It allows you to think and relax. You don't need to multi-task. Just be engaged with the AI like you would a person. It will improve the output.