Update attributes set in trailer

Author: lukasIO

livekit-rtc/livekit/rtc/data_stream.py

@@ -70,7 +70,10 @@ async def _on_chunk_update(self, chunk: proto_DataStream.Chunk):
         await self._queue.put(chunk)
 
     async def _on_stream_close(self, trailer: proto_DataStream.Trailer):
-        self.info.attributes.update(trailer.attributes)
+        if self.info.attributes is None:
+            self.info.attributes = dict(trailer.attributes)
+        else:
+            self.info.attributes.update(trailer.attributes)
         await self._queue.put(None)
 
     def __aiter__(self) -> AsyncIterator[str]:
@@ -119,6 +122,10 @@ async def _on_chunk_update(self, chunk: proto_DataStream.Chunk):
         await self._queue.put(chunk)
 
     async def _on_stream_close(self, trailer: proto_DataStream.Trailer):
+        if self.info.attributes is None:
+            self.info.attributes = dict(trailer.attributes)
+        else:
+            self.info.attributes.update(trailer.attributes)
         await self._queue.put(None)
 
     def __aiter__(self) -> AsyncIterator[bytes]: