From f1d516f0ae690dc5ae6dc7a2828b95bde6ac197a Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Sat, 10 Jan 2026 15:15:41 -0800
Subject: [PATCH] test fixes

---
 tests/proxy_unit_tests/test_proxy_custom_logger.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tests/proxy_unit_tests/test_proxy_custom_logger.py b/tests/proxy_unit_tests/test_proxy_custom_logger.py
index 28de522a4c7..b7e8f6cf499 100644
--- a/tests/proxy_unit_tests/test_proxy_custom_logger.py
+++ b/tests/proxy_unit_tests/test_proxy_custom_logger.py
@@ -91,7 +91,9 @@ def test_embedding(client):
         assert "headers" in metadata
         proxy_server_request = litellm_params.get("proxy_server_request")
         model_info = litellm_params.get("model_info")
-        assert proxy_server_request == {
+        # Remove arrival_time from comparison as it's dynamic
+        proxy_server_request_copy = {k: v for k, v in proxy_server_request.items() if k != "arrival_time"}
+        assert proxy_server_request_copy == {
             "url": "http://testserver/embeddings",
             "method": "POST",
             "headers": {
@@ -188,7 +190,9 @@ def test_chat_completion(client):
         }
 
         assert "authorization" not in proxy_server_request_object["headers"]
-        assert proxy_server_request_object == {
+        # 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": {