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

style(otel): format metadata and tighten test assertions

Browse source
This commit is contained in:
Ege Arhan 2026-11-06 03:07:50 +03:00
parent 25a6223130
commit 089e77ac19
2 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -154,9 +154,7 @@ class RequestIdentity:
)
raw_key_meta: Final = as_str_mapping(get("metadata") or get("user_api_key_auth_metadata"))
key_meta: Final = (
tuple(flatten_metadata({"user_api_key_auth_metadata": raw_key_meta}))
if raw_key_meta is not None
else ()
tuple(flatten_metadata({"user_api_key_auth_metadata": raw_key_meta})) if raw_key_meta is not None else ()
)
request_meta: Final = flatten_metadata(request_metadata) if request_metadata is not None else ()
metadata: Final = MappingProxyType(dict((*request_meta, *auth_meta, *key_meta)))

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

@ -1613,10 +1613,12 @@ def test_request_identity_from_user_api_key_auth_flattens_key_metadata():
class _Auth:
metadata = {"team": "a", "customtag": "b", "nested": {"k": "v"}}
ident = RequestIdentity.from_user_api_key_auth(_Auth())
assert ident.metadata["user_api_key_auth_metadata.team"] == "a"
assert ident.metadata["user_api_key_auth_metadata.customtag"] == "b"
assert ident.metadata["user_api_key_auth_metadata.nested.k"] == "v"
ident: Final = RequestIdentity.from_user_api_key_auth(_Auth())
assert ident.metadata == {
"user_api_key_auth_metadata.team": "a",
"user_api_key_auth_metadata.customtag": "b",
"user_api_key_auth_metadata.nested.k": "v",
}
# --- request-metadata translation layer (RequestContext) -------------------- #