tPjgUdZddlZddlmZmZddlmZddlmZddl m Z m Z e jde jde jd e jd e j d e j"d e j$d iZeeeeeedzeedfgedzffed<e je je je je j"e j$fZeeedfed<dZeeedfed<dZeeedfed<eefdededzdeedfdeedfdeedfdeeeff dZdeeefdzdeedfdedzfdZy)aBaggage promotion: request-identity values carried across child spans. A bounded set of identity values is written into OpenTelemetry Baggage on the LLM-call span so that child spans (guardrail, service) inherit them. ``providers.LiteLLMBaggageSpanProcessor`` reads Baggage at span start and stamps the allowlisted keys onto every span. This module is the single place baggage is defined: ``_PROMOTABLE`` maps each promotable attribute key to how its value is read, and the ``*_KEYS`` defaults select what is promoted unless the config overrides them. ``TEAM_METADATA``'s extractor filters the team's free-form metadata to the sub-keys an operator allowlists via ``baggage_team_metadata_keys`` (default none), so the blob is never promoted whole. N)CallableMapping)Final)RequestIdentity)GenAILiteLLMc|jSN)team_ididentitymodelteam_metadata_keyss l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/otel/model/baggage.pyrs AQAQc|jSr ) team_aliasr s rrrs HDWDWrc.t|j|Sr )_filtered_team_metadata_json team_metadatar s rrrsGc 2Hrc|jSr )key_hashr s rrr! (BSBSrc|jSr )end_userr s rrr"rrc|Sr r s rrr#sUrc|jSr )provider_modelr s rrr$s H_H_r. _PROMOTABLEBAGGAGE_PROMOTED_KEYS)user_api_key_org_iduser_api_key_user_iduser_api_key_aliasuser_api_key_end_user_idrequester_ip_addressDEFAULT_BAGGAGE_METADATA_KEYSr"DEFAULT_BAGGAGE_TEAM_METADATA_KEYSr request_model promoted_keys metadata_keysrreturnci}tjD]\}}||vs ||||}|s|||<|D]7} |jj| }|s!||tj | <9|S)auIdentity values to write into Baggage, filtered to ``promoted_keys``. ``promoted_keys`` selects from ``_PROMOTABLE``; ``metadata_keys`` selects sub-keys of ``identity.metadata`` to promote under ``litellm.metadata.*``; ``team_metadata_keys`` selects sub-keys of the team's metadata to promote under ``litellm.team.metadata``. Empty values are dropped. )r!itemsmetadatagetrMETADATA_PREFIX) r r*r+r,routkeyextractvaluemeta_keys rpromoted_baggager8Ds"$C#))+! W - Hm5GHE C ! "@!!%%h/ :?C7**+H:6 7@ Jrr0 allowed_keysct|tr|sy|Dcic] }||vs|||}}|sytj|tdScc}w)aJSON-serialize only the allowlisted sub-keys of a team's metadata. Returns ``None`` when nothing is allowlisted or no allowlisted key is present, so the empty case is dropped rather than promoting ``"{}"``. Keys are sorted for a stable, diff-friendly value. NT)default sort_keys) isinstancerjsondumpsstr)r0r9r4filtereds rrr_sT h ( 5AUcSH_sHSM)UHU  ::ht <<Vs A A )__doc__r>collections.abcrrtypingr(litellm.integrations.otel.model.metadatar'litellm.integrations.otel.model.semconvrrTEAM_ID TEAM_ALIAS TEAM_METADATAKEY_HASHEND_USER REQUEST_MODELPROVIDER_MODELr!dictr@tuple__annotations__r"r(r)r8objectrrrrrRs  -DB OOQ W  S S J _ g U4Xd E#s(O&TVY\`V`&`aab c   OO      1uU38_-9uU38_5>@"E%S/$:?&C*L :c?c?  c3h   #s(^ 6=c6k"T)=S/= 4Z=r