o pi2@sdZddlmZddlZddlZddlmZddlmZddl m Z ddl Z ddl m Z ddlmZdd lmZdd lmZdd lmZddlZeejd Zee Ze eZWdn1scwYed Ze diZ!e"e Z#edZ$eee!dZ%ee%ddZ&ee&gdZ'e (e'e )e*Z+e+j,ddddZ-dCddZ.e+j,ddddZ/dDd"d#Z0e+j1d$gd%d&d'd(Z2   dEdFd1d2Z3d3ej4vre"ed45ej4d3<dGd6d7Z6dHd9d:Z7dIdJd=d>Z8dKdAdBZ9e9e$dS)LatOpenTelemetry 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 descriptionunitmodel'Model'returnNonecCs*trtdt|ddttddSdS)z Increment the model initialization counter. Parameters ---------- model : Model Instantiated model  _otel_originunknown)originversion session_idN)is_metrics_enabledmodel_init_counteraddgetattrr SESSION_ID)rr%T/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/audio/telemetry/metrics.pytrack_model_init;s  r'zoss-pipeline-initz"Number of pipeline initializationspipeline 'Pipeline'cCs4trtdt|ddt|ddttddSdS)z| Track pipeline initialization Parameters ---------- pipeline : Pipeline Instantiated pipeline rrr _otel_name)rrrrN)r pipeline_init_counterr"r#r r$)r(r%r%r&track_pipeline_initWs   r,zoss-pipeline-apply) <xi,iXiii i@8ipiQzDuration of file in secondss)r#explicit_bucket_boundaries_advisoryrrfile 'AudioFile' num_speakers int | None min_speakers max_speakerskwargsrc Kstr>ddlm}||}ddlm}||||d\}}}t|t|ddt|ddt t ||t |r7|ndd d Sd S) 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)r6r8r9rrr*inf)rrrrr6r8r9N) r pyannote.audio.core.ior; get_duration*pyannote.audio.pipelines.utils.diarizationr<file_duration_bucketrecordr#r r$npisfinite) r(r4r6r8r9r:r;durationr<r%r%r&track_pipeline_applys*      rFPYANNOTE_METRICS_ENABLEDmetrics_enabledboolcCs$dtjvr tdtjddkS)z Check if metrics are enabled. Returns ------- bool True if metrics are enabled, False otherwise. rGz9PYANNOTE_METRICS_ENABLED environment variable is not set.true)osenviron ValueErrorlowerr%r%r%r&r s r enabledcCsvtt }t|}Wdn1swY||d<ttd}t||WddS1s4wYdS)z Save metrics state to the configuration file. Parameters ---------- enabled : bool Whether metrics should be enabled or disabled. NrHw)open CONFIG_FILEyaml safe_load safe_dump)rOfconfigr%r%r&_save_metrics_configs  "rXFsave_choice_as_defaultcCs.t|}t|tjd<|rt|dSdS)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. rGN)rIstrrNrKrLrX)rOrYr%r%r&set_telemetry_metricss   r[levelrZcCs`d}|}||vrtdd|tt|}tjjjD]}|dr-t | |qdS)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) upperrMjoinr#loggingrootmanager loggerDict startswith getLoggersetLevel)r\ valid_levels logging_level logger_namer%r%r&set_opentelemetry_log_levels   ro)rrrr)r(r)rr)NNN)r(r)r4r5r6r7r8r7r9r7r:rrr)rrI)rOrIrr)F)rOrIrYrIrr)r\rZrr):__doc__ __future__rrerKpathlibrtypingruuidrrSrbr6opentelemetry.exporter.otlp.proto.http.metric_exporterropentelemetry.sdk.metricsr opentelemetry.sdk.metrics.exportr pyannote.audior numpyrC__file__parentrRrQconfig_file_streamrTCONFIG OTLP_ENDPOINTget OTLP_HEADERSrZr$DEFAULT_LOG_LEVELexporterreaderproviderset_meter_provider get_meter__name__metercreate_counterr!r'r+r,create_histogramrArFrLrNr rXr[ror%r%r%r&sp                     .     /