o êi @sdZddlmZmZmZmZmZddlZddlm Z ddl m Z dee efde fd d Zdee efdeefd d Zdeefd dZdeefddZdS)z&Real-time WebSocket streaming helpers.)Any AsyncIteratorDictIteratorOptionalN)WebSocketDisconnect)WebSocketErrordatareturncCs|ddko |ddkS)z Check if WebSocket event signals stream should stop. Args: data: Unpacked WebSocket message data Returns: True if stream should stop, False otherwise eventfinishreasonstop)getr rP/home/ubuntu/.local/lib/python3.10/site-packages/fishaudio/resources/realtime.py _should_stop s rcCs@|ddkr |dS|ddkr|ddkrtddS)z Process a WebSocket audio event. Args: data: Unpacked WebSocket message data Returns: Audio bytes if audio event, None for unknown events Raises: WebSocketError: If finish event has error reason r audior rerrorz!WebSocket stream ended with errorN)rr rrrr_process_audio_events  rccsX z|}t|}t|rWdSt|}|dur|VWn ty*tdwq)ae Process WebSocket audio messages (sync). Receives messages from WebSocket, yields audio chunks, handles errors. Unknown events are ignored and iteration continues. Args: ws: WebSocket connection from httpx_ws.connect_ws Yields: Audio bytes Raises: WebSocketError: On disconnect or error finish event TN#WebSocket disconnected unexpectedly receive_bytes ormsgpackunpackbrrrr wsmessager rrrriter_websocket_audio,s  r cCs^ z |IdH}t|}t|rWdSt|}|dur!|VWn ty-tdwq)ag Process WebSocket audio messages (async). Receives messages from WebSocket, yields audio chunks, handles errors. Unknown events are ignored and iteration continues. Args: ws: WebSocket connection from httpx_ws.aconnect_ws Yields: Audio bytes Raises: WebSocketError: On disconnect or error finish event TNrrrrrraiter_websocket_audioLs  r!)__doc__typingrrrrrrhttpx_wsr exceptionsr strboolrbytesrr r!rrrrs