scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

Merge e986b752b9 into d32a6d75f8

Browse source
This commit is contained in:
Ege 2026-11-06 16:37:42 -08:00 • committed by GitHub
commit fe8f1d8959
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 28 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/integrations/otel/model/metadata.py
View file

@ -152,8 +152,9 @@ class RequestIdentity:
)
if (value := get(attr))
)
key_meta: Final = flatten_metadata({"user_api_key_auth_metadata": get("metadata")})
request_meta: Final = flatten_metadata(request_metadata) if request_metadata is not None else ()
metadata: Final = MappingProxyType(dict((*request_meta, *auth_meta)))
metadata: Final = MappingProxyType(dict((*request_meta, *auth_meta, *key_meta)))
return cls(
team_id=as_str(get("team_id")),
team_alias=as_str(get("team_alias")),

26
tests/unit/integrations/otel/test_otel_v2_sources_of_truth.py
View file

@ -1641,6 +1641,32 @@ def test_request_identity_from_user_api_key_auth():
assert ident.metadata["user_api_key_end_user_id"] == "eu9"
def test_virtual_key_metadata_reaches_baggage_only_for_allowlisted_sub_keys():
from litellm.integrations.otel.model.baggage import DEFAULT_BAGGAGE_METADATA_KEYS, promoted_baggage
from litellm.integrations.otel.model.payloads import RequestIdentity
from litellm.proxy._types import UserAPIKeyAuth
auth: Final = UserAPIKeyAuth(
key_alias="cost-key",
metadata={"team": "a", "cost": {"center": "cc-42"}, "owner": "not-allowlisted"},
)
identity: Final = RequestIdentity.from_user_api_key_auth(auth)
allowlisted: Final = promoted_baggage(
identity,
None,
promoted_keys=(),
metadata_keys=("user_api_key_auth_metadata.team", "user_api_key_auth_metadata.cost.center"),
)
default: Final = promoted_baggage(identity, None, promoted_keys=(), metadata_keys=DEFAULT_BAGGAGE_METADATA_KEYS)
assert allowlisted == {
"litellm.metadata.user_api_key_auth_metadata.team": "a",
"litellm.metadata.user_api_key_auth_metadata.cost.center": "cc-42",
}
assert default == {"litellm.metadata.user_api_key_alias": "cost-key"}
# --- request-metadata translation layer (RequestContext) -------------------- #