Reduce type complexity

Author: ladvoc

examples/data_tracks/publisher.py

@@ -33,7 +33,7 @@ async def main(room: rtc.Room):
     await room.connect(URL, TOKEN)
     logger.info("connected to room %s", room.name)
 
-    track = await room.local_participant.publish_data_track("my_sensor_data")
+    track = await room.local_participant.publish_data_track({"name": "my_sensor_data"})
     await push_frames(track)
 
 

livekit-rtc/livekit/rtc/participant.py

@@ -676,13 +676,12 @@ async def send_file(
 
     async def publish_data_track(
         self,
-        options: Union[str, DataTrackOptions],
+        options: DataTrackOptions,
     ) -> LocalDataTrack:
         """Publishes a data track.
 
         Args:
-            options: Either a track name string or a :class:`DataTrackOptions`
-                object. When a string is provided, it is used as the track name.
+            options: Options for publishing the data track.
 
         Returns:
             The published data track. Use :meth:`LocalDataTrack.try_push` to
@@ -691,9 +690,6 @@ async def publish_data_track(
         Raises:
             PublishDataTrackError: If there is an error publishing the data track.
         """
-        if isinstance(options, str):
-            options = DataTrackOptions(name=options)
-
         proto_opts = proto_data_track.DataTrackOptions(name=options["name"])
 
         req = proto_ffi.FfiRequest()

tests/rtc/test_e2e.py

@@ -477,7 +477,7 @@ def on_data_track_unpublished(sid: str):
         await subscriber_room.connect(url, subscriber_token)
         await publisher_room.connect(url, publisher_token)
 
-        local_track = await publisher_room.local_participant.publish_data_track(TRACK_NAME)
+        local_track = await publisher_room.local_participant.publish_data_track({"name": TRACK_NAME})
         assert local_track.info.sid is not None
         assert local_track.info.name == TRACK_NAME
         assert local_track.is_published()