fix types

Author: typester

livekit-rtc/livekit/rtc/audio_filter.py

@@ -1,12 +1,13 @@
 import json
+from typing import Optional
 
 from ._ffi_client import FfiClient, FfiHandle
 from ._proto import ffi_pb2 as proto_ffi
 
 
 class AudioFilter:
     def __init__(
-        self, path: str, url: str, token: str, dependencies: list[str] = None
+        self, path: str, url: str, token: str, dependencies: Optional[list[str]] = None
     ) -> None:
         self._path = path
 

livekit-rtc/livekit/rtc/audio_stream.py

@@ -95,14 +95,17 @@ def __init__(
         self._audio_filter_handle = None
         self._audio_filter_options = None
         if enable_filter is not None:
-            room = track.room()
-            if room is None:
-                raise RuntimeError("Unexpected track")
-            handle = room._filter_handle(enable_filter)
-            if handle is None:
-                raise RuntimeError("audio filter is not enabled for the room")
-            self._audio_filter_handle = handle
-            self._audio_filter_options = enable_filter.filter_options(filter_options)
+            if isinstance(track, RemoteTrack):
+                room = track.room()
+                if room is None:
+                    raise Exception("Unexpected track")
+                handle = room._filter_handle(enable_filter)
+                if handle is None:
+                    raise Exception("audio filter is not enabled for the room")
+                self._audio_filter_handle = handle
+                self._audio_filter_options = enable_filter.filter_options(filter_options)
+            else:
+                raise TypeError("track is not a RemoteTrack")
         self._task = self._loop.create_task(self._run())
         self._task.add_done_callback(task_done_logger)
 

livekit-rtc/livekit/rtc/room.py

@@ -18,6 +18,7 @@
 import logging
 from dataclasses import dataclass, field
 from typing import Callable, Dict, Literal, Optional, cast, Mapping, List, Any
+from types import ModuleType
 
 from .event_emitter import EventEmitter
 from ._ffi_client import FfiClient, FfiHandle
@@ -151,7 +152,7 @@ def __init__(
         self._first_sid_future = asyncio.Future[str]()
         self._local_participant: LocalParticipant | None = None
         self._filters = filters if filters is not None else []
-        self._filter_instances = {}
+        self._filter_instances: Dict[ModuleType, AudioFilter] = {}
 
         self._text_stream_readers: Dict[str, TextStreamReader] = {}
         self._byte_stream_readers: Dict[str, ByteStreamReader] = {}