o êi\ @s4dZddlmZmZGdddZGdddZdS)z0Audio stream wrappers with collection utilities.) AsyncIteratorIteratorc@BeZdZdZdeefddZdeefddZdefdd Zd S) AudioStreama4Wrapper for sync audio byte streams with collection utilities. This class wraps an iterator of audio bytes and provides a convenient `.collect()` method to gather all chunks into a single bytes object. Examples: ```python from fishaudio import FishAudio client = FishAudio(api_key="...") # Collect all audio at once audio = client.tts.stream(text="Hello!").collect() # Or stream chunks manually for chunk in client.tts.stream(text="Hello!"): process_chunk(chunk) ``` iteratorcC ||_dS)z{Initialize the audio iterator wrapper. Args: iterator: The underlying iterator of audio bytes N_iter)selfrr L/home/ubuntu/.local/lib/python3.10/site-packages/fishaudio/core/iterators.py__init__ zAudioStream.__init__returncC|jS)z)Allow direct iteration over audio chunks.rr r r r __iter__#zAudioStream.__iter__cCs$g}|jD]}||qd|S)aCollect all audio chunks into a single bytes object. This consumes the iterator and returns all audio data as bytes. After calling this method, the iterator cannot be used again. Returns: Complete audio data as bytes Examples: ```python audio = client.tts.stream(text="Hello!").collect() with open("output.mp3", "wb") as f: f.write(audio) ``` r appendjoinr chunkschunkr r r collect's   zAudioStream.collectN) __name__ __module__ __qualname____doc__rbytesr rrr r r r rs rc@r) AsyncAudioStreama{Wrapper for async audio byte streams with collection utilities. This class wraps an async iterator of audio bytes and provides a convenient `.collect()` method to gather all chunks into a single bytes object. Examples: ```python from fishaudio import AsyncFishAudio client = AsyncFishAudio(api_key="...") # Collect all audio at once stream = await client.tts.stream(text="Hello!") audio = await stream.collect() # Or stream chunks manually async for chunk in await client.tts.stream(text="Hello!"): await process_chunk(chunk) ``` async_iteratorcCr)zInitialize the async audio iterator wrapper. Args: async_iterator: The underlying async iterator of audio bytes Nr)r r"r r r r SrzAsyncAudioStream.__init__rcCr)z/Allow direct async iteration over audio chunks.rrr r r __aiter__[rzAsyncAudioStream.__aiter__cs0g}|j2z 3dHW}||q6d|S)aCollect all audio chunks into a single bytes object. This consumes the async iterator and returns all audio data as bytes. After calling this method, the iterator cannot be used again. Returns: Complete audio data as bytes Examples: ```python stream = await client.tts.stream(text="Hello!") audio = await stream.collect() with open("output.mp3", "wb") as f: f.write(audio) ``` Nrrrr r r r_s   zAsyncAudioStream.collectN) rrrrrr r r#rr r r r r!=s r!N)rtypingrrrr!r r r r s7