Skip to content

Forward-merge release/0.6 into main#472

Merged
GPUtester merged 1 commit into
mainfrom
release/0.6
Jul 17, 2026
Merged

Forward-merge release/0.6 into main#472
GPUtester merged 1 commit into
mainfrom
release/0.6

Conversation

@rapids-bot

@rapids-bot rapids-bot Bot commented Jul 17, 2026

Copy link
Copy Markdown

Forward-merge triggered by push to release/0.6 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

Resolve RELAY-499 by snapshotting ATOF stream-sink headers sourced from `header_env` when the endpoint is activated.

- [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

- Resolve `headers` and `header_env` once before the endpoint worker is spawned.
- Reuse the activated header map for HTTP POST, WebSocket reconnects, and NDJSON uploads.
- Add core and Python regressions that remove the environment variable after activation and verify all lifecycle events retain the Authorization header.

#### Where should the reviewer start?

Start with `crates/core/src/observability/atof.rs`, where `ActivatedAtofEndpoint` carries the validated configuration and snapshotted headers into each transport worker.

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

- Relates to: RELAY-499



## Summary by CodeRabbit

* **Bug Fixes**
  * Improved streaming observability endpoint setup by validating configurations and resolving headers before workers start.
  * Ensured environment-based headers are captured at activation time and remain available for outgoing requests.
  * Applied consistent header handling across HTTP, WebSocket, and NDJSON transports.
* **Tests**
  * Added coverage verifying environment-based authorization headers are preserved without exposing credentials in reports, payloads, or logs.

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

Approvers:
  - Maryam Najafian (https://github.com/mnajafian-nv)

URL: #463
@rapids-bot
rapids-bot Bot requested a review from a team as a code owner July 17, 2026 17:39
@GPUtester
GPUtester merged commit 27831c4 into main Jul 17, 2026
1 check passed
@rapids-bot

rapids-bot Bot commented Jul 17, 2026

Copy link
Copy Markdown
Author

SUCCESS - forward-merge complete.

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

Labels

lang:python PR changes/introduces Python code 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