o i @sdZddlZddlmZddlmZddlmZzddlZWn e y>Z ze de e de de dZ [ wwdd l mZGd d d eZdS) z/Sentry integration for frame processor metrics.N)Optional)logger)BaseTaskManagerz Exception: zEIn order to use Sentry, you need to `pip install pipecat-ai[sentry]`.zMissing module: )FrameProcessorMetricscseZdZdZfddZdeffdd ZfddZd d d ee d e ffd dZ d ddee ffddZ d d d ee ffddZ d ddee ffddZddZZS) SentryMetricszFrame processor metrics integration with Sentry monitoring. Extends FrameProcessorMetrics to send time-to-first-byte (TTFB) and processing metrics as Sentry transactions for performance monitoring and debugging. cs:td|_d|_t|_|jstdd|_ dS)zInitialize the Sentry metrics collector. Sets up internal state for tracking transactions and verifies Sentry SDK initialization status. Nz=Sentry SDK not initialized. Sentry features will be disabled.) super__init___ttfb_metrics_tx_processing_metrics_tx sentry_sdkis_initialized_sentry_availablerwarning _sentry_taskself __class__U/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/processors/metrics/sentry.pyr"s    zSentryMetrics.__init__ task_managercsHt|IdH|jr"t|_|jj||dd|_ dSdS)zSetup the Sentry metrics system. Args: task_manager: The task manager to use for background operations. Nz::_sentry_task_handler)name) rsetupr asyncioQueue _sentry_queuer create_task_sentry_task_handlerr)rrrrrr0s  zSentryMetrics.setupcs`tIdH|jr.|jdIdH|jIdHd|_t|dtjdddSdS)zClean up Sentry resources and flush pending transactions. Ensures all pending transactions are sent to Sentry before shutdown. Nz Flushing Sentry metricsg@)timeout) rcleanuprrputrtracer flushrrrrr=s zSentryMetrics.cleanupN start_timer$report_only_initial_ttfbcsntj||dIdH|jr3|jr5tjdd|d|_t |d|jj d|jj ddSdSdS) zStart tracking time-to-first-byte metrics. Args: start_time: Optional start timestamp override. report_only_initial_ttfb: Whether to report only the initial TTFB measurement. )r$r%Nttfbz TTFB for opr! Sentry transaction started (ID:  Name: )) rstart_ttfb_metrics_should_report_ttfbr r start_transaction_processor_namer rdebugspan_idr)rr$r%rrrr,Js   z SentryMetrics.start_ttfb_metricsend_timer3cHtj|dIdH|jr |jr"|j|jIdHd|_dSdSdS)zxStop tracking time-to-first-byte metrics. Args: end_time: Optional end timestamp override. r2N)rstop_ttfb_metricsr r rr rr3rrrr5`   zSentryMetrics.stop_ttfb_metricscsbtj|dIdH|jr/tjdd|d|_t|d|jj d|jj ddSdS) z{Start tracking frame processing metrics. Args: start_time: Optional start timestamp override. r#N processingzProcessing for r'r)r*r+) rstart_processing_metricsr r r.r/r rr0r1r)rr$rrrr9ls z&SentryMetrics.start_processing_metricscr4)zvStop tracking frame processing metrics. Args: end_time: Optional end timestamp override. r2N)rstop_processing_metricsr r rr r6rrrr:}r7z%SentryMetrics.stop_processing_metricscsVd}|r)|jIdH}|r|jd|jIdH|du}|j|sdSdS)z;Background task handler for completing Sentry transactions.TN)rgetrget_event_looprun_in_executorfinish task_done)rrunningtxrrrrs  z"SentryMetrics._sentry_task_handler)__name__ __module__ __qualname____doc__rrrrrfloatboolr,r5r9r:r __classcell__rrrrrs    r)rErtypingrlogurur"pipecat.utils.asyncio.task_managerrr ModuleNotFoundErroreerror Exception2pipecat.processors.metrics.frame_processor_metricsrrrrrrs