From e2edeedf555348ea4c05917c24d0228a4f73aa19 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil.hafeez@gmail.com>
Date: Fri, 28 Mar 2025 03:47:09 -0700
Subject: [PATCH] fix rust tests

---
 crates/prompt_gateway/tests/integration.rs | 32 +++++-----------------
 1 file changed, 7 insertions(+), 25 deletions(-)

diff --git a/crates/prompt_gateway/tests/integration.rs b/crates/prompt_gateway/tests/integration.rs
index bbde10b0..91b36c01 100644
--- a/crates/prompt_gateway/tests/integration.rs
+++ b/crates/prompt_gateway/tests/integration.rs
@@ -380,6 +380,7 @@ fn prompt_gateway_request_to_llm_gateway() {
         .expect_log(Some(LogLevel::Warn), None)
         .expect_log(Some(LogLevel::Info), None)
         .expect_log(Some(LogLevel::Debug), None)
+        .expect_log(Some(LogLevel::Info), None)
         .expect_log(Some(LogLevel::Debug), None)
         .expect_log(Some(LogLevel::Info), None)
         .expect_log(Some(LogLevel::Debug), None)
@@ -453,6 +454,7 @@ fn prompt_gateway_request_to_llm_gateway() {
         .expect_log(Some(LogLevel::Info), None)
         .expect_set_buffer_bytes(Some(BufferType::HttpResponseBody), None)
         .expect_log(Some(LogLevel::Info), None)
+        .expect_log(Some(LogLevel::Info), None)
         .expect_log(Some(LogLevel::Debug), None)
         .execute_and_expect(ReturnType::Action(Action::Continue))
         .unwrap();
@@ -493,19 +495,9 @@ fn prompt_gateway_request_no_intent_match() {
             finish_reason: Some("test".to_string()),
             index: Some(0),
             message: Message {
-                role: "system".to_string(),
+                role: "assistant".to_string(),
                 content: None,
-                tool_calls: Some(vec![ToolCall {
-                    id: String::from("test"),
-                    tool_type: ToolType::Function,
-                    function: FunctionCallDetail {
-                        name: String::from("weather_forecast"),
-                        arguments: Some(HashMap::from([(
-                            String::from("city"),
-                            Value::String(String::from("seattle")),
-                        )])),
-                    },
-                }]),
+                tool_calls: None,
                 model: None,
                 tool_call_id: None,
             },
@@ -523,7 +515,7 @@ fn prompt_gateway_request_no_intent_match() {
         .expect_log(Some(LogLevel::Warn), None)
         .expect_log(Some(LogLevel::Info), None)
         .expect_log(Some(LogLevel::Debug), None)
-        .expect_log(Some(LogLevel::Info), Some("intent not matched"))
+        .expect_log(Some(LogLevel::Info), Some("intent matched: false"))
         .expect_log(
             Some(LogLevel::Info),
             Some("no default prompt target found, forwarding request to upstream llm"),
@@ -651,17 +643,7 @@ fn prompt_gateway_request_no_intent_match_default_target() {
             message: Message {
                 role: "system".to_string(),
                 content: None,
-                tool_calls: Some(vec![ToolCall {
-                    id: String::from("test"),
-                    tool_type: ToolType::Function,
-                    function: FunctionCallDetail {
-                        name: String::from("weather_forecast"),
-                        arguments: Some(HashMap::from([(
-                            String::from("city"),
-                            Value::String(String::from("seattle")),
-                        )])),
-                    },
-                }]),
+                tool_calls: None,
                 model: None,
                 tool_call_id: None,
             },
@@ -679,7 +661,7 @@ fn prompt_gateway_request_no_intent_match_default_target() {
         .expect_log(Some(LogLevel::Warn), None)
         .expect_log(Some(LogLevel::Info), None)
         .expect_log(Some(LogLevel::Debug), None)
-        .expect_log(Some(LogLevel::Info), Some("intent not matched"))
+        .expect_log(Some(LogLevel::Info), Some("intent matched: false"))
         .expect_log(
             Some(LogLevel::Info),
             Some("default prompt target found, forwarding request to default prompt target"),