From d6e95a613e5f21a6c441dcc14987113ae9262d6d Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Thu, 26 Feb 2026 10:46:01 +0530 Subject: [PATCH] Fix test_async_gcs_pub_sub_v1 --- .../management_endpoints/usage_endpoints/ai_usage_chat.py | 4 ++-- tests/logging_callback_tests/test_gcs_pub_sub.py | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) 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",