After the AI writes code you'll keep, ask it to document:
"Document this function in JSDoc format. Include: a one-sentence summary, each parameter with type and valid values, return type and possible values, usage examples, and any side effects."
Specific documentation requirements produce specific documentation. "Add comments" produces generic comments.