apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix tests

Browse source
This commit is contained in:
Adil Hafeez 2024-11-07 14:35:43 -08:00
parent 4d9cadd60d
commit 270ff1f5de
2 changed files with 2 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

6
crates/prompt_gateway/src/stream_context.rs
View file

@ -170,7 +170,6 @@ impl StreamContext {
("content-type", "application/json"),
("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", "60000"),
("x-envoy-force-trace", "true"),
];
if self.request_id.is_some() {
@ -300,7 +299,6 @@ impl StreamContext {
("content-type", "application/json"),
("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", "60000"),
("x-envoy-force-trace", "true"),
];
if self.request_id.is_some() {
@ -511,7 +509,6 @@ impl StreamContext {
("content-type", "application/json"),
("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", timeout_str.as_str()),
("x-envoy-force-trace", "true"),
];
if self.request_id.is_some() {
@ -663,7 +660,6 @@ impl StreamContext {
("content-type", "application/json"),
("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", timeout_str.as_str()),
("x-envoy-force-trace", "true"),
];
if self.request_id.is_some() {
@ -844,7 +840,6 @@ impl StreamContext {
("content-type", "application/json"),
("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", "60000"),
("x-envoy-force-trace", "true"),
];
if self.request_id.is_some() {
@ -900,7 +895,6 @@ impl StreamContext {
(":authority", endpoint.name.as_str()),
("content-type", "application/json"),
("x-envoy-max-retries", "3"),
("x-envoy-force-trace", "true"),
];
if self.request_id.is_some() {

2
crates/prompt_gateway/tests/integration.rs
View file

@ -36,6 +36,8 @@ fn request_headers_expectations(module: &mut Tester, http_context: i32) {
.expect_log(Some(LogLevel::Trace), None)
.expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some("x-request-id"))
.returning(None)
.expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some("traceparent"))
.returning(None)
.execute_and_expect(ReturnType::Action(Action::Continue))
.unwrap();
}