From 7dede188f8dbed8b1815791d3d06a29de031b5de Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Fri, 18 Sep 2026 22:30:12 +0000
Subject: [PATCH] test(timing): type the logging object test helper

Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com>
---
 .../llm_response_utils/test_response_metadata.py           | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py b/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py
index 40f964cd3fc..eeccbc719d3 100644
--- a/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py
+++ b/tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py
@@ -233,7 +233,12 @@ class TestResponseTimingMetrics:
     START = datetime.datetime(2025, 1, 1, 0, 0, 0)
     END = datetime.datetime(2025, 1, 1, 0, 0, 1)
 
-    def _make_logging_obj(self, llm_api_duration_ms=None, caching_details=None, received_at=None):
+    def _make_logging_obj(
+        self,
+        llm_api_duration_ms: float | None = None,
+        caching_details: dict[str, object] | None = None,
+        received_at: datetime.datetime | str | None = None,
+    ) -> MagicMock:
         logging_obj = MagicMock()
         logging_obj.model_call_details = {}
         if llm_api_duration_ms is not None: