Update audio_mixer.py

Author: theomonnom

livekit-rtc/livekit/rtc/audio_mixer.py

@@ -2,11 +2,11 @@
 import numpy as np
 import contextlib
 from dataclasses import dataclass
-from typing import AsyncIterator, Optional
+from typing import AsyncIterable, Optional
 from .audio_frame import AudioFrame
 from .log import logger
 
-_Stream = AsyncIterator[AudioFrame]
+_Stream = AsyncIterable[AudioFrame]
 
 
 @dataclass
@@ -21,9 +21,9 @@ class _Contribution:
 class AudioMixer:
     def __init__(
         self,
-        *,
         sample_rate: int,
         num_channels: int,
+        *,
         blocksize: int = 0,
         stream_timeout_ms: int = 100,
         capacity: int = 100,
@@ -62,7 +62,7 @@ def __init__(
         self._ending: bool = False
         self._mixer_task: asyncio.Task = asyncio.create_task(self._mixer())
 
-    def add_stream(self, stream: AsyncIterator[AudioFrame]) -> None:
+    def add_stream(self, stream: AsyncIterable[AudioFrame]) -> None:
         """
         Add an audio stream to the mixer.
 
@@ -79,7 +79,7 @@ def add_stream(self, stream: AsyncIterator[AudioFrame]) -> None:
         if stream not in self._buffers:
             self._buffers[stream] = np.empty((0, self._num_channels), dtype=np.int16)
 
-    def remove_stream(self, stream: AsyncIterator[AudioFrame]) -> None:
+    def remove_stream(self, stream: AsyncIterable[AudioFrame]) -> None:
         """
         Remove an audio stream from the mixer.
 
@@ -169,7 +169,7 @@ async def _mixer(self) -> None:
         await self._queue.put(None)
 
     async def _get_contribution(
-        self, stream: AsyncIterator[AudioFrame], buf: np.ndarray
+        self, stream: AsyncIterable[AudioFrame], buf: np.ndarray
     ) -> _Contribution:
         had_data = buf.shape[0] > 0
         exhausted = False