scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

test: add Final annotations to LIT-5757 regression test variables

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-08-18 23:19:38 +00:00
parent 0585c45cd9
commit 4a43b50800
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/types/test_types_utils.py
View file

@ -1,5 +1,6 @@
import os
import sys
from typing import Final
import pytest
@ -82,19 +83,19 @@ def test_prompt_tokens_details_maps_nested_cache_creation_input_tokens():
overriding an explicitly provided canonical value."""
from litellm.types.utils import PromptTokensDetailsWrapper
nested = PromptTokensDetailsWrapper(
nested: Final = PromptTokensDetailsWrapper(
cached_tokens=0, text_tokens=2059, cache_creation_input_tokens=2048
)
assert nested.cache_write_tokens == 2048
assert nested.cache_creation_tokens == 2048
explicit = PromptTokensDetailsWrapper(
explicit: Final = PromptTokensDetailsWrapper(
cache_write_tokens=100, cache_creation_input_tokens=2048
)
assert explicit.cache_write_tokens == 100
assert explicit.cache_creation_tokens == 100
non_int = PromptTokensDetailsWrapper(cache_creation_input_tokens=None)
non_int: Final = PromptTokensDetailsWrapper(cache_creation_input_tokens=None)
assert not hasattr(non_int, "cache_write_tokens")