I use ai to write diagrams too. But I use d2 instead of mermaid