scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-21 00:21:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 74

test(timing): type the logging object test helper

Browse source 
Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-18 22:30:12 +00:00
parent 00ab2c1be3
commit 7dede188f8
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/litellm_core_utils/llm_response_utils/test_response_metadata.py
View file

@ -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: