viam-sdk 0.45.2__py3-none-win_amd64.whl

viam/gen/app/datasync/v1/data_sync_pb2.pyi

@@ -0,0 +1,425 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+from .... import app
+import builtins
+import collections.abc
+import google.protobuf.any_pb2
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import google.protobuf.struct_pb2
+import google.protobuf.timestamp_pb2
+import sys
+import typing
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _MimeType:
+    ValueType = typing.NewType('ValueType', builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _MimeTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MimeType.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    MIME_TYPE_UNSPECIFIED: _MimeType.ValueType
+    MIME_TYPE_IMAGE_JPEG: _MimeType.ValueType
+    MIME_TYPE_IMAGE_PNG: _MimeType.ValueType
+    MIME_TYPE_APPLICATION_PCD: _MimeType.ValueType
+
+class MimeType(_MimeType, metaclass=_MimeTypeEnumTypeWrapper):
+    ...
+MIME_TYPE_UNSPECIFIED: MimeType.ValueType
+MIME_TYPE_IMAGE_JPEG: MimeType.ValueType
+MIME_TYPE_IMAGE_PNG: MimeType.ValueType
+MIME_TYPE_APPLICATION_PCD: MimeType.ValueType
+global___MimeType = MimeType
+
+class _DataType:
+    ValueType = typing.NewType('ValueType', builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _DataTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DataType.ValueType], builtins.type):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    DATA_TYPE_UNSPECIFIED: _DataType.ValueType
+    DATA_TYPE_BINARY_SENSOR: _DataType.ValueType
+    DATA_TYPE_TABULAR_SENSOR: _DataType.ValueType
+    DATA_TYPE_FILE: _DataType.ValueType
+
+class DataType(_DataType, metaclass=_DataTypeEnumTypeWrapper):
+    """DataType specifies the type of data uploaded."""
+DATA_TYPE_UNSPECIFIED: DataType.ValueType
+DATA_TYPE_BINARY_SENSOR: DataType.ValueType
+DATA_TYPE_TABULAR_SENSOR: DataType.ValueType
+DATA_TYPE_FILE: DataType.ValueType
+global___DataType = DataType
+
+@typing.final
+class DataCaptureUploadRequest(google.protobuf.message.Message):
+    """DataCaptureUploadRequest requests to upload the contents and metadata for tabular data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    METADATA_FIELD_NUMBER: builtins.int
+    SENSOR_CONTENTS_FIELD_NUMBER: builtins.int
+
+    @property
+    def metadata(self) -> global___UploadMetadata:
+        ...
+
+    @property
+    def sensor_contents(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SensorData]:
+        ...
+
+    def __init__(self, *, metadata: global___UploadMetadata | None=..., sensor_contents: collections.abc.Iterable[global___SensorData] | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['metadata', b'metadata']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['metadata', b'metadata', 'sensor_contents', b'sensor_contents']) -> None:
+        ...
+global___DataCaptureUploadRequest = DataCaptureUploadRequest
+
+@typing.final
+class DataCaptureUploadResponse(google.protobuf.message.Message):
+    """DataCaptureUploadResponse returns the file id of the uploaded contents and metadata for tabular data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    FILE_ID_FIELD_NUMBER: builtins.int
+    BINARY_DATA_ID_FIELD_NUMBER: builtins.int
+    file_id: builtins.str
+    binary_data_id: builtins.str
+
+    def __init__(self, *, file_id: builtins.str=..., binary_data_id: builtins.str=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['binary_data_id', b'binary_data_id', 'file_id', b'file_id']) -> None:
+        ...
+global___DataCaptureUploadResponse = DataCaptureUploadResponse
+
+@typing.final
+class FileUploadRequest(google.protobuf.message.Message):
+    """FileUploadRequest requests to upload the contents and metadata for binary (image + file) data.
+    The first packet must be the UploadMetadata associated with the binary data.
+    """
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    METADATA_FIELD_NUMBER: builtins.int
+    FILE_CONTENTS_FIELD_NUMBER: builtins.int
+
+    @property
+    def metadata(self) -> global___UploadMetadata:
+        ...
+
+    @property
+    def file_contents(self) -> global___FileData:
+        ...
+
+    def __init__(self, *, metadata: global___UploadMetadata | None=..., file_contents: global___FileData | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['file_contents', b'file_contents', 'metadata', b'metadata', 'upload_packet', b'upload_packet']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['file_contents', b'file_contents', 'metadata', b'metadata', 'upload_packet', b'upload_packet']) -> None:
+        ...
+
+    def WhichOneof(self, oneof_group: typing.Literal['upload_packet', b'upload_packet']) -> typing.Literal['metadata', 'file_contents'] | None:
+        ...
+global___FileUploadRequest = FileUploadRequest
+
+@typing.final
+class FileUploadResponse(google.protobuf.message.Message):
+    """FileUploadResponse returns the file id of the uploaded contents and metadata for binary (image + file) data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    FILE_ID_FIELD_NUMBER: builtins.int
+    BINARY_DATA_ID_FIELD_NUMBER: builtins.int
+    file_id: builtins.str
+    binary_data_id: builtins.str
+
+    def __init__(self, *, file_id: builtins.str=..., binary_data_id: builtins.str=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['binary_data_id', b'binary_data_id', 'file_id', b'file_id']) -> None:
+        ...
+global___FileUploadResponse = FileUploadResponse
+
+@typing.final
+class StreamingDataCaptureUploadRequest(google.protobuf.message.Message):
+    """StreamingDataCaptureUploadRequest requests to upload the contents and metadata for streaming binary (image + file) data.
+    The first packet must be the DataCaptureUploadMetadata associated with the data.
+    """
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    METADATA_FIELD_NUMBER: builtins.int
+    DATA_FIELD_NUMBER: builtins.int
+    data: builtins.bytes
+
+    @property
+    def metadata(self) -> global___DataCaptureUploadMetadata:
+        ...
+
+    def __init__(self, *, metadata: global___DataCaptureUploadMetadata | None=..., data: builtins.bytes=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['data', b'data', 'metadata', b'metadata', 'upload_packet', b'upload_packet']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['data', b'data', 'metadata', b'metadata', 'upload_packet', b'upload_packet']) -> None:
+        ...
+
+    def WhichOneof(self, oneof_group: typing.Literal['upload_packet', b'upload_packet']) -> typing.Literal['metadata', 'data'] | None:
+        ...
+global___StreamingDataCaptureUploadRequest = StreamingDataCaptureUploadRequest
+
+@typing.final
+class StreamingDataCaptureUploadResponse(google.protobuf.message.Message):
+    """StreamingDataCaptureUploadResponse returns the file id of the uploaded contents and metadata for streaming binary (image + file) data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    FILE_ID_FIELD_NUMBER: builtins.int
+    BINARY_DATA_ID_FIELD_NUMBER: builtins.int
+    file_id: builtins.str
+    binary_data_id: builtins.str
+
+    def __init__(self, *, file_id: builtins.str=..., binary_data_id: builtins.str=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['binary_data_id', b'binary_data_id', 'file_id', b'file_id']) -> None:
+        ...
+global___StreamingDataCaptureUploadResponse = StreamingDataCaptureUploadResponse
+
+@typing.final
+class SensorMetadata(google.protobuf.message.Message):
+    """SensorMetadata contains the time the sensor data was requested and was
+    received.
+    """
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    TIME_REQUESTED_FIELD_NUMBER: builtins.int
+    TIME_RECEIVED_FIELD_NUMBER: builtins.int
+    MIME_TYPE_FIELD_NUMBER: builtins.int
+    ANNOTATIONS_FIELD_NUMBER: builtins.int
+    mime_type: global___MimeType.ValueType
+
+    @property
+    def time_requested(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        ...
+
+    @property
+    def time_received(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        ...
+
+    @property
+    def annotations(self) -> app.data.v1.data_pb2.Annotations:
+        ...
+
+    def __init__(self, *, time_requested: google.protobuf.timestamp_pb2.Timestamp | None=..., time_received: google.protobuf.timestamp_pb2.Timestamp | None=..., mime_type: global___MimeType.ValueType=..., annotations: app.data.v1.data_pb2.Annotations | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['annotations', b'annotations', 'time_received', b'time_received', 'time_requested', b'time_requested']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['annotations', b'annotations', 'mime_type', b'mime_type', 'time_received', b'time_received', 'time_requested', b'time_requested']) -> None:
+        ...
+global___SensorMetadata = SensorMetadata
+
+@typing.final
+class SensorData(google.protobuf.message.Message):
+    """SensorData contains the contents and metadata for tabular data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    METADATA_FIELD_NUMBER: builtins.int
+    STRUCT_FIELD_NUMBER: builtins.int
+    BINARY_FIELD_NUMBER: builtins.int
+    binary: builtins.bytes
+
+    @property
+    def metadata(self) -> global___SensorMetadata:
+        ...
+
+    @property
+    def struct(self) -> google.protobuf.struct_pb2.Struct:
+        ...
+
+    def __init__(self, *, metadata: global___SensorMetadata | None=..., struct: google.protobuf.struct_pb2.Struct | None=..., binary: builtins.bytes=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['binary', b'binary', 'data', b'data', 'metadata', b'metadata', 'struct', b'struct']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['binary', b'binary', 'data', b'data', 'metadata', b'metadata', 'struct', b'struct']) -> None:
+        ...
+
+    def WhichOneof(self, oneof_group: typing.Literal['data', b'data']) -> typing.Literal['struct', 'binary'] | None:
+        ...
+global___SensorData = SensorData
+
+@typing.final
+class FileData(google.protobuf.message.Message):
+    """FileData contains the contents of binary (image + file) data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    DATA_FIELD_NUMBER: builtins.int
+    data: builtins.bytes
+
+    def __init__(self, *, data: builtins.bytes=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['data', b'data']) -> None:
+        ...
+global___FileData = FileData
+
+@typing.final
+class UploadMetadata(google.protobuf.message.Message):
+    """UploadMetadata contains the metadata for binary (image + file) data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class MethodParametersEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+
+        @property
+        def value(self) -> google.protobuf.any_pb2.Any:
+            ...
+
+        def __init__(self, *, key: builtins.str=..., value: google.protobuf.any_pb2.Any | None=...) -> None:
+            ...
+
+        def HasField(self, field_name: typing.Literal['value', b'value']) -> builtins.bool:
+            ...
+
+        def ClearField(self, field_name: typing.Literal['key', b'key', 'value', b'value']) -> None:
+            ...
+    PART_ID_FIELD_NUMBER: builtins.int
+    COMPONENT_TYPE_FIELD_NUMBER: builtins.int
+    COMPONENT_NAME_FIELD_NUMBER: builtins.int
+    METHOD_NAME_FIELD_NUMBER: builtins.int
+    TYPE_FIELD_NUMBER: builtins.int
+    FILE_NAME_FIELD_NUMBER: builtins.int
+    METHOD_PARAMETERS_FIELD_NUMBER: builtins.int
+    FILE_EXTENSION_FIELD_NUMBER: builtins.int
+    TAGS_FIELD_NUMBER: builtins.int
+    part_id: builtins.str
+    component_type: builtins.str
+    component_name: builtins.str
+    method_name: builtins.str
+    type: global___DataType.ValueType
+    file_name: builtins.str
+    file_extension: builtins.str
+
+    @property
+    def method_parameters(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, google.protobuf.any_pb2.Any]:
+        ...
+
+    @property
+    def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        ...
+
+    def __init__(self, *, part_id: builtins.str=..., component_type: builtins.str=..., component_name: builtins.str=..., method_name: builtins.str=..., type: global___DataType.ValueType=..., file_name: builtins.str=..., method_parameters: collections.abc.Mapping[builtins.str, google.protobuf.any_pb2.Any] | None=..., file_extension: builtins.str=..., tags: collections.abc.Iterable[builtins.str] | None=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['component_name', b'component_name', 'component_type', b'component_type', 'file_extension', b'file_extension', 'file_name', b'file_name', 'method_name', b'method_name', 'method_parameters', b'method_parameters', 'part_id', b'part_id', 'tags', b'tags', 'type', b'type']) -> None:
+        ...
+global___UploadMetadata = UploadMetadata
+
+@typing.final
+class CaptureInterval(google.protobuf.message.Message):
+    """CaptureInterval specifies the start and end times of the data capture."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    START_FIELD_NUMBER: builtins.int
+    END_FIELD_NUMBER: builtins.int
+
+    @property
+    def start(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        ...
+
+    @property
+    def end(self) -> google.protobuf.timestamp_pb2.Timestamp:
+        ...
+
+    def __init__(self, *, start: google.protobuf.timestamp_pb2.Timestamp | None=..., end: google.protobuf.timestamp_pb2.Timestamp | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['end', b'end', 'start', b'start']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['end', b'end', 'start', b'start']) -> None:
+        ...
+global___CaptureInterval = CaptureInterval
+
+@typing.final
+class DataCaptureMetadata(google.protobuf.message.Message):
+    """DataCaptureMetadata contains the metadata for data captured by collectors."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
+    class MethodParametersEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: builtins.str
+
+        @property
+        def value(self) -> google.protobuf.any_pb2.Any:
+            ...
+
+        def __init__(self, *, key: builtins.str=..., value: google.protobuf.any_pb2.Any | None=...) -> None:
+            ...
+
+        def HasField(self, field_name: typing.Literal['value', b'value']) -> builtins.bool:
+            ...
+
+        def ClearField(self, field_name: typing.Literal['key', b'key', 'value', b'value']) -> None:
+            ...
+    COMPONENT_TYPE_FIELD_NUMBER: builtins.int
+    COMPONENT_NAME_FIELD_NUMBER: builtins.int
+    METHOD_NAME_FIELD_NUMBER: builtins.int
+    TYPE_FIELD_NUMBER: builtins.int
+    METHOD_PARAMETERS_FIELD_NUMBER: builtins.int
+    FILE_EXTENSION_FIELD_NUMBER: builtins.int
+    TAGS_FIELD_NUMBER: builtins.int
+    component_type: builtins.str
+    component_name: builtins.str
+    method_name: builtins.str
+    type: global___DataType.ValueType
+    file_extension: builtins.str
+
+    @property
+    def method_parameters(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, google.protobuf.any_pb2.Any]:
+        ...
+
+    @property
+    def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        ...
+
+    def __init__(self, *, component_type: builtins.str=..., component_name: builtins.str=..., method_name: builtins.str=..., type: global___DataType.ValueType=..., method_parameters: collections.abc.Mapping[builtins.str, google.protobuf.any_pb2.Any] | None=..., file_extension: builtins.str=..., tags: collections.abc.Iterable[builtins.str] | None=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['component_name', b'component_name', 'component_type', b'component_type', 'file_extension', b'file_extension', 'method_name', b'method_name', 'method_parameters', b'method_parameters', 'tags', b'tags', 'type', b'type']) -> None:
+        ...
+global___DataCaptureMetadata = DataCaptureMetadata
+
+@typing.final
+class DataCaptureUploadMetadata(google.protobuf.message.Message):
+    """DataCaptureUploadMetadata contains the metadata for streaming binary (image + file) data."""
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    UPLOAD_METADATA_FIELD_NUMBER: builtins.int
+    SENSOR_METADATA_FIELD_NUMBER: builtins.int
+
+    @property
+    def upload_metadata(self) -> global___UploadMetadata:
+        ...
+
+    @property
+    def sensor_metadata(self) -> global___SensorMetadata:
+        ...
+
+    def __init__(self, *, upload_metadata: global___UploadMetadata | None=..., sensor_metadata: global___SensorMetadata | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['sensor_metadata', b'sensor_metadata', 'upload_metadata', b'upload_metadata']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['sensor_metadata', b'sensor_metadata', 'upload_metadata', b'upload_metadata']) -> None:
+        ...
+global___DataCaptureUploadMetadata = DataCaptureUploadMetadata

viam/gen/app/mlinference/v1/ml_inference_grpc.py

@@ -0,0 +1,28 @@
+import abc
+import typing
+import grpclib.const
+import grpclib.client
+import grpclib.exceptions
+if typing.TYPE_CHECKING:
+    import grpclib.server
+from .... import app
+from .... import service
+
+class MLInferenceServiceBase(abc.ABC):
+
+    @abc.abstractmethod
+    async def GetInference(self, stream: 'grpclib.server.Stream[app.mlinference.v1.ml_inference_pb2.GetInferenceRequest, app.mlinference.v1.ml_inference_pb2.GetInferenceResponse]') -> None:
+        pass
+
+    def __mapping__(self) -> typing.Dict[str, grpclib.const.Handler]:
+        return {'/viam.app.mlinference.v1.MLInferenceService/GetInference': grpclib.const.Handler(self.GetInference, grpclib.const.Cardinality.UNARY_UNARY, app.mlinference.v1.ml_inference_pb2.GetInferenceRequest, app.mlinference.v1.ml_inference_pb2.GetInferenceResponse)}
+
+class UnimplementedMLInferenceServiceBase(MLInferenceServiceBase):
+
+    async def GetInference(self, stream: 'grpclib.server.Stream[app.mlinference.v1.ml_inference_pb2.GetInferenceRequest, app.mlinference.v1.ml_inference_pb2.GetInferenceResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+class MLInferenceServiceStub:
+
+    def __init__(self, channel: grpclib.client.Channel) -> None:
+        self.GetInference = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mlinference.v1.MLInferenceService/GetInference', app.mlinference.v1.ml_inference_pb2.GetInferenceRequest, app.mlinference.v1.ml_inference_pb2.GetInferenceResponse)

viam/gen/app/mlinference/v1/ml_inference_pb2.py

@@ -0,0 +1,23 @@
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(_runtime_version.Domain.PUBLIC, 5, 29, 2, '', 'app/mlinference/v1/ml_inference.proto')
+_sym_db = _symbol_database.Default()
+from ....app.data.v1 import data_pb2 as app_dot_data_dot_v1_dot_data__pb2
+from ....service.mlmodel.v1 import mlmodel_pb2 as service_dot_mlmodel_dot_v1_dot_mlmodel__pb2
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%app/mlinference/v1/ml_inference.proto\x12\x17viam.app.mlinference.v1\x1a\x16app/data/v1/data.proto\x1a service/mlmodel/v1/mlmodel.proto"\xfb\x01\n\x13GetInferenceRequest\x12(\n\x10registry_item_id\x18\x01 \x01(\tR\x0eregistryItemId\x122\n\x15registry_item_version\x18\x02 \x01(\tR\x13registryItemVersion\x127\n\tbinary_id\x18\x03 \x01(\x0b2\x1a.viam.app.data.v1.BinaryIDR\x08binaryId\x12$\n\x0ebinary_data_id\x18\x05 \x01(\tR\x0cbinaryDataId\x12\'\n\x0forganization_id\x18\x04 \x01(\tR\x0eorganizationId"\xa4\x01\n\x14GetInferenceResponse\x12K\n\x0eoutput_tensors\x18\x01 \x01(\x0b2$.viam.service.mlmodel.v1.FlatTensorsR\routputTensors\x12?\n\x0bannotations\x18\x02 \x01(\x0b2\x1d.viam.app.data.v1.AnnotationsR\x0bannotations2\x81\x01\n\x12MLInferenceService\x12k\n\x0cGetInference\x12,.viam.app.mlinference.v1.GetInferenceRequest\x1a-.viam.app.mlinference.v1.GetInferenceResponseB$Z"go.viam.com/api/app/mlinference/v1b\x06proto3')
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'app.mlinference.v1.ml_inference_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals['DESCRIPTOR']._loaded_options = None
+    _globals['DESCRIPTOR']._serialized_options = b'Z"go.viam.com/api/app/mlinference/v1'
+    _globals['_GETINFERENCEREQUEST']._serialized_start = 125
+    _globals['_GETINFERENCEREQUEST']._serialized_end = 376
+    _globals['_GETINFERENCERESPONSE']._serialized_start = 379
+    _globals['_GETINFERENCERESPONSE']._serialized_end = 543
+    _globals['_MLINFERENCESERVICE']._serialized_start = 546
+    _globals['_MLINFERENCESERVICE']._serialized_end = 675

viam/gen/app/mlinference/v1/ml_inference_pb2.pyi

@@ -0,0 +1,63 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+from .... import app
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+from .... import service
+import typing
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class GetInferenceRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    REGISTRY_ITEM_ID_FIELD_NUMBER: builtins.int
+    REGISTRY_ITEM_VERSION_FIELD_NUMBER: builtins.int
+    BINARY_ID_FIELD_NUMBER: builtins.int
+    BINARY_DATA_ID_FIELD_NUMBER: builtins.int
+    ORGANIZATION_ID_FIELD_NUMBER: builtins.int
+    registry_item_id: builtins.str
+    'The model framework and model type are inferred from the ML model registry item;\n    For valid model types (classification, detections) we will return the formatted\n    labels or annotations from the associated inference outputs.\n    '
+    registry_item_version: builtins.str
+    binary_data_id: builtins.str
+    organization_id: builtins.str
+
+    @property
+    def binary_id(self) -> app.data.v1.data_pb2.BinaryID:
+        ...
+
+    def __init__(self, *, registry_item_id: builtins.str=..., registry_item_version: builtins.str=..., binary_id: app.data.v1.data_pb2.BinaryID | None=..., binary_data_id: builtins.str=..., organization_id: builtins.str=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['binary_id', b'binary_id']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['binary_data_id', b'binary_data_id', 'binary_id', b'binary_id', 'organization_id', b'organization_id', 'registry_item_id', b'registry_item_id', 'registry_item_version', b'registry_item_version']) -> None:
+        ...
+global___GetInferenceRequest = GetInferenceRequest
+
+@typing.final
+class GetInferenceResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    OUTPUT_TENSORS_FIELD_NUMBER: builtins.int
+    ANNOTATIONS_FIELD_NUMBER: builtins.int
+
+    @property
+    def output_tensors(self) -> service.mlmodel.v1.mlmodel_pb2.FlatTensors:
+        ...
+
+    @property
+    def annotations(self) -> app.data.v1.data_pb2.Annotations:
+        ...
+
+    def __init__(self, *, output_tensors: service.mlmodel.v1.mlmodel_pb2.FlatTensors | None=..., annotations: app.data.v1.data_pb2.Annotations | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['annotations', b'annotations', 'output_tensors', b'output_tensors']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['annotations', b'annotations', 'output_tensors', b'output_tensors']) -> None:
+        ...
+global___GetInferenceResponse = GetInferenceResponse

viam/gen/app/mltraining/v1/ml_training_grpc.py

@@ -0,0 +1,78 @@
+import abc
+import typing
+import grpclib.const
+import grpclib.client
+import grpclib.exceptions
+if typing.TYPE_CHECKING:
+    import grpclib.server
+import google.protobuf.timestamp_pb2
+import google.rpc.status_pb2
+from .... import tagger
+from .... import app
+
+class MLTrainingServiceBase(abc.ABC):
+
+    @abc.abstractmethod
+    async def SubmitTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def SubmitCustomTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def GetTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def ListTrainingJobs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def CancelTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def DeleteCompletedTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse]') -> None:
+        pass
+
+    @abc.abstractmethod
+    async def GetTrainingJobLogs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse]') -> None:
+        pass
+
+    def __mapping__(self) -> typing.Dict[str, grpclib.const.Handler]:
+        return {'/viam.app.mltraining.v1.MLTrainingService/SubmitTrainingJob': grpclib.const.Handler(self.SubmitTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/SubmitCustomTrainingJob': grpclib.const.Handler(self.SubmitCustomTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJob': grpclib.const.Handler(self.GetTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/ListTrainingJobs': grpclib.const.Handler(self.ListTrainingJobs, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse), '/viam.app.mltraining.v1.MLTrainingService/CancelTrainingJob': grpclib.const.Handler(self.CancelTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/DeleteCompletedTrainingJob': grpclib.const.Handler(self.DeleteCompletedTrainingJob, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse), '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJobLogs': grpclib.const.Handler(self.GetTrainingJobLogs, grpclib.const.Cardinality.UNARY_UNARY, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse)}
+
+class UnimplementedMLTrainingServiceBase(MLTrainingServiceBase):
+
+    async def SubmitTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+    async def SubmitCustomTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+    async def GetTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+    async def ListTrainingJobs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+    async def CancelTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+    async def DeleteCompletedTrainingJob(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+    async def GetTrainingJobLogs(self, stream: 'grpclib.server.Stream[app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse]') -> None:
+        raise grpclib.exceptions.GRPCError(grpclib.const.Status.UNIMPLEMENTED)
+
+class MLTrainingServiceStub:
+
+    def __init__(self, channel: grpclib.client.Channel) -> None:
+        self.SubmitTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/SubmitTrainingJob', app.mltraining.v1.ml_training_pb2.SubmitTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitTrainingJobResponse)
+        self.SubmitCustomTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/SubmitCustomTrainingJob', app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobRequest, app.mltraining.v1.ml_training_pb2.SubmitCustomTrainingJobResponse)
+        self.GetTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJob', app.mltraining.v1.ml_training_pb2.GetTrainingJobRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobResponse)
+        self.ListTrainingJobs = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/ListTrainingJobs', app.mltraining.v1.ml_training_pb2.ListTrainingJobsRequest, app.mltraining.v1.ml_training_pb2.ListTrainingJobsResponse)
+        self.CancelTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/CancelTrainingJob', app.mltraining.v1.ml_training_pb2.CancelTrainingJobRequest, app.mltraining.v1.ml_training_pb2.CancelTrainingJobResponse)
+        self.DeleteCompletedTrainingJob = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/DeleteCompletedTrainingJob', app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobRequest, app.mltraining.v1.ml_training_pb2.DeleteCompletedTrainingJobResponse)
+        self.GetTrainingJobLogs = grpclib.client.UnaryUnaryMethod(channel, '/viam.app.mltraining.v1.MLTrainingService/GetTrainingJobLogs', app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsRequest, app.mltraining.v1.ml_training_pb2.GetTrainingJobLogsResponse)