scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix(datadog): use .get() for safe team tag extraction

Browse source
This commit is contained in:
Harshit Jain 2026-02-18 18:52:56 +05:30
parent 02e4530bfe
commit 57554833a3
No known key found for this signature in database
GPG key ID: 36C392CD4415B4CF
1 changed files with 13 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
litellm/integrations/datadog/datadog_cost_management.py
View file

@ -169,16 +169,20 @@ class DatadogCostManagementLogger(CustomBatchLogger):
metadata = log.get("metadata", {})
if metadata:
# Add user info
if "user_api_key_alias" in metadata:
# Add user info
if metadata.get("user_api_key_alias"):
tags["user"] = str(metadata["user_api_key_alias"])
if "user_api_key_team_alias" in metadata:
tags["team"] = str(metadata["user_api_key_team_alias"])
elif "team_alias" in metadata:
tags["team"] = str(metadata["team_alias"])
elif "user_api_key_team_id" in metadata:
tags["team"] = str(metadata["user_api_key_team_id"])
elif "team_id" in metadata:
tags["team"] = str(metadata["team_id"])
# Add Team Tag
team_tag = (
metadata.get("user_api_key_team_alias")
or metadata.get("team_alias") # type: ignore
or metadata.get("user_api_key_team_id")
or metadata.get("team_id") # type: ignore
)
if team_tag:
tags["team"] = str(team_tag)
# model_group is not in StandardLoggingMetadata TypedDict, so we need to access it via dict.get()
model_group = metadata.get("model_group") # type: ignore[misc]
if model_group: