From a8111eb25802f91393fbdcf7a2010572d027bfb9 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil@katanemo.com>
Date: Fri, 18 Oct 2024 14:35:38 -0700
Subject: [PATCH] fix int tests

---
 crates/prompt_gateway/tests/integration.rs | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/crates/prompt_gateway/tests/integration.rs b/crates/prompt_gateway/tests/integration.rs
index 14ca1aa2..7f7322f8 100644
--- a/crates/prompt_gateway/tests/integration.rs
+++ b/crates/prompt_gateway/tests/integration.rs
@@ -78,10 +78,10 @@ fn normal_flow(module: &mut Tester, filter_context: i32, http_context: i32) {
         .expect_http_call(
             Some("arch_internal"),
             Some(vec![
-                ("x-arch-upstream", "model_server"),
+                ("x-arch-upstream", "guard"),
                 (":method", "POST"),
                 (":path", "/guard"),
-                (":authority", "model_server"),
+                (":authority", "guard"),
                 ("content-type", "application/json"),
                 ("x-envoy-max-retries", "3"),
                 ("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(
             Some("arch_internal"),
             Some(vec![
-                ("x-arch-upstream", "model_server"),
+                ("x-arch-upstream", "embeddings"),
                 (":method", "POST"),
                 (":path", "/embeddings"),
-                (":authority", "model_server"),
+                (":authority", "embeddings"),
                 ("content-type", "application/json"),
                 ("x-envoy-max-retries", "3"),
                 ("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(
             Some("arch_internal"),
             Some(vec![
-                ("x-arch-upstream", "model_server"),
+                ("x-arch-upstream", "zeroshot"),
                 (":method", "POST"),
                 (":path", "/zeroshot"),
-                (":authority", "model_server"),
+                (":authority", "zeroshot"),
                 ("content-type", "application/json"),
                 ("x-envoy-max-retries", "3"),
                 ("x-envoy-upstream-rq-timeout-ms", "60000"),
@@ -249,10 +249,10 @@ fn setup_filter(module: &mut Tester, config: &str) -> i32 {
         .expect_http_call(
             Some("arch_internal"),
             Some(vec![
-                ("x-arch-upstream", "model_server"),
+                ("x-arch-upstream", "embeddings"),
                 (":method", "POST"),
                 (":path", "/embeddings"),
-                (":authority", "model_server"),
+                (":authority", "embeddings"),
                 ("content-type", "application/json"),
                 ("x-envoy-upstream-rq-timeout-ms", "60000"),
             ]),
@@ -568,10 +568,10 @@ fn request_to_llm_gateway() {
         .expect_http_call(
             Some("arch_internal"),
             Some(vec![
-                ("x-arch-upstream", "model_server"),
+                ("x-arch-upstream", "hallucination"),
                 (":method", "POST"),
                 (":path", "/hallucination"),
-                (":authority", "model_server"),
+                (":authority", "hallucination"),
                 ("content-type", "application/json"),
                 ("x-envoy-max-retries", "3"),
                 ("x-envoy-upstream-rq-timeout-ms", "60000"),