litellm/tests/test_litellm_rust/messages
devin-ai-integration[bot] 571ada0b0f
feat(rust_bridge): mark native streams with the x-litellm-rust header (#42758)
Non-streaming responses served by the Rust core already carry
x-litellm-rust: true through _hidden_params.additional_headers, which the
SDK exposes and the gateway renders as a response header. Native streams
did not, because the lifecycle Stream and SyncStream objects had nowhere
to hold hidden params and the marker writer skips objects without them.

Give both stream classes the same _hidden_params bag every other litellm
response has, so the existing marker attaches without wrapping the stream
or changing its identity.

Co-authored-by: Yujong Lee <yujong@berri.ai>
2026-09-24 09:36:25 -07:00
..
__init__.py add streaming message 2026-09-18 15:45:08 -07:00
test_callbacks.py feat(rust_bridge): mark native streams with the x-litellm-rust header (#42758) 2026-09-24 09:36:25 -07:00