Implement `agent_step_budget` in Python. Test cases verify edge cases and functional requirements.
agent_step_budget(*args, **kwargs)