You should try writing a design doc with the AI before you have it write the code. It will make random extrapolations from your first prompt, some good and some bad. Then you get a chance to argue back and forth with yourself with the robot as a helper.
“Are there any ambiguities, open questions, or hidden assumptions in spec? Please think deeply.”
Oh yes! I find an argumentative rubber duck to be one of the best uses of AI for myself!