Skip to content

fix(BA-3721): Ensure request_id propagation for non-HTTP entry points#8160

Draft
jopemachine wants to merge 19 commits intofeat/BA-3721from
BA-3722
Draft

fix(BA-3721): Ensure request_id propagation for non-HTTP entry points#8160
jopemachine wants to merge 19 commits intofeat/BA-3721from
BA-3722

Conversation

@jopemachine
Copy link
Member

@jopemachine jopemachine commented Jan 20, 2026 •
edited
Loading

resolves #7749 (BA-3721)

Checklist: (if applicable)

  • Milestone metadata specifying the target backport version
  • Mention to the original issue
  • Installer updates including:
    • Fixtures for db schema changes
    • New mandatory config options
  • Update of end-to-end CLI integration tests in ai.backend.test
  • API server-client counterparts (e.g., manager API -> client SDK)
  • Test case(s) to:
    • Demonstrate the difference of before/after
    • Demonstrate the flow of abstract/conceptual models with a concrete implementation
  • Documentation
    • Contents in the docs directory
    • docstrings in public interfaces and type annotations

@github-actions github-actions bot added size:L 100~500 LoC comp:manager Related to Manager component comp:agent Related to Agent component comp:common Related to Common component comp:cli Related to CLI component labels Jan 20, 2026
@jopemachine jopemachine added this to the 26.1 milestone Jan 20, 2026
@jopemachine jopemachine changed the title fix(BA-3721): Add missing request_id propagation to message queue event fix(BA-3721): Ensure request_id propagation for event handlers, background tasks, and sweepers Jan 20, 2026
jopemachine and others added 4 commits January 20, 2026 06:53
@jopemachine @claude
fix: Add type annotations for headers dict and remove unused import
93c268a 
- Add explicit `dict[str, str]` type annotations to headers variables
  used with bind_request_id() to satisfy type checker
- Remove unused `current_request_id` import from agent/server.py

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
@jopemachine
fix: Add missing request_id propagation to message queue event
5c2ba74
@jopemachine
docs: Add news fragment
1540616
@jopemachine
fix: Apply ensure_request_id to bgtask
d9298c6
@jopemachine jopemachine changed the title fix(BA-3721): Ensure request_id propagation for event handlers, background tasks, and sweepers fix(BA-3721): Ensure request_id propagation for non-HTTP entry points Jan 20, 2026
@github-actions github-actions bot added size:M 30~100 LoC and removed size:L 100~500 LoC labels Jan 20, 2026
HyeockJinKim
HyeockJinKim reviewed Jan 20, 2026
View reviewed changes
src/ai/backend/common/bgtask/bgtask.py
last_message=last_message,
)

@ensure_request_id
Copy link
Collaborator

@HyeockJinKim HyeockJinKim Jan 20, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Middleware might not guarantee the use of the existing request ID.

@jopemachine jopemachine modified the milestones: 26.1, 26.2 Jan 26, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@HyeockJinKim HyeockJinKim HyeockJinKim left review comments

Assignees

@jopemachine jopemachine

Labels

comp:agent Related to Agent component comp:cli Related to CLI component comp:common Related to Common component comp:manager Related to Manager component size:M 30~100 LoC

Projects

None yet

Milestone

26.2

Development

Successfully merging this pull request may close these issues.

2 participants

@jopemachine @HyeockJinKim