qiVj XddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZddlmZmZddlmZmZddlmZddlmZdd lmZdd lmZdd l m!Z!dd l"m#Z#dd l$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl,m2Z2ddl3m4Z4m5Z5m6Z6m7Z7Gdde8Z9Gdde8Z:Gdde8Z;Gdde8Z<Gdde8Z=Gdd eZ>e d!ge e e e?e e?ffZ@ed"e e2ge e e e?e e?ff#ZAGd$d%e>ZBGd&d'e>ZCy)() annotationsN) ListUnionCallableDict AwaitableOptionalMappingcastTypeVar)abstractmethodABC) FfiClient FfiHandle)ffi_pb2)participant_pb2)TrackPublishOptions)TranscriptionSegment)ParticipantTrackPermission)BroadcastQueue) LocalTrack)LocalTrackPublicationRemoteTrackPublicationTrackPublication) Transcription)RpcError)"RpcMethodInvocationResponseRequest)logger)RpcInvocationData)TextStreamWriterByteStreamWriterByteStreamInfoSTREAM_CHUNK_SIZEceZdZddZy)PublishTrackErrorc||_yNmessageselfr*s I/opt/pipecat/venv/lib/python3.12/site-packages/livekit/rtc/participant.py__init__zPublishTrackError.__init__;  Nr*strreturnNone__name__ __module__ __qualname__r.r0r-r&r&:r0r&ceZdZddZy)UnpublishTrackErrorc||_yr(r)r+s r-r.zUnpublishTrackError.__init__@r/r0Nr1r5r9r0r-r<r<?r:r0r<ceZdZddZy)PublishDataErrorc||_yr(r)r+s r-r.zPublishDataError.__init__Er/r0Nr1r5r9r0r-r?r?Dr:r0r?ceZdZddZy)PublishDTMFErrorc||_yr(r)r+s r-r.zPublishDTMFError.__init__Jr/r0Nr1r5r9r0r-rBrBIr:r0rBceZdZddZy)PublishTranscriptionErrorc||_yr(r)r+s r-r.z"PublishTranscriptionError.__init__Or/r0Nr1r5r9r0r-rErENr:r0rEceZdZd dZeed dZeddZeddZeddZ eddZ eddZ eddZ edd Z e dd Zy ) Participantcn|j|_t|jj|_yr()info_inforhandleid _ffi_handle)r, owned_infos r-r.zParticipant.__init__Ts&__ $Z%6%6%9%9:r0cy)U A dictionary of track publications associated with the participant. Nr9r,s r-track_publicationszParticipant.track_publicationsXs r0c.|jjSr()rKsidrRs r-rUzParticipant.sid`szz~~r0c.|jjSr()rKnamerRs r-rWzParticipant.namedszzr0c.|jjSr()rKidentityrRs r-rYzParticipant.identityhzz"""r0c.|jjSr()rKmetadatarRs r-r\zParticipant.metadatalrZr0c@t|jjS)z2Custom attributes associated with the participant.)dictrK attributesrRs r-r_zParticipant.attributespsDJJ))**r0c.|jjS)zLParticipant's kind (e.g., regular participant, ingress, egress, sip, agent).)rKkindrRs r-razParticipant.kinduszzr0c.|jjS)z.The participant's permissions within the room.)rK permissionrRs r- permissionszParticipant.permissionszszz$$$r0c|jjtjjk(ry|jjS)aReason for the participant's disconnection. Returns one of DisconnectReasons or None if the participant isn't disconnected. Common reasons are: - CLIENT_INITIATED - the client initiated the disconnect - DUPLICATE_IDENTITY - another participant with the same identity has joined the room - SERVER_SHUTDOWN - the server instance is shutting down - PARTICIPANT_REMOVED - RoomService.RemoveParticipant was called - ROOM_DELETED - RoomService.DeleteRoom was called - STATE_MISMATCH - the client is attempting to resume a session, but server is not aware of it - JOIN_FAILURE - client was unable to connect fully When dialing a participant via SIP, you may see the following reasons: - USER_UNAVAILABLE - SIP callee did not respond in time - USER_REJECTED - SIP callee rejected the call (busy) - SIP_TRUNK_FAILURE - SIP protocol failure or unexpected response N)rKdisconnect_reasonproto_participantDisconnectReasonUNKNOWN_REASONrRs r-rfzParticipant.disconnect_reasons6( :: ' '+<+M+M+\+\ \zz+++r0NrO"proto_participant.OwnedParticipantr3r4)r3zMapping[str, TrackPublication]r3r2)r3dict[str, str])r3z+proto_participant.ParticipantKind.ValueType)r3z'proto_participant.ParticipantPermission)r3z6Optional[proto_participant.DisconnectReason.ValueType])r6r7r8r.propertyr rSrUrWrYr\r_rardrfr9r0r-rHrHSs;  ####++%%, ?,,r0rHr F)boundceZdZdZ d!fd Zed"dZdgdd d#dZd$dZd%d Z d d  d&d Z d' d(d Z d)dZ d d d*dZ d+dZd,dZd-dZd.dZd dd d d d d d d/dZd dd d d d0dZd dd d d dd d1dZdd d d d d2dZef d3dZd4dZd5d ZxZS)6LocalParticipantz+Represents the local participant in a room.cNt||||_i|_i|_yr()superr. _room_queue_track_publications _rpc_handlers)r, room_queuerO __class__s r-r.zLocalParticipant.__init__s* $%EG 46r0c|jSrQrvrRs r-rSz#LocalParticipant.track_publications '''r0T)reliabledestination_identitiestopicc Kt|tr|jd}t|}t j |z|}t j}|jj|j_ t j||j_ ||j_||j_||j_|jj"j%|t&j(j*j-} t&j(j/| |j1 fdd{} t&j(j*j3|| jj4rt7| jj4y7c#t&j(j*j3|wxYww)a Publish arbitrary data to the room. Args: payload (Union[bytes, str]): The data to publish. reliable (bool, optional): Whether to send reliably or not. Defaults to True. destination_identities (List[str], optional): List of participant identities to send to. Defaults to []. topic (str, optional): The topic under which to publish the data. Defaults to "". Raises: PublishDataError: If there is an error in publishing data. zutf-8c^|jjjjk(Sr() publish_dataasync_ideresps r-z/LocalParticipant.publish_data..s"!..11T5F5F5O5OOr0N) isinstancer2encodelenctypesc_byte proto_ffi FfiRequestrNrLrlocal_participant_handle addressofdata_ptrdata_lenrrrextendrinstancequeue subscriberequestwait_for unsubscribeerrorr?) r,payloadrrrrcdatareqrcbrs @r-rzLocalParticipant.publish_datass( gs #nnW-Gw<)G4""$484D4D4K4K1$*$4$4U$;!$,!$,!!& //667MN""((224 8%%--c2D+0>>O,&B    $ $ 0 0 7 ?? "2??#8#89 9 ! &    $ $ 0 0 7s1DG-#6F?F=F?AG-=F??+G**G-cKtj}|jj|j_||j_||j_tjjj} tjj||jfdd{}tjjj||jjrt!|jjy7c#tjjj|wxYww)z Publish SIP DTMF message. Args: code (int): DTMF code. digit (str): DTMF digit. Raises: PublishDTMFError: If there is an error in publishing SIP DTMF message. c^|jjjjk(Sr()publish_sip_dtmfrrs r-rz/LocalParticipant.publish_dtmf..s$!,,559N9N9W9WWr0N)rrrNrLrrcodedigitrrrrrrrrrB)r,rrrrrrs @r- publish_dtmfzLocalParticipant.publish_dtmfs""$8<8H8H8O8O5$(!%*"""((224 8%%--c2D+0>>W,&B    $ $ 0 0 7   $ $"2#6#6#<#<= = % &    $ $ 0 0 7s1BE6D$>D"?D$AE"D$$+EEc Ktj}|jDcgc]O}t|j|j |j |j|j|jQ}}|jj|j_ |j|j_|jjj||j |j_t$j&j(j+} t$j&j-||j/fdd{}t$j&j(j1||jj2rt5|jj2ycc}w7h#t$j&j(j1|wxYww)a Publish transcription data to the room. Args: transcription (Transcription): The transcription data to publish. Raises: PublishTranscriptionError: If there is an error in publishing transcription. )rMtext start_timeend_timefinallanguagec^|jjjjk(Sr()publish_transcriptionrrs r-rz8LocalParticipant.publish_transcription..s$!11::d>X>X>a>aar0N)rrsegmentsProtoTranscriptionSegmentrMrrrrrrNrLrrparticipant_identityr track_sidtrack_idrrrrrrrrrE)r, transcriptionrsproto_segmentsrrrs @r-rz&LocalParticipant.publish_transcriptions""$#++   &44VV<<gg      >B=M=M=T=T!!:9F9[9[!!6 !!**11.A-:-D-D!!*""((224 8%%--c2D+0>>a,&B    $ $ 0 0 7 # # ) )+B,D,D,J,JK K *5  (&    $ $ 0 0 7s=#G4AF?:B*G4%6GGG A$G4G+G11G4N)response_timeoutcPKtj}|jj|j_||j_||j_||j_|t|dz|j_ tjjj} tjj||j!fdd{}tjjj#||jj%dr)t'j(|jj*|jjS7#tjjj#|wxYww)a Initiate an RPC call to a remote participant. Args: destination_identity (str): The `identity` of the destination participant method (str): The method name to call payload (str): The method payload response_timeout (Optional[float]): Timeout for receiving a response after initial connection Returns: str: The response payload Raises: RpcError: On failure. Details in `message`. Nic^|jjjjk(Sr() perform_rpcrrs r-rz.LocalParticipant.perform_rpc..Fs" 0F0F$JZJZJcJc0cr0r)rrrNrLrrdestination_identitymethodrintresponse_timeout_msrrrrrrrHasFieldr _from_protor) r,rrrrrrrrs @r-rzLocalParticipant.perform_rpc$s/.""$373C3C3J3J0/C,!'")  '256F6M2NCOO /""((224 8%%--c2D~~&cddB    $ $ 0 0 7 >> " "7 +&&r~~';';< <~~%%%e    $ $ 0 0 7s1B4F&86E8.E6/E83BF&6E88+F##F&c,dfd }|||S|S)a Establishes the participant as a receiver for calls of the specified RPC method. Can be used either as a decorator or a regular method. The handler will receive one argument of type `RpcInvocationData` and should return a string response which will be forwarded back to the caller. The handler may be synchronous or asynchronous. If unable to respond within `response_timeout`, the caller will hang up and receive an error on their side. You may raise errors of type `RpcError` in the handler, and they will be forwarded to the caller. Other errors raised in your handler will be caught and forwarded to the caller as "1500 Application Error". Args: method_name (str): The name of the indicated RPC method. handler (Optional[Callable]): Handler to be invoked whenever an RPC request for this method is received. Omit this argument to use the decorator syntax. Returns: None (when used as a decorator it returns the decorator function) Example: # As a decorator: @room.local_participant.register_rpc_method("greet") async def greet_handler(data: RpcInvocationData) -> str: print(f"Received greeting from {data.caller_identity}: {data.payload}") return f"Hello, {data.caller_identity}!" # As a regular method: async def greet_handler(data: RpcInvocationData) -> str: print(f"Received greeting from {data.caller_identity}: {data.payload}") return f"Hello, {data.caller_identity}!" room.local_participant.register_rpc_method('greet', greet_handler) c|j<tj}jj|j _|j _tjj||Sr() rwrrrNrLregister_rpc_methodrrrrr) handler_funcr method_namer,s r-registerz6LocalParticipant.register_rpc_method..registerxsc.:D  { +&&(C?C?O?O?V?VC # # <-8C # # *    & &s + r0)rror3ror9)r,rhandlerrs`` r-rz$LocalParticipant.register_rpc_methodOs!R   G$ $Or0c|jj|dtj}|jj |j _||j _tjj|y)z Unregisters a previously registered RPC method. Args: method (str): The name of the RPC method to unregister N) rwpoprrrNrLunregister_rpc_methodrrrrr)r,rrs r-rz&LocalParticipant.unregister_rpc_methodsd vt,""$=A=M=M=T=T!!:+1!!(""3'r0)participant_permissionsc(|g}tj}|jj|j_||j_|jjj|tjj|y)a Set the track subscription permissions for the local participant. Args: allow_all_participants (bool): Whether to allow all participants to subscribe to this participant's tracks. participant_permissions (List[ParticipantTrackPermission]): Participant-specific track subscription permissions, ignored if `allow_all_participants` is True. N) rrrNrL"set_track_subscription_permissionsrall_participants_allowedrdrrrr)r,allow_all_participantsrrs r-rz3LocalParticipant.set_track_subscription_permissionssu # *&( #""$JNJZJZJaJa..GJ`..G ..::AABYZ""3'r0cKd}d}t||||} |jj|} | s.tjtj j }nZ tj| r' tj| | |d{}nttt | | }t+j,t/|j0j2||r|j5nd|} t6j8j;| } | j<j>r/| j<j>}t%j>d|yy7#tj$r.tjtj jtj$r.tjtj jwxYw#t$r } | }Yd} ~ Dd} ~ wt"$rJt%j&d|dtjtj j(}YwxYww)N)timeoutz+Uncaught error returned by RPC handler for z'. Returning APPLICATION_ERROR instead. )r invocation_idrr)rpc_method_invocation_responsez.error sending rpc method invocation response: ) r rwgetr _built_in ErrorCodeUNSUPPORTED_METHODasyncioiscoroutinefunctionr TimeoutErrorRESPONSE_TIMEOUTCancelledErrorRECIPIENT_DISCONNECTEDr r r2 Exceptionr exceptionAPPLICATION_ERRORrrrrNrL _to_protorrrrr)r,rr request_idcaller_identityrrresponse_errorresponse_payloadparamsrrrreserrs r-_handle_rpc_method_invocationz.LocalParticipant._handle_rpc_method_invocations.2*.":IYZ$$((0%//0B0B0U0UVN Z..w7\181A1A#FO5E2,((,HSM76?'K$""+M)-)9)9)@)@+4Bn..0( ,   ((-  - - 3 344::C LLI#O P 4;,#//V&001C1C1T1TUU"11\&001C1C1Z1Z[[\ '!& Z  A&J<<"*!3!3H4F4F4X4X!Y  ZsiAI!G-7 E*E(E*G-:B.I(E**BG**G-- I6G>8I>AIIIIcKtj}|jj|j_||j_tjjj} tjj||jfdd{tjjj|y7.#tjjj|wxYww)z Set the metadata for the local participant. Note: this requires `canUpdateOwnMetadata` permission. Args: metadata (str): The new metadata. c^|jjjjk(Sr()set_local_metadatarrs r-rz/LocalParticipant.set_metadata..s$!..774;R;R;[;[[r0N)rrrNrLrrr\rrrrrrr)r,r\rrrs @r- set_metadatazLocalParticipant.set_metadatas""$:>:J:J:Q:Q7*2'""((224 8%%--c2D..[      $ $ 0 0 7      $ $ 0 0 70A3D 76C-C.C2*D C+D  D cKtj}|jj|j_||j_tjjj} tjj||jfdd{tjjj|y7.#tjjj|wxYww)z Set the name for the local participant. Note: this requires `canUpdateOwnMetadata` permission. Args: name (str): The new name. c^|jjjjk(Sr()set_local_namerrs r-rz+LocalParticipant.set_name..s$!**33t7J7J7S7SSr0N)rrrNrLrrrWrrrrrrr)r,rWrrrs @r-set_namezLocalParticipant.set_names""$6:6F6F6M6M3"&""((224 8%%--c2D..S      $ $ 0 0 7      $ $ 0 0 7rc*Ktj}|jj|j_|jj Dcic]}|j|j}}|j||jD]7\}}|jj j}||_||_9tjjj} tjj!||j#fdd{tjjj%|ycc}w73#tjjj%|wxYww)z Set custom attributes for the local participant. Note: this requires `canUpdateOwnMetadata` permission. Args: attributes (dict[str, str]): A dictionary of attributes to set. c^|jjjjk(Sr()set_local_attributesrrs r-rz1LocalParticipant.set_attributes..&s$!0099T=V=V=_=__r0N)rrrNrLrrr_keyvalueupdateitemsaddrrrrrrr) r,r_rentryexisting_attributesrrrrs @r-set_attributeszLocalParticipant.set_attributes sL""$<@.s"!//22d6H6H6Q6QQr0N)rrrNrLr# track_handleroptionsCopyFromrurrrrrrr&r publication_trackrUrKrv task_doner)r,trackr%rrrtrack_publicationrs @r-r#zLocalParticipant.publish_tracksW ""$).):):)A)A&595E5E5L5L2 !!**73  **, 0%%--c2D+0>>Q,&B%%'(8(8(>(>?? 5b6F6F6R6R S ',  $/33EKKO>OD $ $%6%:%: ; OO $    ( ( /&    ( ( /s1BF;"6FFB$F=F;FF88F;cKtj}|jj|j_||j_d|j_|jj} tjj||jfdd{}|jjrt|jj|j j#|}d|_|j'|jj)|y7#|jj)|wxYww)z Unpublish a track from the room. Args: track_sid (str): The SID of the track to unpublish. Raises: UnpublishTrackError: If there is an error in unpublishing the track. Tc^|jjjjk(Sr()unpublish_trackrrs r-rz2LocalParticipant.unpublish_track..s$!++448L8L8U8UUr0N)rrrNrLr.rrstop_on_unpublishrurrrrrrr<rvrr(r)r)r,rrrrr'rs @r-r.z LocalParticipant.unpublish_tracks""$7;7G7G7N7N4(1%04-  **, 0%%--c2D+0>>U,&B!!'')"*<*<*B*BCC2266yAK!%K  OO     ( ( /&    ( ( /s1A6E:6D:0D81A+D:E8D::EEcVd|jd|jd|jdS)Nzrtc.LocalParticipant(sid= , identity=, name=)rUrYrWrRs r-__repr__zLocalParticipant.__repr__s.*488*K gVZV_V_U``abbr0)rxz"BroadcastQueue[proto_ffi.FfiEvent]rOrkr3r4)r3z#Mapping[str, LocalTrackPublication]) rzUnion[bytes, str]rboolrz List[str]rr2r3r4)rrrr2r3r4)rrr3r4) rr2rr2rr2rzOptional[float]r3r2r()rr2rz Optional[F]r3zUnion[F, Callable[[F], F]])rr2r3r4)rr6rz*Optional[List[ParticipantTrackPermission]]r3r4)rrrr2rr2rr2rr2rfloatr3r4)r\r2r3r4)rWr2r3r4)r_rmr3r4)rOptional[List[str]]rr2r_Optional[Dict[str, str]]r str | Nonerr:r int | Nonerr:r3r!) rr2rr8rr2r_r9rr:)rWr2rr;r r2r_r9rr:rr8rr2r3r") rr2rr2rr8r_r9rr:r3r#)r*rr%rr3r)rr2r3r4rl)r6r7r8__doc__r.rnrSrrrrrrrrrrrrr rr rr#r.r5 __classcell__rys@r-rrrrs576777  7((,. ,:",: ,: !* ,:  ,: ,:\>8&L\-1 )&")& )&  )& * )& )&\ $444 $ 4l (&OS (!%("L (  (,5Q5Q5Q 5Q  5Q  5Q 5Q 5Qn8,8,8B7;/3 $"&!%&*!4  -    $ D7;/3"&!4    -  6"&3/3 $6:    - !4 D6:/3 $  !4  -  BAT@U'0'0*='0 '0R0>cr0rrc:eZdZdfd ZeddZddZxZS)RemoteParticipantc2t||i|_yr()rtr.rv)r,rOrys r-r.zRemoteParticipant.__init__s $FH r0c|jSr{r|rRs r-rSz$RemoteParticipant.track_publicationsr}r0cVd|jd|jd|jdS)Nzrtc.RemoteParticipant(sid=r1r2r3r4rRs r-r5zRemoteParticipant.__repr__s.+DHH:[wW[W`W`Vaabccr0rj)r3z$Mapping[str, RemoteTrackPublication]rl)r6r7r8r.rnrSr5r=r>s@r-r@r@s#I(( dr0r@)D __future__rrrrrrtypingrrrrrr r r r abcr r _ffi_clientrr_protorrrrg_proto.room_pb2rrr_proto.track_pb2r_utilsrr*rr+rrrrrrpcr_proto.rpc_pb2rlogrr data_streamr!r"r#r$rr&r<r?rBrErHr2 RpcHandlerrorrr@r9r0r-rQs6#  [[[#-(8# )>"  ) y y   B,#B,J*+U9Xc]3KXVY]3Z-[[ \  x*+U9Xc]3KXVY]3Z-[[\ I c{I cX d dr0