Matt Miller's book is filled with great ideas on how to bring AI into your classroom. But, his policy view is incredible because it's simple for any teacher to digest.
Use this diagram to help you think through what is best for your room. Still need a little help? Replace "AI" with "best friend," or "classmate." Now when you think about the spectrum above, it may be more clear.
Would you let your best friend do your work and you submit it? NO! (We at least hope not). Would you let your best friend proof read your answer and give you suggestions before turning in your assignment? Probably! That's really nice of them to help!