The short answer
Keep responsibility for the purpose, audience, main idea and reasons for choosing a direction. Use AI to explore or draft, then compare the result with those decisions. Explain what you kept, rejected or changed before treating the content as ready.
A closer look
Try writing a short explanation of the idea before you request finished prose, or describe what remains uncertain if you are still exploring. After reviewing AI's suggestions, state why a particular direction deserves further work. Return to that explanation when the draft sounds polished but no longer makes the point you intended.
For practice, occasionally outline a different approach without looking at the generated draft and compare the choices. This is a practical exercise, not a proven cure for a measured loss of creativity. The distinction between producing an answer and understanding the work is explored in learning with AI without giving up the thinking.