tPj, 4UdZddlmZddlmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZddlmZdd lmZmZdd lmZdd lmZmZmZmZmZmZdd lm Z m!Z!m"Z"dd l#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*ddl+m,Z,ddl-m.Z.e%j^e'e%j`e)e%jbe(e%jde&e%jfe*e%jhe*iZ5ee6e%ede7ffe8d<dZ9ee8d<de de7de7ddfdZ:de deddfdZ;dddde dedeGd!d"Z?y)#zEThe span engine: dedup, start, run the mapper chain, set status, end.) OrderedDict)CallableSequence)Final)Context)LinkSpanTracer)Status StatusCode)resolve_mappers)AttributeMapperSpanData)OpenTelemetryV2Config)GuardrailSpanDataLLMCallSpanDataMCPListToolsSpanDataMCPToolCallSpanDataServiceSpanData SpanError)ErrorExceptionEvent LiteLLMError) SPAN_REGISTRYSpanRoleguardrail_span_namellm_call_span_namemcp_list_tools_span_namemcp_tool_call_span_nameservice_span_name)GenAIEventRecorder)to_otel_span_kind._NAME_BUILDERSi'_DEDUP_CACHE_MAXspan error_typeresolved_messagereturnNc|jtj||jtj|y)zStamp the OTel-semconv error attributes (``error.type`` + ``error.message``). ``error_type`` and ``resolved_message`` are ``finish_span``'s already-computed fallback chains, so the pair on the status, event, and attributes stays in lockstep.N) set_attributerTYPEMESSAGE)r%r&r's f/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/otel/emitter.py_stamp_otel_error_attributesr.7s-  uzz:.u}}&67errorcJ|jr*|jtj|j|jr*|jtj |j|j r+|jtj|j yy)zStamp litellm-specific error detail attributes. Emitted only when the corresponding field is populated so guardrail-shape errors carrying only a message aren't polluted with empty detail keys.N)coder*rCODE stack_trace STACK_TRACE llm_provider LLM_PROVIDER)r%r0s r-_stamp_litellm_error_attributesr8@st zz <,,ejj9  <33U5F5FG  <44e6H6HIr/T) record_event set_statusr9r:c|js |jsy|jxsd}|jxs|jxsd}t|||t|||r)|j t t j||r@|jtjtj|tj|i||fS)a7Stamp the full v2 error attribute set on ``span`` and return the resolved ``(error_type, message)`` pair, or ``None`` when the error carries neither a type nor a message. Shared by the LLM-call span (``finish_span``) and the proxy-level failure spans (the FastAPI SERVER span and the ``auth`` phase span) so every v2 error span carries identical keys. The semconv ``exception`` event rides alongside the attributes so backends that map unknown string attrs to a truncated ``keyword`` (e.g. Elasticsearch's 1024-char ``ignore_above``) still see the full untruncated message on the recognized event field. ``record_event`` and ``set_status`` are opt-outs for callers whose span lifecycle (``use_span``) or owner (the FastAPI instrumentor) already records the event or the status. Nr0) r&messager.r8r:r r ERROR add_eventrNAMEr+r,)r%r0r9r:r&r<s r- stamp_errorr@Ls(    ((3GJ]]Ae&6&6A'G z7;#D%0 z//9:      *n.D.Dg N  w r/cNeZdZ ddededeedzdedzddf dZ ddddd e d e d e dzd e dzdedzd ee dzdefdZde dzd e ddfdZde dzd e defdZ ddddddd e ded e dzd e dzde dzdedzd ee dzdedzfdZddd e dedede dzddf dZy) SpanEmitterNtracerconfigmappersevent_recorderr(c||_||_||_| t|nt |j |_t|_yN) _tracer_config_event_recorderlistr mapper_names_mappersr_emitted)selfrCrDrErFs r-__init__zSpanEmitter.__init__qsI  -%0DMofFYFY6Z BM r/)rClinksrolenameparent_context start_time_nsrRc |xs |jj||tt|j||rt |SdS)aStart a span for ``role`` without dedup or attribute mapping. For callers that own and manage their own span lifecycle. ``tracer`` overrides the bound tracer for this span only, used for per-request multi-tenant credential routing. ``links`` records related-but-not-parent spans (e.g. the trace context an MCP client propagated in ``params._meta``). N)contextkind start_timerR)rI start_spanr"rrYrL)rPrSrTrUrVrCrRs r-r[zSpanEmitter.start_spans^"&$,,22 ""=#6#;#;<$!&$u+ 3  -1 3  r/ dedup_keyc(|j||y)zRegister a span emitted outside :meth:`emit` (the boundary-opened LLM-call span closed via :meth:`finish_span`) so a later :meth:`emit` for the same ``(dedup_key, role)`` deduplicates against it.N)_seen)rPr\rSs r- mark_emittedzSpanEmitter.mark_emitteds 9d#r/c|sy||f}||jvr|jj|yd|j|<t|jtkDr|jj dy)zReturn True once a ``(dedup_key, role)`` pair has been emitted. Guards against emitting the same span twice when a streaming call fires both a sync and an async logging callback. FTN)last)rO move_to_endlenr$popitem)rPr\rSmarkers r-r^zSpanEmitter._seensm "D) T]] " MM % %f - $ f t}}  0 0 MM ! !u ! -r/)rV end_time_nsrCrRdatarfct|tttfr|jj nd}|j ||ry|j|t||||||} |j|| ||| S)aoEmit one complete span: dedup, start, map attributes, status, end. Return the span, or ``None`` if it was deduplicated away. ``tracer`` overrides the bound tracer for this span, used for per-request routing. ``links`` records related-but-not-parent spans (e.g. the trace context an MCP client propagated in ``params._meta``). N)rUrVrCrRrf) isinstancerrridentitycall_idr^r[r# finish_span) rPrSrgrUrVrfrCrRr\r%s r-emitzSpanEmitter.emits.$2EG[ \] MM ! !  ::i &oo  4  &)' &  tT{C r/rir%c2|jt|||jD];}|j|j D]\}}|j ||=t |tttttfr |jnd}|rit||} | [|jO|tj ur=| \} } |jj#|j%| | |j&||j)|y)u#Stamp attributes + status on an already-started ``span`` and end it. The counterpart to :meth:`start_span` for callers that own a span's lifecycle — the LLM-call span is opened at the request's ``pre_call`` boundary (so it parents to the live server span via real ambient context, never a span threaded through a metadata dict) and closed here once the typed payload is available. The span name is (re)built from the now-known data, since the boundary opener only has a provisional name. N) span_contextr&r<r4 timestamp_ns)end_time) update_namer#rNmapitemsr*rjrrrrrr0r@rKrLLM_CALLrecord_operation_exceptionget_span_contextr4end) rPrSr%rgrfmapperkeyvaluer0stampedr&r<s r-rmzSpanEmitter.finish_spans" -d34mm /F$jj.446 / U""3. / / #'(#%   JJ  (u5G"t';';'GDT\TeTeLe&-# G$$??!%!6!6!8)# % 1 1!, @ +&r/)NNrH)__name__ __module__ __qualname__r rrrr!rQrstrrintrr r[r_boolr^rrnrmr/r-rBrBps 5948 OO&O/*T1 O +T1 O  O2*.$(  !%'+    $  Tz    ~$   2$cDj$$T$ sTzd.*. % %)"& $'+%%% $ % Tz %4Z% %~$% %Z#' 2'2'2' 2' 4Z 2' 2'r/rB)@__doc__ collectionsrcollections.abcrrtypingropentelemetry.contextropentelemetry.tracerr r opentelemetry.trace.statusr r !litellm.integrations.otel.mappersr &litellm.integrations.otel.mappers.baserr&litellm.integrations.otel.model.configr(litellm.integrations.otel.model.payloadsrrrrrr'litellm.integrations.otel.model.semconvrrr%litellm.integrations.otel.model.spansrrrrrrr )litellm.integrations.otel.plumbing.eventsr!,litellm.integrations.otel.plumbing.providersr"rv MCP_TOOL_CALLMCP_LIST_TOOLS GUARDRAILDB_CALLSERVICEr#dictr__annotations__r$r.r8rtupler@rBrr/r-rsfK#.)229=LHXWIJ  ) 3 5 + ' ' =d8Xc3h%7789 !% 8t88PS8X\8 J$ Jy JT J  ! ! ! !  !  38_t !Ha'a'r/