Update utils.py (#5530)

fix KeyError (cause by typo?)
This commit is contained in:
OrangeWolf 2024-09-05 23:21:41 +08:00 committed by Ishaan Jaff
parent 936e486c95
commit 542b333c95

View file

@ -504,7 +504,7 @@ class Usage(CompletionUsage):
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=0"]
self._cache_read_input_tokens = params["prompt_cache_hit_tokens"]
for k, v in params.items():
setattr(self, k, v)