Update internal code generator to reflect the new layout

[leofang]

[leofang]

@vzhurba01 I will provide needed code change (as part of #91) for you to incorporate into the codegen 🙂

[vzhurba01]

Both the code gen and the public repo is now updated. Closing.