diff --git a/src/config/src/router.rs b/src/config/src/router.rs index be843142d9..3bf2d673a2 100644 --- a/src/config/src/router.rs +++ b/src/config/src/router.rs @@ -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" diff --git a/src/router/http/mod.rs b/src/router/http/mod.rs index 5b81c1c249..d81cb5647a 100644 --- a/src/router/http/mod.rs +++ b/src/router/http/mod.rs @@ -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"