Fixed litellm incomplete usage

piotrpenar · web-flow · commit 97bdeabfad04 · 2025-05-15T18:49:53.000+02:00
diff --git a/libs/agno/agno/models/litellm/chat.py b/libs/agno/agno/models/litellm/chat.py
@@ -214,12 +214,8 @@ def parse_provider_response(self, response: Any, **kwargs) -> ModelResponse:
                     }
                 )
 
-        if hasattr(response, "usage"):
-            model_response.response_usage = {
-                "input_tokens": response.usage.prompt_tokens,
-                "output_tokens": response.usage.completion_tokens,
-                "total_tokens": response.usage.total_tokens,
-            }
+        if response.usage is not None:
+            model_response.response_usage = response.usage
 
         return model_response
 

Original file line number	Diff line number	Diff line change
`@@ -214,12 +214,8 @@ def parse_provider_response(self, response: Any, **kwargs) -> ModelResponse:`
`214`	`214`	`}`
`215`	`215`	`)`
`216`	`216`
`217`		`- if hasattr(response, "usage"):`
`218`		`- model_response.response_usage = {`
`219`		`- "input_tokens": response.usage.prompt_tokens,`
`220`		`- "output_tokens": response.usage.completion_tokens,`
`221`		`- "total_tokens": response.usage.total_tokens,`
`222`		`- }`
	`217`	`+ if response.usage is not None:`
	`218`	`+ model_response.response_usage = response.usage`
`223`	`219`
`224`	`220`	`return model_response`
`225`	`221`