Balancing AI Assistance with Human Input