Can't you just force it to do structured output via constrained generation?