From a7b9458e5aae0f17feeaa39ed9c7a627c76b26b8 Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Fri, 17 Jan 2025 11:00:36 -0800 Subject: [PATCH] fix rust tests --- crates/llm_gateway/tests/integration.rs | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/crates/llm_gateway/tests/integration.rs b/crates/llm_gateway/tests/integration.rs index cbaedad3..c43bfcf1 100644 --- a/crates/llm_gateway/tests/integration.rs +++ b/crates/llm_gateway/tests/integration.rs @@ -23,11 +23,15 @@ fn request_headers_expectations(module: &mut Tester, http_context: i32) { Some("x-arch-llm-provider-hint"), ) .returning(Some("default")) - .expect_log(Some(LogLevel::Debug), None) + .expect_log( + Some(LogLevel::Debug), + Some("llm provider hint: Some(Default)"), + ) + .expect_log(Some(LogLevel::Debug), Some("selected llm: open-ai-gpt-4")) .expect_add_header_map_value( Some(MapType::HttpRequestHeaders), Some("x-arch-llm-provider"), - Some("open-ai-gpt-4"), + Some("openai"), ) .expect_replace_header_map_value( Some(MapType::HttpRequestHeaders), @@ -46,8 +50,6 @@ fn request_headers_expectations(module: &mut Tester, http_context: i32) { .returning(None) .expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some(":path")) .returning(Some("/v1/chat/completions")) - .expect_get_header_map_pairs(Some(MapType::HttpRequestHeaders)) - .returning(None) .expect_log(Some(LogLevel::Debug), None) .expect_get_header_map_value(Some(MapType::HttpRequestHeaders), Some("x-request-id")) .returning(None)