From dbb592967893bb765ef3513686875eb2b21144f5 Mon Sep 17 00:00:00 2001
From: Gauthier Piarrette <gauthierpiarrette@gmail.com>
Date: Tue, 4 Aug 2026 05:12:37 -0700
Subject: [PATCH] fix(mlflow): type the cache token extraction helper

---
 litellm/integrations/mlflow.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/integrations/mlflow.py b/litellm/integrations/mlflow.py
index aadf7d49d17..80766c3fee6 100644
--- a/litellm/integrations/mlflow.py
+++ b/litellm/integrations/mlflow.py
@@ -228,7 +228,9 @@ class MlflowLogger(CustomLogger):
             )
         return attributes
 
-    def _extract_cache_token_usage(self, standard_obj) -> dict:
+    def _extract_cache_token_usage(
+        self, standard_obj: StandardLoggingPayload
+    ) -> dict[str, int]:
         """
         Extract cache token counts from the raw response usage.