fix: include search_multi_stream in is_streaming_endpoint function (#11194)

Co-authored-by: Hengfei Yang <hengfei.yang@gmail.com>
This commit is contained in:
Subhra264
2026-04-09 08:03:19 +00:00
committed by GitHub
co-authored by Hengfei Yang
parent 4f8d50a474
commit 8d1fc4e05d
2 changed files with 6 additions and 0 deletions
+4
View File
@@ -164,8 +164,12 @@ mod tests {
fn test_is_querier_route_by_body() {
assert!(is_querier_route_by_body("/_search"));
assert!(is_querier_route_by_body("/_search?foo=bar"));
assert!(is_querier_route_by_body("/_search_multi_stream"));
assert!(is_querier_route_by_body("/_search_multi_stream?foo=bar"));
assert!(is_querier_route_by_body("/_search_stream"));
assert!(is_querier_route_by_body("/_search_stream?foo=bar"));
assert!(is_querier_route_by_body("/_values_stream"));
assert!(is_querier_route_by_body("/_values_stream?foo=bar"));
assert!(is_querier_route_by_body("/prometheus/api/v1/query_range"));
assert!(is_querier_route_by_body(
"/prometheus/api/v1/query_exemplars"
+2
View File
@@ -615,6 +615,7 @@ fn build_request_headers(headers: &HeaderMap, is_streaming: bool) -> reqwest::he
const STREAMING_ENDPOINTS: &[&str] = &[
"/_search_stream",
"/_values_stream",
"/_search_multi_stream",
"/ai/chat_stream",
"/prometheus/api/v1/query_range",
];
@@ -685,6 +686,7 @@ mod tests {
fn test_is_streaming_endpoint() {
assert!(is_streaming_endpoint("/api/org/_search_stream"));
assert!(is_streaming_endpoint("/api/org/_values_stream"));
assert!(is_streaming_endpoint("/api/org/_search_multi_stream"));
assert!(is_streaming_endpoint("/api/org/ai/chat_stream"));
assert!(is_streaming_endpoint(
"/api/org/prometheus/api/v1/query_range"