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

fix(mlflow): type the cache token extraction helper

Browse source
This commit is contained in:
Gauthier Piarrette 2026-08-04 05:12:37 -07:00
parent 7ea2cf73f1
commit dbb5929678
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/integrations/mlflow.py
View file

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