o i@sdddlmZddlZddlmZddlmZdddd dZddddddZddddddZ dS)) annotationsN)rtc)ATTRIBUTE_AGENT_NAME agent_nameroomrtc.Roomr str | Nonereturnrtc.RemoteParticipantc s|s tdtdfdd dfd d }dfdd }dfdd }|d||d||d|z;|jD]}|r`|W|d||d||d|SqCIdHW|d||d||d|S|d||d||d|w)a Wait for an agent participant to join the room. Args: room: The room to wait for the agent in. agent_name: If provided, waits for an agent with matching lk.agent.name attribute. If None, returns the first agent participant found. Returns: The agent participant. Raises: RuntimeError: If the room is not connected. room is not connectedpr r boolcs.|jtjjkr dSdurdS|jtkS)NFT)kindrParticipantKindPARTICIPANT_KIND_AGENT attributesgetrrrb/var/www/html/livekit_bhavya/venv/lib/python3.10/site-packages/livekit/agents/utils/participant.py matches_agent"s z%wait_for_agent..matches_agentNonecs&|rs|dSdSdSN)done set_resultrfutrrron_participant_connected)sz0wait_for_agent..on_participant_connectedchanged list[str]rtc.Participantcs6t|tjr|rs|dSdSdSdSr) isinstancerRemoteParticipantrr)r rrrron_attributes_changed-s z-wait_for_agent..on_attributes_changedstateintc.|tjjkrstddSdSdS)Nz5room disconnected while waiting for agent participantrConnectionStateCONN_DISCONNECTEDr set_exception RuntimeErrorr&rrron_connection_state_changed1z3wait_for_agent..on_connection_state_changedparticipant_connectedparticipant_attributes_changedconnection_state_changedNrr r rrr r r)r r!rr"r rr&r'r r) isconnectedr-asyncioFutureonremote_participantsvaluesoff)rrrr%r0rr)rrrrwait_for_agent s4         r?)identityrr@rJlist[rtc.ParticipantKind.ValueType] | rtc.ParticipantKind.ValueType | Nonec s|s tdtjtjdfdd dfd d }dfd d }|d||d|z$|jD] }|| rEnq9IdHW| d|| d|S| d|| d|w)z Returns a participant that matches the given identity. If identity is None, the first participant that joins the room will be returned. If the participant has already joined, the function will return immediately. r rr r rcs*durdSttr|jvS|jkSNT)r#listrrrrr kind_matchVs    z(wait_for_participant..kind_matchrcs<dus |jkr|rs|dSdSdSdSr)r@rrrrr@rErr_on_participant_connected_s z7wait_for_participant.._on_participant_connectedr&r'cr()Nz/room disconnected while waiting for participantr)r.r/rr_on_connection_state_changeddr1z:wait_for_participant.._on_connection_state_changedr2r4Nr5r6r7) r8r-r9r:rr$r;r<r=rr>)rr@rrGrHrrrr@rrErwait_for_participantFs(       rJ>list[rtc.TrackKind.ValueType] | rtc.TrackKind.ValueType | Nonertc.RemoteTrackPublicationc s|s tdtjtjdfdd dfd d }dfdd }|d||d|z-|jD]}|j D] }||| rMnq@q9IdHW| d|| d|S| d|| d|w)zReturns a remote track matching the given identity and kind. If identity is None, the first track matching the kind will be returned. If the track has already been published, the function will return immediately. r krtc.TrackKind.ValueTyper rcs&durdSttr|vS|kSrB)r#rC)rMrDrrrEs  z.wait_for_track_publication..kind_match publicationrL participantr rcs>rdSdus|jkr|jr|dSdSdSr)rr@rr)rOrPrFrr_on_track_publisheds z7wait_for_track_publication.._on_track_publishedr&r'cr()Nz5room disconnected while waiting for track publicationr)r.r/rrrHr1z@wait_for_track_publication.._on_connection_state_changedtrack_publishedr4N)rMrNr r)rOrLrPr r rr7) r8r-r9r:rRemoteTrackPublicationr;r<r=track_publicationsrr>)rr@rrQrHrrOrrIrwait_for_track_publicationws,        rU)rr rr r r )rr r@r rrAr r )rr r@r rrKr rL) __future__rr9livekitrtypesrr?rJrUrrrrs   ?4