fix import by AgentDiagramg

Ref gemini-fd api

upd made with AgentDiagram https://github.com/Lelekhoa1812/AgentDiagram