fix: satisfy DeepSeek usage lint rules

litellm/types/utils.py

@@ -1742,20 +1742,22 @@ class Usage(SafeAttributeModel, CompletionUsage):
             elif isinstance(prompt_tokens_details, PromptTokensDetailsWrapper):
                 _prompt_tokens_details = prompt_tokens_details
 
-        deepseek_cache_hit_tokens = _non_negative_token_count(params.get("prompt_cache_hit_tokens"))
-        deepseek_cache_miss_tokens = _non_negative_token_count(params.get("prompt_cache_miss_tokens"))
+        deepseek_cache_hit_tokens: Final = _non_negative_token_count(params.get("prompt_cache_hit_tokens"))
+        deepseek_cache_miss_tokens: Final = _non_negative_token_count(params.get("prompt_cache_miss_tokens"))
 
         ## DEEPSEEK MAPPING ##
         if deepseek_cache_hit_tokens is not None:
             if _prompt_tokens_details is None:
-                _prompt_tokens_details = PromptTokensDetailsWrapper(cached_tokens=deepseek_cache_hit_tokens)
+                _prompt_tokens_details = PromptTokensDetailsWrapper(
+                    cached_tokens=deepseek_cache_hit_tokens
+                )  # rebind-ok: builds prompt details from provider cache metadata
             else:
                 _prompt_tokens_details.cached_tokens = deepseek_cache_hit_tokens
 
-        if not prompt_tokens:
-            deepseek_prompt_tokens = (deepseek_cache_hit_tokens or 0) + (deepseek_cache_miss_tokens or 0)
-            if deepseek_prompt_tokens:
-                prompt_tokens = deepseek_prompt_tokens
+        deepseek_prompt_tokens: Final = (deepseek_cache_hit_tokens or 0) + (deepseek_cache_miss_tokens or 0)
+        effective_prompt_tokens: Final = (
+            deepseek_prompt_tokens if not prompt_tokens and deepseek_prompt_tokens else prompt_tokens or 0
+        )
 
         ## ANTHROPIC MAPPING ##
         if "cache_read_input_tokens" in params and isinstance(params["cache_read_input_tokens"], int):
@@ -1773,7 +1775,7 @@ class Usage(SafeAttributeModel, CompletionUsage):
                 _prompt_tokens_details.cache_write_tokens = params["cache_creation_input_tokens"]
 
         super().__init__(
-            prompt_tokens=prompt_tokens or 0,
+            prompt_tokens=effective_prompt_tokens,
             completion_tokens=completion_tokens or 0,
             total_tokens=total_tokens or 0,
             completion_tokens_details=_completion_tokens_details or None,
@@ -1801,8 +1803,8 @@ class Usage(SafeAttributeModel, CompletionUsage):
             self._cache_read_input_tokens = params["cache_read_input_tokens"]
 
         ## DEEPSEEK MAPPING ##
-        if "prompt_cache_hit_tokens" in params and isinstance(params["prompt_cache_hit_tokens"], int):
-            self._cache_read_input_tokens = params["prompt_cache_hit_tokens"]
+        if deepseek_cache_hit_tokens is not None:
+            self._cache_read_input_tokens = deepseek_cache_hit_tokens
 
         for k, v in params.items():
             setattr(self, k, v)

tests/test_litellm/test_cost_calculator.py

@@ -3330,61 +3330,6 @@ def test_extract_cache_creation_tokens_zero_when_missing():
     )
 
 
-def test_usage_maps_deepseek_cache_hit_and_miss_tokens():
-    usage = Usage(
-        completion_tokens=50,
-        prompt_cache_hit_tokens=75,
-        prompt_cache_miss_tokens=25,
-    )
-
-    assert usage.prompt_tokens == 100
-    assert usage.prompt_tokens_details is not None
-    assert usage.prompt_tokens_details.cached_tokens == 75
-    assert usage._cache_read_input_tokens == 75
-
-
-def test_usage_does_not_override_reported_prompt_tokens():
-    usage = Usage(
-        prompt_tokens=120,
-        completion_tokens=50,
-        prompt_cache_hit_tokens=75,
-        prompt_cache_miss_tokens=25,
-    )
-
-    assert usage.prompt_tokens == 120
-    assert usage.prompt_tokens_details is not None
-    assert usage.prompt_tokens_details.cached_tokens == 75
-
-
-def test_usage_ignores_malformed_deepseek_cache_bucket_tokens():
-    usage = Usage(
-        completion_tokens=50,
-        prompt_cache_hit_tokens="75",
-        prompt_cache_miss_tokens=True,
-    )
-
-    assert usage.prompt_tokens == 0
-    assert usage.prompt_tokens_details is None
-    assert usage._cache_read_input_tokens == 0
-
-
-def test_usage_preserves_cache_read_and_write_mappings():
-    usage = Usage(
-        prompt_tokens=100,
-        completion_tokens=50,
-        cache_read_input_tokens=30,
-        cache_creation_input_tokens=10,
-    )
-
-    assert usage.prompt_tokens == 100
-    assert usage.prompt_tokens_details is not None
-    assert usage.prompt_tokens_details.cached_tokens == 30
-    assert usage.prompt_tokens_details.cache_write_tokens == 10
-    assert usage.prompt_tokens_details.cache_creation_tokens == 10
-    assert usage._cache_read_input_tokens == 30
-    assert usage._cache_creation_input_tokens == 10
-
-
 def test_custom_pricing_anthropic_style_cache_tokens_not_double_counted():
     """
     Anthropic providers report cache tokens at the top level of Usage, and

tests/test_litellm/types/test_types_utils.py

@@ -113,6 +113,69 @@ def test_usage_converts_server_tool_use_dict():
     assert round_trip.server_tool_use.tool_search_requests == 1
 
 
+def test_usage_maps_deepseek_cache_hit_and_miss_tokens():
+    from litellm.types.utils import Usage
+
+    usage = Usage(
+        completion_tokens=50,
+        prompt_cache_hit_tokens=75,
+        prompt_cache_miss_tokens=25,
+    )
+
+    assert usage.prompt_tokens == 100
+    assert usage.prompt_tokens_details is not None
+    assert usage.prompt_tokens_details.cached_tokens == 75
+    assert usage._cache_read_input_tokens == 75
+
+
+def test_usage_does_not_override_reported_prompt_tokens():
+    from litellm.types.utils import Usage
+
+    usage = Usage(
+        prompt_tokens=120,
+        completion_tokens=50,
+        prompt_cache_hit_tokens=75,
+        prompt_cache_miss_tokens=25,
+    )
+
+    assert usage.prompt_tokens == 120
+    assert usage.prompt_tokens_details is not None
+    assert usage.prompt_tokens_details.cached_tokens == 75
+
+
+def test_usage_ignores_malformed_deepseek_cache_bucket_tokens():
+    from litellm.types.utils import Usage
+
+    usage = Usage(
+        completion_tokens=50,
+        prompt_cache_hit_tokens=True,
+        prompt_cache_miss_tokens="25",
+    )
+
+    assert usage.prompt_tokens == 0
+    assert usage.prompt_tokens_details is None
+    assert usage._cache_read_input_tokens == 0
+
+
+def test_usage_preserves_cache_read_and_write_mappings():
+    from litellm.types.utils import Usage
+
+    usage = Usage(
+        prompt_tokens=100,
+        completion_tokens=50,
+        cache_read_input_tokens=30,
+        cache_creation_input_tokens=10,
+    )
+
+    assert usage.prompt_tokens == 100
+    assert usage.prompt_tokens_details is not None
+    assert usage.prompt_tokens_details.cached_tokens == 30
+    assert usage.prompt_tokens_details.cache_write_tokens == 10
+    assert usage.prompt_tokens_details.cache_creation_tokens == 10
+    assert usage._cache_read_input_tokens == 30
+    assert usage._cache_creation_input_tokens == 10
+
+
 def test_usage_completion_tokens_details_text_tokens():
     from litellm.types.utils import Usage