qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/builds/builds_pb2.pyi

@@ -631,32 +631,6 @@ class ParameterCategory(google.protobuf.message.Message):
 
 global___ParameterCategory = ParameterCategory
 
-class RegisterBuildRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_SPEC_FIELD_NUMBER: builtins.int
-    MODEL_SCHEMA_FIELD_NUMBER: builtins.int
-    QWAK_CALLING_USER_ID_FIELD_NUMBER: builtins.int
-    @property
-    def build_spec(self) -> global___BuildSpec:
-        """Build spec to register"""
-    @property
-    def model_schema(self) -> global___ModelSchema:
-        """Deprecated - persist model scheme"""
-    qwak_calling_user_id: builtins.str
-    """User id of the caller - used to pass the user id whom called the build model process through fitness service"""
-    def __init__(
-        self,
-        *,
-        build_spec: global___BuildSpec | None = ...,
-        model_schema: global___ModelSchema | None = ...,
-        qwak_calling_user_id: builtins.str = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "model_schema", b"model_schema"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "model_schema", b"model_schema", "qwak_calling_user_id", b"qwak_calling_user_id"]) -> None: ...
-
-global___RegisterBuildRequest = RegisterBuildRequest
-
 class RegisterModelSchemaRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -825,17 +799,21 @@ class BuildModelUri(google.protobuf.message.Message):
     URI_FIELD_NUMBER: builtins.int
     GIT_BRANCH_FIELD_NUMBER: builtins.int
     MAIN_DIR_FIELD_NUMBER: builtins.int
+    DEPENDENCY_REQUIRED_FOLDERS_FIELD_NUMBER: builtins.int
     uri: builtins.str
     git_branch: builtins.str
     main_dir: builtins.str
+    @property
+    def dependency_required_folders(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
     def __init__(
         self,
         *,
         uri: builtins.str = ...,
         git_branch: builtins.str = ...,
         main_dir: builtins.str = ...,
+        dependency_required_folders: collections.abc.Iterable[builtins.str] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["git_branch", b"git_branch", "main_dir", b"main_dir", "uri", b"uri"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["dependency_required_folders", b"dependency_required_folders", "git_branch", b"git_branch", "main_dir", b"main_dir", "uri", b"uri"]) -> None: ...
 
 global___BuildModelUri = BuildModelUri
 
@@ -1100,168 +1078,3 @@ class Param(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
 
 global___Param = Param
-
-class RegisterModelSchemaResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    def __init__(
-        self,
-    ) -> None: ...
-
-global___RegisterModelSchemaResponse = RegisterModelSchemaResponse
-
-class RegisterBuildResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_ID_FIELD_NUMBER: builtins.int
-    BUILD_STATUS_FIELD_NUMBER: builtins.int
-    build_id: builtins.str
-    """Build id of the registered build"""
-    build_status: global___BuildStatus.ValueType
-    """Build status of the registered build. Will return IN_PROGRESS"""
-    def __init__(
-        self,
-        *,
-        build_id: builtins.str = ...,
-        build_status: global___BuildStatus.ValueType = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "build_status", b"build_status"]) -> None: ...
-
-global___RegisterBuildResponse = RegisterBuildResponse
-
-class UpdateBuildStatusRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_ID_FIELD_NUMBER: builtins.int
-    BUILD_STATUS_FIELD_NUMBER: builtins.int
-    QWAK_CALLING_USER_ID_FIELD_NUMBER: builtins.int
-    AVAILABLE_ENVIRONMENTS_FIELD_NUMBER: builtins.int
-    build_id: builtins.str
-    """Build id to update"""
-    build_status: global___BuildStatus.ValueType
-    """Build status to update to"""
-    qwak_calling_user_id: builtins.str
-    """User id of the caller - used to pass the user id whom called the build model process through fitness service"""
-    @property
-    def available_environments(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
-        """Environments the build is available on"""
-    def __init__(
-        self,
-        *,
-        build_id: builtins.str = ...,
-        build_status: global___BuildStatus.ValueType = ...,
-        qwak_calling_user_id: builtins.str = ...,
-        available_environments: collections.abc.Iterable[builtins.str] | None = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["available_environments", b"available_environments", "build_id", b"build_id", "build_status", b"build_status", "qwak_calling_user_id", b"qwak_calling_user_id"]) -> None: ...
-
-global___UpdateBuildStatusRequest = UpdateBuildStatusRequest
-
-class UpdateBuildStatusResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    def __init__(
-        self,
-    ) -> None: ...
-
-global___UpdateBuildStatusResponse = UpdateBuildStatusResponse
-
-class UpdateBuildBranchRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_ID_FIELD_NUMBER: builtins.int
-    BRANCH_NAME_FIELD_NUMBER: builtins.int
-    build_id: builtins.str
-    """Build id to update"""
-    branch_name: builtins.str
-    """New branch name"""
-    def __init__(
-        self,
-        *,
-        build_id: builtins.str = ...,
-        branch_name: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["branch_name", b"branch_name", "build_id", b"build_id"]) -> None: ...
-
-global___UpdateBuildBranchRequest = UpdateBuildBranchRequest
-
-class UpdateBuildBranchResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    def __init__(
-        self,
-    ) -> None: ...
-
-global___UpdateBuildBranchResponse = UpdateBuildBranchResponse
-
-class GetBuildRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_ID_FIELD_NUMBER: builtins.int
-    build_id: builtins.str
-    """Build id of the requested build"""
-    def __init__(
-        self,
-        *,
-        build_id: builtins.str = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
-
-global___GetBuildRequest = GetBuildRequest
-
-class GetBuildResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILD_FIELD_NUMBER: builtins.int
-    @property
-    def build(self) -> global___Build:
-        """Requested build information"""
-    def __init__(
-        self,
-        *,
-        build: global___Build | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build", b"build"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build", b"build"]) -> None: ...
-
-global___GetBuildResponse = GetBuildResponse
-
-class ListBuildsRequest(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BRANCH_ID_FIELD_NUMBER: builtins.int
-    MODEL_UUID_FIELD_NUMBER: builtins.int
-    BUILD_IDS_FIELD_NUMBER: builtins.int
-    branch_id: builtins.str
-    """Branch id of the requested builds"""
-    model_uuid: builtins.str
-    """Model uuid"""
-    @property
-    def build_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
-        """Build ids"""
-    def __init__(
-        self,
-        *,
-        branch_id: builtins.str = ...,
-        model_uuid: builtins.str = ...,
-        build_ids: collections.abc.Iterable[builtins.str] | None = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["branch_id", b"branch_id", "build_ids", b"build_ids", "model_uuid", b"model_uuid"]) -> None: ...
-
-global___ListBuildsRequest = ListBuildsRequest
-
-class ListBuildsResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    BUILDS_FIELD_NUMBER: builtins.int
-    @property
-    def builds(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Build]:
-        """List of the branches' builds information"""
-    def __init__(
-        self,
-        *,
-        builds: collections.abc.Iterable[global___Build] | None = ...,
-    ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["builds", b"builds"]) -> None: ...
-
-global___ListBuildsResponse = ListBuildsResponse

_qwak_proto/qwak/builds/builds_pb2_grpc.py

@@ -2,236 +2,3 @@
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
 
-from _qwak_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)

_qwak_proto/qwak/builds/internal_builds_orchestrator_service_pb2.py

@@ -4,9 +4,8 @@
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 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()
@@ -17,43 +16,19 @@ from _qwak_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')
 
-
-
-_BUILDINITDETAILS = DESCRIPTOR.message_types_by_name['BuildInitDetails']
-_INTERNALBUILDMODELREQUEST = DESCRIPTOR.message_types_by_name['InternalBuildModelRequest']
-_INTERNALBUILDMODELRESPONSE = DESCRIPTOR.message_types_by_name['InternalBuildModelResponse']
-BuildInitDetails = _reflection.GeneratedProtocolMessageType('BuildInitDetails', (_message.Message,), {
-  'DESCRIPTOR' : _BUILDINITDETAILS,
-  '__module__' : 'qwak.builds.internal_builds_orchestrator_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.builds.orchestrator.BuildInitDetails)
-  })
-_sym_db.RegisterMessage(BuildInitDetails)
-
-InternalBuildModelRequest = _reflection.GeneratedProtocolMessageType('InternalBuildModelRequest', (_message.Message,), {
-  'DESCRIPTOR' : _INTERNALBUILDMODELREQUEST,
-  '__module__' : 'qwak.builds.internal_builds_orchestrator_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.builds.orchestrator.InternalBuildModelRequest)
-  })
-_sym_db.RegisterMessage(InternalBuildModelRequest)
-
-InternalBuildModelResponse = _reflection.GeneratedProtocolMessageType('InternalBuildModelResponse', (_message.Message,), {
-  'DESCRIPTOR' : _INTERNALBUILDMODELRESPONSE,
-  '__module__' : 'qwak.builds.internal_builds_orchestrator_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.builds.orchestrator.InternalBuildModelResponse)
-  })
-_sym_db.RegisterMessage(InternalBuildModelResponse)
-
-_INTERNALBUILDSORCHESTRATORSERVICE = DESCRIPTOR.services_by_name['InternalBuildsOrchestratorService']
+_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'
-  _BUILDINITDETAILS._serialized_start=110
-  _BUILDINITDETAILS._serialized_end=252
-  _INTERNALBUILDMODELREQUEST._serialized_start=254
-  _INTERNALBUILDMODELREQUEST._serialized_end=353
-  _INTERNALBUILDMODELRESPONSE._serialized_start=355
-  _INTERNALBUILDMODELRESPONSE._serialized_end=383
-  _INTERNALBUILDSORCHESTRATORSERVICE._serialized_start=386
-  _INTERNALBUILDSORCHESTRATORSERVICE._serialized_end=542
+  _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)

_qwak_proto/qwak/data_versioning/data_versioning_pb2.py

@@ -4,9 +4,8 @@
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 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()
@@ -14,22 +13,17 @@ _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(\tB/\n+com.qwak.ai.management.model_versioning.apiP\x01\x62\x06proto3')
-
-
-
-_DATATAGSPEC = DESCRIPTOR.message_types_by_name['DataTagSpec']
-DataTagSpec = _reflection.GeneratedProtocolMessageType('DataTagSpec', (_message.Message,), {
-  'DESCRIPTOR' : _DATATAGSPEC,
-  '__module__' : 'qwak.data_versioning.data_versioning_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.management.data_versioning.DataTagSpec)
-  })
-_sym_db.RegisterMessage(DataTagSpec)
+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'
-  _DATATAGSPEC._serialized_start=79
-  _DATATAGSPEC._serialized_end=203
+  _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)

_qwak_proto/qwak/data_versioning/data_versioning_pb2.pyi

@@ -48,3 +48,24 @@ class DataTagSpec(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "environment_id", b"environment_id", "extension_type", b"extension_type", "model_id", b"model_id", "path", b"path", "tag", b"tag"]) -> None: ...
 
 global___DataTagSpec = DataTagSpec
+
+class DataTagFilter(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TAG_CONTAINS_FIELD_NUMBER: builtins.int
+    TAG_PREFIX_FIELD_NUMBER: builtins.int
+    tag_contains: builtins.str
+    """Filter data tags by tag contains"""
+    tag_prefix: builtins.str
+    """Filter data tags by tag prefix"""
+    def __init__(
+        self,
+        *,
+        tag_contains: builtins.str = ...,
+        tag_prefix: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["filter", b"filter", "tag_contains", b"tag_contains", "tag_prefix", b"tag_prefix"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["filter", b"filter", "tag_contains", b"tag_contains", "tag_prefix", b"tag_prefix"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["filter", b"filter"]) -> typing_extensions.Literal["tag_contains", "tag_prefix"] | None: ...
+
+global___DataTagFilter = DataTagFilter