tPjy UdZddlZddlmZmZmZddlmZddlm Z m Z ddl m Z m Z dZeed< ed zZeed < d ed efd Zdeee dzfd e fdZdeeegefdee deeee ge dzffded e f dZdeeefded e fdZdeeefd edzfdZded edzfdZded edzfdZdeed edzfdZded edzfd Zd!e d efd"Zy)#uShared helpers for the attribute mappers. Small, mapper-agnostic utilities — JSON serialization, message extraction, and extractor-table application — pulled out of the individual mapper modules so they live in one place. N)CallableMappingSequence)Final) AttributeMap AttrValue)LLMCallSpanDataToolDefinitionDEFAULT_SPAN_ATTRIBUTE_LIMIT"MAX_TOOL_DEFINITION_ATTRS_PER_SPAN vocabulariesreturnc(tt|dzS)zGSplit the span-wide tool-definition ceiling across active vocabularies.)rmax)rs l/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/otel/mappers/utils.pytool_attr_budgetr#s -\11E EEvaluesc\|jDcic] \}}| || c}}Scc}}w)z7Return ``values`` with ``None``-valued entries removed.)items)rkvs r drop_noner(s(#\\^ =TQq}AqD == =s ((key_fortools extractors attr_budgetc t|tt|dzd}tt|d|Dcic]-\}}|j D]\}}|||||/c}}}}Scc}}}}w)aSPer-index attributes for as many tools as ``attr_budget`` affords. ``key_for`` builds a vocabulary's key from the tool's index and the field name, so each mapper keeps its own naming while sharing the budget. One tool always keeps its detail, so the family stays legible even when many vocabularies split the ceiling. rN)rlenr enumerater) rrrr max_toolsidxtoolsuffixextracts rtool_definition_attrsr)-s;#c*oq*AA1EI 'uZi'89  T#-#3#3#5   C '$- /    s2A7tablesourcec tt|jDcic]\}}|||c}}Scc}}w)zBApply an extractor table to ``source``, dropping ``None`` results.)rr)r*r+keyr(s rcollectr.Ds. u{{}M|sGc76?*M NNMs4 payloadc4|rtj|SdS)zCJSON-serialize ``payload`` only when it's non-empty; else ``None``.N)jsondumps)r/s rjson_ifr3Is")4::g 3t3rvaluecX tj|tS#t$rYywxYw)zHJSON-serialize ``value`` (falling back to ``str``); ``None`` on failure.defaultN)r1r2str Exception)r4s r json_or_noner:Ns*zz%-- s  ))messageczt|tsy tj|tS#t $rYywxYw)zIJSON-serialize a chat message dict; ``None`` if not a dict or on failure.Nr6) isinstancedictr1r2r8r9)r;s rstringify_messager?Vs7 gt $zz'3// s . ::messagescd|DDcgc]}|tj|}}|rtj|SdScc}w)zERound-trip a sequence of message dicts through ``stringify_message``.c32K|]}t|ywN)r?).0ms r z%serialize_messages..bs0X!1B11E0XsN)r1loadsr2)r@s serializeds rserialize_messagesrJ`sD0Xx0Xj1\]\iAjJj%/4::j !9T9ks AAcXt|tsy|jd}t|tr|St|tra|Dcgc]:}t|ts|jddk(s)|jdd<}}dj d|DxsdSycc}w)z9Extract the textual ``content`` from a chat message dict.Ncontenttypetextc3BK|]}t|ts|ywrC)r=r8)rDps rrFz"message_content..ps>Q:a+=q>s)r=r>getr8listjoin)r;rLpartpartss rmessage_contentrWfs gt $[[+G'3'4 29s$Zd=SX\X`X`agXhlrXr&"%ssww>%>>F$F ts B'B'4B'datac|jDcgc]$}t|ts|jd&c}Scc}w)z0The ``message`` payload of each response choice.r;) choices_outr=r>rR)rXcs routput_messagesr\ts.&*&6&6 N*Q:MAEE)  NN Ns;;)__doc__r1collections.abcrrrtypingr&litellm.integrations.otel.mappers.baserr(litellm.integrations.otel.model.payloadsr r r __annotations__rintrr8rr)objectr.r3r:r?rJrWrSr\rrrfs 77JT&)e)<,HA,M"EM  F3F3F >gc9t#334>>  sCj#o & N #X~&6 D8H&HIIJ   .O73=)O6OlO 4WS&[)4cDj4 3:v#*:&!1:cDj: V d O/OdOr