frogml-core 0.0.116__py3-none-any.whl → 0.0.118__py3-none-any.whl

_proto/qwak/builds/builds_pb2_grpc.py

@@ -0,0 +1,237 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.builds import builds_pb2 as qwak_dot_builds_dot_builds__pb2
+
+
+class BuildsManagementServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.RegisterBuild = channel.unary_unary(
+                '/qwak.builds.management.BuildsManagementService/RegisterBuild',
+                request_serializer=qwak_dot_builds_dot_builds__pb2.RegisterBuildRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__pb2.RegisterBuildResponse.FromString,
+                )
+        self.UpdateBuildStatus = channel.unary_unary(
+                '/qwak.builds.management.BuildsManagementService/UpdateBuildStatus',
+                request_serializer=qwak_dot_builds_dot_builds__pb2.UpdateBuildStatusRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__pb2.UpdateBuildStatusResponse.FromString,
+                )
+        self.RegisterModelSchema = channel.unary_unary(
+                '/qwak.builds.management.BuildsManagementService/RegisterModelSchema',
+                request_serializer=qwak_dot_builds_dot_builds__pb2.RegisterModelSchemaRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__pb2.RegisterModelSchemaResponse.FromString,
+                )
+        self.RegisterExperimentTrackingValues = channel.unary_unary(
+                '/qwak.builds.management.BuildsManagementService/RegisterExperimentTrackingValues',
+                request_serializer=qwak_dot_builds_dot_builds__pb2.RegisterExperimentTrackingValuesRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__pb2.RegisterExperimentTrackingValuesResponse.FromString,
+                )
+        self.GetBuild = channel.unary_unary(
+                '/qwak.builds.management.BuildsManagementService/GetBuild',
+                request_serializer=qwak_dot_builds_dot_builds__pb2.GetBuildRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__pb2.GetBuildResponse.FromString,
+                )
+        self.ListBuilds = channel.unary_unary(
+                '/qwak.builds.management.BuildsManagementService/ListBuilds',
+                request_serializer=qwak_dot_builds_dot_builds__pb2.ListBuildsRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_builds__pb2.ListBuildsResponse.FromString,
+                )
+
+
+class BuildsManagementServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def RegisterBuild(self, request, context):
+        """Register a Qwak build
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UpdateBuildStatus(self, request, context):
+        """Update Build Status
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RegisterModelSchema(self, request, context):
+        """Register model schema (interface) as part of a build
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RegisterExperimentTrackingValues(self, request, context):
+        """Register Qwak metrics and params
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetBuild(self, request, context):
+        """Get build
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ListBuilds(self, request, context):
+        """List registered builds for the given branch
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_BuildsManagementServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'RegisterBuild': grpc.unary_unary_rpc_method_handler(
+                    servicer.RegisterBuild,
+                    request_deserializer=qwak_dot_builds_dot_builds__pb2.RegisterBuildRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__pb2.RegisterBuildResponse.SerializeToString,
+            ),
+            'UpdateBuildStatus': grpc.unary_unary_rpc_method_handler(
+                    servicer.UpdateBuildStatus,
+                    request_deserializer=qwak_dot_builds_dot_builds__pb2.UpdateBuildStatusRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__pb2.UpdateBuildStatusResponse.SerializeToString,
+            ),
+            'RegisterModelSchema': grpc.unary_unary_rpc_method_handler(
+                    servicer.RegisterModelSchema,
+                    request_deserializer=qwak_dot_builds_dot_builds__pb2.RegisterModelSchemaRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__pb2.RegisterModelSchemaResponse.SerializeToString,
+            ),
+            'RegisterExperimentTrackingValues': grpc.unary_unary_rpc_method_handler(
+                    servicer.RegisterExperimentTrackingValues,
+                    request_deserializer=qwak_dot_builds_dot_builds__pb2.RegisterExperimentTrackingValuesRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__pb2.RegisterExperimentTrackingValuesResponse.SerializeToString,
+            ),
+            'GetBuild': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetBuild,
+                    request_deserializer=qwak_dot_builds_dot_builds__pb2.GetBuildRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__pb2.GetBuildResponse.SerializeToString,
+            ),
+            'ListBuilds': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListBuilds,
+                    request_deserializer=qwak_dot_builds_dot_builds__pb2.ListBuildsRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_builds__pb2.ListBuildsResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.builds.management.BuildsManagementService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class BuildsManagementService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def RegisterBuild(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.builds.management.BuildsManagementService/RegisterBuild',
+            qwak_dot_builds_dot_builds__pb2.RegisterBuildRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__pb2.RegisterBuildResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def UpdateBuildStatus(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.builds.management.BuildsManagementService/UpdateBuildStatus',
+            qwak_dot_builds_dot_builds__pb2.UpdateBuildStatusRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__pb2.UpdateBuildStatusResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def RegisterModelSchema(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.builds.management.BuildsManagementService/RegisterModelSchema',
+            qwak_dot_builds_dot_builds__pb2.RegisterModelSchemaRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__pb2.RegisterModelSchemaResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def RegisterExperimentTrackingValues(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.builds.management.BuildsManagementService/RegisterExperimentTrackingValues',
+            qwak_dot_builds_dot_builds__pb2.RegisterExperimentTrackingValuesRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__pb2.RegisterExperimentTrackingValuesResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetBuild(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.builds.management.BuildsManagementService/GetBuild',
+            qwak_dot_builds_dot_builds__pb2.GetBuildRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__pb2.GetBuildResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ListBuilds(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.builds.management.BuildsManagementService/ListBuilds',
+            qwak_dot_builds_dot_builds__pb2.ListBuildsRequest.SerializeToString,
+            qwak_dot_builds_dot_builds__pb2.ListBuildsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_proto/qwak/builds/internal_builds_orchestrator_service_pb2.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.builds/internal_builds_orchestrator_service.proto
+"""Generated protocol buffer code."""
+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
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml._proto.qwak.builds import build_pb2 as qwak_dot_builds_dot_build__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6qwak/builds/internal_builds_orchestrator_service.proto\x12\x18qwak.builds.orchestrator\x1a\x17qwak/builds/build.proto\"\x8e\x01\n\x10\x42uildInitDetails\x12=\n\nbuild_spec\x18\x01 \x01(\x0b\x32).qwak.builds.orchestrator.RemoteBuildSpec\x12;\n\tinitiator\x18\x02 \x01(\x0b\x32(.qwak.builds.orchestrator.BuildInitiator\"c\n\x19InternalBuildModelRequest\x12\x46\n\x12\x62uild_init_details\x18\x01 \x01(\x0b\x32*.qwak.builds.orchestrator.BuildInitDetails\"\x1c\n\x1aInternalBuildModelResponse2\x9c\x01\n!InternalBuildsOrchestratorService\x12w\n\nBuildModel\x12\x33.qwak.builds.orchestrator.InternalBuildModelRequest\x1a\x34.qwak.builds.orchestrator.InternalBuildModelResponseB\'\n#com.qwak.ai.builds.orchestrator.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.builds.internal_builds_orchestrator_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n#com.qwak.ai.builds.orchestrator.apiP\001'
+  _globals['_BUILDINITDETAILS']._serialized_start=110
+  _globals['_BUILDINITDETAILS']._serialized_end=252
+  _globals['_INTERNALBUILDMODELREQUEST']._serialized_start=254
+  _globals['_INTERNALBUILDMODELREQUEST']._serialized_end=353
+  _globals['_INTERNALBUILDMODELRESPONSE']._serialized_start=355
+  _globals['_INTERNALBUILDMODELRESPONSE']._serialized_end=383
+  _globals['_INTERNALBUILDSORCHESTRATORSERVICE']._serialized_start=386
+  _globals['_INTERNALBUILDSORCHESTRATORSERVICE']._serialized_end=542
+# @@protoc_insertion_point(module_scope)

_proto/qwak/builds/internal_builds_orchestrator_service_pb2.pyi

@@ -0,0 +1,64 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import frogml._proto.qwak.builds.build_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class BuildInitDetails(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_SPEC_FIELD_NUMBER: builtins.int
+    INITIATOR_FIELD_NUMBER: builtins.int
+    @property
+    def build_spec(self) -> frogml._proto.qwak.builds.build_pb2.RemoteBuildSpec:
+        """Requested build model spec"""
+    @property
+    def initiator(self) -> frogml._proto.qwak.builds.build_pb2.BuildInitiator:
+        """override to the initiator of the build"""
+    def __init__(
+        self,
+        *,
+        build_spec: frogml._proto.qwak.builds.build_pb2.RemoteBuildSpec | None = ...,
+        initiator: frogml._proto.qwak.builds.build_pb2.BuildInitiator | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "initiator", b"initiator"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "initiator", b"initiator"]) -> None: ...
+
+global___BuildInitDetails = BuildInitDetails
+
+class InternalBuildModelRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_INIT_DETAILS_FIELD_NUMBER: builtins.int
+    @property
+    def build_init_details(self) -> global___BuildInitDetails:
+        """The request of build"""
+    def __init__(
+        self,
+        *,
+        build_init_details: global___BuildInitDetails | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["build_init_details", b"build_init_details"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_init_details", b"build_init_details"]) -> None: ...
+
+global___InternalBuildModelRequest = InternalBuildModelRequest
+
+class InternalBuildModelResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___InternalBuildModelResponse = InternalBuildModelResponse

_proto/qwak/builds/internal_builds_orchestrator_service_pb2_grpc.py

@@ -0,0 +1,67 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.builds import internal_builds_orchestrator_service_pb2 as qwak_dot_builds_dot_internal__builds__orchestrator__service__pb2
+
+
+class InternalBuildsOrchestratorServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.BuildModel = channel.unary_unary(
+                '/qwak.builds.orchestrator.InternalBuildsOrchestratorService/BuildModel',
+                request_serializer=qwak_dot_builds_dot_internal__builds__orchestrator__service__pb2.InternalBuildModelRequest.SerializeToString,
+                response_deserializer=qwak_dot_builds_dot_internal__builds__orchestrator__service__pb2.InternalBuildModelResponse.FromString,
+                )
+
+
+class InternalBuildsOrchestratorServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def BuildModel(self, request, context):
+        """Build a serving model image
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_InternalBuildsOrchestratorServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'BuildModel': grpc.unary_unary_rpc_method_handler(
+                    servicer.BuildModel,
+                    request_deserializer=qwak_dot_builds_dot_internal__builds__orchestrator__service__pb2.InternalBuildModelRequest.FromString,
+                    response_serializer=qwak_dot_builds_dot_internal__builds__orchestrator__service__pb2.InternalBuildModelResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.builds.orchestrator.InternalBuildsOrchestratorService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class InternalBuildsOrchestratorService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def BuildModel(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.builds.orchestrator.InternalBuildsOrchestratorService/BuildModel',
+            qwak_dot_builds_dot_internal__builds__orchestrator__service__pb2.InternalBuildModelRequest.SerializeToString,
+            qwak_dot_builds_dot_internal__builds__orchestrator__service__pb2.InternalBuildModelResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_proto/qwak/data_versioning/data_versioning_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.data_versioning/data_versioning.proto
+"""Generated protocol buffer code."""
+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
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*qwak/data_versioning/data_versioning.proto\x12\x1fqwak.management.data_versioning\"|\n\x0b\x44\x61taTagSpec\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12\x16\n\x0e\x65xtension_type\x18\x04 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\x12\x0c\n\x04path\x18\x06 \x01(\t\"G\n\rDataTagFilter\x12\x16\n\x0ctag_contains\x18\x01 \x01(\tH\x00\x12\x14\n\ntag_prefix\x18\x02 \x01(\tH\x00\x42\x08\n\x06\x66ilterB/\n+com.qwak.ai.management.model_versioning.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.data_versioning.data_versioning_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.management.model_versioning.apiP\001'
+  _globals['_DATATAGSPEC']._serialized_start=79
+  _globals['_DATATAGSPEC']._serialized_end=203
+  _globals['_DATATAGFILTER']._serialized_start=205
+  _globals['_DATATAGFILTER']._serialized_end=276
+# @@protoc_insertion_point(module_scope)

_proto/qwak/data_versioning/data_versioning_service_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.data_versioning/data_versioning_service.proto
+"""Generated protocol buffer code."""
+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
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml._proto.qwak.data_versioning import data_versioning_pb2 as qwak_dot_data__versioning_dot_data__versioning__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n2qwak/data_versioning/data_versioning_service.proto\x12\x1fqwak.management.data_versioning\x1a*qwak/data_versioning/data_versioning.proto\"]\n\x16RegisterDataTagRequest\x12\x43\n\rdata_tag_spec\x18\x01 \x01(\x0b\x32,.qwak.management.data_versioning.DataTagSpec\"\x19\n\x17RegisterDataTagResponse\"}\n\x17GetModelDataTagsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12>\n\x06\x66ilter\x18\x03 \x01(\x0b\x32..qwak.management.data_versioning.DataTagFilter\"[\n\x18GetModelDataTagsResponse\x12?\n\tdata_tags\x18\x01 \x03(\x0b\x32,.qwak.management.data_versioning.DataTagSpec2\xb2\x02\n\x1f\x44\x61taVersioningManagementService\x12\x84\x01\n\x0fRegisterDataTag\x12\x37.qwak.management.data_versioning.RegisterDataTagRequest\x1a\x38.qwak.management.data_versioning.RegisterDataTagResponse\x12\x87\x01\n\x10GetModelDataTags\x12\x38.qwak.management.data_versioning.GetModelDataTagsRequest\x1a\x39.qwak.management.data_versioning.GetModelDataTagsResponseB/\n+com.qwak.ai.management.model_versioning.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.data_versioning.data_versioning_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.management.model_versioning.apiP\001'
+  _globals['_REGISTERDATATAGREQUEST']._serialized_start=131
+  _globals['_REGISTERDATATAGREQUEST']._serialized_end=224
+  _globals['_REGISTERDATATAGRESPONSE']._serialized_start=226
+  _globals['_REGISTERDATATAGRESPONSE']._serialized_end=251
+  _globals['_GETMODELDATATAGSREQUEST']._serialized_start=253
+  _globals['_GETMODELDATATAGSREQUEST']._serialized_end=378
+  _globals['_GETMODELDATATAGSRESPONSE']._serialized_start=380
+  _globals['_GETMODELDATATAGSRESPONSE']._serialized_end=471
+  _globals['_DATAVERSIONINGMANAGEMENTSERVICE']._serialized_start=474
+  _globals['_DATAVERSIONINGMANAGEMENTSERVICE']._serialized_end=780
+# @@protoc_insertion_point(module_scope)

_proto/qwak/data_versioning/data_versioning_service_pb2.pyi

@@ -0,0 +1,83 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+import frogml._proto.qwak.data_versioning.data_versioning_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class RegisterDataTagRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_TAG_SPEC_FIELD_NUMBER: builtins.int
+    @property
+    def data_tag_spec(self) -> frogml._proto.qwak.data_versioning.data_versioning_pb2.DataTagSpec: ...
+    def __init__(
+        self,
+        *,
+        data_tag_spec: frogml._proto.qwak.data_versioning.data_versioning_pb2.DataTagSpec | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["data_tag_spec", b"data_tag_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data_tag_spec", b"data_tag_spec"]) -> None: ...
+
+global___RegisterDataTagRequest = RegisterDataTagRequest
+
+class RegisterDataTagResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___RegisterDataTagResponse = RegisterDataTagResponse
+
+class GetModelDataTagsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    BUILD_ID_FIELD_NUMBER: builtins.int
+    FILTER_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    """The model id to get data tags for"""
+    build_id: builtins.str
+    """The build id to get data tags for"""
+    @property
+    def filter(self) -> frogml._proto.qwak.data_versioning.data_versioning_pb2.DataTagFilter:
+        """The filter to apply to the data tags"""
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        build_id: builtins.str = ...,
+        filter: frogml._proto.qwak.data_versioning.data_versioning_pb2.DataTagFilter | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["filter", b"filter"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "filter", b"filter", "model_id", b"model_id"]) -> None: ...
+
+global___GetModelDataTagsRequest = GetModelDataTagsRequest
+
+class GetModelDataTagsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_TAGS_FIELD_NUMBER: builtins.int
+    @property
+    def data_tags(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.data_versioning.data_versioning_pb2.DataTagSpec]: ...
+    def __init__(
+        self,
+        *,
+        data_tags: collections.abc.Iterable[qwak.data_versioning.data_versioning_pb2.DataTagSpec] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data_tags", b"data_tags"]) -> None: ...
+
+global___GetModelDataTagsResponse = GetModelDataTagsResponse

_proto/qwak/data_versioning/data_versioning_service_pb2_grpc.py

@@ -0,0 +1,101 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.data_versioning import data_versioning_service_pb2 as qwak_dot_data__versioning_dot_data__versioning__service__pb2
+
+
+class DataVersioningManagementServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.RegisterDataTag = channel.unary_unary(
+                '/qwak.management.data_versioning.DataVersioningManagementService/RegisterDataTag',
+                request_serializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.RegisterDataTagRequest.SerializeToString,
+                response_deserializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.RegisterDataTagResponse.FromString,
+                )
+        self.GetModelDataTags = channel.unary_unary(
+                '/qwak.management.data_versioning.DataVersioningManagementService/GetModelDataTags',
+                request_serializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.GetModelDataTagsRequest.SerializeToString,
+                response_deserializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.GetModelDataTagsResponse.FromString,
+                )
+
+
+class DataVersioningManagementServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def RegisterDataTag(self, request, context):
+        """Register data tag
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetModelDataTags(self, request, context):
+        """Get the models tags
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_DataVersioningManagementServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'RegisterDataTag': grpc.unary_unary_rpc_method_handler(
+                    servicer.RegisterDataTag,
+                    request_deserializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.RegisterDataTagRequest.FromString,
+                    response_serializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.RegisterDataTagResponse.SerializeToString,
+            ),
+            'GetModelDataTags': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetModelDataTags,
+                    request_deserializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.GetModelDataTagsRequest.FromString,
+                    response_serializer=qwak_dot_data__versioning_dot_data__versioning__service__pb2.GetModelDataTagsResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.management.data_versioning.DataVersioningManagementService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class DataVersioningManagementService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def RegisterDataTag(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.management.data_versioning.DataVersioningManagementService/RegisterDataTag',
+            qwak_dot_data__versioning_dot_data__versioning__service__pb2.RegisterDataTagRequest.SerializeToString,
+            qwak_dot_data__versioning_dot_data__versioning__service__pb2.RegisterDataTagResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetModelDataTags(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.management.data_versioning.DataVersioningManagementService/GetModelDataTags',
+            qwak_dot_data__versioning_dot_data__versioning__service__pb2.GetModelDataTagsRequest.SerializeToString,
+            qwak_dot_data__versioning_dot_data__versioning__service__pb2.GetModelDataTagsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)