My understanding is that OK is correct. It’s also OK to use okay (which I often do as, though less correct, it looks less disruptive in a sentence) and it can be better to use okay when using it as a verb (so okayed rather than OKed or OK’d). Ok is always wrong though.