scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

fix _success_handler_helper_fn

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-08 11:06:12 -08:00
parent bbcdf6f996
commit 9e3a8c4b0d
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/litellm_core_utils/litellm_logging.py
View file

@ -1571,12 +1571,14 @@ class Logging(LiteLLMLoggingBaseClass):
# MAP RESPONSES API USAGE OBJECT TO LITELLM USAGE OBJECT
if isinstance(result, ResponsesAPIResponse):
result = result.model_copy()
transformed_usage = ResponseAPILoggingUtils._transform_response_api_usage_to_chat_usage(
result.usage
)
# Set as dict instead of Usage object so model_dump() serializes it correctly
setattr(
result,
"usage",
ResponseAPILoggingUtils._transform_response_api_usage_to_chat_usage(
result.usage
),
transformed_usage.model_dump() if hasattr(transformed_usage, 'model_dump') else dict(transformed_usage),
)
if (