add explicit return types

livekit · Jan 16, 2025 · 15f100c · 15f100c
1 parent 318b8d4
commit 15f100c
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/livekit-rtc/livekit/rtc/participant.py b/livekit-rtc/livekit/rtc/participant.py
@@ -45,7 +45,12 @@
 from .log import logger
 
 from .rpc import RpcInvocationData
-from .data_stream import TextStreamWriter, FileStreamWriter, STREAM_CHUNK_SIZE
+from .data_stream import (
+    TextStreamWriter,
+    FileStreamWriter,
+    FileStreamInfo,
+    STREAM_CHUNK_SIZE,
+)
 
 
 class PublishTrackError(Exception):
@@ -550,7 +555,7 @@ async def stream_text(
         topic: str = "",
         extensions: Dict[str, str] = {},
         reply_to_id: str | None = None,
-    ):
+    ) -> TextStreamWriter:
         writer = TextStreamWriter(
             self,
             topic=topic,
@@ -571,7 +576,7 @@ async def stream_file(
         extensions: Dict[str, str] = {},
         stream_id: str | None = None,
         destination_identities: List[str] = [],
-    ):
+    ) -> FileStreamWriter:
         writer = FileStreamWriter(
             self,
             file_name=file_name,
@@ -592,7 +597,7 @@ async def send_file(
         destination_identities: List[str] = [],
         extensions: Dict[str, str] = {},
         stream_id: str | None = None,
-    ):
+    ) -> FileStreamInfo:
         file_size = os.path.getsize(file_path)
         file_name = os.path.basename(file_path)
         mime_type, _ = mimetypes.guess_type(file_path)
@@ -615,6 +620,8 @@ async def send_file(
                 await writer.write(bytes)
         await writer.close()
 
+        return writer.info
+
     async def publish_track(
         self, track: LocalTrack, options: TrackPublishOptions = TrackPublishOptions()
     ) -> LocalTrackPublication: