From 04bf5ad3726c11d0524f2df83ff7ef1473341532 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?The=CC=81o=20Monnom?= Date: Thu, 13 Mar 2025 14:14:45 +0100 Subject: [PATCH 1/2] VideoBufferType str --- livekit-rtc/livekit/rtc/video_frame.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/livekit-rtc/livekit/rtc/video_frame.py b/livekit-rtc/livekit/rtc/video_frame.py index b3fd65a5..9277a34e 100644 --- a/livekit-rtc/livekit/rtc/video_frame.py +++ b/livekit-rtc/livekit/rtc/video_frame.py @@ -217,7 +217,7 @@ def validate_video_frame(value: Any) -> "VideoFrame": return VideoFrame( width=value["width"], height=value["height"], - type=proto_video.VideoBufferType.ValueType(value["type"]), + type=proto_video.VideoBufferType.Value(value["type"]), data=base64.b64decode(value["data"]), ) @@ -230,7 +230,7 @@ def validate_video_frame(value: Any) -> "VideoFrame": { "width": core_schema.model_field(core_schema.int_schema()), "height": core_schema.model_field(core_schema.int_schema()), - "type": core_schema.model_field(core_schema.int_schema()), + "type": core_schema.model_field(core_schema.str_schema()), "data": core_schema.model_field(core_schema.str_schema()), }, ), @@ -242,7 +242,7 @@ def validate_video_frame(value: Any) -> "VideoFrame": lambda instance: { "width": instance.width, "height": instance.height, - "type": instance.type, + "type": proto_video.VideoBufferType.Name(instance.type), "data": base64.b64encode(instance.data).decode("utf-8"), } ), From cc8b7cf5061222e19703b1eab6184594b6df3ff0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?The=CC=81o=20Monnom?= Date: Thu, 13 Mar 2025 14:17:28 +0100 Subject: [PATCH 2/2] better repr --- livekit-rtc/livekit/rtc/video_frame.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/livekit-rtc/livekit/rtc/video_frame.py b/livekit-rtc/livekit/rtc/video_frame.py index 9277a34e..c2762576 100644 --- a/livekit-rtc/livekit/rtc/video_frame.py +++ b/livekit-rtc/livekit/rtc/video_frame.py @@ -199,7 +199,7 @@ def convert( return VideoFrame._from_owned_info(resp.video_convert.buffer) def __repr__(self) -> str: - return f"rtc.VideoFrame(width={self.width}, height={self.height}, type={self.type})" + return f"rtc.VideoFrame(width={self.width}, height={self.height}, type={proto_video.VideoBufferType.Name(self.type)})" @classmethod def __get_pydantic_core_schema__(cls, *_: Any):