Closing loop - we merged in a PR that lifted the codex toml modification logic from the codex source code, so it should be safe to use. We weren't able to recreate cases were the previous approach broke our test cases, but by porting codex logic we figure is the best way to structurally ensure there aren't unforseen changes.
If you run into any other issues -- let us know.