o i'@sdZddlmZddlmZmZmZmZddlZddl m Z ddl m Z m Z ddlmZmZGdd d eeZGd d d eeZGd d d eeZGddde ZGddde ZGddde ZGddde ZGdddeZGdddeZdS)z;HeyGen API. API to communicate with HeyGen Streaming API. )Enum)AnyDictLiteralOptionalN)logger) BaseModelField) BaseAvatarApiStandardSessionResponsec@seZdZdZdZdZdZdS) AvatarQualityz2Enum representing different avatar quality levels.lowmediumhighN)__name__ __module__ __qualname____doc__r rrrrb/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/heygen/api_interactive_avatar.pyr s r c@seZdZdZdZdZdS) VideoEncodingz%Enum representing the video encoding.H264VP8N)rrrrrrrrrrrsrc@s$eZdZdZdZdZdZdZdZdS) VoiceEmotionz0Enum representing different voice emotion types.excitedseriousfriendlysoothing broadcasterN) rrrrEXCITEDSERIOUSFRIENDLYSOOTHING BROADCASTERrrrrr%src@sbeZdZUdZdZeeed<dZeeed<dZ ee ed<dZ ee ed<dZ eeed<dS)ElevenLabsSettingsaSettings for ElevenLabs voice configuration. Parameters: stability (Optional[float]): Stability of the voice synthesis. similarity_boost (Optional[float]): Adjustment for similarity in voice performance. model_id (Optional[str]): Identifier for the ElevenLabs model to use. style (Optional[int]): Style metric to apply for the voice. use_speaker_boost (Optional[bool]): Flag to enable speaker boost. N stabilitysimilarity_boostmodel_idstyleuse_speaker_boost)rrrrr%rfloat__annotations__r&r'strr(intr)boolrrrrr$/s  r$c@sbeZdZUdZedddZeeed<dZ ee ed<dZ ee ed<edddZ eeed <dS) VoiceSettingsaVVoice configuration settings. Parameters: voice_id (Optional[str]): ID of the voice to be used. rate (Optional[float]): Speaking rate for the voice. emotion (Optional[VoiceEmotion]): Emotion tone for the voice. elevenlabs_settings (Optional[ElevenLabsSettings]): Details for ElevenLabs configuration. NvoiceId)aliasvoice_idrateemotionelevenlabsSettingselevenlabs_settings)rrrrr r2rr,r+r3r*r4rr6r$rrrrr/As  r/c@seZdZUdZdZeeed<dZee ed<dZ ee ed<dZ ee ed<dZee ed<dZee ed<d Zed ed <dZeeed <dZeeed <dS) NewSessionRequestaRequesting model for creating a new HeyGen session. Parameters: quality (Optional[AvatarQuality]): Desired quality of the avatar. avatar_id (Optional[str]): Unique identifier for the avatar. voice (Optional[VoiceSettings]): Voice configurations for the session. video_encoding (Optional[VideoEncoding]): Desired encoding for the video stream. knowledge_id (Optional[str]): Identifier for the knowledge base (if applicable). knowledge_base (Optional[str]): Details of any external knowledge base. version (Literal["v2"]): API version to use. disable_idle_timeout (Optional[bool]): Flag to disable automatic idle timeout. activity_idle_timeout (Optional[int]): Timeout in seconds for activity-based idle detection. Nquality avatar_idvoicevideo_encoding knowledge_idknowledge_basev2versiondisable_idle_timeoutactivity_idle_timeout)rrrrr8rr r+r9r,r:r/r;rr<r=r?rr@r.rAr-rrrrr7Qs r7c@s:eZdZUdZeed<eed<eed<eed<eed<dS) HeyGenSessionuResponse model for a HeyGen session. Parameters: session_id (str): Unique identifier for the streaming session. access_token (str): Token for accessing the session securely. livekit_agent_token (str): Token for HeyGen’s audio agents(Pipecat). realtime_endpoint (str): Real-time communication endpoint URL. url (str): Direct URL for the session. session_id access_tokenlivekit_agent_tokenrealtime_endpointurlN)rrrrr,r+rrrrrBks   rBcs2eZdZdZdedededdffdd ZZS) HeygenApiErrorz'Custom exception for HeyGen API errors.messagestatus response_textreturnNcst|||_||_dS)zInitialize the HeyGen API error. Args: message: Error message status: HTTP status code response_text: Raw response text from the API N)super__init__rJrK)selfrIrJrK __class__rrrNs  zHeygenApiError.__init__)rrrrr,r-rN __classcell__rrrPrrH}s&rHc @seZdZdZdZdedejddfddZdd ed e ee fd e de fd dZ de defddZdede fddZdede fddZdefddZdS) HeyGenApizHeyGen Streaming API client.zhttps://api.heygen.com/v1api_keysessionrLNcCs||_||_dS)zInitialize the HeyGen API. Args: api_key: HeyGen API key session: Optional aiohttp client session N)rTrU)rOrTrUrrrrNs zHeyGenApi.__init__Tpathparams expect_datac s*|j|}|jdd}td|zf|jj|||d4IdHK}|jsB|IdH}td|t d|j |j ||r]| IdH}| d} | WdIdHWS|IdHWdIdHWS1IdHsuwYWdSt jy} z td t| d} ~ ww) aMake a POST request to the HeyGen API. Args: path: API endpoint path. params: JSON-serializable parameters. expect_data: Whether to expect and extract 'data' field from response (default: True). Returns: Parsed JSON response data. Raises: HeygenApiError: If the API response is not successful or data is missing when expected. aiohttp.ClientError: For network-related errors. zapplication/json)z x-api-keyz Content-TypezHeyGen API request: )jsonheadersNzHeyGen API error: zAPI request failed with status dataz(Network error while calling HeyGen API: )BASE_URLrTrdebugrUpostoktexterrorrHrJrYgetaiohttp ClientErrorr,) rOrVrWrXrGrZresponserK json_datar[errr_requests8   4zHeyGenApi._request request_datac s|j|j|jr |jjnd|jr|jjnd|jr|jjnd|jr$|jjndd|j|j|j |j |j |j d }| d|IdH}td|t|}||jIdHt|j|j|j|j|j|dS)a5Create a new streaming session and start it immediately. https://docs.heygen.com/reference/new-session Args: request_data: Session configuration parameters. Returns: StandardSessionResponse: Standardized session information with HeyGen raw response. N)r2r3r4r6) r8r9r:r<r=r?r;r@rAz/streaming.newzheygen session info)rCrD livekit_urlrEws_url raw_response)r8r9r:r0r3r4r5r<r=r?r;r@rArhprintrBmodel_validate_start_sessionrCr rDrGrErF)rOrirW session_infoheygen_sessionrrr new_sessions6   zHeyGenApi.new_sessionrCcs(|stdd|i}|d|IdHS)a1Start the streaming session. https://docs.heygen.com/reference/start-session Args: session_id: ID of the session to start. Returns: Response data from the start session API call. Raises: ValueError: If session ID is not set. .Session ID is not set. Call new_session first.rCz/streaming.startN ValueErrorrhrOrCrWrrrros zHeyGenApi._start_sessioncs,|stdd|i}|jd|ddIdHS)a=Terminate an active the streaming session. https://docs.heygen.com/reference/close-session Args: session_id: ID of the session to stop. Returns: Response data from the stop session API call. Raises: ValueError: If session ID is not set. rsrCz/streaming.stopF)rXNrtrvrrr close_session s zHeyGenApi.close_sessioncs|diIdH}|dS)zCreate a streaming token. https://docs.heygen.com/reference/streaming-token Returns: str: The generated access token for the streaming session z/streaming.create_tokenNtoken)rh)rO token_inforrr create_token szHeyGenApi.create_token)T)rrrrr\r,rc ClientSessionrNrrr.rhr7r rrrorwrzrrrrrSs$ *.rS)renumrtypingrrrrrclogururpydanticrr pipecat.services.heygen.base_apir r r,r rrr$r/r7rB ExceptionrHrSrrrrs