o iq)@sdZddlZddlZddlmZddlZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZddlmZmZmZmZmZmZmZmZdd lmZGd d d eZdS) z.Telnyx WebSocket frame serializer for Pipecat.N)Optional)logger) BaseModel) KeypadEntry) alaw_to_pcmcreate_stream_resampler pcm_to_alaw pcm_to_ulaw ulaw_to_pcm) AudioRawFrame CancelFrameEndFrameFrameInputAudioRawFrameInputDTMFFrameInterruptionFrame StartFrame)FrameSerializerc@seZdZdZGdddeZ   ddedededeed eed eef d d Zd e fddZ d e dee BdBfddZ ddZdee Bde dBfddZdS)TelnyxFrameSerializeraSerializer for Telnyx WebSocket protocol. This serializer handles converting between Pipecat frames and Telnyx's WebSocket media streams protocol. It supports audio conversion, DTMF events, and automatic call termination. When auto_hang_up is enabled (default), the serializer will automatically terminate the Telnyx call when an EndFrame or CancelFrame is processed, but requires Telnyx credentials to be provided. c@sReZdZUdZdZeed<dZeeed<dZ e ed<dZ e ed<d Z e ed <dS) z!TelnyxFrameSerializer.InputParamsaConfiguration parameters for TelnyxFrameSerializer. Parameters: telnyx_sample_rate: Sample rate used by Telnyx, defaults to 8000 Hz. sample_rate: Optional override for pipeline input sample rate. inbound_encoding: Audio encoding for data sent to Telnyx (e.g., "PCMU"). outbound_encoding: Audio encoding for data received from Telnyx (e.g., "PCMU"). auto_hang_up: Whether to automatically terminate call on EndFrame. i@telnyx_sample_rateN sample_ratePCMUinbound_encodingoutbound_encodingT auto_hang_up)__name__ __module__ __qualname____doc__rint__annotations__rrrstrrrboolr#r#N/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/serializers/telnyx.py InputParams2s    r%N stream_idrrcall_control_idapi_keyparamscCsZ||_||_||_|pt|_||j_||j_|jj|_ d|_ t |_ t |_ d|_dS)aInitialize the TelnyxFrameSerializer. Args: stream_id: The Stream ID for Telnyx. outbound_encoding: The encoding type for outbound audio (e.g., "PCMU"). inbound_encoding: The encoding type for inbound audio (e.g., "PCMU"). call_control_id: The Call Control ID for the Telnyx call (optional, but required for auto hang-up). api_key: Your Telnyx API key (required for auto hang-up). params: Configuration parameters. rFN) _stream_id_call_control_id_api_keyrr%_paramsrrr_telnyx_sample_rate _sample_rater_input_resampler_output_resampler_hangup_attempted)selfr&rrr'r(r)r#r#r$__init__Cs  zTelnyxFrameSerializer.__init__framecs|jjp|j|_dS)zSets up the serializer with pipeline configuration. Args: frame: The StartFrame containing pipeline configuration. N)r-raudio_in_sample_rater/)r3r5r#r#r$setupdszTelnyxFrameSerializer.setupreturncs|jjr|jst|ttfrd|_|IdHdSt|tr)ddi}t |St|t r|j }|jj dkrEt ||j|j|jIdH}n|jj dkrYt||j|j|jIdH}n td|jj |duslt|dkrndSt|d }d d |id }t |SdS) aSerializes a Pipecat frame to Telnyx WebSocket format. Handles conversion of various frame types to Telnyx WebSocket messages. For EndFrames and CancelFrames, initiates call termination if auto_hang_up is enabled. Args: frame: The Pipecat frame to serialize. Returns: Serialized data as string or bytes, or None if the frame isn't handled. Raises: ValueError: If an unsupported encoding is specified. TNeventclearrPCMAUnsupported encoding: rzutf-8mediapayload)r9r=)r-rr2 isinstancer r _hang_up_callrjsondumpsr audiorr rr.r1r ValueErrorlenbase64 b64encodedecode)r3r5answerdataserialized_datar>r#r#r$ serializels@        zTelnyxFrameSerializer.serializec sz|j}|j}|r |stdWdSd|d}dd|d}t4IdH}|j||d4IdH}|jd krEtd |nq|jd krz5| IdH}t d d | dgDr~t d|dWWdIdHWdIdHWdSWn t yYnw|IdH}td|d|jd|n|IdH}td|d|jd|WdIdHn 1IdHswYWdIdHWdS1IdHswYWdSt y} ztd| WYd} ~ dSd} ~ ww)z0Hang up the Telnyx call using Telnyx's REST API.zHCannot hang up Telnyx call: call_control_id and api_key must be providedNz https://api.telnyx.com/v2/calls/z/actions/hangupzapplication/jsonzBearer )z Content-Type Authorization)headersz$Successfully terminated Telnyx call icss|] }|ddkVqdS)code90018N)get).0errorr#r#r$ s  z6TelnyxFrameSerializer._hang_up_call..errorsz Telnyx call z was already terminatedz Failed to terminate Telnyx call z : Status z , Response: zFailed to hang up Telnyx call: )r+r,rwarningaiohttp ClientSessionpoststatusinforAanyrRdebug ExceptiontextrT) r3r'r(endpointrNsessionresponse error_data error_texter#r#r$r@sl      *2#z#TelnyxFrameSerializer._hang_up_callrJc st|}|ddkr^|dd}t|}|jjdkr+t||j|j|j IdH}n|jjdkr?t ||j|j|j IdH}n t d|jj|dusRt |dkrTdSt |d |jd }|S|dd kr|d id }ztt|WSt y}zWYd}~dSd}~wwdS) aDeserializes Telnyx WebSocket data to Pipecat frames. Handles conversion of Telnyx media events to appropriate Pipecat frames, including audio data and DTMF keypresses. Args: data: The raw WebSocket data from Telnyx. Returns: A Pipecat frame corresponding to the Telnyx event, or None if unhandled. Raises: ValueError: If an unsupported encoding is specified. r9r=r>rNr;r<r)rC num_channelsrdtmfdigit)rAloadsrF b64decoder-rr r.r/r0rrDrErrRrr) r3rJmessagepayload_base64r>deserialized_data audio_framerjrfr#r#r$ deserializesF        z!TelnyxFrameSerializer.deserialize)NNN)rrrrrr%r!rr4rr7rbytesrLr@rqr#r#r#r$r&s.  !89r)rrFrAtypingrrXlogururpydanticrpipecat.audio.dtmf.typesrpipecat.audio.utilsrrrr r pipecat.frames.framesr r r rrrrr#pipecat.serializers.base_serializerrrr#r#r#r$s    (