o h @sddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z dZ e ddfdddZeGdddZGdddZdS)) annotationsN) dataclass) UserError)npnpti]buffer/npt.NDArray[np.int16 | np.float32 | np.float64] frame_rateint sample_widthchannelsreturntuple[str, io.BytesIO, str]cCs|jtjkrt|dd}|dtj}n |jtjkr tdt}t |d#}| || || ||||dWdn1sPwYd|dfS) N?0Buffer must be a numpy array of int16 or float32wrz audio.wavz audio/wav)dtyperfloat32clipastypeint16rioBytesIOwaveopen setnchannels setsampwidth setframerate writeframestobytesseek)r r r r audio_filewav_filer'S/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/voice/input.py_buffer_to_audio_files       r)c@sZeZdZUdZded< eZded< dZded< dZded < dd d Z dddZ dS) AudioInputz7Static audio to be used as input for the VoicePipeline.z"npt.NDArray[np.int16 | np.float32]r r r rr rrrrcCst|j|j|j|jS)z2Returns a tuple of (filename, bytes, content_type))r)r r r rselfr'r'r( to_audio_file:szAudioInput.to_audio_filestrcCsb|jjtjkrt|jdd|_|jdtj|_n |jjtjkr&tdt |j  dS)z2Returns the audio data as a base64 encoded string.rrrrzutf-8) r rrrrrrrbase64 b64encoder#decoder+r'r'r( to_base64>s zAudioInput.to_base64N)rr)rr.) __name__ __module__ __qualname____doc____annotations__DEFAULT_SAMPLE_RATEr r rr-r2r'r'r'r(r*(s     r*c@s"eZdZdZddZd ddZdS) StreamedAudioInputzAudio input represented as a stream of audio data. You can pass this to the `VoicePipeline` and then push audio data into the queue using the `add_audio` method. cCst|_dS)N)asyncioQueuequeuer+r'r'r(__init__OszStreamedAudioInput.__init__audio)npt.NDArray[np.int16 | np.float32] | Nonecs|j|IdHdS)zAdds more audio data to the stream. Args: audio: The audio data to add. Must be a numpy array of int16 or float32 or None. If None passed, it indicates the end of the stream. N)r<put)r,r>r'r'r( add_audioRszStreamedAudioInput.add_audioN)r>r?)r3r4r5r6r=rAr'r'r'r(r9Jsr9) r r r r r r rr rr) __future__rr:r/rr dataclassesr exceptionsrimportsrrr8r)r*r9r'r'r'r(s    !