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

fix int tests

Browse source
This commit is contained in:
Adil Hafeez 2024-10-18 14:35:38 -07:00
parent e2d9b6a008
commit a8111eb258
1 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -78,10 +78,10 @@ fn normal_flow(module: &mut Tester, filter_context: i32, http_context: i32) {
.expect_http_call( .expect_http_call(
Some("arch_internal"), Some("arch_internal"),
Some(vec![ Some(vec![
("x-arch-upstream", "model_server"), ("x-arch-upstream", "guard"),
(":method", "POST"), (":method", "POST"),
(":path", "/guard"), (":path", "/guard"),
(":authority", "model_server"), (":authority", "guard"),
("content-type", "application/json"), ("content-type", "application/json"),
("x-envoy-max-retries", "3"), ("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", "60000"), ("x-envoy-upstream-rq-timeout-ms", "60000"),
@ -119,10 +119,10 @@ fn normal_flow(module: &mut Tester, filter_context: i32, http_context: i32) {
.expect_http_call( .expect_http_call(
Some("arch_internal"), Some("arch_internal"),
Some(vec![ Some(vec![
("x-arch-upstream", "model_server"), ("x-arch-upstream", "embeddings"),
(":method", "POST"), (":method", "POST"),
(":path", "/embeddings"), (":path", "/embeddings"),
(":authority", "model_server"), (":authority", "embeddings"),
("content-type", "application/json"), ("content-type", "application/json"),
("x-envoy-max-retries", "3"), ("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", "60000"), ("x-envoy-upstream-rq-timeout-ms", "60000"),
@ -164,10 +164,10 @@ fn normal_flow(module: &mut Tester, filter_context: i32, http_context: i32) {
.expect_http_call( .expect_http_call(
Some("arch_internal"), Some("arch_internal"),
Some(vec![ Some(vec![
("x-arch-upstream", "model_server"), ("x-arch-upstream", "zeroshot"),
(":method", "POST"), (":method", "POST"),
(":path", "/zeroshot"), (":path", "/zeroshot"),
(":authority", "model_server"), (":authority", "zeroshot"),
("content-type", "application/json"), ("content-type", "application/json"),
("x-envoy-max-retries", "3"), ("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", "60000"), ("x-envoy-upstream-rq-timeout-ms", "60000"),
@ -249,10 +249,10 @@ fn setup_filter(module: &mut Tester, config: &str) -> i32 {
.expect_http_call( .expect_http_call(
Some("arch_internal"), Some("arch_internal"),
Some(vec![ Some(vec![
("x-arch-upstream", "model_server"), ("x-arch-upstream", "embeddings"),
(":method", "POST"), (":method", "POST"),
(":path", "/embeddings"), (":path", "/embeddings"),
(":authority", "model_server"), (":authority", "embeddings"),
("content-type", "application/json"), ("content-type", "application/json"),
("x-envoy-upstream-rq-timeout-ms", "60000"), ("x-envoy-upstream-rq-timeout-ms", "60000"),
]), ]),
@ -568,10 +568,10 @@ fn request_to_llm_gateway() {
.expect_http_call( .expect_http_call(
Some("arch_internal"), Some("arch_internal"),
Some(vec![ Some(vec![
("x-arch-upstream", "model_server"), ("x-arch-upstream", "hallucination"),
(":method", "POST"), (":method", "POST"),
(":path", "/hallucination"), (":path", "/hallucination"),
(":authority", "model_server"), (":authority", "hallucination"),
("content-type", "application/json"), ("content-type", "application/json"),
("x-envoy-max-retries", "3"), ("x-envoy-max-retries", "3"),
("x-envoy-upstream-rq-timeout-ms", "60000"), ("x-envoy-upstream-rq-timeout-ms", "60000"),