Updated Codex Query prototype for new search API