Skip to content

Forward-merge release/0.8 into main - #903

Merged
GPUtester merged 1 commit into
mainfrom
release/0.8
Aug 26, 2026
Merged

Forward-merge release/0.8 into main#903
GPUtester merged 1 commit into
mainfrom
release/0.8

Conversation

@rapids-bot

@rapids-bot rapids-bot Bot commented Aug 26, 2026

Copy link
Copy Markdown

Forward-merge triggered by push to release/0.8 that creates a PR to keep main up-to-date. If this PR is unable to be immediately merged due to conflicts, it will remain open for the team to manually merge. See forward-merger docs for more info.

#### Overview

Add transparent forwarding for OpenAI image-generation requests so Codex Pets and other clients can use `POST /v1/images/generations` through the NeMo Relay gateway.

- [x] I confirm this contribution is my own work, or I have the right to submit it under this project's license.
- [x] I searched existing issues and open pull requests, and this does not duplicate existing work.

#### Details

- Register `/v1/images/generations` as an OpenAI gateway route.
- Reuse the configured OpenAI base URL, authentication normalization, query-string forwarding, and request-size limit.
- Forward image requests outside the managed text-generation pipeline because Relay does not have image request/response codecs.
- Preserve upstream status, headers, and response bytes without synthesizing LLM observability events.
- Add route, authentication, URL-normalization, and end-to-end forwarding coverage.

Validation:

- `just test-rust`
- `cargo fmt --all`
- `cargo clippy --workspace --all-targets -- -D warnings`
- `uv run pre-commit run --all-files`

#### Where should the reviewer start?

Start with `crates/cli/src/gateway/mod.rs` and the `images_generations` handler. The key design decision is to use transparent upstream forwarding rather than treating the distinct image API payload as a managed text-generation call without a compatible codec.

#### Related Issues: (use one of the action keywords Closes / Fixes / Resolves / Relates to)

- Relates to: [RELAY-788](https://linear.app/nvidia/issue/RELAY-788/codex-pets-error-with-nemo-relay)



## Summary by CodeRabbit

- **New Features**
  - Added support for OpenAI-compatible image generation requests.
  - Image-generation requests are transparently forwarded while preserving request details and authorization.
  - Added compatible OpenAI and ChatGPT authentication for image-generation requests.

- **Bug Fixes**
  - Codex gateway URLs now consistently include the required `/v1` path.
  - Existing Codex configurations with legacy or trailing-slash URLs remain supported.

- **Documentation**
  - Added upgrade guidance for refreshing persistent Codex integration settings.

- **Tests**
  - Added coverage for routing, forwarding, authentication, URL normalization, and upstream request handling.

Authors:
  - Will Killian (https://github.com/willkill07)

Approvers:
  - Bryan Bednarski (https://github.com/bbednarski9)

URL: #900
@rapids-bot
rapids-bot Bot requested review from a team as code owners August 26, 2026 03:13
@GPUtester
GPUtester merged commit 2beb454 into main Aug 26, 2026
2 checks passed
@rapids-bot

rapids-bot Bot commented Aug 26, 2026

Copy link
Copy Markdown
Author

SUCCESS - forward-merge complete.

@github-actions github-actions Bot added size:M PR is medium lang:rust PR changes/introduces Rust code labels Aug 26, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

lang:rust PR changes/introduces Rust code size:M PR is medium

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants