Data management for AI agents