scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 100

fix(proxy): preserve rust spend metadata key

Browse source
This commit is contained in:
Ishaan Jaff 2026-06-25 22:18:04 -07:00
parent cdfe9fa501
commit 82d0a9f058
No known key found for this signature in database
1 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -108,13 +108,12 @@ def _get_spend_logs_metadata(
eval_information=None,
cold_storage_object_key=cold_storage_object_key,
litellm_overhead_time_ms=None,
litellm_rust=litellm_rust,
attempted_retries=None,
max_retries=None,
cost_breakdown=None,
litellm_call_id=litellm_call_id,
)
if litellm_rust is not None:
clean_metadata["litellm_rust"] = litellm_rust
return clean_metadata
verbose_proxy_logger.debug(
"getting payload for SpendLogs, available keys in metadata: "
@ -138,10 +137,7 @@ def _get_spend_logs_metadata(
clean_metadata["model_map_information"] = model_map_information
clean_metadata["cold_storage_object_key"] = cold_storage_object_key
clean_metadata["litellm_overhead_time_ms"] = litellm_overhead_time_ms
if litellm_rust is not None:
clean_metadata["litellm_rust"] = litellm_rust
else:
clean_metadata.pop("litellm_rust", None)
clean_metadata["litellm_rust"] = litellm_rust
clean_metadata["cost_breakdown"] = cost_breakdown
clean_metadata["litellm_call_id"] = litellm_call_id