gimlet-api 0.0.6__py3-none-any.whl → 0.0.7__py3-none-any.whl

{gimlet_api-0.0.6.dist-info → gimlet_api-0.0.7.dist-info}/METADATA

@@ -10,9 +10,10 @@ Requires-Python: >=3
 Requires-Dist: protobuf
 Requires-Dist: grpcio
 Requires-Dist: torch>=2.3.0
-Requires-Dist: torch_mlir_gml==0.0.2
+Requires-Dist: torch-mlir-gml
 Requires-Dist: numpy<2.0.0
 Requires-Dist: transformers>=4.43.3
-Version: 0.0.6
+Requires-Dist: safetensors-mlir
+Version: 0.0.7
 
 UNKNOWN

{gimlet_api-0.0.6.dist-info → gimlet_api-0.0.7.dist-info}/RECORD

@@ -1,12 +1,13 @@
 gml/__init__.py,sha256=H3WQZ_RaN7VNeb__qeHEbKLEwkaG7gpL5FQ8s1IotUA,773
 gml/_utils.py,sha256=mSCWHhCdzcUvHqmJIB2FS215K1LMgJCWcZ6e6FWK3hQ,1184
+gml/asset_manager.py,sha256=VnbqUZHPOgPrAh6ri9C0EuNhS8tAHIrbUyJPAJuD9po,2053
 gml/client.py,sha256=5QDKljltBeBTCd2hH38--fTSP0bVVcAvSnWsA9YEFQc,13819
-gml/compile.py,sha256=hR8u3LaMiIW8d12FHrvtmtgzUNQq48DYxe8bW-wJ_VY,6054
-gml/device.py,sha256=VUZc6m8QalJ7G9KBKjCY4cIcv2VBd6zAT3ysnh_m1Z0,2585
-gml/hf.py,sha256=GRvEEl9zSIv0iWN91Z6ykFYZ2VdNAVABjZrrzYWUFw4,17792
-gml/model.py,sha256=nXUV6-L4TIkQHCWUpWyG7QJ6YKTZb7eauW9F4pzVTII,6566
+gml/compile.py,sha256=K4WdC01WkyLlbcoSJzdF2LaVmOgxIkIdc3YNjRFRw9s,10849
+gml/device.py,sha256=sMILurG02aDjL8wrdBW3ftC44WoAPUeZ4Y0yQ0DtaBk,2665
+gml/hf.py,sha256=hi0Af0Q3FM7VvfLB1PkrNai1j7siH6Ouwc_sXf8QE8c,17900
+gml/model.py,sha256=cHFjIEplWCDeSCSl_IPHYzNyv-KOenx-OAsEd-5TpTs,7260
 gml/model_utils.py,sha256=vZvE5cHZIDkUkeZ4Pk4hhV-zOYMiREluv4b8kdqQ3Ig,1375
-gml/pipelines.py,sha256=Bha8J3b5uW8COIejiH12NNF0Tc0XDBt2B3Dez5Jxt4s,5314
+gml/pipelines.py,sha256=6tujvMpAACwmEmUfZGFuRn0N8zKvjSVIsIX7FKcPvEU,4301
 gml/preprocessing.py,sha256=STQDSA1_jXPTenJotNtsNMXOc9h1x_wJyQ100LXS6-g,3209
 gml/proto/gogoproto/gogo_pb2.py,sha256=WVMIAR8K--mCUkTPM7mEeeXGpQlRRtt_kco10iP3CZs,15728
 gml/proto/mediapipe/framework/calculator_contract_test_pb2.py,sha256=hNjyZCBz3RYa6rN4xR3FOCZKA24gq_LsJ3EMegl5wK4,2031
@@ -24,9 +25,11 @@ gml/proto/mediapipe/framework/thread_pool_executor_pb2.py,sha256=9TJ66fqSo1BiJmE
 gml/proto/opentelemetry/proto/common/v1/common_pb2.py,sha256=wQjeDti-C8JiNwRn-z5M5p-Fqxm-SmnbPaoitJcSK-4,2860
 gml/proto/opentelemetry/proto/metrics/v1/metrics_pb2.py,sha256=t2Far6oVcUFQIimzgAkZ8vQd0asMIlvECp4osC0ujgg,9735
 gml/proto/opentelemetry/proto/resource/v1/resource_pb2.py,sha256=cbNmE12Nm3PjW4NXU7-Z-9m_0Zs3Ab8R1xLkDnvclCg,1730
-gml/proto/src/api/corepb/v1/controlplane_pb2.py,sha256=FPNx5fRXj-bnN5mkDUXVz17M33vuHV_hmxH0ggkAUVs,5536
-gml/proto/src/api/corepb/v1/cp_edge_pb2.py,sha256=u41Sohshi6gBfeZO5VnQzfRStFADFzT1Um5mDY9chcg,15309
-gml/proto/src/api/corepb/v1/mediastream_pb2.py,sha256=1qA-ElTgWeGv3oevYlIjK1TIRSgWbR1TTWxA6Q3SOXk,7224
+gml/proto/src/api/corepb/v1/controlplane_pb2.py,sha256=ChS3EbDhgtywVrI1fU-CqqynhRpRU-YrdK7gFCaN46w,6008
+gml/proto/src/api/corepb/v1/cp_edge_pb2.py,sha256=O0YhmiQrwy-bFuqYt5Vno2M8m99-qlwyhkfeLBB14WQ,14903
+gml/proto/src/api/corepb/v1/device_info_pb2.py,sha256=j4gvzhM1MsDLMHCAwah1X5xxpfG_5nODs7K83mTY0zI,3425
+gml/proto/src/api/corepb/v1/gem_config_pb2.py,sha256=yyEqUqq3-YiX-ByAhbTbZfdh09KuNzEtIYhgk_noJVM,3367
+gml/proto/src/api/corepb/v1/mediastream_pb2.py,sha256=_WV7zav0uaPHzP-yvjRtUtrwexWiz4eqVclIZmhqdcY,7193
 gml/proto/src/api/corepb/v1/model_exec_pb2.py,sha256=1DM58lSFgfoHk0ui3ZTjDfifgp4dhE7nHvhMwmInpsA,27103
 gml/proto/src/common/typespb/jwt_pb2.py,sha256=lxy-bqbyg96i9n_xr2JbkuWX-ldnoJavXPMnApzVSio,5580
 gml/proto/src/common/typespb/status_pb2.py,sha256=IbBJnbsAlvsuTtyT285ZuW6k5VaPfl5kRSOnBxD_H8M,2109
@@ -42,6 +45,6 @@ gml/proto/src/controlplane/logicalpipeline/lppb/v1/lppb_pb2_grpc.py,sha256=-snjW
 gml/proto/src/controlplane/model/mpb/v1/mpb_pb2.py,sha256=RVedXkNYu2iF5OHiXoYyRw9AGRCUWG7qNyY-5QY71Go,3762
 gml/proto/src/controlplane/model/mpb/v1/mpb_pb2_grpc.py,sha256=KSdb6V04qUHDsb1R2o3wixwTyZgrhwnPYobjnRgWX4I,4735
 gml/tensor.py,sha256=753IsMFYZD7p_f0cQPt4nTIBo5p5S5ELqwCuoHORdMk,14823
-gimlet_api-0.0.6.dist-info/WHEEL,sha256=sobxWSyDDkdg_rinUth-jxhXHqoNqlmNMJY3aTZn2Us,91
-gimlet_api-0.0.6.dist-info/METADATA,sha256=mF73_t-Tn5NPVxLnJBOTCQkKvb0weobtQLmOqSkc4B0,506
-gimlet_api-0.0.6.dist-info/RECORD,,
+gimlet_api-0.0.7.dist-info/WHEEL,sha256=sobxWSyDDkdg_rinUth-jxhXHqoNqlmNMJY3aTZn2Us,91
+gimlet_api-0.0.7.dist-info/METADATA,sha256=IhN5ODabyw5hfMJDD-LCeGoiZSmqn1kWCKDNay8or5s,531
+gimlet_api-0.0.7.dist-info/RECORD,,

gml/asset_manager.py

@@ -0,0 +1,75 @@
+# Copyright 2023- Gimlet Labs, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import abc
+import tempfile
+from pathlib import Path
+from typing import Dict
+
+
+class AssetManager:
+    @abc.abstractmethod
+    def add_asset(self, name: str) -> Path:
+        pass
+
+    @abc.abstractmethod
+    def assets(self) -> Dict[str, Path]:
+        pass
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc, value, tb):
+        pass
+
+
+class DirectoryAssetManager(AssetManager):
+    def __init__(self, path: str | Path):
+        self.path = Path(path)
+        self._asset_paths: Dict[str, Path] = dict()
+
+    def add_asset(self, name: str) -> Path:
+        path = self.path / name
+        self._asset_paths[name] = path
+        return path
+
+    def assets(self) -> Dict[str, Path]:
+        return self._asset_paths
+
+
+class TempFileAssetManager(AssetManager):
+    def __init__(self):
+        self._assets = dict()
+        self._asset_paths = dict()
+
+    def add_asset(self, name: str) -> Path:
+        tmp = tempfile.NamedTemporaryFile(mode="w")
+        self._assets[name] = tmp
+        file = tmp.__enter__()
+        self._asset_paths[name] = Path(file.name)
+        return self._asset_paths[name]
+
+    def assets(self) -> Dict[str, Path]:
+        return self._asset_paths
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc, value, tb):
+        for tmp in self._assets.values():
+            tmp.__exit__(exc, value, tb)
+        self._assets.clear()
+        self._asset_paths.clear()

gml/compile.py

@@ -16,19 +16,31 @@
 
 import contextlib
 import functools
-from typing import Dict, List, Optional, Sequence, Union
+from typing import Any, Dict, List, Optional, Sequence, Union
 
-import gml.proto.src.api.corepb.v1.model_exec_pb2 as modelexecpb
+import safetensors_mlir
 import torch
-import torch_mlir  # noqa
-
-try:
-    import torch_mlir.fx  # noqa
-    from torch.export import export  # noqa
-
-    has_fx_importer_torch_export = True
-except ImportError:
-    has_fx_importer_torch_export = False
+import torch_mlir
+from gml.asset_manager import AssetManager
+from mlir.ir import (
+    BF16Type,
+    ComplexType,
+    Context,
+    F16Type,
+    F32Type,
+    F64Type,
+    IntegerType,
+    Operation,
+    RankedTensorType,
+    Value,
+)
+from safetensors.torch import save_file
+from torch._decomp import remove_decompositions
+from torch.export._trace import _export
+from torch_mlir.dialects import torch as torch_d
+from torch_mlir.extras.fx_decomp_util import get_decomposition_table
+from torch_mlir.extras.fx_importer import FxImporter, FxImporterHooks
+from torch_mlir.fx import export_and_import
 
 
 def _default_decomposition_denylist():
@@ -61,19 +73,194 @@ def _patch_aot_export_module():
     torch._functorch.aot_autograd.aot_export_module = orig
 
 
-def to_torch_mlir_w_torch_export(
+_torch_dtype_to_builtin_element_type = {
+    torch.float16: lambda: F16Type.get(),
+    torch.bfloat16: lambda: BF16Type.get(),
+    torch.float32: lambda: F32Type.get(),
+    torch.float64: lambda: F64Type.get(),
+    torch.uint8: lambda: IntegerType.get_unsigned(8),
+    torch.int8: lambda: IntegerType.get_signless(8),
+    torch.int16: lambda: IntegerType.get_signless(16),
+    torch.int32: lambda: IntegerType.get_signless(32),
+    torch.int64: lambda: IntegerType.get_signless(64),
+    torch.bool: lambda: IntegerType.get_signless(1),
+    torch.qint8: lambda: IntegerType.get_signless(8),
+    torch.quint8: lambda: IntegerType.get_unsigned(8),
+    torch.complex32: lambda: ComplexType.get(F16Type.get()),
+    torch.complex64: lambda: ComplexType.get(F32Type.get()),
+    torch.complex128: lambda: ComplexType.get(F64Type.get()),
+}
+
+
+def _get_unique_(tensors, name):
+    index = 0
+    name = "{}_{}".format(name, index)
+    while name in tensors:
+        index += 1
+        name = "{}_{}".format(name, index)
+    return name
+
+
+class TensorSet:
+    def __init__(self):
+        self._tensors: Dict[str, torch.Tensor] = dict()
+
+    def add(self, tensor: torch.Tensor) -> str:
+        shape_desc = "_".join([str(d) for d in tensor.shape])
+        base_name = f"torch_tensor_{shape_desc}_{str(tensor.dtype)}"
+
+        index = 0
+        name = "{}_{}".format(base_name, index)
+        while name in self._tensors and not torch.equal(tensor, self._tensors[name]):
+            index += 1
+            name = "{}_{}".format(base_name, index)
+
+        self._tensors[name] = tensor
+        return name
+
+    def tensors(self) -> Dict[str, torch.Tensor]:
+        return self._tensors
+
+
+class SafetensorImporterHooks(FxImporterHooks):
+    def __init__(self, asset_manager: AssetManager):
+        self._asset_mgr = asset_manager
+        # TODO(james): shard weights into multiple shards.
+        self.asset_name = "weights.shard0"
+        self._tensors = TensorSet()
+
+    def resolve_literal(
+        self, gni: "torch_mlir.extras.fx_importer.GraphNodeImporter", literal: Any
+    ) -> Optional[Value]:
+        if not isinstance(literal, torch.Tensor):
+            return None
+        tensor = literal
+        ctx = gni._c
+
+        tensor_name = self._tensors.add(tensor)
+
+        file_attr = safetensors_mlir.FileAttr.get(ctx, self.asset_name)
+
+        if tensor.dtype not in _torch_dtype_to_builtin_element_type:
+            raise ValueError("unsupported torch dtype: {}".format(tensor.dtype))
+        elem_type = _torch_dtype_to_builtin_element_type[tensor.dtype]()
+        tensor_type = RankedTensorType.get(tuple(tensor.size()), elem_type)
+
+        tensor_attr = safetensors_mlir.TensorAttr.get(
+            tensor_type, file_attr, tensor_name
+        )
+        builtin_tensor = safetensors_mlir.tensor_ref(tensor_type, tensor_attr)
+
+        vtensor_type = gni._cc.tensor_to_vtensor_type(tensor)
+        return Operation.create(
+            name="torch_c.from_builtin_tensor",
+            results=[vtensor_type],
+            operands=[builtin_tensor],
+        ).result
+
+    def save_tensors(self):
+        file_path = self._asset_mgr.add_asset(self.asset_name)
+        tensors = self._tensors.tensors()
+        for k in tensors:
+            tensors[k] = tensors[k].contiguous()
+        save_file(tensors, file_path)
+
+
+def _to_module_list(val):
+    if isinstance(val, torch.nn.Module):
+        return val
+
+    converted = []
+    for item in val:
+        c = _to_module_container(item)
+        if c is None:
+            return None
+        converted.append(c)
+    if not converted:
+        return None
+    return torch.nn.ModuleList(converted)
+
+
+def _to_module_dict(val):
+    if isinstance(val, torch.nn.Module):
+        return val
+
+    converted = dict()
+    for k, v in val.items():
+        c = _to_module_container(v)
+        if c is None:
+            return None
+        converted[k] = v
+    if not converted:
+        return None
+    return torch.nn.ModuleDict(converted)
+
+
+def _to_module_container(val, root=False):
+    if isinstance(val, torch.nn.Module) and not root:
+        return val
+    if isinstance(val, dict):
+        return _to_module_dict(val)
+    if isinstance(val, list) or isinstance(val, tuple):
+        return _to_module_list(val)
+
+    return None
+
+
+def _replace_containers_with_torch_containers(mod: torch.nn.Module):
+    """Replaces any lists, dict, or nested combinations of lists/dicts that are attributes of `mod` with torch.nn.ModuleList/torch.nn.ModuleDict
+
+    This fixes some `module is not installed as a submodule` errors.
+    ."""
+    _excludes = set(["_modules"])
+    replacements = dict()
+    for name, val in mod.__dict__.items():
+        if name in _excludes:
+            continue
+        c = _to_module_container(val, root=True)
+        if c is None:
+            continue
+        replacements[name] = c
+
+    for name, repl in replacements.items():
+        setattr(mod, name, repl)
+
+
+def _ensure_submodules_accessed_through_getattr(mod: torch.nn.Module):
+    """This removes any registered modules from `mod.__dict__`.
+
+    This ensures that all accesses of submodules go through torch's __getattr__ infra,
+    preventing certain cases of `module is not installed as a submodule` errors.
+    """
+    if not hasattr(mod, "_modules"):
+        return
+    for name in mod._modules:
+        if name in mod.__dict__:
+            del mod.__dict__[name]
+
+
+def _submodule_registration_workarounds(mod: torch.nn.Module):
+    """Apply submodule registration workarounds recursively to all submodules of `mod`."""
+    _ensure_submodules_accessed_through_getattr(mod)
+    _replace_containers_with_torch_containers(mod)
+    # We intentionally don't use `mod.modules()` (which returns all recursive submodules) here because we want only
+    # the direct dependencies of `mod`. So that we get a pre-order traversal, ensuring the workarounds are applied
+    # before we check for submodules.
+    for submod in mod._modules.values():
+        if submod is mod:
+            continue
+        _submodule_registration_workarounds(submod)
+
+
+def to_torch_mlir(
     model: torch.nn.Module,
     example_inputs: Sequence[torch.Tensor],
     dynamic_shapes: Optional[
         Sequence[Dict[int, Union[str, "torch.export.dynamic_shapes._Dim"]]]
     ] = None,
     decomposition_denylist: Optional[List[torch._ops.OperatorBase]] = None,
+    weight_manager: Optional[AssetManager] = None,
 ):
-    from torch._decomp import remove_decompositions
-    from torch.export._trace import _export
-    from torch_mlir.extras.fx_decomp_util import get_decomposition_table
-    from torch_mlir.fx import export_and_import
-
     if dynamic_shapes is not None:
         for shape in dynamic_shapes:
             if not isinstance(shape, dict):
@@ -88,6 +275,8 @@ def to_torch_mlir_w_torch_export(
 
     model = model.eval().to("cpu")
 
+    _submodule_registration_workarounds(model)
+
     try:
         # Running the model a few times on the inputs, leads to more consistent compiled results.
         for _ in range(2):
@@ -105,76 +294,31 @@ def to_torch_mlir_w_torch_export(
     )
     decomp_table = get_decomposition_table()
     remove_decompositions(decomp_table, decomposition_denylist)
+    hooks = None
+    if weight_manager is not None:
+        hooks = SafetensorImporterHooks(weight_manager)
+
+    context = Context()
+    torch_d.register_dialect(context)
+    safetensors_mlir.register_dialect(context)
+    fx_importer = FxImporter(context=context, hooks=hooks)
+
     with _patch_aot_export_module():
-        return export_and_import(
+        module = export_and_import(
             prog,
             *example_inputs,
             decomposition_table=decomp_table,
+            fx_importer=fx_importer,
         )
 
-
-def to_torch_mlir_fallback(model, example_inputs):
-    from torch.fx.experimental.proxy_tensor import make_fx
-    from torch_mlir import ExampleArgs, OutputType
-    from torch_mlir import compile as torch_mlir_compile
-    from torch_mlir.dynamo import _get_decomposition_table
-
-    example_args = ExampleArgs.get(example_inputs)
-    args = example_args._get_for_tracing(use_tracing=True, ignore_traced_shapes=True)[
-        "forward"
-    ]
-    try:
-        # Running the model a few times on the inputs, leads to more consistent compiled results.
-        for _ in range(2):
-            _ = model(*args)
-    except Exception:
-        # Ignore errors running the model. This can happen when the model has data dependent branches.
-        pass
+    if hooks is not None:
+        hooks.save_tensors()
 
     try:
-        compiled = torch_mlir_compile(
-            model,
-            example_inputs,
-            use_tracing=False,
-            ignore_traced_shapes=False,
-            output_type=OutputType.RAW,
-            use_make_fx=False,
-        )
-        return compiled
-    except Exception:
-        pass
-
-    # If the module can't be exported directly, we try to create an FX graph and then export it.
-    model = make_fx(
-        model, pre_dispatch=True, decomposition_table=_get_decomposition_table()
-    )(*args)
-    compiled = torch_mlir_compile(
-        model,
-        example_inputs,
-        use_tracing=False,
-        ignore_traced_shapes=False,
-        output_type=OutputType.RAW,
-        use_make_fx=False,
-    )
-
-    return compiled
-
-
-def to_torch_mlir(
-    model,
-    example_inputs,
-    dynamic_shapes: Optional[
-        Sequence[Dict[int, Union[str, "torch.export.dynamic_shapes._Dim"]]]
-    ] = None,
-):
-    if has_fx_importer_torch_export:
-        return to_torch_mlir_w_torch_export(model, example_inputs, dynamic_shapes)
-    else:
-        return to_torch_mlir_fallback(model, example_inputs)
-
+        module.operation.verify()
+    except Exception as exc:
+        raise Exception(
+            "failed to verify converted torch model MLIR module: {}".format(module)
+        ) from exc
 
-def torch_mlir_output_kind():
-    if has_fx_importer_torch_export:
-        return modelexecpb.ModelInfo.MODEL_KIND_TORCH
-    else:
-        return modelexecpb.ModelInfo.MODEL_KIND_TORCHSCRIPT
+    return module

gml/device.py

@@ -17,6 +17,7 @@
 from typing import List
 
 import gml.proto.src.api.corepb.v1.cp_edge_pb2 as cpedgepb
+import gml.proto.src.api.corepb.v1.device_info_pb2 as deviceinfopb
 
 
 class DeviceCapabilities:
@@ -27,32 +28,38 @@ class DeviceCapabilities:
     def to_proto(self) -> cpedgepb.DeviceCapabilities:
         return cpedgepb.DeviceCapabilities(
             model_runtimes=[
-                cpedgepb.DeviceCapabilities.ModelRuntimeInfo(
+                deviceinfopb.ModelRuntimeInfo(
                     type=_runtime_str_to_runtime_protos(runtime)
                 )
                 for runtime in self.runtimes
             ],
             cameras=[
-                cpedgepb.DeviceCapabilities.CameraInfo(
+                deviceinfopb.CameraInfo(
                     driver=_camera_driver_str_to_camera_driver_protos(camera),
                     camera_id=str(idx),
                 )
                 for idx, camera in enumerate(self.cameras)
             ],
+            camera_drivers=[
+                deviceinfopb.CameraDriverInfo(
+                    driver=_camera_driver_str_to_camera_driver_protos(camera)
+                )
+                for camera in self.cameras
+            ],
         )
 
 
 def _runtime_str_to_runtime_protos(
     runtime: str,
-) -> cpedgepb.DeviceCapabilities.ModelRuntimeInfo.ModelRuntimeType:
+) -> deviceinfopb.ModelRuntimeInfo.ModelRuntimeType:
     match runtime.lower():
         case "tensorrt":
             return (
-                cpedgepb.DeviceCapabilities.ModelRuntimeInfo.ModelRuntimeType.MODEL_RUNTIME_TYPE_TENSORRT
+                deviceinfopb.ModelRuntimeInfo.ModelRuntimeType.MODEL_RUNTIME_TYPE_TENSORRT
             )
         case "openvino":
             return (
-                cpedgepb.DeviceCapabilities.ModelRuntimeInfo.ModelRuntimeType.MODEL_RUNTIME_TYPE_OPENVINO
+                deviceinfopb.ModelRuntimeInfo.ModelRuntimeType.MODEL_RUNTIME_TYPE_OPENVINO
             )
         case _:
             raise ValueError("invalid runtime: {}".format(runtime))
@@ -60,15 +67,11 @@ def _runtime_str_to_runtime_protos(
 
 def _camera_driver_str_to_camera_driver_protos(
     driver: str,
-) -> cpedgepb.DeviceCapabilities.CameraInfo.CameraDriver:
+) -> deviceinfopb.CameraDriver:
     match driver.lower():
         case "argus":
-            return (
-                cpedgepb.DeviceCapabilities.CameraInfo.CameraDriver.CAMERA_DRIVER_ARGUS
-            )
+            return deviceinfopb.CameraDriver.CAMERA_DRIVER_ARGUS
         case "v4l2":
-            return (
-                cpedgepb.DeviceCapabilities.CameraInfo.CameraDriver.CAMERA_DRIVER_V4L2
-            )
+            return deviceinfopb.CameraDriver.CAMERA_DRIVER_V4L2
         case _:
             raise ValueError("invalid driver: {}".format(driver))

gml/hf.py

@@ -23,6 +23,7 @@ from typing import Any, BinaryIO, Dict, List, Optional, TextIO, Tuple
 import gml.proto.src.api.corepb.v1.model_exec_pb2 as modelexecpb
 import torch
 import transformers
+from gml.asset_manager import AssetManager
 from gml.model import GenerationConfig, Model, TorchModel
 from gml.preprocessing import (
     ImagePreprocessingStep,
@@ -66,7 +67,9 @@ class HuggingFaceTokenizer(Model):
         )
         self.tokenizer = tokenizer
 
-    def _collect_assets(self) -> Dict[str, TextIO | BinaryIO | Path]:
+    def _collect_assets(
+        self, weight_manager: Optional[AssetManager] = None
+    ) -> Dict[str, TextIO | BinaryIO | Path]:
         with tempfile.TemporaryDirectory() as tmpdir:
             self.tokenizer.save_pretrained(tmpdir)
             paths = [Path(f) for f in glob.glob(tmpdir + "/*")]
@@ -342,7 +345,7 @@ class HuggingFaceImageProcessor:
     def output_spec_object_detection(self) -> Dict[str, Any]:
         if not hasattr(self.processor, "post_process_object_detection"):
             raise NotImplementedError(
-                "only semantic segmentation is currently supported"
+                "processor must have post_process_object_detection set"
             )
 
         id_to_label = self.model.config.id2label

gml/model.py

@@ -23,7 +23,8 @@ from typing import BinaryIO, Dict, List, Literal, Optional, Sequence, TextIO, Tu
 
 import gml.proto.src.api.corepb.v1.model_exec_pb2 as modelexecpb
 import torch
-from gml.compile import to_torch_mlir, torch_mlir_output_kind
+from gml.asset_manager import AssetManager, TempFileAssetManager
+from gml.compile import to_torch_mlir
 from gml.preprocessing import ImagePreprocessingStep
 from gml.tensor import TensorSemantics
 
@@ -90,12 +91,14 @@ class Model(abc.ABC):
         )
 
     @abc.abstractmethod
-    def _collect_assets(self) -> Dict[str, TextIO | BinaryIO | Path]:
+    def _collect_assets(
+        self, weight_manager: Optional[AssetManager] = None
+    ) -> Dict[str, TextIO | BinaryIO | Path]:
         pass
 
     @contextlib.contextmanager
-    def collect_assets(self):
-        yield from self._collect_assets()
+    def collect_assets(self, weight_manager: Optional[AssetManager] = None):
+        yield from self._collect_assets(weight_manager)
 
 
 class TorchModel(Model):
@@ -111,7 +114,7 @@ class TorchModel(Model):
     ):
         super().__init__(
             name,
-            torch_mlir_output_kind(),
+            modelexecpb.ModelInfo.MODEL_KIND_TORCH,
             modelexecpb.ModelInfo.MODEL_STORAGE_FORMAT_MLIR_TEXT,
             **kwargs,
         )
@@ -130,17 +133,27 @@ class TorchModel(Model):
                 for shape, dtype in zip(self.input_shapes, self.input_dtypes)
             ]
 
-    def _convert_to_torch_mlir(self):
+    def _convert_to_torch_mlir(self, weight_manager: Optional[AssetManager] = None):
         return to_torch_mlir(
             self.torch_module,
             self.example_inputs,
             self.dynamic_shapes,
+            weight_manager=weight_manager,
         )
 
-    def _collect_assets(self) -> Dict[str, TextIO | BinaryIO | Path]:
-        compiled = self._convert_to_torch_mlir()
-        file = io.BytesIO(str(compiled).encode("utf-8"))
-        yield {"": file}
+    def _collect_assets(
+        self, weight_manager: Optional[AssetManager] = None
+    ) -> Dict[str, TextIO | BinaryIO | Path]:
+        if weight_manager is None:
+            # If the user does not provide a weight manager, use temp files.
+            weight_manager = TempFileAssetManager()
+
+        with weight_manager as weight_mgr:
+            compiled = self._convert_to_torch_mlir(weight_mgr)
+            file = io.BytesIO(str(compiled).encode("utf-8"))
+            assets = {"": file}
+            assets.update(weight_mgr.assets())
+            yield assets
 
 
 def _kind_str_to_kind_format_protos(
@@ -178,5 +191,7 @@ class ModelFromFiles(Model):
         super().__init__(name=name, kind=kind, storage_format=storage_format, **kwargs)
         self.files = files
 
-    def _collect_assets(self) -> Dict[str, TextIO | BinaryIO | Path]:
+    def _collect_assets(
+        self, weight_manager: Optional[AssetManager] = None
+    ) -> Dict[str, TextIO | BinaryIO | Path]:
         yield self.files

gml/pipelines.py

@@ -66,27 +66,6 @@ nodes:
     frame: .camera_source.frame
   outputs:
   - detections
-- name: frame_metrics_sink
-  kind: FrameMetricsSink
-  attributes:
-    frame_rate_limit: 30
-  inputs:
-    frame: .camera_source.frame
-  outputs:
-  - frame_metrics
-- name: detection_metrics_sink
-  kind: DetectionsMetricsSink
-  attributes:
-  inputs:
-    detections: .detect.detections
-- name: pipeline_latency_metrics_sink
-  kind: LatencyMetricsSink
-  attributes:
-    name: model
-  inputs:
-    reference: .camera_source.frame
-    detections: .detect.detections
-    frame_metrics: .frame_metrics_sink.frame_metrics
 - name: video_stream_sink
   kind: VideoStreamSink
   attributes:
@@ -94,7 +73,6 @@ nodes:
   inputs:
     frame: .camera_source.frame
     detections: .detect.detections
-    frame_metrics: .frame_metrics_sink.frame_metrics
 """
 
 
@@ -122,29 +100,12 @@ nodes:
     frame: .camera_source.frame
   outputs:
   - segmentation
-- name: frame_metrics_sink
-  kind: FrameMetricsSink
-  attributes:
-    frame_rate_limit: 30
-  inputs:
-    frame: .camera_source.frame
-  outputs:
-  - frame_metrics
-- name: pipeline_latency_metrics_sink
-  kind: LatencyMetricsSink
-  attributes:
-    name: model
-  inputs:
-    reference: .camera_source.frame
-    segmentation: .segment.segmentation
-    frame_metrics: .frame_metrics_sink.frame_metrics
 - name: video_stream_sink
   kind: VideoStreamSink
   attributes:
     frame_rate_limit: 30
   inputs:
     frame: .camera_source.frame
-    frame_metrics: .frame_metrics_sink.frame_metrics
     segmentation: .segment.segmentation
 """
 

gml/proto/src/api/corepb/v1/controlplane_pb2.py

@@ -17,7 +17,7 @@ from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$src/api/corepb/v1/controlplane.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x19google/protobuf/any.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xc4\x01\n\nCPMetadata\x12\x37\n\x05topic\x18\x01 \x01(\x0e\x32!.gml.internal.api.core.v1.CPTopicR\x05topic\x12:\n\tentity_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x45ntityIDR\x08\x65ntityId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"v\n\tCPMessage\x12@\n\x08metadata\x18\x01 \x01(\x0b\x32$.gml.internal.api.core.v1.CPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"M\n\x0f\x44\x65viceConnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"J\n\x0c\x44\x65viceUpdate\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"P\n\x12\x44\x65viceDisconnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\xd8\x01\n\x1ePhysicalPipelineReconciliation\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12Y\n\x14physical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x1f\n\x0b\x66orce_apply\x18\x03 \x01(\x08R\nforceApply\"\xbc\x01\n PipelineDeploymentReconciliation\x12_\n\x16pipeline_deployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12\x37\n\x08\x66leet_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId*\xdc\x01\n\x07\x43PTopic\x12\x14\n\x10\x43P_TOPIC_UNKNOWN\x10\x00\x12\x1d\n\x19\x43P_TOPIC_DEVICE_CONNECTED\x10\x01\x12-\n)CP_TOPIC_PHYSICAL_PIPELINE_RECONCILIATION\x10\x02\x12 \n\x1c\x43P_TOPIC_DEVICE_DISCONNECTED\x10\x03\x12/\n+CP_TOPIC_PIPELINE_DEPLOYMENT_RECONCILIATION\x10\x04\x12\x1a\n\x16\x43P_TOPIC_DEVICE_UPDATE\x10\x05\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$src/api/corepb/v1/controlplane.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x19google/protobuf/any.proto\x1a\x1fgoogle/protobuf/timestamp.proto\"\xc4\x01\n\nCPMetadata\x12\x37\n\x05topic\x18\x01 \x01(\x0e\x32!.gml.internal.api.core.v1.CPTopicR\x05topic\x12:\n\tentity_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x45ntityIDR\x08\x65ntityId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"v\n\tCPMessage\x12@\n\x08metadata\x18\x01 \x01(\x0b\x32$.gml.internal.api.core.v1.CPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"M\n\x0f\x44\x65viceConnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"J\n\x0c\x44\x65viceUpdate\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"P\n\x12\x44\x65viceDisconnected\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\"\xd8\x01\n\x1ePhysicalPipelineReconciliation\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12Y\n\x14physical_pipeline_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x1f\n\x0b\x66orce_apply\x18\x03 \x01(\x08R\nforceApply\"\xbc\x01\n PipelineDeploymentReconciliation\x12_\n\x16pipeline_deployment_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x18\xe2\xde\x1f\x14PipelineDeploymentIDR\x14pipelineDeploymentId\x12\x37\n\x08\x66leet_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0b\xe2\xde\x1f\x07\x46leetIDR\x07\x66leetId\"U\n\x17\x42\x61seConfigUpdateRequest\x12:\n\tdevice_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId*\xf8\x01\n\x07\x43PTopic\x12\x14\n\x10\x43P_TOPIC_UNKNOWN\x10\x00\x12\x1d\n\x19\x43P_TOPIC_DEVICE_CONNECTED\x10\x01\x12-\n)CP_TOPIC_PHYSICAL_PIPELINE_RECONCILIATION\x10\x02\x12 \n\x1c\x43P_TOPIC_DEVICE_DISCONNECTED\x10\x03\x12/\n+CP_TOPIC_PIPELINE_DEPLOYMENT_RECONCILIATION\x10\x04\x12\x1a\n\x16\x43P_TOPIC_DEVICE_UPDATE\x10\x05\x12\x1a\n\x16\x43P_TOPIC_DEVICE_CONFIG\x10\x06\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.controlplane_pb2', globals())
@@ -41,8 +41,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PIPELINEDEPLOYMENTRECONCILIATION.fields_by_name['pipeline_deployment_id']._serialized_options = b'\342\336\037\024PipelineDeploymentID'
   _PIPELINEDEPLOYMENTRECONCILIATION.fields_by_name['fleet_id']._options = None
   _PIPELINEDEPLOYMENTRECONCILIATION.fields_by_name['fleet_id']._serialized_options = b'\342\336\037\007FleetID'
-  _CPTOPIC._serialized_start=1146
-  _CPTOPIC._serialized_end=1366
+  _BASECONFIGUPDATEREQUEST.fields_by_name['device_id']._options = None
+  _BASECONFIGUPDATEREQUEST.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
+  _CPTOPIC._serialized_start=1233
+  _CPTOPIC._serialized_end=1481
   _CPMETADATA._serialized_start=180
   _CPMETADATA._serialized_end=376
   _CPMESSAGE._serialized_start=378
@@ -57,4 +59,6 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _PHYSICALPIPELINERECONCILIATION._serialized_end=952
   _PIPELINEDEPLOYMENTRECONCILIATION._serialized_start=955
   _PIPELINEDEPLOYMENTRECONCILIATION._serialized_end=1143
+  _BASECONFIGUPDATEREQUEST._serialized_start=1145
+  _BASECONFIGUPDATEREQUEST._serialized_end=1230
 # @@protoc_insertion_point(module_scope)

gml/proto/src/api/corepb/v1/cp_edge_pb2.py

@@ -17,10 +17,12 @@ from gml.proto.src.common.typespb import status_pb2 as src_dot_common_dot_typesp
 from google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 from gml.proto.opentelemetry.proto.metrics.v1 import metrics_pb2 as opentelemetry_dot_proto_dot_metrics_dot_v1_dot_metrics__pb2
+from gml.proto.src.api.corepb.v1 import device_info_pb2 as src_dot_api_dot_corepb_dot_v1_dot_device__info__pb2
 from gml.proto.src.api.corepb.v1 import model_exec_pb2 as src_dot_api_dot_corepb_dot_v1_dot_model__exec__pb2
+from gml.proto.src.api.corepb.v1 import gem_config_pb2 as src_dot_api_dot_corepb_dot_v1_dot_gem__config__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fsrc/api/corepb/v1/cp_edge.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fsrc/common/typespb/status.proto\x1a\x19google/protobuf/any.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a,opentelemetry/proto/metrics/v1/metrics.proto\x1a\"src/api/corepb/v1/model_exec.proto\"1\n\rEdgeHeartbeat\x12 \n\x06seq_id\x18\x01 \x01(\x03\x42\t\xe2\xde\x1f\x05SeqIDR\x05seqId\"4\n\x10\x45\x64geHeartbeatAck\x12 \n\x06seq_id\x18\x01 \x01(\x03\x42\t\xe2\xde\x1f\x05SeqIDR\x05seqId\"\xbb\x01\n\x1aPhysicalPipelineSpecUpdate\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x42\n\x04spec\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.PhysicalPipelineSpecR\x04spec\"\xdd\x01\n\x1cPhysicalPipelineStatusUpdate\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12H\n\x06status\x18\x03 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PhysicalPipelineStatusR\x06status\"\x0c\n\nCPRunModel\"\x0f\n\rCPRunModelAck\"\xb2\x01\n\x12\x45xecutionGraphSpec\x12=\n\x05graph\x18\x01 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05graph\x12\x43\n\x05state\x18\x02 \x01(\x0e\x32-.gml.internal.api.core.v1.ExecutionGraphStateR\x05state\x12\x18\n\x07version\x18\x03 \x01(\x03R\x07version\"\x8d\x01\n\x14\x45xecutionGraphStatus\x12\x43\n\x05state\x18\x01 \x01(\x0e\x32-.gml.internal.api.core.v1.ExecutionGraphStateR\x05state\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\x12\x18\n\x07version\x18\x03 \x01(\x03R\x07version\"\x8a\x02\n\x13\x41pplyExecutionGraph\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12V\n\x13logical_pipeline_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12@\n\x04spec\x18\x02 \x01(\x0b\x32,.gml.internal.api.core.v1.ExecutionGraphSpecR\x04spec\"q\n\x14\x44\x65leteExecutionGraph\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\"\xbf\x01\n\x1a\x45xecutionGraphStatusUpdate\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x46\n\x06status\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.ExecutionGraphStatusR\x06status\"\x12\n\x10VideoStreamStart\"\x11\n\x0fVideoStreamStop\"\x16\n\x14VideoStreamKeepAlive\"\x12\n\x10MediaStreamStart\"\x11\n\x0fMediaStreamStop\"\x16\n\x14MediaStreamKeepAlive\"q\n\x12MediaStreamControl\x12[\n\x13text_stream_control\x18\x01 \x01(\x0b\x32+.gml.internal.api.core.v1.TextStreamControlR\x11textStreamControl\"+\n\x11TextStreamControl\x12\x16\n\x06prompt\x18\x01 \x01(\tR\x06prompt\"\x7f\n\x18\x45\x64geCPMediaStreamMessage\x12:\n\tstream_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08StreamIDR\x08streamId\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"\x7f\n\x18\x43PEdgeMediaStreamMessage\x12:\n\tstream_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08StreamIDR\x08streamId\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"m\n\x0f\x45\x64geOTelMetrics\x12Z\n\x10resource_metrics\x18\x01 \x01(\x0b\x32/.opentelemetry.proto.metrics.v1.ResourceMetricsR\x0fresourceMetrics\"\x94\x01\n\x13\x46ileTransferRequest\x12\x34\n\x07\x66ile_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x12*\n\x11\x63hunk_start_bytes\x18\x02 \x01(\x03R\x0f\x63hunkStartBytes\x12\x1b\n\tnum_bytes\x18\x03 \x01(\x03R\x08numBytes\"\x8f\x02\n\x14\x46ileTransferResponse\x12)\n\x06status\x18\x01 \x01(\x0b\x32\x11.gml.types.StatusR\x06status\x12N\n\x05\x63hunk\x18\x02 \x01(\x0b\x32\x38.gml.internal.api.core.v1.FileTransferResponse.FileChunkR\x05\x63hunk\x12\x34\n\x07\x66ile_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x1a\x46\n\tFileChunk\x12\x1f\n\x0bstart_bytes\x18\x01 \x01(\x03R\nstartBytes\x12\x18\n\x07payload\x18\x02 \x01(\x0cR\x07payload\"\xb0\x05\n\x12\x44\x65viceCapabilities\x12\x64\n\x0emodel_runtimes\x18\x01 \x03(\x0b\x32=.gml.internal.api.core.v1.DeviceCapabilities.ModelRuntimeInfoR\rmodelRuntimes\x12Q\n\x07\x63\x61meras\x18\x02 \x03(\x0b\x32\x37.gml.internal.api.core.v1.DeviceCapabilities.CameraInfoR\x07\x63\x61meras\x1a\xec\x01\n\x10ModelRuntimeInfo\x12\x62\n\x04type\x18\x01 \x01(\x0e\x32N.gml.internal.api.core.v1.DeviceCapabilities.ModelRuntimeInfo.ModelRuntimeTypeR\x04type\"t\n\x10ModelRuntimeType\x12\x1e\n\x1aMODEL_RUNTIME_TYPE_UNKNOWN\x10\x00\x12\x1f\n\x1bMODEL_RUNTIME_TYPE_TENSORRT\x10\x01\x12\x1f\n\x1bMODEL_RUNTIME_TYPE_OPENVINO\x10\x02\x1a\xf1\x01\n\nCameraInfo\x12\\\n\x06\x64river\x18\x01 \x01(\x0e\x32\x44.gml.internal.api.core.v1.DeviceCapabilities.CameraInfo.CameraDriverR\x06\x64river\x12)\n\tcamera_id\x18\x02 \x01(\tB\x0c\xe2\xde\x1f\x08\x43\x61meraIDR\x08\x63\x61meraId\"Z\n\x0c\x43\x61meraDriver\x12\x19\n\x15\x43\x41MERA_DRIVER_UNKNOWN\x10\x00\x12\x17\n\x13\x43\x41MERA_DRIVER_ARGUS\x10\x01\x12\x16\n\x12\x43\x41MERA_DRIVER_V4L2\x10\x02\"\xcc\x01\n\x0e\x45\x64geCPMetadata\x12;\n\x05topic\x18\x01 \x01(\x0e\x32%.gml.internal.api.core.v1.EdgeCPTopicR\x05topic\x12:\n\tdevice_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"~\n\rEdgeCPMessage\x12\x44\n\x08metadata\x18\x01 \x01(\x0b\x32(.gml.internal.api.core.v1.EdgeCPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"\xcc\x01\n\x0e\x43PEdgeMetadata\x12;\n\x05topic\x18\x01 \x01(\x0e\x32%.gml.internal.api.core.v1.CPEdgeTopicR\x05topic\x12:\n\tdevice_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"~\n\rCPEdgeMessage\x12\x44\n\x08metadata\x18\x01 \x01(\x0b\x32(.gml.internal.api.core.v1.CPEdgeMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg*\xbe\x02\n\x13\x45xecutionGraphState\x12!\n\x1d\x45XECUTION_GRAPH_STATE_UNKNOWN\x10\x00\x12*\n&EXECUTION_GRAPH_STATE_UPDATE_REQUESTED\x10\n\x12%\n!EXECUTION_GRAPH_STATE_DOWNLOADING\x10\x14\x12#\n\x1f\x45XECUTION_GRAPH_STATE_COMPILING\x10\x1e\x12\x1f\n\x1b\x45XECUTION_GRAPH_STATE_READY\x10(\x12\"\n\x1e\x45XECUTION_GRAPH_STATE_DEPLOYED\x10\x32\x12%\n!EXECUTION_GRAPH_STATE_TERMINATING\x10<\x12 \n\x1c\x45XECUTION_GRAPH_STATE_FAILED\x10\x64*\xe0\x01\n\x0b\x45\x64geCPTopic\x12\x19\n\x15\x45\x44GE_CP_TOPIC_UNKNOWN\x10\x00\x12\x18\n\x14\x45\x44GE_CP_TOPIC_STATUS\x10\x01\x12\x17\n\x13\x45\x44GE_CP_TOPIC_VIDEO\x10\x02\x12\x16\n\x12\x45\x44GE_CP_TOPIC_EXEC\x10\x03\x12\x19\n\x15\x45\x44GE_CP_TOPIC_METRICS\x10\x04\x12\x1f\n\x1b\x45\x44GE_CP_TOPIC_FILE_TRANSFER\x10\x05\x12\x16\n\x12\x45\x44GE_CP_TOPIC_INFO\x10\x06\x12\x17\n\x13\x45\x44GE_CP_TOPIC_MEDIA\x10\x07*\xe0\x01\n\x0b\x43PEdgeTopic\x12\x19\n\x15\x43P_EDGE_TOPIC_UNKNOWN\x10\x00\x12\x18\n\x14\x43P_EDGE_TOPIC_STATUS\x10\x01\x12\x17\n\x13\x43P_EDGE_TOPIC_VIDEO\x10\x02\x12\x16\n\x12\x43P_EDGE_TOPIC_EXEC\x10\x03\x12\x19\n\x15\x43P_EDGE_TOPIC_METRICS\x10\x04\x12\x1f\n\x1b\x43P_EDGE_TOPIC_FILE_TRANSFER\x10\x05\x12\x16\n\x12\x43P_EDGE_TOPIC_INFO\x10\x06\x12\x17\n\x13\x43P_EDGE_TOPIC_MEDIA\x10\x07\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fsrc/api/corepb/v1/cp_edge.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x1dsrc/common/typespb/uuid.proto\x1a\x1fsrc/common/typespb/status.proto\x1a\x19google/protobuf/any.proto\x1a\x1fgoogle/protobuf/timestamp.proto\x1a,opentelemetry/proto/metrics/v1/metrics.proto\x1a#src/api/corepb/v1/device_info.proto\x1a\"src/api/corepb/v1/model_exec.proto\x1a\"src/api/corepb/v1/gem_config.proto\"1\n\rEdgeHeartbeat\x12 \n\x06seq_id\x18\x01 \x01(\x03\x42\t\xe2\xde\x1f\x05SeqIDR\x05seqId\"4\n\x10\x45\x64geHeartbeatAck\x12 \n\x06seq_id\x18\x01 \x01(\x03\x42\t\xe2\xde\x1f\x05SeqIDR\x05seqId\"\xbb\x01\n\x1aPhysicalPipelineSpecUpdate\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x42\n\x04spec\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.PhysicalPipelineSpecR\x04spec\"\xdd\x01\n\x1cPhysicalPipelineStatusUpdate\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\x12H\n\x06status\x18\x03 \x01(\x0b\x32\x30.gml.internal.api.core.v1.PhysicalPipelineStatusR\x06status\"\x0c\n\nCPRunModel\"\x0f\n\rCPRunModelAck\"\xb2\x01\n\x12\x45xecutionGraphSpec\x12=\n\x05graph\x18\x01 \x01(\x0b\x32\'.gml.internal.api.core.v1.ExecutionSpecR\x05graph\x12\x43\n\x05state\x18\x02 \x01(\x0e\x32-.gml.internal.api.core.v1.ExecutionGraphStateR\x05state\x12\x18\n\x07version\x18\x03 \x01(\x03R\x07version\"\x8d\x01\n\x14\x45xecutionGraphStatus\x12\x43\n\x05state\x18\x01 \x01(\x0e\x32-.gml.internal.api.core.v1.ExecutionGraphStateR\x05state\x12\x16\n\x06reason\x18\x02 \x01(\tR\x06reason\x12\x18\n\x07version\x18\x03 \x01(\x03R\x07version\"\x8a\x02\n\x13\x41pplyExecutionGraph\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12V\n\x13logical_pipeline_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\x15\xe2\xde\x1f\x11LogicalPipelineIDR\x11logicalPipelineId\x12@\n\x04spec\x18\x02 \x01(\x0b\x32,.gml.internal.api.core.v1.ExecutionGraphSpecR\x04spec\"q\n\x14\x44\x65leteExecutionGraph\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\"\xbf\x01\n\x1a\x45xecutionGraphStatusUpdate\x12Y\n\x14physical_pipeline_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x16\xe2\xde\x1f\x12PhysicalPipelineIDR\x12physicalPipelineId\x12\x46\n\x06status\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.ExecutionGraphStatusR\x06status\"\x12\n\x10VideoStreamStart\"\x11\n\x0fVideoStreamStop\"\x16\n\x14VideoStreamKeepAlive\"\x12\n\x10MediaStreamStart\"\x11\n\x0fMediaStreamStop\"\x16\n\x14MediaStreamKeepAlive\"q\n\x12MediaStreamControl\x12[\n\x13text_stream_control\x18\x01 \x01(\x0b\x32+.gml.internal.api.core.v1.TextStreamControlR\x11textStreamControl\"+\n\x11TextStreamControl\x12\x16\n\x06prompt\x18\x01 \x01(\tR\x06prompt\"\x7f\n\x18\x45\x64geCPMediaStreamMessage\x12:\n\tstream_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08StreamIDR\x08streamId\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"\x7f\n\x18\x43PEdgeMediaStreamMessage\x12:\n\tstream_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08StreamIDR\x08streamId\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"m\n\x0f\x45\x64geOTelMetrics\x12Z\n\x10resource_metrics\x18\x01 \x01(\x0b\x32/.opentelemetry.proto.metrics.v1.ResourceMetricsR\x0fresourceMetrics\"\x94\x01\n\x13\x46ileTransferRequest\x12\x34\n\x07\x66ile_id\x18\x01 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x12*\n\x11\x63hunk_start_bytes\x18\x02 \x01(\x03R\x0f\x63hunkStartBytes\x12\x1b\n\tnum_bytes\x18\x03 \x01(\x03R\x08numBytes\"\x8f\x02\n\x14\x46ileTransferResponse\x12)\n\x06status\x18\x01 \x01(\x0b\x32\x11.gml.types.StatusR\x06status\x12N\n\x05\x63hunk\x18\x02 \x01(\x0b\x32\x38.gml.internal.api.core.v1.FileTransferResponse.FileChunkR\x05\x63hunk\x12\x34\n\x07\x66ile_id\x18\x03 \x01(\x0b\x32\x0f.gml.types.UUIDB\n\xe2\xde\x1f\x06\x46ileIDR\x06\x66ileId\x1a\x46\n\tFileChunk\x12\x1f\n\x0bstart_bytes\x18\x01 \x01(\x03R\nstartBytes\x12\x18\n\x07payload\x18\x02 \x01(\x0cR\x07payload\"\xfa\x01\n\x12\x44\x65viceCapabilities\x12Q\n\x0emodel_runtimes\x18\x01 \x03(\x0b\x32*.gml.internal.api.core.v1.ModelRuntimeInfoR\rmodelRuntimes\x12>\n\x07\x63\x61meras\x18\x02 \x03(\x0b\x32$.gml.internal.api.core.v1.CameraInfoR\x07\x63\x61meras\x12Q\n\x0e\x63\x61mera_drivers\x18\x03 \x03(\x0b\x32*.gml.internal.api.core.v1.CameraDriverInfoR\rcameraDrivers\"\x89\x01\n\x17\x44\x65viceConfigStateUpdate\x12>\n\x05state\x18\x01 \x01(\x0b\x32(.gml.internal.api.core.v1.GEMConfigStateR\x05state\x12.\n\x13\x62\x61se_config_version\x18\x02 \x01(\x03R\x11\x62\x61seConfigVersion\"o\n\x16\x44\x65viceBaseConfigUpdate\x12;\n\x06\x63onfig\x18\x01 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigR\x06\x63onfig\x12\x18\n\x07version\x18\x02 \x01(\x03R\x07version\"\xcc\x01\n\x0e\x45\x64geCPMetadata\x12;\n\x05topic\x18\x01 \x01(\x0e\x32%.gml.internal.api.core.v1.EdgeCPTopicR\x05topic\x12:\n\tdevice_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"~\n\rEdgeCPMessage\x12\x44\n\x08metadata\x18\x01 \x01(\x0b\x32(.gml.internal.api.core.v1.EdgeCPMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg\"\xcc\x01\n\x0e\x43PEdgeMetadata\x12;\n\x05topic\x18\x01 \x01(\x0e\x32%.gml.internal.api.core.v1.CPEdgeTopicR\x05topic\x12:\n\tdevice_id\x18\x02 \x01(\x0b\x32\x0f.gml.types.UUIDB\x0c\xe2\xde\x1f\x08\x44\x65viceIDR\x08\x64\x65viceId\x12\x41\n\x0erecv_timestamp\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\rrecvTimestamp\"~\n\rCPEdgeMessage\x12\x44\n\x08metadata\x18\x01 \x01(\x0b\x32(.gml.internal.api.core.v1.CPEdgeMetadataR\x08metadata\x12\'\n\x03msg\x18\xe8\x07 \x01(\x0b\x32\x14.google.protobuf.AnyR\x03msg*\xbe\x02\n\x13\x45xecutionGraphState\x12!\n\x1d\x45XECUTION_GRAPH_STATE_UNKNOWN\x10\x00\x12*\n&EXECUTION_GRAPH_STATE_UPDATE_REQUESTED\x10\n\x12%\n!EXECUTION_GRAPH_STATE_DOWNLOADING\x10\x14\x12#\n\x1f\x45XECUTION_GRAPH_STATE_COMPILING\x10\x1e\x12\x1f\n\x1b\x45XECUTION_GRAPH_STATE_READY\x10(\x12\"\n\x1e\x45XECUTION_GRAPH_STATE_DEPLOYED\x10\x32\x12%\n!EXECUTION_GRAPH_STATE_TERMINATING\x10<\x12 \n\x1c\x45XECUTION_GRAPH_STATE_FAILED\x10\x64*\xfa\x01\n\x0b\x45\x64geCPTopic\x12\x19\n\x15\x45\x44GE_CP_TOPIC_UNKNOWN\x10\x00\x12\x18\n\x14\x45\x44GE_CP_TOPIC_STATUS\x10\x01\x12\x17\n\x13\x45\x44GE_CP_TOPIC_VIDEO\x10\x02\x12\x16\n\x12\x45\x44GE_CP_TOPIC_EXEC\x10\x03\x12\x19\n\x15\x45\x44GE_CP_TOPIC_METRICS\x10\x04\x12\x1f\n\x1b\x45\x44GE_CP_TOPIC_FILE_TRANSFER\x10\x05\x12\x16\n\x12\x45\x44GE_CP_TOPIC_INFO\x10\x06\x12\x17\n\x13\x45\x44GE_CP_TOPIC_MEDIA\x10\x07\x12\x18\n\x14\x45\x44GE_CP_TOPIC_CONFIG\x10\x08*\xfa\x01\n\x0b\x43PEdgeTopic\x12\x19\n\x15\x43P_EDGE_TOPIC_UNKNOWN\x10\x00\x12\x18\n\x14\x43P_EDGE_TOPIC_STATUS\x10\x01\x12\x17\n\x13\x43P_EDGE_TOPIC_VIDEO\x10\x02\x12\x16\n\x12\x43P_EDGE_TOPIC_EXEC\x10\x03\x12\x19\n\x15\x43P_EDGE_TOPIC_METRICS\x10\x04\x12\x1f\n\x1b\x43P_EDGE_TOPIC_FILE_TRANSFER\x10\x05\x12\x16\n\x12\x43P_EDGE_TOPIC_INFO\x10\x06\x12\x17\n\x13\x43P_EDGE_TOPIC_MEDIA\x10\x07\x12\x18\n\x14\x43P_EDGE_TOPIC_CONFIG\x10\x08\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.cp_edge_pb2', globals())
@@ -52,84 +54,78 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _FILETRANSFERREQUEST.fields_by_name['file_id']._serialized_options = b'\342\336\037\006FileID'
   _FILETRANSFERRESPONSE.fields_by_name['file_id']._options = None
   _FILETRANSFERRESPONSE.fields_by_name['file_id']._serialized_options = b'\342\336\037\006FileID'
-  _DEVICECAPABILITIES_CAMERAINFO.fields_by_name['camera_id']._options = None
-  _DEVICECAPABILITIES_CAMERAINFO.fields_by_name['camera_id']._serialized_options = b'\342\336\037\010CameraID'
   _EDGECPMETADATA.fields_by_name['device_id']._options = None
   _EDGECPMETADATA.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
   _CPEDGEMETADATA.fields_by_name['device_id']._options = None
   _CPEDGEMETADATA.fields_by_name['device_id']._serialized_options = b'\342\336\037\010DeviceID'
-  _EXECUTIONGRAPHSTATE._serialized_start=4184
-  _EXECUTIONGRAPHSTATE._serialized_end=4502
-  _EDGECPTOPIC._serialized_start=4505
-  _EDGECPTOPIC._serialized_end=4729
-  _CPEDGETOPIC._serialized_start=4732
-  _CPEDGETOPIC._serialized_end=4956
-  _EDGEHEARTBEAT._serialized_start=289
-  _EDGEHEARTBEAT._serialized_end=338
-  _EDGEHEARTBEATACK._serialized_start=340
-  _EDGEHEARTBEATACK._serialized_end=392
-  _PHYSICALPIPELINESPECUPDATE._serialized_start=395
-  _PHYSICALPIPELINESPECUPDATE._serialized_end=582
-  _PHYSICALPIPELINESTATUSUPDATE._serialized_start=585
-  _PHYSICALPIPELINESTATUSUPDATE._serialized_end=806
-  _CPRUNMODEL._serialized_start=808
-  _CPRUNMODEL._serialized_end=820
-  _CPRUNMODELACK._serialized_start=822
-  _CPRUNMODELACK._serialized_end=837
-  _EXECUTIONGRAPHSPEC._serialized_start=840
-  _EXECUTIONGRAPHSPEC._serialized_end=1018
-  _EXECUTIONGRAPHSTATUS._serialized_start=1021
-  _EXECUTIONGRAPHSTATUS._serialized_end=1162
-  _APPLYEXECUTIONGRAPH._serialized_start=1165
-  _APPLYEXECUTIONGRAPH._serialized_end=1431
-  _DELETEEXECUTIONGRAPH._serialized_start=1433
-  _DELETEEXECUTIONGRAPH._serialized_end=1546
-  _EXECUTIONGRAPHSTATUSUPDATE._serialized_start=1549
-  _EXECUTIONGRAPHSTATUSUPDATE._serialized_end=1740
-  _VIDEOSTREAMSTART._serialized_start=1742
-  _VIDEOSTREAMSTART._serialized_end=1760
-  _VIDEOSTREAMSTOP._serialized_start=1762
-  _VIDEOSTREAMSTOP._serialized_end=1779
-  _VIDEOSTREAMKEEPALIVE._serialized_start=1781
-  _VIDEOSTREAMKEEPALIVE._serialized_end=1803
-  _MEDIASTREAMSTART._serialized_start=1805
-  _MEDIASTREAMSTART._serialized_end=1823
-  _MEDIASTREAMSTOP._serialized_start=1825
-  _MEDIASTREAMSTOP._serialized_end=1842
-  _MEDIASTREAMKEEPALIVE._serialized_start=1844
-  _MEDIASTREAMKEEPALIVE._serialized_end=1866
-  _MEDIASTREAMCONTROL._serialized_start=1868
-  _MEDIASTREAMCONTROL._serialized_end=1981
-  _TEXTSTREAMCONTROL._serialized_start=1983
-  _TEXTSTREAMCONTROL._serialized_end=2026
-  _EDGECPMEDIASTREAMMESSAGE._serialized_start=2028
-  _EDGECPMEDIASTREAMMESSAGE._serialized_end=2155
-  _CPEDGEMEDIASTREAMMESSAGE._serialized_start=2157
-  _CPEDGEMEDIASTREAMMESSAGE._serialized_end=2284
-  _EDGEOTELMETRICS._serialized_start=2286
-  _EDGEOTELMETRICS._serialized_end=2395
-  _FILETRANSFERREQUEST._serialized_start=2398
-  _FILETRANSFERREQUEST._serialized_end=2546
-  _FILETRANSFERRESPONSE._serialized_start=2549
-  _FILETRANSFERRESPONSE._serialized_end=2820
-  _FILETRANSFERRESPONSE_FILECHUNK._serialized_start=2750
-  _FILETRANSFERRESPONSE_FILECHUNK._serialized_end=2820
-  _DEVICECAPABILITIES._serialized_start=2823
-  _DEVICECAPABILITIES._serialized_end=3511
-  _DEVICECAPABILITIES_MODELRUNTIMEINFO._serialized_start=3031
-  _DEVICECAPABILITIES_MODELRUNTIMEINFO._serialized_end=3267
-  _DEVICECAPABILITIES_MODELRUNTIMEINFO_MODELRUNTIMETYPE._serialized_start=3151
-  _DEVICECAPABILITIES_MODELRUNTIMEINFO_MODELRUNTIMETYPE._serialized_end=3267
-  _DEVICECAPABILITIES_CAMERAINFO._serialized_start=3270
-  _DEVICECAPABILITIES_CAMERAINFO._serialized_end=3511
-  _DEVICECAPABILITIES_CAMERAINFO_CAMERADRIVER._serialized_start=3421
-  _DEVICECAPABILITIES_CAMERAINFO_CAMERADRIVER._serialized_end=3511
-  _EDGECPMETADATA._serialized_start=3514
-  _EDGECPMETADATA._serialized_end=3718
-  _EDGECPMESSAGE._serialized_start=3720
-  _EDGECPMESSAGE._serialized_end=3846
-  _CPEDGEMETADATA._serialized_start=3849
-  _CPEDGEMETADATA._serialized_end=4053
-  _CPEDGEMESSAGE._serialized_start=4055
-  _CPEDGEMESSAGE._serialized_end=4181
+  _EXECUTIONGRAPHSTATE._serialized_start=4072
+  _EXECUTIONGRAPHSTATE._serialized_end=4390
+  _EDGECPTOPIC._serialized_start=4393
+  _EDGECPTOPIC._serialized_end=4643
+  _CPEDGETOPIC._serialized_start=4646
+  _CPEDGETOPIC._serialized_end=4896
+  _EDGEHEARTBEAT._serialized_start=362
+  _EDGEHEARTBEAT._serialized_end=411
+  _EDGEHEARTBEATACK._serialized_start=413
+  _EDGEHEARTBEATACK._serialized_end=465
+  _PHYSICALPIPELINESPECUPDATE._serialized_start=468
+  _PHYSICALPIPELINESPECUPDATE._serialized_end=655
+  _PHYSICALPIPELINESTATUSUPDATE._serialized_start=658
+  _PHYSICALPIPELINESTATUSUPDATE._serialized_end=879
+  _CPRUNMODEL._serialized_start=881
+  _CPRUNMODEL._serialized_end=893
+  _CPRUNMODELACK._serialized_start=895
+  _CPRUNMODELACK._serialized_end=910
+  _EXECUTIONGRAPHSPEC._serialized_start=913
+  _EXECUTIONGRAPHSPEC._serialized_end=1091
+  _EXECUTIONGRAPHSTATUS._serialized_start=1094
+  _EXECUTIONGRAPHSTATUS._serialized_end=1235
+  _APPLYEXECUTIONGRAPH._serialized_start=1238
+  _APPLYEXECUTIONGRAPH._serialized_end=1504
+  _DELETEEXECUTIONGRAPH._serialized_start=1506
+  _DELETEEXECUTIONGRAPH._serialized_end=1619
+  _EXECUTIONGRAPHSTATUSUPDATE._serialized_start=1622
+  _EXECUTIONGRAPHSTATUSUPDATE._serialized_end=1813
+  _VIDEOSTREAMSTART._serialized_start=1815
+  _VIDEOSTREAMSTART._serialized_end=1833
+  _VIDEOSTREAMSTOP._serialized_start=1835
+  _VIDEOSTREAMSTOP._serialized_end=1852
+  _VIDEOSTREAMKEEPALIVE._serialized_start=1854
+  _VIDEOSTREAMKEEPALIVE._serialized_end=1876
+  _MEDIASTREAMSTART._serialized_start=1878
+  _MEDIASTREAMSTART._serialized_end=1896
+  _MEDIASTREAMSTOP._serialized_start=1898
+  _MEDIASTREAMSTOP._serialized_end=1915
+  _MEDIASTREAMKEEPALIVE._serialized_start=1917
+  _MEDIASTREAMKEEPALIVE._serialized_end=1939
+  _MEDIASTREAMCONTROL._serialized_start=1941
+  _MEDIASTREAMCONTROL._serialized_end=2054
+  _TEXTSTREAMCONTROL._serialized_start=2056
+  _TEXTSTREAMCONTROL._serialized_end=2099
+  _EDGECPMEDIASTREAMMESSAGE._serialized_start=2101
+  _EDGECPMEDIASTREAMMESSAGE._serialized_end=2228
+  _CPEDGEMEDIASTREAMMESSAGE._serialized_start=2230
+  _CPEDGEMEDIASTREAMMESSAGE._serialized_end=2357
+  _EDGEOTELMETRICS._serialized_start=2359
+  _EDGEOTELMETRICS._serialized_end=2468
+  _FILETRANSFERREQUEST._serialized_start=2471
+  _FILETRANSFERREQUEST._serialized_end=2619
+  _FILETRANSFERRESPONSE._serialized_start=2622
+  _FILETRANSFERRESPONSE._serialized_end=2893
+  _FILETRANSFERRESPONSE_FILECHUNK._serialized_start=2823
+  _FILETRANSFERRESPONSE_FILECHUNK._serialized_end=2893
+  _DEVICECAPABILITIES._serialized_start=2896
+  _DEVICECAPABILITIES._serialized_end=3146
+  _DEVICECONFIGSTATEUPDATE._serialized_start=3149
+  _DEVICECONFIGSTATEUPDATE._serialized_end=3286
+  _DEVICEBASECONFIGUPDATE._serialized_start=3288
+  _DEVICEBASECONFIGUPDATE._serialized_end=3399
+  _EDGECPMETADATA._serialized_start=3402
+  _EDGECPMETADATA._serialized_end=3606
+  _EDGECPMESSAGE._serialized_start=3608
+  _EDGECPMESSAGE._serialized_end=3734
+  _CPEDGEMETADATA._serialized_start=3737
+  _CPEDGEMETADATA._serialized_end=3941
+  _CPEDGEMESSAGE._serialized_start=3943
+  _CPEDGEMESSAGE._serialized_end=4069
 # @@protoc_insertion_point(module_scope)

gml/proto/src/api/corepb/v1/device_info_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: src/api/corepb/v1/device_info.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/api/corepb/v1/device_info.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\"R\n\x10\x43\x61meraDriverInfo\x12>\n\x06\x64river\x18\x01 \x01(\x0e\x32&.gml.internal.api.core.v1.CameraDriverR\x06\x64river\"\xb4\x02\n\nCameraMode\x12\x16\n\x06\x66ourcc\x18\x01 \x01(\tR\x06\x66ourcc\x12Q\n\x0bresolutions\x18\x02 \x03(\x0b\x32/.gml.internal.api.core.v1.CameraMode.ResolutionR\x0bresolutions\x1a\xba\x01\n\nResolution\x12\x14\n\x05width\x18\x01 \x01(\x05R\x05width\x12\x16\n\x06height\x18\x02 \x01(\x05R\x06height\x12V\n\tintervals\x18\x03 \x03(\x0b\x32\x38.gml.internal.api.core.v1.CameraMode.Resolution.IntervalR\tintervals\x1a&\n\x08Interval\x12\x1a\n\x08interval\x18\x01 \x01(\x01R\x08interval\"\xb3\x01\n\nCameraInfo\x12>\n\x06\x64river\x18\x01 \x01(\x0e\x32&.gml.internal.api.core.v1.CameraDriverR\x06\x64river\x12)\n\tcamera_id\x18\x02 \x01(\tB\x0c\xe2\xde\x1f\x08\x43\x61meraIDR\x08\x63\x61meraId\x12:\n\x05modes\x18\x03 \x03(\x0b\x32$.gml.internal.api.core.v1.CameraModeR\x05modes\"\xf7\x01\n\x10ModelRuntimeInfo\x12O\n\x04type\x18\x01 \x01(\x0e\x32;.gml.internal.api.core.v1.ModelRuntimeInfo.ModelRuntimeTypeR\x04type\"\x91\x01\n\x10ModelRuntimeType\x12\x1e\n\x1aMODEL_RUNTIME_TYPE_UNKNOWN\x10\x00\x12\x1f\n\x1bMODEL_RUNTIME_TYPE_TENSORRT\x10\x01\x12\x1f\n\x1bMODEL_RUNTIME_TYPE_OPENVINO\x10\x02\x12\x1b\n\x17MODEL_RUNTIME_TYPE_IREE\x10\x03*Z\n\x0c\x43\x61meraDriver\x12\x19\n\x15\x43\x41MERA_DRIVER_UNKNOWN\x10\x00\x12\x17\n\x13\x43\x41MERA_DRIVER_ARGUS\x10\x01\x12\x16\n\x12\x43\x41MERA_DRIVER_V4L2\x10\x02\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.device_info_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepb'
+  _CAMERAINFO.fields_by_name['camera_id']._options = None
+  _CAMERAINFO.fields_by_name['camera_id']._serialized_options = b'\342\336\037\010CameraID'
+  _CAMERADRIVER._serialized_start=914
+  _CAMERADRIVER._serialized_end=1004
+  _CAMERADRIVERINFO._serialized_start=87
+  _CAMERADRIVERINFO._serialized_end=169
+  _CAMERAMODE._serialized_start=172
+  _CAMERAMODE._serialized_end=480
+  _CAMERAMODE_RESOLUTION._serialized_start=294
+  _CAMERAMODE_RESOLUTION._serialized_end=480
+  _CAMERAMODE_RESOLUTION_INTERVAL._serialized_start=442
+  _CAMERAMODE_RESOLUTION_INTERVAL._serialized_end=480
+  _CAMERAINFO._serialized_start=483
+  _CAMERAINFO._serialized_end=662
+  _MODELRUNTIMEINFO._serialized_start=665
+  _MODELRUNTIMEINFO._serialized_end=912
+  _MODELRUNTIMEINFO_MODELRUNTIMETYPE._serialized_start=767
+  _MODELRUNTIMEINFO_MODELRUNTIMETYPE._serialized_end=912
+# @@protoc_insertion_point(module_scope)

gml/proto/src/api/corepb/v1/gem_config_pb2.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: src/api/corepb/v1/gem_config.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import builder as _builder
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/api/corepb/v1/gem_config.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\"w\n\x0bVideoSource\x12=\n\x04type\x18\x01 \x01(\x0e\x32).gml.internal.api.core.v1.VideoSourceTypeR\x04type\x12)\n\tsource_id\x18\x02 \x01(\tB\x0c\xe2\xde\x1f\x08SourceIDR\x08sourceId\"\xe4\x01\n\tGEMConfig\x12j\n\x13video_source_claims\x18\x01 \x03(\x0b\x32:.gml.internal.api.core.v1.GEMConfig.VideoSourceClaimsEntryR\x11videoSourceClaims\x1ak\n\x16VideoSourceClaimsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12;\n\x05value\x18\x02 \x01(\x0b\x32%.gml.internal.api.core.v1.VideoSourceR\x05value:\x02\x38\x01\"$\n\x0c\x43onfigStatus\x12\x14\n\x05\x65rror\x18\x01 \x01(\tR\x05\x65rror\"\x95\x02\n\x0eGEMConfigState\x12J\n\x0e\x61pplied_config\x18\x01 \x01(\x0b\x32#.gml.internal.api.core.v1.GEMConfigR\rappliedConfig\x12R\n\x08statuses\x18\x02 \x03(\x0b\x32\x36.gml.internal.api.core.v1.GEMConfigState.StatusesEntryR\x08statuses\x1a\x63\n\rStatusesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12<\n\x05value\x18\x02 \x01(\x0b\x32&.gml.internal.api.core.v1.ConfigStatusR\x05value:\x02\x38\x01*\xa4\x01\n\x0fVideoSourceType\x12\x1d\n\x19VIDEO_SOURCE_TYPE_UNKNOWN\x10\x00\x12\x1b\n\x17VIDEO_SOURCE_TYPE_ARGUS\x10\x01\x12\x1a\n\x16VIDEO_SOURCE_TYPE_V4L2\x10\x02\x12\x1a\n\x16VIDEO_SOURCE_TYPE_FILE\x10\x03\x12\x1d\n\x19VIDEO_SOURCE_TYPE_NETWORK\x10\x04\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
+
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.gem_config_pb2', globals())
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepb'
+  _VIDEOSOURCE.fields_by_name['source_id']._options = None
+  _VIDEOSOURCE.fields_by_name['source_id']._serialized_options = b'\342\336\037\010SourceID'
+  _GEMCONFIG_VIDEOSOURCECLAIMSENTRY._options = None
+  _GEMCONFIG_VIDEOSOURCECLAIMSENTRY._serialized_options = b'8\001'
+  _GEMCONFIGSTATE_STATUSESENTRY._options = None
+  _GEMCONFIGSTATE_STATUSESENTRY._serialized_options = b'8\001'
+  _VIDEOSOURCETYPE._serialized_start=757
+  _VIDEOSOURCETYPE._serialized_end=921
+  _VIDEOSOURCE._serialized_start=86
+  _VIDEOSOURCE._serialized_end=205
+  _GEMCONFIG._serialized_start=208
+  _GEMCONFIG._serialized_end=436
+  _GEMCONFIG_VIDEOSOURCECLAIMSENTRY._serialized_start=329
+  _GEMCONFIG_VIDEOSOURCECLAIMSENTRY._serialized_end=436
+  _CONFIGSTATUS._serialized_start=438
+  _CONFIGSTATUS._serialized_end=474
+  _GEMCONFIGSTATE._serialized_start=477
+  _GEMCONFIGSTATE._serialized_end=754
+  _GEMCONFIGSTATE_STATUSESENTRY._serialized_start=655
+  _GEMCONFIGSTATE_STATUSESENTRY._serialized_end=754
+# @@protoc_insertion_point(module_scope)

gml/proto/src/api/corepb/v1/mediastream_pb2.py

@@ -15,7 +15,7 @@ from gml.proto.gogoproto import gogo_pb2 as gogoproto_dot_gogo__pb2
 from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/api/corepb/v1/mediastream.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x1egoogle/protobuf/wrappers.proto\"3\n\x05Label\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x14\n\x05score\x18\x02 \x01(\x02R\x05score\"d\n\x14NormalizedCenterRect\x12\x0e\n\x02xc\x18\x01 \x01(\x02R\x02xc\x12\x0e\n\x02yc\x18\x02 \x01(\x02R\x02yc\x12\x14\n\x05width\x18\x03 \x01(\x02R\x05width\x12\x16\n\x06height\x18\x04 \x01(\x02R\x06height\"G\n\x0e\x43lassification\x12\x35\n\x05label\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.LabelR\x05label\"\xcd\x01\n\tDetection\x12\x35\n\x05label\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.LabelR\x05label\x12Q\n\x0c\x62ounding_box\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.NormalizedCenterRectR\x0b\x62oundingBox\x12\x36\n\x08track_id\x18\x03 \x01(\x0b\x32\x1b.google.protobuf.Int64ValueR\x07trackId\"R\n\rDetectionList\x12\x41\n\tdetection\x18\x01 \x03(\x0b\x32#.gml.internal.api.core.v1.DetectionR\tdetection\"<\n\x0eTracksMetadata\x12*\n\x11removed_track_ids\x18\x01 \x03(\x03R\x0fremovedTrackIds\"X\n\x10SegmentationMask\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12.\n\x13run_length_encoding\x18\x02 \x03(\x05R\x11runLengthEncoding\"~\n\x0cSegmentation\x12@\n\x05masks\x18\x01 \x03(\x0b\x32*.gml.internal.api.core.v1.SegmentationMaskR\x05masks\x12\x14\n\x05width\x18\x02 \x01(\x03R\x05width\x12\x16\n\x06height\x18\x03 \x01(\x03R\x06height\"8\n\nRegression\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x14\n\x05value\x18\x02 \x01(\x01R\x05value\"\xbb\x01\n\x0eImageHistogram\x12\x45\n\x07\x63hannel\x18\x01 \x01(\x0e\x32+.gml.internal.api.core.v1.ImageColorChannelR\x07\x63hannel\x12\x10\n\x03min\x18\x02 \x01(\x01R\x03min\x12\x10\n\x03max\x18\x03 \x01(\x01R\x03max\x12\x10\n\x03num\x18\x04 \x01(\x03R\x03num\x12\x10\n\x03sum\x18\x05 \x01(\x01R\x03sum\x12\x1a\n\x06\x62ucket\x18\x06 \x03(\x03\x42\x02\x10\x01R\x06\x62ucket\"_\n\x13ImageHistogramBatch\x12H\n\nhistograms\x18\x01 \x03(\x0b\x32(.gml.internal.api.core.v1.ImageHistogramR\nhistograms\"e\n\x13ImageQualityMetrics\x12#\n\rbrisque_score\x18\x01 \x01(\x01R\x0c\x62risqueScore\x12)\n\x10\x62lurriness_score\x18\x02 \x01(\x01R\x0f\x62lurrinessScore\"\xae\x03\n\x11ImageOverlayChunk\x12\x31\n\x08\x66rame_ts\x18\x01 \x01(\x03\x42\x16\xe2\xde\x1f\x07\x46rameTS\xea\xde\x1f\x07\x66rameTSR\x07\x66rameTS\x12\x19\n\x03\x65of\x18\x02 \x01(\x08\x42\x07\xe2\xde\x1f\x03\x45OFR\x03\x65of\x12I\n\ndetections\x18\x64 \x01(\x0b\x32\'.gml.internal.api.core.v1.DetectionListH\x00R\ndetections\x12L\n\x0csegmentation\x18\x65 \x01(\x0b\x32&.gml.internal.api.core.v1.SegmentationH\x00R\x0csegmentation\x12P\n\nhistograms\x18\xc8\x01 \x01(\x0b\x32-.gml.internal.api.core.v1.ImageHistogramBatchH\x00R\nhistograms\x12U\n\rimage_quality\x18\xac\x02 \x01(\x0b\x32-.gml.internal.api.core.v1.ImageQualityMetricsH\x00R\x0cimageQualityB\t\n\x07overlay\"\x81\x01\n\tH264Chunk\x12\x31\n\x08\x66rame_ts\x18\x01 \x01(\x03\x42\x16\xe2\xde\x1f\x07\x46rameTS\xea\xde\x1f\x07\x66rameTSR\x07\x66rameTS\x12\x19\n\x03\x65of\x18\x02 \x01(\x08\x42\x07\xe2\xde\x1f\x03\x45OFR\x03\x65of\x12&\n\x08nal_data\x18\x03 \x01(\x0c\x42\x0b\xe2\xde\x1f\x07NALDataR\x07nalData\"Z\n\x0bVideoHeader\x12\x14\n\x05width\x18\x01 \x01(\x03R\x05width\x12\x16\n\x06height\x18\x02 \x01(\x03R\x06height\x12\x1d\n\nframe_rate\x18\x03 \x01(\x01R\tframeRate\":\n\tTextBatch\x12\x12\n\x04text\x18\x01 \x01(\tR\x04text\x12\x19\n\x03\x65os\x18\x02 \x01(\x08\x42\x07\xe2\xde\x1f\x03\x45OSR\x03\x65os*\xac\x01\n\x11ImageColorChannel\x12\x1f\n\x1bIMAGE_COLOR_CHANNEL_UNKNOWN\x10\x00\x12\x1c\n\x18IMAGE_COLOR_CHANNEL_GRAY\x10\x01\x12\x1b\n\x17IMAGE_COLOR_CHANNEL_RED\x10\x02\x12\x1d\n\x19IMAGE_COLOR_CHANNEL_GREEN\x10\x03\x12\x1c\n\x18IMAGE_COLOR_CHANNEL_BLUE\x10\x04\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/api/corepb/v1/mediastream.proto\x12\x18gml.internal.api.core.v1\x1a\x14gogoproto/gogo.proto\x1a\x1egoogle/protobuf/wrappers.proto\"3\n\x05Label\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x14\n\x05score\x18\x02 \x01(\x02R\x05score\"d\n\x14NormalizedCenterRect\x12\x0e\n\x02xc\x18\x01 \x01(\x02R\x02xc\x12\x0e\n\x02yc\x18\x02 \x01(\x02R\x02yc\x12\x14\n\x05width\x18\x03 \x01(\x02R\x05width\x12\x16\n\x06height\x18\x04 \x01(\x02R\x06height\"G\n\x0e\x43lassification\x12\x35\n\x05label\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.LabelR\x05label\"\xcd\x01\n\tDetection\x12\x35\n\x05label\x18\x01 \x03(\x0b\x32\x1f.gml.internal.api.core.v1.LabelR\x05label\x12Q\n\x0c\x62ounding_box\x18\x02 \x01(\x0b\x32..gml.internal.api.core.v1.NormalizedCenterRectR\x0b\x62oundingBox\x12\x36\n\x08track_id\x18\x03 \x01(\x0b\x32\x1b.google.protobuf.Int64ValueR\x07trackId\"R\n\rDetectionList\x12\x41\n\tdetection\x18\x01 \x03(\x0b\x32#.gml.internal.api.core.v1.DetectionR\tdetection\"<\n\x0eTracksMetadata\x12*\n\x11removed_track_ids\x18\x01 \x03(\x03R\x0fremovedTrackIds\"X\n\x10SegmentationMask\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12.\n\x13run_length_encoding\x18\x02 \x03(\x05R\x11runLengthEncoding\"~\n\x0cSegmentation\x12@\n\x05masks\x18\x01 \x03(\x0b\x32*.gml.internal.api.core.v1.SegmentationMaskR\x05masks\x12\x14\n\x05width\x18\x02 \x01(\x03R\x05width\x12\x16\n\x06height\x18\x03 \x01(\x03R\x06height\"8\n\nRegression\x12\x14\n\x05label\x18\x01 \x01(\tR\x05label\x12\x14\n\x05value\x18\x02 \x01(\x01R\x05value\"\xbb\x01\n\x0eImageHistogram\x12\x45\n\x07\x63hannel\x18\x01 \x01(\x0e\x32+.gml.internal.api.core.v1.ImageColorChannelR\x07\x63hannel\x12\x10\n\x03min\x18\x02 \x01(\x01R\x03min\x12\x10\n\x03max\x18\x03 \x01(\x01R\x03max\x12\x10\n\x03num\x18\x04 \x01(\x03R\x03num\x12\x10\n\x03sum\x18\x05 \x01(\x01R\x03sum\x12\x1a\n\x06\x62ucket\x18\x06 \x03(\x03\x42\x02\x10\x01R\x06\x62ucket\"_\n\x13ImageHistogramBatch\x12H\n\nhistograms\x18\x01 \x03(\x0b\x32(.gml.internal.api.core.v1.ImageHistogramR\nhistograms\"e\n\x13ImageQualityMetrics\x12#\n\rbrisque_score\x18\x01 \x01(\x01R\x0c\x62risqueScore\x12)\n\x10\x62lurriness_score\x18\x02 \x01(\x01R\x0f\x62lurrinessScore\"\xae\x03\n\x11ImageOverlayChunk\x12\x31\n\x08\x66rame_ts\x18\x01 \x01(\x03\x42\x16\xe2\xde\x1f\x07\x46rameTS\xea\xde\x1f\x07\x66rameTSR\x07\x66rameTS\x12\x19\n\x03\x65of\x18\x02 \x01(\x08\x42\x07\xe2\xde\x1f\x03\x45OFR\x03\x65of\x12I\n\ndetections\x18\x64 \x01(\x0b\x32\'.gml.internal.api.core.v1.DetectionListH\x00R\ndetections\x12L\n\x0csegmentation\x18\x65 \x01(\x0b\x32&.gml.internal.api.core.v1.SegmentationH\x00R\x0csegmentation\x12P\n\nhistograms\x18\xc8\x01 \x01(\x0b\x32-.gml.internal.api.core.v1.ImageHistogramBatchH\x00R\nhistograms\x12U\n\rimage_quality\x18\xac\x02 \x01(\x0b\x32-.gml.internal.api.core.v1.ImageQualityMetricsH\x00R\x0cimageQualityB\t\n\x07overlay\"\x81\x01\n\tH264Chunk\x12\x31\n\x08\x66rame_ts\x18\x01 \x01(\x03\x42\x16\xe2\xde\x1f\x07\x46rameTS\xea\xde\x1f\x07\x66rameTSR\x07\x66rameTS\x12\x19\n\x03\x65of\x18\x02 \x01(\x08\x42\x07\xe2\xde\x1f\x03\x45OFR\x03\x65of\x12&\n\x08nal_data\x18\x03 \x01(\x0c\x42\x0b\xe2\xde\x1f\x07NALDataR\x07nalData\"A\n\x0bVideoHeader\x12\x14\n\x05width\x18\x01 \x01(\x03R\x05width\x12\x16\n\x06height\x18\x02 \x01(\x03R\x06heightJ\x04\x08\x03\x10\x04\":\n\tTextBatch\x12\x12\n\x04text\x18\x01 \x01(\tR\x04text\x12\x19\n\x03\x65os\x18\x02 \x01(\x08\x42\x07\xe2\xde\x1f\x03\x45OSR\x03\x65os*\xac\x01\n\x11ImageColorChannel\x12\x1f\n\x1bIMAGE_COLOR_CHANNEL_UNKNOWN\x10\x00\x12\x1c\n\x18IMAGE_COLOR_CHANNEL_GRAY\x10\x01\x12\x1b\n\x17IMAGE_COLOR_CHANNEL_RED\x10\x02\x12\x1d\n\x19IMAGE_COLOR_CHANNEL_GREEN\x10\x03\x12\x1c\n\x18IMAGE_COLOR_CHANNEL_BLUE\x10\x04\x42/Z-gimletlabs.ai/gimlet/src/api/corepb/v1;corepbb\x06proto3')
 
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'src.api.corepb.v1.mediastream_pb2', globals())
@@ -37,8 +37,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _H264CHUNK.fields_by_name['nal_data']._serialized_options = b'\342\336\037\007NALData'
   _TEXTBATCH.fields_by_name['eos']._options = None
   _TEXTBATCH.fields_by_name['eos']._serialized_options = b'\342\336\037\003EOS'
-  _IMAGECOLORCHANNEL._serialized_start=2085
-  _IMAGECOLORCHANNEL._serialized_end=2257
+  _IMAGECOLORCHANNEL._serialized_start=2060
+  _IMAGECOLORCHANNEL._serialized_end=2232
   _LABEL._serialized_start=119
   _LABEL._serialized_end=170
   _NORMALIZEDCENTERRECT._serialized_start=172
@@ -68,7 +68,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _H264CHUNK._serialized_start=1801
   _H264CHUNK._serialized_end=1930
   _VIDEOHEADER._serialized_start=1932
-  _VIDEOHEADER._serialized_end=2022
-  _TEXTBATCH._serialized_start=2024
-  _TEXTBATCH._serialized_end=2082
+  _VIDEOHEADER._serialized_end=1997
+  _TEXTBATCH._serialized_start=1999
+  _TEXTBATCH._serialized_end=2057
 # @@protoc_insertion_point(module_scope)