Made tool use work! check out demo here: https://x.com/chxy/status/1861684254297727299