scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix(spend_tracking): narrow the provider response id with isinstance instead of cast

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-01 23:48:03 +00:00
parent 509e6a4e31
commit cc910fdf24
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/spend_tracking/spend_tracking_utils.py
View file

@ -397,6 +397,7 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time) -> SpendLogs
usage = _combined_usage.model_dump()
id = get_spend_logs_id(call_type or "acompletion", response_obj_dict, kwargs)
raw_response_id: Final = response_obj_dict.get("id")
standard_logging_payload: Final = cast(StandardLoggingPayload | None, kwargs.get("standard_logging_object", None))
end_user_id = get_end_user_id_for_cost_tracking(litellm_params)
@ -522,7 +523,7 @@ def get_logging_payload(kwargs, response_obj, start_time, end_time) -> SpendLogs
standard_logging_payload.get("autorouter_savings", None) if standard_logging_payload is not None else None
),
litellm_call_id=litellm_call_id,
response_id=cast(str | None, response_obj_dict.get("id")),
response_id=raw_response_id if isinstance(raw_response_id, str) else None,
router_metadata=_get_router_metadata_for_spend_log(
metadata=metadata,
requested_model=_model_group,