Add callbacks for streaming and one-shot uploads