Reasoning Token Allocation

There is a problem where LLMs spend a lot of reasoning tokens if not capped, getting stuck in loops or over-optimizng. The key of this envornment is for an agent to learn to set a cap on reasoning token spending and achieve near same accuracy on gsm8k dataset