-`i@UddlZddlmZddlmZddlmZddgZeeZ dZ dZ e dze d< dd lmZdd lmZdd lmZdd lmZdd lmZmZmZddlmZdZ nR#e$rJddlZejZ GddZGddZGddZGddZYnwxYwde fdZ!de de dedzfdZ"dZ#dee e fdzdedzfdZ$dee e fdee e ffd Z%Gd!d"Z&dee e fde fd#Z'ed%d$Z(dS)&N)Mapping) init_logger)run_once traceparent tracestateFotel_import_error_traceback)Context)"OTEL_EXPORTER_OTLP_TRACES_PROTOCOL)TracerProvider)BatchSpanProcessor)SpanKindTracerset_tracer_provider)TraceContextTextMapPropagatorTceZdZdS)r N__name__ __module__ __qualname__`/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/tracing.pyr r & rr ceZdZdS)BaseSpanAttributesNrrrrrr)rrrceZdZdS)r Nrrrrr r ,rrr ceZdZdS)rNrrrrrr/rrrreturnctSN)_is_otel_importedrrris_otel_availabler"3s rinstrumenting_module_nameotlp_traces_endpointctstdtt}t |}|t |t|||}|S)Nz|OpenTelemetry is not available. Unable to initialize a tracer. Ensure OpenTelemetry packages are installed. Original error: ) r" ValueErrorrr get_span_exporteradd_span_processorr r get_tracer)r#r$trace_provider span_exportertracers r init_tracerr-7s     > ; > >   $%%N%&:;;M%%&8&G&GHHH'''  & &'@ A AF Mrctjtd}|dkrddlm}n |dkrddlm}ntd|d||S)Ngrpcr)OTLPSpanExporterz http/protobufzUnsupported OTLP protocol 'z' is configured)endpoint)osenvirongetr 5opentelemetry.exporter.otlp.proto.grpc.trace_exporterr05opentelemetry.exporter.otlp.proto.http.trace_exporterr&)r1protocolr0s rr'r'Jsz~~@&IIH6        _ $ $       PxPPPQQQ  X . . ..rheaderscltr%|pi}t|SdSr )r"rextractr8s rextract_trace_contextr<Zs8-R,..66w???trc*fdtDS)Nc*i|]}|v||Srr.0hr8s r z)extract_trace_headers..cs$ A A AaALLAwqzLLLr) TRACE_HEADERSr;s`rextract_trace_headersrDbs A A A A= A A AArcVeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdS)SpanAttributeszgen_ai.usage.completion_tokenszgen_ai.usage.prompt_tokenszgen_ai.request.max_tokenszgen_ai.request.top_pzgen_ai.request.temperaturezgen_ai.response.modelzgen_ai.request.idzgen_ai.request.nzgen_ai.usage.num_sequenceszgen_ai.latency.time_in_queuez"gen_ai.latency.time_to_first_tokenzgen_ai.latency.e2ez gen_ai.latency.time_in_schedulerz$gen_ai.latency.time_in_model_forwardz$gen_ai.latency.time_in_model_executez$gen_ai.latency.time_in_model_prefillz#gen_ai.latency.time_in_model_decodez&gen_ai.latency.time_in_model_inferenceN)rrrGEN_AI_USAGE_COMPLETION_TOKENSGEN_AI_USAGE_PROMPT_TOKENSGEN_AI_REQUEST_MAX_TOKENSGEN_AI_REQUEST_TOP_PGEN_AI_REQUEST_TEMPERATUREGEN_AI_RESPONSE_MODELGEN_AI_REQUEST_IDGEN_AI_REQUEST_NGEN_AI_USAGE_NUM_SEQUENCESGEN_AI_LATENCY_TIME_IN_QUEUE"GEN_AI_LATENCY_TIME_TO_FIRST_TOKENGEN_AI_LATENCY_E2E GEN_AI_LATENCY_TIME_IN_SCHEDULER$GEN_AI_LATENCY_TIME_IN_MODEL_FORWARD$GEN_AI_LATENCY_TIME_IN_MODEL_EXECUTE$GEN_AI_LATENCY_TIME_IN_MODEL_PREFILL#GEN_AI_LATENCY_TIME_IN_MODEL_DECODE&GEN_AI_LATENCY_TIME_IN_MODEL_INFERENCErrrrFrFfs&F"!= ;1!=3+)!=#A )M&-'I$+Q(,R(+Q(*O'-U***rrFcDtfdtDS)Nc3 K|]}|vV dSr rr?s r z)contains_trace_headers..s'33qG|333333r)anyrCr;s`rcontains_trace_headersr]s& 3333]333 3 33rc:tddS)Nz=Received a request with trace context but tracing is disabled)loggerwarningrrrlog_tracing_disabled_warningras NNRSSSSSr)rN))r2collections.abcr vllm.loggerrvllm.utils.func_utilsrrCrr_r!rstr__annotations__opentelemetry.context.contextr 'opentelemetry.sdk.environment_variablesr opentelemetry.sdk.tracer opentelemetry.sdk.trace.exportr opentelemetry.tracer rr,opentelemetry.trace.propagation.tracecontextr ImportError traceback format_excrboolr"r-r'r<rDrFr]rarrrrqsU ############****** - X  *.S4Z...  555555766666AAAAAAIIIIIIIIII   "6)"6"8"8                                  % ,4":= d]& / / / 738#4t#;$B738#4Bc9JBBBBVVVVVVVV64GCH$54$4444 TTT TTTs*AA B.-B.