dataset agent updates for the migration to mi