I think this is roughly solved. Tell the agent to do all of its calculations in python.