Diagram-wise I had better luck in telling it to make a text version for diagram generator, not only AI edits tend to be more correct, it's easier to fix any errors too.

As a bonus you can then prompt AI to validate whether the diagram matches article and vice versa, I've caught some small bugs like that when we migrated our docs