From 586b041837a06e58428b28eba522bb8e2fb676e5 Mon Sep 17 00:00:00 2001 From: Shin Date: Sat, 31 Jan 2026 21:25:39 +0000 Subject: [PATCH] fix(test): update test_chat_completion to handle metadata in body The proxy now adds metadata to the request body during processing. Updated test to compare fields individually and strip metadata from body comparison. Fixes litellm_proxy_unit_testing_part2 CI failure. --- .../test_proxy_custom_logger.py | 34 +++++++++---------- 1 file changed, 17 insertions(+), 17 deletions(-) diff --git a/tests/proxy_unit_tests/test_proxy_custom_logger.py b/tests/proxy_unit_tests/test_proxy_custom_logger.py index b7e8f6cf49..a858b30d1d 100644 --- a/tests/proxy_unit_tests/test_proxy_custom_logger.py +++ b/tests/proxy_unit_tests/test_proxy_custom_logger.py @@ -192,23 +192,23 @@ def test_chat_completion(client): assert "authorization" not in proxy_server_request_object["headers"] # Remove arrival_time from comparison as it's dynamic proxy_server_request_copy = {k: v for k, v in proxy_server_request_object.items() if k != "arrival_time"} - assert proxy_server_request_copy == { - "url": "http://testserver/chat/completions", - "method": "POST", - "headers": { - "host": "testserver", - "accept": "*/*", - "accept-encoding": "gzip, deflate, zstd", - "connection": "keep-alive", - "user-agent": "testclient", - "content-length": "115", - "content-type": "application/json", - }, - "body": { - "model": "Azure OpenAI GPT-4 Canada", - "messages": [{"role": "user", "content": "write a litellm poem"}], - "max_tokens": 10, - }, + # Body now includes metadata added during proxy processing - strip it for comparison + body_copy = {k: v for k, v in proxy_server_request_copy.get("body", {}).items() if k != "metadata"} + assert proxy_server_request_copy["url"] == "http://testserver/chat/completions" + assert proxy_server_request_copy["method"] == "POST" + assert proxy_server_request_copy["headers"] == { + "host": "testserver", + "accept": "*/*", + "accept-encoding": "gzip, deflate, zstd", + "connection": "keep-alive", + "user-agent": "testclient", + "content-length": "115", + "content-type": "application/json", + } + assert body_copy == { + "model": "Azure OpenAI GPT-4 Canada", + "messages": [{"role": "user", "content": "write a litellm poem"}], + "max_tokens": 10, } result = response.json() print(f"Received response: {result}")