Right now since the repo is having two different implementations of codex, flake was updated to work with both typescript implementation and rust implementation
