o ۾i @sUddlZddlmZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZddlmZmZmZmZgdZegefZeed<ed efZeed <ed efZeed <ed efZeed <ed efZeed <de e e e e fiZeeeeeeeeffed< d+dededeeefdBfddZ dededefddZ! d+dddddedBdedeeefdBdefd d!Z"    d,ded"e#d#e#dBdeeefdBd$ed%ef d&d'Z$d(efd)d*Z%dS)-N)Callable)Any TypeAlias)SpanKindextract_trace_contextinit_otel_tracerinit_otel_worker_tracerinstrument_otelis_otel_availablemanual_instrument_otelotel_import_error_traceback)SpanAttributescontains_trace_headersextract_trace_headerslog_tracing_disabled_warning) instrumentinstrument_manual init_tracermaybe_init_worker_traceris_tracing_availablerrrrrrr BackendAvailableFunc.InstrumentFuncInstrumentManualFuncInitTracerFuncInitWorkerTracerFuncotel_REGISTERED_TRACING_BACKENDSinstrumenting_module_nameotlp_traces_endpointextra_attributescCs(td\}}}}}|r||||SdSNrr)rrr is_availableinit_tracer_fn_r&I/home/ubuntu/.local/lib/python3.10/site-packages/vllm/tracing/__init__.pyrB r process_kind process_namecCs(td\}}}}}|r||||SdSr!r")rr)r*r#r%init_worker_tracer_fnr&r&r'rNr(rT span_name attributesrecord_exceptionobjr.r/r0cCsF|dur tjt|||dStd\}}}}}|r!|||||dS|S)z4 Generic decorator to instrument functions. Nr-r)funcr.r/r0) functoolspartialrr)r1r.r/r0r#r%otel_instrumentr&r&r'rZs  r start_timeend_timecontextkindc Cs.td\}}}}}|r|||||||SdS)aManually create a span with explicit timestamps. Args: span_name: Name of the span to create. start_time: Start time in nanoseconds since epoch. end_time: Optional end time in nanoseconds. If None, ends immediately. attributes: Optional dict of span attributes. context: Optional trace context (e.g., from extract_trace_context). kind: Optional SpanKind (e.g., SpanKind.SERVER). rNr") r.r6r7r/r8r9r#r%manual_instrument_fnr&r&r'rys  rreturncCsddtD}t|S)z Returns True if any tracing backend (OTel, Profiler, etc.) is available. Use this to guard expensive tracing logic in the main code. cSsg|] \}}}}}|qSr&r&).0r#r%r&r&r' s z(is_tracing_available..)rvaluesany)check_availabler&r&r'rsr)N)NNNN)&r3collections.abcrtypingrrrrrrr r r r r utilsrrrr__all__boolr__annotations__rrrrrdictstrtuplerrrintrrr&r&r&r's  (       "