Optimized API for multi-step AI programs