You should take a look at guidance [0] to help with JSON returns from LLMs. Have been using it for some side projects and it seems to work pretty reliably.

[0] https://github.com/microsoft/guidance