From 82d0a9f0582e2e0796727d3c7f14da7d0a2a5606 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 25 Jun 2026 22:18:04 -0700 Subject: [PATCH] fix(proxy): preserve rust spend metadata key --- litellm/proxy/spend_tracking/spend_tracking_utils.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/litellm/proxy/spend_tracking/spend_tracking_utils.py b/litellm/proxy/spend_tracking/spend_tracking_utils.py index 3be8482f06b..dc07445872d 100644 --- a/litellm/proxy/spend_tracking/spend_tracking_utils.py +++ b/litellm/proxy/spend_tracking/spend_tracking_utils.py @@ -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