:jdZddlmZddlZddlZddlmZddlmZddl m Z ddl Z ddl Z ddlmZddlmZdd lmZdd lmZeej.d z Zee5Ze j6eZddded Zej=d iZe e Z!edZ"eeedZ#edZ$e$jKee#dde$jMe'Z(ddZ)d dZ* d! d"dZ+dejXvr&e edj[ejXd<d#dZ.d$dZ/d%d&dZ0d'dZ1e1e"y#1swYxYw)(atOpenTelemetry metrics for tracking function usage. Metrics can be enabled/disabled for the current python kernel: >>> from pyannote.audio.telemetry.metrics import set_telemetry_metrics >>> set_telemetry_metrics(True) # enable >>> set_telemetry_metrics(False) # disable To activate / deactivate globally: >>> set_telemetry_metrics(True, save_choice_as_default=True) ) annotationsN)Path)Any)uuid4)OTLPSpanExporter)TracerProvider)BatchSpanProcessor) __version__z config.yaml otlp_endpoint otlp_headerstelemetry_log_level)endpointheaderstimeoutF)shutdown_on_exitii)schedule_delay_millisexport_timeout_millisctrGtjd5}|jt |ddt t ddddyy#1swYyxYw)z Increment the model initialization counter. Parameters ---------- model : Model Instantiated model zoss-model-init _otel_originunknown)originversion session_idNis_metrics_enabledtracerstart_as_current_spanset_attributesgetattrr SESSION_ID)modelspans u/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/pyannote/audio/telemetry/metrics.pytrack_model_initr%2s_  ) )*: ; t   %e^YG*",     s )AAc trStjd5}|jt |ddt |ddt t ddddyy#1swYyxYw)z| Track pipeline initialization Parameters ---------- pipeline : Pipeline Instantiated pipeline zoss-pipeline-initrr _otel_name)rnamerrNr)pipeliner#s r$track_pipeline_initr*Fsk  ) )*= > $   %h J#HlIF*",      s 5AA(c ntrddlm}|j|}ddlm}||||\}}}t jd5} | jt|ddt|ddtt|||tj|r|nd d d d d y y #1swYy xYw) ad Track duration of files processed by pipelines Parameters ---------- pipeline : Pipeline The pipeline being applied. file : AudioFile The audio file being processed by the pipeline. num_speakers, min_speakers, max_speakers : int, optional Requested number of speakers in the audio file, if applicable. r)Audio)set_num_speakers) num_speakers min_speakers max_speakerszoss-pipeline-applyrrr'inf)rr(rrdurationr.r/r0N)rpyannote.audio.core.ior, get_duration*pyannote.audio.pipelines.utils.diarizationr-rrrr r r!npisfinite) r)filer.r/r0kwargsr,r2r-r#s r$track_pipeline_applyr:[s*0'..t4O3C%%%4 0 lL  ) )*> ? 4   %h J#HlIF*", ($0$046KK 4MLSX      s AB++B4PYANNOTE_METRICS_ENABLEDmetrics_enabledcdtjvr tdtjdjdvS)z Check if metrics are enabled. Returns ------- bool True if metrics are enabled, False otherwise. r;z9PYANNOTE_METRICS_ENABLED environment variable is not set.)true1)osenviron ValueErrorlowerr$rrs:"3TUU ::0 1 7 7 9] JJrEctt5}tj|}ddd|d<ttd5}tj||dddy#1swY>xYw#1swYyxYw)z Save metrics state to the configuration file. Parameters ---------- enabled : bool Whether metrics should be enabled or disabled. Nr<w)open CONFIG_FILEyaml safe_load safe_dump)enabledfconfigs r$_save_metrics_configrPsr k #a"#!(F  k3 "1 vq!"" ## ""sA#A/#A,/A8ct|}t|jtjd<|r t |yy)a Enable or disable metrics globally. Parameters ---------- enabled : bool Whether to enable or disable the metrics. save_choice_as_default : bool, optional If True, saves the state to the configuration file as the default value. r;N)boolstrrCr@rArP)rMsave_choice_as_defaults r$set_telemetry_metricsrUs97mG-0\-?-?-ABJJ)*W%rEcHd}|j}||vrtddj|tt|}tj j jD]8}|jdst j|j|:y)ac Set the log level for OpenTelemetry loggers to control error visibility. This can be used to suppress error messages when telemetry fails. Parameters ---------- level : str The log level to set. Must be one of: 'DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL'. - 'DEBUG': Show all messages including detailed debugging information - 'INFO': Show informational messages, warnings and errors - 'WARNING': Show only warnings and errors (default) - 'ERROR': Show only errors - 'CRITICAL': Show only critical errors Raises ------ ValueError If the provided level is not a valid logging level. Examples -------- To suppress most error messages: >>> set_opentelemetry_log_level('ERROR') To show all messages: >>> set_opentelemetry_log_level('DEBUG') )DEBUGINFOWARNINGERRORCRITICALzLog level must be one of: z, opentelemetryN) upperrBjoinr loggingrootmanager loggerDict startswith getLoggersetLevel)level valid_levels logging_level logger_names r$set_opentelemetry_log_levelrjs@EL KKME L 5dii 6M5NOPPGU+M||++66C  ! !/ 2   k * 3 3M BCrE)r"z'Model'returnNone)r) 'Pipeline'rkrl)NNN)r)rmr8z 'AudioFile'r. int | Noner/rnr0rnr9rrkrl)rkrR)rMrRrkrl)F)rMrRrTrRrkrl)rfrSrkrl)2__doc__ __future__rr_r@pathlibrtypingruuidrnumpyr6rJ5opentelemetry.exporter.otlp.proto.http.trace_exporterropentelemetry.sdk.traceropentelemetry.sdk.trace.exportr pyannote.audior __file__parentrIrHconfig_file_streamrKCONFIG OTLP_ENDPOINTget OTLP_HEADERSrSr!DEFAULT_LOG_LEVELexporterprovideradd_span_processor get_tracer__name__rr%r*r:rArCrrPrUrjrDrEr$rs #  R2=&8n##m3  +0, T^^. /F0' zz."- \ 01 ]LRS T 5 1.xsjnop   X &(0 $## ,, ,, ,  ,  , ,`RZZ/-08I1J-K-Q-Q-SBJJ)* K "$&",C^-.]00s EE