Codex recently built this in - "Prevent sleep while running" toggle