+ ~j5^RIHt^RIt^RIt^RIHt^RIt^RIH t ^RI H t ^RI H t ^RIHt^RIHtHtHtHt^R IHt^R IHtHtHtHtHt!R R ]4t!R R]4t]!RR44t!RR4t R#)) annotationsN) dataclass)rtc)utils) LanguageCode)logger)DEFAULT_API_CONNECT_OPTIONS NOT_GIVENAPIConnectOptions NotGivenOr)AudioByteStream)STTRecognizeStream SpeechData SpeechEventSpeechEventTypec va]tRt^tRRRRR]RRRR/R V3R llltR ]R ]/R RlltR ]R ]/RRlltRt V;t #)MultiSpeakerAdapterdetect_primary_speakerTsuppress_background_speakerFprimary_detection_optionsprimary_format{text}background_formatc 0V^8dQhRRRRRRRRRR R R /#) sttrrboolrr*NotGivenOr[PrimarySpeakerDetectionOptions]rstrr)formats"}/Users/mitch_tango/dev/rabbit-r1-livekit/agent/.venv/lib/python3.14/site-packages/livekit/agents/stt/multi_speaker_adapter.py __annotate__ MultiSpeakerAdapter.__annotate__sI%4%4%4!% %4 &* %4 $N %4%4%4c <VPP'g \R4h\SV`VPR7WnW nW0nT;'g \4Vn WPn W`n R#)aMultiSpeakerAdapter is an adapter that allows to detect and suppress background speakers. It needs STT with diarization capability and works for a single audio track. Args: stt (STT): STT instance to wrap detect_primary_speaker (bool, optional): Whether to detect primary speaker. Defaults to True. suppress_background_speaker (bool, optional): Whether to suppress background speaker. Defaults to False. primary_detection_options (NotGivenOr[PrimarySpeakerDetectionOptions], optional): Primary speaker detection options. If not provided, the default options will be used. primary_format (str, optional): Format for primary speaker. Supports {text} and {speaker_id} placeholders. Defaults to "{text}". background_format (str, optional): Format for background speaker. Supports {text} and {speaker_id} placeholders. Defaults to "{text}". Raises: ValueError: If the STT does not support diarization. z9MultiSpeakerAdapter needs STT with diarization capability) capabilitiesN) r( diarization ValueErrorsuper__init___stt_detect_primary_suppress_backgroundPrimarySpeakerDetectionOptions_opt_primary_format_background_format)selfrrrrrr __class__s&$$$$$$r#r,MultiSpeakerAdapter.__init__si6+++XY Y c&6&67 5$?!-QQ1O1Q -"3r&language conn_optionsc(V^8dQhRRRRRRRR/#) rbufferzutils.AudioBufferr7NotGivenOr[str]r8r returnrr!)r"s"r#r$r%;s8__!_" _ ( _  _r&c X"VPPWVR7GRjxL #L5i)r7r8N)r- recognize)r4r:r7r8s&&$$r#_recognize_impl#MultiSpeakerAdapter._recognize_impl;s(YY((Q](^^^^s !*(*c$V^8dQhRRRRRR/#)rr7r;r8r r<rr!)r"s"r#r$r%Ds*  " (    r&c 0\WPWR7#))r wrapped_sttr7r8)MultiSpeakerAdapterWrapperr-)r4r7r8s&$$r#streamMultiSpeakerAdapter.streamDs *))h  r&)r3r.r1r2r-r/) __name__ __module__ __qualname____firstlineno__r r,r r@rF__static_attributes__ __classcell__r5s@r#rrs~%4(, %4 -2 %4 QZ %4'%4"*%4%4N_%. _ +F _ %. +F   r&rc:a]tRt^OtRV3RlltRRltRtV;t#)rEc(V^8dQhRRRRRRRR/#) rrrrDrr7r;r8r r!)r"s"r#r$'MultiSpeakerAdapterWrapper.__annotate__Ps2     "  (  r&c <\SV`WR7W nW0n\ VP VP VPVPVPR7Vn R#))rr8)rrrrrN) r+r, _wrapped_stt _language_PrimarySpeakerDetectorr.r/r1r2r3 _detector)r4rrDr7r8r5s&&$$$r#r,#MultiSpeakerAdapterWrapper.__init__PsW S<'!0##6(+(@(@&)hh..!44  r&cV^8dQhRR/#)rr<Noner!)r"s"r#r$rQds("("D("r&c Za"RV3RllpRV3RllpSPPSPSPR7p\P !V!V4RR7\P !V!V4RR7.p\P !V!GRjxL \PP!V!GRjxL VP4GRjxL R#LDL L \PP!T!GRjxL TP4GRjxL i;i5i) c V^8dQhRRRR/#rrFrr<rYr!)r"s"r#r$5MultiSpeakerAdapterWrapper._run..__annotate__es # # #T #r&c<"SPRjxL p\V\P4'd/VP V4SP P V4KX\VSP4'gKvVP4KLD\P!\4;_uu_4TP4RRR4R# +'giR#;i5iN) _input_ch isinstancer AudioFrame push_framerV push_audio_FlushSentinelflush contextlibsuppress RuntimeError end_input)rFframer4s& r#_forward_input7MultiSpeakerAdapterWrapper._run.._forward_inputes#~~ # #eeS^^44%%e,NN--e4t':':;;LLN  #~$$\22  "3222sD C,BBBA)C,C,B#C,=C C, C) # C,c V^8dQhRRRR/#r\r!)r"s"r#r$r]ps  / d r&c ~<"VRjxL pSPPV4pVeSPPV4KFVP\ P 8XgKgSPP\\ P \\R4RR7.R74KLDR#5i)N)r7text)type alternatives) rV on_stt_event _event_ch send_nowaitrrrFINAL_TRANSCRIPTrrr)rFev updated_evr4s& r#_forward_output8MultiSpeakerAdapterWrapper._run.._forward_outputps"  b!^^88< )NN..z:WW @ @@NN..#!0!A!A*4l2>NUW*X)Y Fs-B=B:B8B:AB=-A B=8B::B=r>z'DiarizationAdapterWrapper.forward_input)namez(DiarizationAdapterWrapper.forward_outputN) rSrFrT _conn_optionsasyncio create_taskgatherraiocancel_and_waitaclose)r4rlrzrFtaskssf r#_runMultiSpeakerAdapterWrapper._runds # #  ""))4>>PTPbPb)c   v&-V    '.X    "..%( ( ())++U3 3 3--/ ! ! ) 3 !))++U3 3 3--/ ! !slBD+C&C C& !D+C"D+C$D+ C&"D+$D+&"D(D  D(!D$"D((D+)rVrTrS)rHrIrJrKr,rrLrMrNs@r#rErEOs  (("("r&rEc]tRt^t$Rt^dtR]R&RtR]R&^tR]R&Rt R]R &R t R]R &^]tRt^t$R]R&RtR]R&RtR]R&RtR#) #_PrimarySpeakerDetector.SpeakerDatar speaker_idrlast_activity_timermsr!N)rHrIrJrKrrrrLr!r&r# SpeakerDatars$'E'Ur&rrTrFrrrrc ,V^8dQhRRRRRRRRRR/#) rrrrrrrr rr!)r"s"r#r$$_PrimarySpeakerDetector.__annotate__sF(S(S!%(S&* (S $N (S  (S(Sr&c W@nWPnWnW nT;'g \ 4VnVP'd0VP'g\ P!R4RVnRVnRVn /Vn RVn .Vn VP PR, Vn\VP P VP, 4VnR#)aPrimary speaker detector. It detects the primary speaker based on RMS, formats the primary and background speakers separately, or suppresses the background speaker. Args: detect_primary_speaker (bool, optional): Whether to detect primary speaker. Defaults to True. suppress_background_speaker (bool, optional): Whether to suppress background speaker. Defaults to False. primary_detection_options (PrimaryDetectionOptions, optional): Primary speaker detection options. primary_format (str, optional): Format for primary speaker. Supports {text} and {speaker_id} placeholders. Defaults to "{text}". background_format (str, optional): Format for background speaker. Supports {text} and {speaker_id} placeholders. Defaults to "{text}". zVSuppressing background speaker is not supported when `detect_primary_speaker` is FalseFrNi)r2r3r.r/r0r1rwarning_pushed_duration_primary_speaker _speaker_data_bstream _rms_bufferr _frame_sizerr _max_rms_size)r4rrrrrs&$$$$$r#r, _PrimarySpeakerDetector.__init__s* ."35$?!-QQ1O1Q  $ $ $T-A-A-A NNh ).D %'*,0MO04 (*9922T9 !>!>AQAQ!QRr&c V^8dQhRRRR/#)rrkrtc.AudioFramer<rYr!)r"s"r#r$rsGGG4Gr&c VP'g&V;PVP, unR#VP'ge\ VP VP ,4p\VP VPVR7VnW!P , VnVPPVP4FRpVPV4pVPPV4V;PVP, unKT \VP4VP8d"VPVP)RVn R#R#)N) sample_rate num_channelssamples_per_channel)r.rdurationrrrrr rpushdata _compute_rmsrappendlenr)r4rksample_per_channelfrs&& r#rd"_PrimarySpeakerDetector.push_audios###  ! !U^^ 3 ! }}}!$U%6%69I9I%I!J +!--"//$6DM 24E4EED ##EJJ/A##A&C    # #C (  ! !QZZ / !0 t 4#5#5 5#//1C1C0C0EFD  6r&c V^8dQhRRRR/#)rrxrr<zSpeechEvent | Noner!)r"s"r#r$rs{/Ar&c TVP'gV#VP^,pVP\P8XdVP V4VP eVP fV#VP VP 8HVnVP'd9VPPVPVP R7Vn V#VP'dR#VPPVPVP R7Vn V#)rN)rqr) rsrrrrw_update_primary_speakerrris_primary_speakerr2r"rqr/r3)r4rxsds&& r#rt$_PrimarySpeakerDetector.on_stt_eventsI __Q  77o66 6  ( ( , == D$9$9$AI " 1F1F F  **11rww2==1YBG  (((--44"''bmm4\BG r&c V^8dQhRRRR/#)rrkrr<rr!)r"s"r#r$rs.Ur&c 8\P!VP\PR7p\ V4^8XdR#\P !\P !VP\P4^,44p\V4#))dtyper) np frombufferrint16rsqrtmeanastypefloat32r)r4rk audio_datars&& r#r$_PrimarySpeakerDetector._compute_rmss^]]5::RXX> z?a ggbggj// ;q@ABSzr&c$V^8dQhRRRRRR/#)r start_timerend_timer<z float | Noner!)r"s"r#r$r s!===%=L=r&c VVP'gR#\VPV, VP, 4p\VPV, VP, 4p\ VP4V, ^, p\ VP4V, pV^8gV\ VP48dR#\ V^4pWC, VP P8dR#\\P!VPW444#r_) rrrrrmaxr1rrrmedian)r4rrstartends&&& r#_get_rms_for_timerange._PrimarySpeakerDetector._get_rms_for_timerange sT**Z74;K;KKL4((83t7G7GGHD$$%-1$""#c) 7es4#3#344E1  ;22 2RYYt//:;<C>C*>C>Cr&c  8VPeVP'g RVnR#VPVPVP 4pVfR#VPpVP PV4;p'dhVP VnVPVPP,V^VPP, ,,Vn M/\PVVP VR7VP V&VPV8XdR#VPe*VP PVP4;pf%W0n\P!RRVRV/R7R#VP VP, pVPP"R8d:VPP"R, VPP$, pMRp\'VPP"Wv,, VPP(4pVPV,p RVRVRV R VR V/p W)8d!W0n\P!R V R7R#\P!R V R7R#) N)rrrzset first primary speakerrr)extrag?r rms_thresholdsilence_duration multiplierzprimary speaker switchedzprimary speaker unchanged)rr.rrrrrgetrrr1rrUrrdebugrrrrr) r4rrrrprimaryr decay_raterrrs && r#r/_PrimarySpeakerDetector._update_primary_speakers# == (<(<(<$(D ! ))"--E ; ]] %%))*5 54 5&(kkD #xx$))"@"@@3DII222DDH.E-P-P%#%;;.Q.D  z *  J .   ! ! )--11$2G2GHHQ$. ! LL4\:W\^a$))B_B__JJ II * *j.K L II . .  j0 * 3 ]  0 *    $. ! LL35 A LL4E Br&) r3rr.rrr1r2rrrrr/N)rHrIrJrKrrr r,rdrtrrrrLr!r&r#rUrUsz (S(,(S-2 (S QZ (S ' (S"*(STG,.=$>C>Cr&rU)! __future__rr~rg dataclassesrnumpyrlivekitrrprr7rlogrtypesr r r r utils.audior rrrrrrrrEr0rUr!r&r#rss"!#YY)OO9 #9 x="="@ QQ Q*vCvCr&