mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-14 23:21:35 +00:00
fix(proxy): omit empty rust spend metadata
This commit is contained in:
parent
5a414a3522
commit
c2265b95c0
1 changed files with 8 additions and 3 deletions
|
|
@ -83,7 +83,7 @@ def _get_spend_logs_metadata(
|
|||
litellm_call_id: Optional[str] = None,
|
||||
) -> SpendLogsMetadata:
|
||||
if metadata is None:
|
||||
return SpendLogsMetadata(
|
||||
clean_metadata = SpendLogsMetadata(
|
||||
user_api_key=None,
|
||||
user_api_key_alias=None,
|
||||
user_api_key_team_id=None,
|
||||
|
|
@ -108,12 +108,14 @@ 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: "
|
||||
+ str(list(metadata.keys()))
|
||||
|
|
@ -136,7 +138,10 @@ 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
|
||||
clean_metadata["litellm_rust"] = litellm_rust
|
||||
if litellm_rust is not None:
|
||||
clean_metadata["litellm_rust"] = litellm_rust
|
||||
else:
|
||||
clean_metadata.pop("litellm_rust", None)
|
||||
clean_metadata["cost_breakdown"] = cost_breakdown
|
||||
clean_metadata["litellm_call_id"] = litellm_call_id
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue