VideoBufferType str

Author: theomonnom

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"),
                 }
             ),