Token-Budget Aware Reinforcement Learning Loss - Coding problem | OneNoughtOne