qi&1ddZddlmZddlmZmZmZddlZddlm Z ddl m Z ddl m Z mZGdd e ZGd d e ZGd d eeZGddeeZGdde ZGdde ZGdde ZGdde ZGdde ZGdde ZGddeZGdde Zy) zCLiveAvatar API. API to communicate with LiveAvatar Streaming API. )Enum)AnyDictOptionalN)logger) BaseModel) BaseAvatarApiStandardSessionResponsecHeZdZUdZdZeeed<dZeeed<dZ eed<y) AvatarPersonaaAvatar persona settings for LiveAvatar. Parameters: voice_id (Optional[str]): ID of the voice to be used. context_id (Optional[str]): Context ID for the avatar. language (str): Language code for the avatar (default: "en"). Nvoice_id context_idenlanguage) __name__ __module__ __qualname____doc__r rstr__annotations__rrX/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/heygen/api_liveavatar.pyr r s/#Hhsm" $J $Hcrr c0eZdZUdZeed<eed<eed<y)CustomSDKLiveKitConfigzCustom LiveKit configuration. Parameters: livekit_url (str): LiveKit server URL. livekit_room (str): LiveKit room name. livekit_client_token (str): LiveKit client access token. livekit_url livekit_roomlivekit_client_tokenNrrrrrrrrrrr$srrceZdZdZdZdZy) VideoEncodingz%Enum representing the video encoding.H264VP8N)rrrrr"r#rrrr!r!2s/ D Crr!c eZdZdZdZdZdZdZy) VideoQualityz2Enum representing different avatar quality levels.lowmediumhigh very_highN)rrrrr&r'r(r)rrrr%r%9s< C F DIrr%c>eZdZUdZeed<ejZeed<y) VideoSettingsz2Video encoding settings for session configuration.encodingqualityN) rrrrr!rr%r(r-rrrr+r+Bs<(--G\-rr+ceZdZUdZdZeed<eed<eejZ e eed<dZ e e ed<d Ze eed <d Ze eed <y ) LiveAvatarNewSessionRequestaRequest model for creating a LiveAvatar session token. Parameters: mode (str): Session mode (default: "LITE"). avatar_id (str): Unique identifier for the avatar. video_settings (VideoSettings): Video encoding settings. is_sandbox (bool): Enable sandbox mode (default: False). avatar_persona (AvatarPersona): Avatar persona configuration. livekit_config (CustomSDKLiveKitConfig): Custom LiveKit configuration. LITEmode avatar_id)r,video_settingsF is_sandboxNavatar_personalivekit_config)rrrrr1rrr+r!r#r3rr4boolr5r r6rrrrr/r/Is_ D#N.;]EVEV.WNH]+W!&J&.2NH]+27;NH34;rr/c&eZdZUdZeed<eed<y)SessionTokenDatazData model for session token response. Parameters: session_id (str): Unique identifier for the session. session_token (str): Session token for authentication. session_id session_tokenNrrrrr9r9]sOrr9c0eZdZUdZeed<eed<eed<y)SessionTokenResponsezResponse model for LiveAvatar session token. Parameters: code (int): Response status code. data (SessionTokenData): Session token data containing session_id and session_token. message (str): Response message. codedatamessageN)rrrrintrr9rrrrr=r=is I  Lrr=cNeZdZUdZeed<eed<eed<eed<eed<eed<y) LiveAvatarSessionDataaData model for LiveAvatar session response. Parameters: session_id (str): Unique identifier for the streaming session. livekit_url (str): LiveKit server URL for the session. livekit_client_token (str): Access token for LiveKit user. livekit_agent_token (str): Access token for LiveKit Agent (Pipecat). max_session_duration (int): Maximum session duration in seconds. ws_url (str): WebSocket URL for the session. r:rrlivekit_agent_tokenmax_session_durationws_urlN)rrrrrrrArrrrCrCws+ O KrrCc0eZdZUdZeed<eed<eed<y)LiveAvatarSessionResponsezResponse model for LiveAvatar session start. Parameters: code (int): Response status code. data (LiveAvatarSessionData): Session data containing connection details. message (str): Response message. r>r?r@N)rrrrrArrCrrrrrHrHs I  LrrHc4eZdZdZdedededdffd ZxZS)LiveAvatarApiErrorz+Custom exception for LiveAvatar API errors.r@status response_textreturnNc@t||||_||_y)zInitialize the LiveAvatar API error. Args: message: Error message status: HTTP status code response_text: Raw response text from the API N)super__init__rKrL)selfr@rKrL __class__s rrPzLiveAvatarApiError.__init__s! ! *r)rrrrrrArP __classcell__)rRs@rrJrJs,5 + +S + + + +rrJc eZdZdZdZdedejddfdZ dded ed e e ee fd e ede f d Z d e defdZdedefdZdedede fdZd e defdZdede fdZy) LiveAvatarApiz LiveAvatar Streaming API client.zhttps://api.liveavatar.com/v1api_keysessionrMNc.||_||_d|_y)zInitialize the LiveAvatar API. Args: api_key: LiveAvatar API key session: aiohttp client session N)_api_key_session_session_token)rQrVrWs rrPzLiveAvatarApi.__init__s   "rmethodpathparams bearer_tokencK|j|}ddi}|r d||d<n|j|d<|d|d<tjd|d | |jj |||| 4d{}|j sT|jd{}tjd |td |j|j||jd{cdddd{S77k77 #1d{7swYyxYw#tj$r'} tjd t| d} ~ wwxYww)aMake a request to the LiveAvatar API. Args: method: HTTP method (GET, POST, etc.). path: API endpoint path. params: JSON-serializable parameters. bearer_token: Optional bearer token for authorization. Returns: Parsed JSON response data. Raises: LiveAvatarApiError: If the API response is not successful. aiohttp.ClientError: For network-related errors. acceptzapplication/jsonzBearer authorizationz X-API-KEYNz content-typezLiveAvatar API request:  )jsonheaderszLiveAvatar API error: zAPI request failed with status z,Network error while calling LiveAvatar API: )BASE_URLrYrdebugrZrequestoktexterrorrJrKrdaiohttp ClientErrorr) rQr\r]r^r_urlreresponserLes r_requestzLiveAvatarApi._requests`,v& (  )0'?GO $#'==GK  &8GN # /xq>? }},,VSvw,W - -[c{{*2--/$9MLL#9-!IJ,9(//9JK % &]]_, - - -$9- - - - -""  LLGAxP Q  sAE#D"7D8D"; D DAD /D 0D 3 D"?D D"ED"D  D  D" DD DD"ED""E5"EEE request_datac.K|j |jnd|jd}|jm|jj|jj|jj d}|j Dcic] \}}| || }}}||d<|j|j|d<|jG|jjj|jjjd}||d<nd tjji|d<tjd ||j!d d |d{}tjd t"j%|Scc}}w74w)aCreate a session token for LiveAvatar. https://docs.liveavatar.com/reference/create_session_token_v1_sessions_token_post Args: request_data: Session token configuration parameters. Returns: Session token information. Nr0)r1r2)r rrr5r4)r,r-r3r,z/Creating LiveAvatar session token with params: POSTz/sessions/tokenz LiveAvatar session token created)r1r2r5r rritemsr4r3r,valuer-r!r#rrgrqr=model_validate)rQrrr^r5kvr3ros rcreate_session_tokenz"LiveAvatarApi.create_session_tokens}*6):):)FL%%F%//"   & & 2(77@@*99DD(77@@N 0>/C/C/EWtq!adWNW'5F# $  " " .#/#:#:F<  & & 2(77@@FF'66>>DDN(6F# $)3M4E4E4K4K'LF# $ FvhOPv/@&II 79#228<<)X"Js%B F F F CFF5Fr;cK|jdd|d{}tjdtj |S7.w)a-Start a new LiveAvatar session. https://docs.liveavatar.com/reference/start_session_v1_sessions_start_post Args: session_token: Session token obtained from create_session_token. Returns: Session information including room URL and session ID. rtz/sessions/start)r_NzLiveAvatar session started)rqrrgrHrw)rQr;ros r start_sessionzLiveAvatarApi.start_sessionsDv/@}]] 13(77AA^sA A /A r:cnK|s tdd|i}|jdd||d{}|S7w)aStop an active LiveAvatar session. https://docs.liveavatar.com/reference/stop_session_v1_sessions_stop_post Args: session_id: ID of the session to stop. session_token: Session token for authentication. Returns: Response data from the stop session API call. Raises: ValueError: If session ID is not set. zSession ID is not set.r:rtz/sessions/stop)r^r_N) ValueErrorrq)rQr:r;r^ros r stop_sessionzLiveAvatarApi.stop_session-sO56 6 + $V-'   s *535cK|j|d{}|jj|_|j |jjd{}t |jj |jj|jj|jj|jj|S77{w)aaCreate and start a new LiveAvatar session (convenience method). This combines create_session_token and start_session into a single call. Args: request_data: Session token configuration parameters. Returns: StandardSessionResponse: Standardized session information with LiveAvatar raw response. N)r: access_tokenrrDrF raw_response) rzr?r;r[r|r r:rrrDrF)rQrrtoken_responsesession_responses r new_sessionzLiveAvatarApi.new_sessionFs $88FF,11??"&!3!3N4G4G4U4U!VV'',,77)..CC(--99 0 5 5 I I#((//)   GWs#CCACC A:CCcK|js td|j||jd{S7w)aClose an active LiveAvatar session (convenience method). This is a convenience method that closes a session using the stored session token from the most recent `new_session()` call. It automatically uses the internally stored session token, eliminating the need to manually track tokens. Args: session_id: ID of the session to close. Returns: Response data from the stop session API call. Raises: ValueError: If no session token is available (i.e., `new_session()` hasn't been called yet or the stored token is None). Note: This method requires that `new_session()` has been called previously to establish a stored session token. For more control over session tokens, use `stop_session()` directly with an explicit token parameter. z1Session token is not set. Call new_session first.N)r[r~r)rQr:s r close_sessionzLiveAvatarApi.close_sessionds<,""PQ Q&&z43F3FGGGGs7A>A)NN)rrrrrfrrl ClientSessionrPrrrrqr/r=rzrHr|rr rrrrrrUrUs*.H # #g.C.C # #,0&* 222c3h( 2 sm 2 2h.=7.= .=`BB9RB S2 7  <HcHcHrrU)renumrtypingrrrrllogururpydanticr pipecat.services.heygen.base_apir r r rrr!r%r+r/r9r=rCrH ExceptionrJrUrrrrs &&S I  Y C3.I.<)<( y  9 I(   + + THMTHr