o i&1@sdZddlmZddlmZmZmZddlZddlm Z ddl m Z ddl m Z mZGdd d e ZGd d d e ZGd d d eeZGdddeeZGddde ZGddde ZGddde ZGddde ZGddde ZGddde ZGdddeZGddde ZdS) zCLiveAvatar API. API to communicate with LiveAvatar Streaming API. )Enum)AnyDictOptionalN)logger) BaseModel) BaseAvatarApiStandardSessionResponsec@s>eZdZUdZdZeeed<dZeeed<dZ eed<dS) 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__r rrrZ/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/heygen/api_liveavatar.pyr s r c@s*eZdZUdZeed<eed<eed<dS)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$  rc@seZdZdZdZdZdS) VideoEncodingz%Enum representing the video encoding.H264VP8N)rrrrrrrrrrr2src@s eZdZdZdZdZdZdZdS) VideoQualityz2Enum representing different avatar quality levels.lowmediumhigh very_highN)rrrrr!r"r#r$rrrrr 9s r c@s(eZdZUdZeed<ejZeed<dS) VideoSettingsz2Video encoding settings for session configuration.encodingqualityN) rrrrrrr r#r'rrrrr%Bs r%c@sneZdZUdZdZeed<eed<eej dZ e eed<dZ e e ed<d Ze eed <d Ze eed <d S) 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)rrrrr*rrr%rrr,rr-boolr.r r/rrrrrr(Is  r(c@s"eZdZUdZeed<eed<dS)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_tokenNrrrrrr1]s  r1c@*eZdZUdZeed<eed<eed<dS)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)rrrrintrr1rrrrrr5irr5c@sBeZdZUdZeed<eed<eed<eed<eed<eed<dS) 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. r2rrlivekit_agent_tokenmax_session_durationws_urlN)rrrrrrr9rrrrr:ws   r:c@r4)LiveAvatarSessionResponsezResponse model for LiveAvatar session start. Parameters: code (int): Response status code. data (LiveAvatarSessionData): Session data containing connection details. message (str): Response message. r6r7r8N)rrrrr9rr:rrrrrr>rr>cs2eZdZdZdedededdffdd ZZS) LiveAvatarApiErrorz+Custom exception for LiveAvatar API errors.r8status response_textreturnNcst|||_||_dS)zInitialize the LiveAvatar API error. Args: message: Error message status: HTTP status code response_text: Raw response text from the API N)super__init__r@rA)selfr8r@rA __class__rrrDs  zLiveAvatarApiError.__init__)rrrrrr9rD __classcell__rrrFrr?s&r?c @seZdZdZdZdedejddfddZ  dd ed ed e e ee fd e ede f d dZ de defddZdedefddZdedede fddZde defddZdede fddZdS) LiveAvatarApiz LiveAvatar Streaming API client.zhttps://api.liveavatar.com/v1api_keysessionrBNcCs||_||_d|_dS)zInitialize the LiveAvatar API. Args: api_key: LiveAvatar API key session: aiohttp client session N)_api_key_session_session_token)rErJrKrrrrDs zLiveAvatarApi.__init__methodpathparams bearer_tokenc s&|j|}ddi}|rd||d<n|j|d<|dur#d|d<td|d |zL|jj||||d 4IdH0}|js[|IdH}td |t d |j |j || IdHWdIdHWS1IdHsswYWdSt j y} z td t| d} ~ ww)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_URLrLrdebugrMrequestoktexterrorr?r@rVaiohttp ClientErrorr) rErOrPrQrRurlrWresponserAerrr_requests6   4 zLiveAvatarApi._request request_datacs|jdur |jnd|jd}|jdur-|jj|jj|jjd}dd|D}||d<|jdur7|j|d<|jdurL|jj j |jj j d }||d <nd t j j i|d <td ||d d|IdH}tdt|S)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. Nr))r*r+)r r rcSsi|] \}}|dur||qS)Nr).0kvrrr sz6LiveAvatarApi.create_session_token..r.r-)r&r'r,r&z/Creating LiveAvatar session token with params: POSTz/sessions/tokenz LiveAvatar session token created)r*r+r.r r ritemsr-r,r&valuer'rrrrYrcr5model_validate)rErdrQr.r,rarrrcreate_session_tokens.       z"LiveAvatarApi.create_session_tokenr3cs,|jdd|dIdH}tdt|S)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. riz/sessions/start)rRNzLiveAvatar session started)rcrrYr>rl)rEr3rarrr start_sessions  zLiveAvatarApi.start_sessionr2cs2|stdd|i}|jdd||dIdH}|S)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.r2riz/sessions/stop)rQrRN) ValueErrorrc)rEr2r3rQrarrr stop_session-s zLiveAvatarApi.stop_sessioncsX||IdH}|jj|_||jjIdH}t|jj|jj|jj|jj |jj |dS)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)r2 access_tokenrr;r= raw_response) rmr7r3rNrnr r2rrr;r=)rErdtoken_responsesession_responserrr new_sessionFs zLiveAvatarApi.new_sessioncs$|jstd|||jIdHS)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)rNrorp)rEr2rrr close_sessiondszLiveAvatarApi.close_session)NN)rrrrrXrr^ ClientSessionrDrrrrcr(r5rmr>rnrpr rurvrrrrrIs< 4 0 rI)renumrtypingrrrr^logururpydanticr pipecat.services.heygen.base_apirr r rrrr r%r(r1r5r:r> Exceptionr?rIrrrrs&