i2FdZddlmZddlZddlZddlmZddlmZddl m Z ddl Z ddl m Z ddlmZdd lmZdd lmZdd lmZddlZeej2d z Zee5Ze j:eZddded ZejAdiZ!e"e Z#edZ$eee!Z%ee%dZ&ee&gZ'e jPe'e jRe*Z+e+jYdddZ-d(dZ.e+jYdddZ/d)dZ0e+jcdgddd Z2 d* d+d!Z3d"ejhvr&e"ed#jkejhd"<d,d$Z6d-d%Z7d.d/d&Z8d0d'Z9e9e$y#1swYxYw)1atOpenTelemetry 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)metrics)OTLPMetricExporter) MeterProvider)PeriodicExportingMetricReader) __version__z config.yaml otlp_endpoint otlp_headerstelemetry_log_level)endpointheadersi`)export_interval_millis)metric_readerszoss-model-initzNumber of model initializations1)name descriptionunitcttr.tjdt|ddtt dyy)z Increment the model initialization counter. Parameters ---------- model : Model Instantiated model  _otel_originunknown)originversion session_idN)is_metrics_enabledmodel_init_counteraddgetattrr SESSION_ID)models u/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/pyannote/audio/telemetry/metrics.pytrack_model_initr%;s7 !%C&(  zoss-pipeline-initz"Number of pipeline initializationsc tr:tjdt|ddt|ddtt dyy)z| Track pipeline initialization Parameters ---------- pipeline : Pipeline Instantiated pipeline rrr _otel_name)rrrrN)rpipeline_init_counterr r!r r")pipelines r$track_pipeline_initr+WsC!! !(NIF, B&(   r&zoss-pipeline-apply) <xi,iXiii i@8ipiQzDuration of file in secondss)r#explicit_bucket_boundaries_advisoryrrc "trddlm}|j|}ddlm}||||\}}}t j|t|ddt|ddtt||tj|r|ndd y y ) 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_speakersrrr(inf)rrrrr6r7r8N) rpyannote.audio.core.ior4 get_duration*pyannote.audio.pipelines.utils.diarizationr5file_duration_bucketrecordr!r r"npisfinite) r*filer6r7r8kwargsr4durationr5s r$track_pipeline_applyrDs*0'..t4O3C%%%4 0 lL ## !(NIF, B&( , ,02 L0I u  r&PYANNOTE_METRICS_ENABLEDmetrics_enabledcdtjvr tdtjdjdk(S)z Check if metrics are enabled. Returns ------- bool True if metrics are enabled, False otherwise. rEz9PYANNOTE_METRICS_ENABLED environment variable is not set.true)osenviron ValueErrorlowerr&r$rrs:"3TUU ::0 1 7 7 9V CCr&ctt5}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. NrFw)open CONFIG_FILEyaml safe_load safe_dump)enabledfconfigs r$_save_metrics_configrXsb 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. rEN)boolstrrLrIrJrX)rUsave_choice_as_defaults r$set_telemetry_metricsr]s97mG-0\-?-?-ABJJ)*W%r&cHd}|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) upperrKjoinr!loggingrootmanager loggerDict startswith getLoggersetLevel)level valid_levels logging_level logger_names r$set_opentelemetry_log_levelrrs@EL KKME L 5dii 6M5NOPPGU+M||++66  ! !/ 2   k * 3 3M B7r&)r#z'Model'returnNone)r* 'Pipeline'rsrt)NNN)r*rurAz 'AudioFile'r6 int | Noner7rvr8rvrBrrsrt)rsrZ)rUrZrsrt)F)rUrZr\rZrsrt)rnr[rsrt):__doc__ __future__rrgrIpathlibrtypingruuidrrRrdr6opentelemetry.exporter.otlp.proto.http.metric_exporterropentelemetry.sdk.metricsr opentelemetry.sdk.metrics.exportr pyannote.audior numpyr?__file__parentrQrPconfig_file_streamrSCONFIG OTLP_ENDPOINTget OTLP_HEADERSr[r"DEFAULT_LOG_LEVELexporterreaderproviderset_meter_provider get_meter__name__metercreate_counterrr%r)r+create_histogramr=rDrJrLrrXr]rrrMr&r$rs  #  !U3J&8n##m3  +, T^^. /F' zz."- \ 01 }l K &x N  18$(#)) 1 *  *,, 4 - *-- ). %.0 $## + + + +  +  +  +  + \RZZ/-08I1J-K-Q-Q-SBJJ)* D "$&",C^-.cs &FF