Jupyter notebooks would be a great tool for AI agents. You have separate cell blocks for text prompting, code, and execution. All in one place and segmented.