400 - 128 = 272. Codex cli source.
If you want to be able to generate up to 128k tokens in one go successfully, then yes, that math checks out.
If you want to be able to generate up to 128k tokens in one go successfully, then yes, that math checks out.