What you're looking for can be easily done using the tool calling and prompting but I'd still recommend doing it post call.