refactor: simplify Codex context handling