Keep format

Author: hexbabe

src/viam/components/camera/service.py

@@ -6,6 +6,7 @@
 from viam.proto.common import DoCommandRequest, DoCommandResponse, GetGeometriesRequest, GetGeometriesResponse
 from viam.proto.component.camera import (
     CameraServiceBase,
+    Format,
     GetImageRequest,
     GetImageResponse,
     GetImagesRequest,
@@ -19,6 +20,7 @@
 )
 from viam.resource.rpc_service_base import ResourceRPCServiceBase
 from viam.utils import dict_to_struct, struct_to_dict
+from viam.media.video import CameraMimeType
 
 from . import Camera
 
@@ -56,8 +58,14 @@ async def GetImages(self, stream: Stream[GetImagesRequest, GetImagesResponse]) -
         )
         img_bytes_lst = []
         for img in images:
+            mime_type = CameraMimeType.from_string(img.mime_type)
+            if isinstance(mime_type, CameraMimeType):
+                fmt = mime_type.to_proto()
+            else:
+                fmt = Format.FORMAT_UNSPECIFIED
+
             img_bytes = img.data
-            img_bytes_lst.append(Image(source_name=name, mime_type=img.mime_type, image=img_bytes))
+            img_bytes_lst.append(Image(source_name=name, mime_type=img.mime_type, format=fmt, image=img_bytes))
         response = GetImagesResponse(images=img_bytes_lst, response_metadata=metadata)
         await stream.send_message(response)