diff --git a/litellm/proxy/management_endpoints/usage_endpoints/ai_usage_chat.py b/litellm/proxy/management_endpoints/usage_endpoints/ai_usage_chat.py
index 56c3c5f0477..4de29e04092 100644
--- a/litellm/proxy/management_endpoints/usage_endpoints/ai_usage_chat.py
+++ b/litellm/proxy/management_endpoints/usage_endpoints/ai_usage_chat.py
@@ -7,6 +7,8 @@ import json
 from datetime import date
 from typing import Any, AsyncIterator, Callable, Dict, List, Literal, Optional, cast
 
+from typing_extensions import TypedDict
+
 import litellm
 from litellm._logging import verbose_proxy_logger
 from litellm.constants import DEFAULT_COMPETITOR_DISCOVERY_MODEL
@@ -14,8 +16,6 @@ from litellm.types.proxy.management_endpoints.common_daily_activity import (
     SpendAnalyticsPaginatedResponse,
 )
 
-from typing_extensions import TypedDict
-
 # ---------------------------------------------------------------------------
 # Constants
 # ---------------------------------------------------------------------------
diff --git a/tests/logging_callback_tests/test_gcs_pub_sub.py b/tests/logging_callback_tests/test_gcs_pub_sub.py
index 8ffbc8eedd5..540fb59ab01 100644
--- a/tests/logging_callback_tests/test_gcs_pub_sub.py
+++ b/tests/logging_callback_tests/test_gcs_pub_sub.py
@@ -36,6 +36,7 @@ ignored_keys = [
     "endTime",
     "completionStartTime",
     "endTime",
+    "request_duration_ms",
     "metadata.model_map_information",
     "metadata.usage_object",
     "metadata.cold_storage_object_key",