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

_qwak_proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py

@@ -44,6 +44,11 @@ class DataSourceServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesResponse.FromString,
                 )
+        self.ListDataSourcesByAccount = channel.unary_unary(
+                '/qwak.feature.store.sources.DataSourceService/ListDataSourcesByAccount',
+                request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.FromString,
+                )
         self.CreateDataSourceUploadURL = channel.unary_unary(
                 '/qwak.feature.store.sources.DataSourceService/CreateDataSourceUploadURL',
                 request_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.CreateDataSourceUploadURLRequest.SerializeToString,
@@ -101,6 +106,13 @@ class DataSourceServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListDataSourcesByAccount(self, request, context):
+        """List of registered data sources
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def CreateDataSourceUploadURL(self, request, context):
         """Create a url for uploading data source configuration
         """
@@ -148,6 +160,11 @@ def add_DataSourceServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesResponse.SerializeToString,
             ),
+            'ListDataSourcesByAccount': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListDataSourcesByAccount,
+                    request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.SerializeToString,
+            ),
             'CreateDataSourceUploadURL': grpc.unary_unary_rpc_method_handler(
                     servicer.CreateDataSourceUploadURL,
                     request_deserializer=qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.CreateDataSourceUploadURLRequest.FromString,
@@ -270,6 +287,23 @@ class DataSourceService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def ListDataSourcesByAccount(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.feature.store.sources.DataSourceService/ListDataSourcesByAccount',
+            qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountRequest.SerializeToString,
+            qwak_dot_feature__store_dot_sources_dot_data__source__service__pb2.ListDataSourcesByAccountResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def CreateDataSourceUploadURL(request,
             target,

_qwak_proto/qwak/feature_store/sources/streaming_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/feature_store/sources/streaming.proto
 """Generated protocol buffer code."""
-from google.protobuf.internal import enum_type_wrapper
 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,119 +15,9 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*qwak/feature_store/sources/streaming.proto\x12\x1aqwak.feature.store.sources\"\xc4\x01\n\x0fStreamingSource\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x42\n\x0bkafkaSource\x18\x03 \x01(\x0b\x32\'.qwak.feature.store.sources.KafkaSourceB\x02\x18\x01H\x00\x12\x42\n\rkafkaSourceV1\x18\x04 \x01(\x0b\x32).qwak.feature.store.sources.KafkaSourceV1H\x00\x42\x06\n\x04type\"\x9f\x01\n\x0bKafkaSource\x12\x18\n\x10\x62ootstrap_server\x18\x01 \x01(\t\x12\x16\n\x0e\x63onsumer_topic\x18\x02 \x01(\t\x12\x1c\n\x14username_secret_name\x18\x03 \x01(\t\x12\x1c\n\x14password_secret_name\x18\x04 \x01(\t\x12\x1e\n\x16\x63onsumer_configuration\x18\x05 \x01(\t:\x02\x18\x01\"\xbd\x04\n\rKafkaSourceV1\x12\x19\n\x11\x62ootstrap_servers\x18\x01 \x01(\t\x12\x10\n\x06\x61ssign\x18\x02 \x01(\tH\x00\x12\x13\n\tsubscribe\x18\x03 \x01(\tH\x00\x12\x1b\n\x11subscribe_pattern\x18\x04 \x01(\tH\x00\x12T\n\x0esecret_configs\x18\x07 \x03(\x0b\x32<.qwak.feature.store.sources.KafkaSourceV1.SecretConfigsEntry\x12^\n\x13passthrough_configs\x18\x08 \x03(\x0b\x32\x41.qwak.feature.store.sources.KafkaSourceV1.PassthroughConfigsEntry\x12\x44\n\x0f\x64\x65serialization\x18\t \x01(\x0b\x32+.qwak.feature.store.sources.Deserialization\x12I\n\x15\x61uthentication_method\x18\n \x01(\x0b\x32*.qwak.feature.store.sources.Authentication\x1a\x34\n\x12SecretConfigsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\x39\n\x17PassthroughConfigsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x15\n\x13topic_configuration\"\xd8\x01\n\x0e\x41uthentication\x12>\n\x12sasl_configuration\x18\x01 \x01(\x0b\x32 .qwak.feature.store.sources.SaslH\x00\x12@\n\x13plain_configuration\x18\x02 \x01(\x0b\x32!.qwak.feature.store.sources.PlainH\x00\x12<\n\x11ssl_configuration\x18\x03 \x01(\x0b\x32\x1f.qwak.feature.store.sources.SslH\x00\x42\x06\n\x04type\"\x05\n\x03Ssl\"\x07\n\x05Plain\"\xc4\x01\n\x04Sasl\x12\x17\n\x0fusername_secret\x18\x01 \x01(\t\x12\x17\n\x0fpassword_secret\x18\x02 \x01(\t\x12\x41\n\x0esasl_mechanism\x18\x03 \x01(\x0e\x32).qwak.feature.store.sources.SaslMechanism\x12G\n\x11security_protocol\x18\x04 \x01(\x0e\x32,.qwak.feature.store.sources.SecurityProtocol\"\xb9\x01\n\x0f\x44\x65serialization\x12M\n\x13\x63ustom_deserializer\x18\x01 \x01(\x0b\x32..qwak.feature.store.sources.CustomDeserializerH\x00\x12O\n\x14generic_deserializer\x18\x02 \x01(\x0b\x32/.qwak.feature.store.sources.GenericDeserializerH\x00\x42\x06\n\x04type\"B\n\x12\x43ustomDeserializer\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x15\n\rartifact_path\x18\x02 \x01(\t\"m\n\x13GenericDeserializer\x12\x46\n\x13\x64\x65serializer_format\x18\x01 \x01(\x0e\x32).qwak.feature.store.sources.MessageFormat\x12\x0e\n\x06schema\x18\x02 \x01(\t*<\n\rSaslMechanism\x12\x0f\n\x0bSCRAMSHA256\x10\x00\x12\x0f\n\x0bSCRAMSHA512\x10\x01\x12\t\n\x05PLAIN\x10\x02* \n\x10SecurityProtocol\x12\x0c\n\x08SASL_SSL\x10\x00*0\n\rMessageFormat\x12\x0b\n\x07INVALID\x10\x00\x12\x08\n\x04JSON\x10\x01\x12\x08\n\x04\x41VRO\x10\x02\x42X\n%com.qwak.ai.feature.store.sources.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
 
-_SASLMECHANISM = DESCRIPTOR.enum_types_by_name['SaslMechanism']
-SaslMechanism = enum_type_wrapper.EnumTypeWrapper(_SASLMECHANISM)
-_SECURITYPROTOCOL = DESCRIPTOR.enum_types_by_name['SecurityProtocol']
-SecurityProtocol = enum_type_wrapper.EnumTypeWrapper(_SECURITYPROTOCOL)
-_MESSAGEFORMAT = DESCRIPTOR.enum_types_by_name['MessageFormat']
-MessageFormat = enum_type_wrapper.EnumTypeWrapper(_MESSAGEFORMAT)
-SCRAMSHA256 = 0
-SCRAMSHA512 = 1
-PLAIN = 2
-SASL_SSL = 0
-INVALID = 0
-JSON = 1
-AVRO = 2
-
-
-_STREAMINGSOURCE = DESCRIPTOR.message_types_by_name['StreamingSource']
-_KAFKASOURCE = DESCRIPTOR.message_types_by_name['KafkaSource']
-_KAFKASOURCEV1 = DESCRIPTOR.message_types_by_name['KafkaSourceV1']
-_KAFKASOURCEV1_SECRETCONFIGSENTRY = _KAFKASOURCEV1.nested_types_by_name['SecretConfigsEntry']
-_KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY = _KAFKASOURCEV1.nested_types_by_name['PassthroughConfigsEntry']
-_AUTHENTICATION = DESCRIPTOR.message_types_by_name['Authentication']
-_SSL = DESCRIPTOR.message_types_by_name['Ssl']
-_PLAIN = DESCRIPTOR.message_types_by_name['Plain']
-_SASL = DESCRIPTOR.message_types_by_name['Sasl']
-_DESERIALIZATION = DESCRIPTOR.message_types_by_name['Deserialization']
-_CUSTOMDESERIALIZER = DESCRIPTOR.message_types_by_name['CustomDeserializer']
-_GENERICDESERIALIZER = DESCRIPTOR.message_types_by_name['GenericDeserializer']
-StreamingSource = _reflection.GeneratedProtocolMessageType('StreamingSource', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGSOURCE,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.StreamingSource)
-  })
-_sym_db.RegisterMessage(StreamingSource)
-
-KafkaSource = _reflection.GeneratedProtocolMessageType('KafkaSource', (_message.Message,), {
-  'DESCRIPTOR' : _KAFKASOURCE,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSource)
-  })
-_sym_db.RegisterMessage(KafkaSource)
-
-KafkaSourceV1 = _reflection.GeneratedProtocolMessageType('KafkaSourceV1', (_message.Message,), {
-
-  'SecretConfigsEntry' : _reflection.GeneratedProtocolMessageType('SecretConfigsEntry', (_message.Message,), {
-    'DESCRIPTOR' : _KAFKASOURCEV1_SECRETCONFIGSENTRY,
-    '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSourceV1.SecretConfigsEntry)
-    })
-  ,
-
-  'PassthroughConfigsEntry' : _reflection.GeneratedProtocolMessageType('PassthroughConfigsEntry', (_message.Message,), {
-    'DESCRIPTOR' : _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY,
-    '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSourceV1.PassthroughConfigsEntry)
-    })
-  ,
-  'DESCRIPTOR' : _KAFKASOURCEV1,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.KafkaSourceV1)
-  })
-_sym_db.RegisterMessage(KafkaSourceV1)
-_sym_db.RegisterMessage(KafkaSourceV1.SecretConfigsEntry)
-_sym_db.RegisterMessage(KafkaSourceV1.PassthroughConfigsEntry)
-
-Authentication = _reflection.GeneratedProtocolMessageType('Authentication', (_message.Message,), {
-  'DESCRIPTOR' : _AUTHENTICATION,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Authentication)
-  })
-_sym_db.RegisterMessage(Authentication)
-
-Ssl = _reflection.GeneratedProtocolMessageType('Ssl', (_message.Message,), {
-  'DESCRIPTOR' : _SSL,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Ssl)
-  })
-_sym_db.RegisterMessage(Ssl)
-
-Plain = _reflection.GeneratedProtocolMessageType('Plain', (_message.Message,), {
-  'DESCRIPTOR' : _PLAIN,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Plain)
-  })
-_sym_db.RegisterMessage(Plain)
-
-Sasl = _reflection.GeneratedProtocolMessageType('Sasl', (_message.Message,), {
-  'DESCRIPTOR' : _SASL,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Sasl)
-  })
-_sym_db.RegisterMessage(Sasl)
-
-Deserialization = _reflection.GeneratedProtocolMessageType('Deserialization', (_message.Message,), {
-  'DESCRIPTOR' : _DESERIALIZATION,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.Deserialization)
-  })
-_sym_db.RegisterMessage(Deserialization)
-
-CustomDeserializer = _reflection.GeneratedProtocolMessageType('CustomDeserializer', (_message.Message,), {
-  'DESCRIPTOR' : _CUSTOMDESERIALIZER,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.CustomDeserializer)
-  })
-_sym_db.RegisterMessage(CustomDeserializer)
-
-GenericDeserializer = _reflection.GeneratedProtocolMessageType('GenericDeserializer', (_message.Message,), {
-  'DESCRIPTOR' : _GENERICDESERIALIZER,
-  '__module__' : 'qwak.feature_store.sources.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.GenericDeserializer)
-  })
-_sym_db.RegisterMessage(GenericDeserializer)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.sources.streaming_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -142,34 +30,34 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _KAFKASOURCEV1_SECRETCONFIGSENTRY._serialized_options = b'8\001'
   _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._options = None
   _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._serialized_options = b'8\001'
-  _SASLMECHANISM._serialized_start=1812
-  _SASLMECHANISM._serialized_end=1872
-  _SECURITYPROTOCOL._serialized_start=1874
-  _SECURITYPROTOCOL._serialized_end=1906
-  _MESSAGEFORMAT._serialized_start=1908
-  _MESSAGEFORMAT._serialized_end=1956
-  _STREAMINGSOURCE._serialized_start=75
-  _STREAMINGSOURCE._serialized_end=271
-  _KAFKASOURCE._serialized_start=274
-  _KAFKASOURCE._serialized_end=433
-  _KAFKASOURCEV1._serialized_start=436
-  _KAFKASOURCEV1._serialized_end=1009
-  _KAFKASOURCEV1_SECRETCONFIGSENTRY._serialized_start=875
-  _KAFKASOURCEV1_SECRETCONFIGSENTRY._serialized_end=927
-  _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._serialized_start=929
-  _KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY._serialized_end=986
-  _AUTHENTICATION._serialized_start=1012
-  _AUTHENTICATION._serialized_end=1228
-  _SSL._serialized_start=1230
-  _SSL._serialized_end=1235
-  _PLAIN._serialized_start=1237
-  _PLAIN._serialized_end=1244
-  _SASL._serialized_start=1247
-  _SASL._serialized_end=1443
-  _DESERIALIZATION._serialized_start=1446
-  _DESERIALIZATION._serialized_end=1631
-  _CUSTOMDESERIALIZER._serialized_start=1633
-  _CUSTOMDESERIALIZER._serialized_end=1699
-  _GENERICDESERIALIZER._serialized_start=1701
-  _GENERICDESERIALIZER._serialized_end=1810
+  _globals['_SASLMECHANISM']._serialized_start=1812
+  _globals['_SASLMECHANISM']._serialized_end=1872
+  _globals['_SECURITYPROTOCOL']._serialized_start=1874
+  _globals['_SECURITYPROTOCOL']._serialized_end=1906
+  _globals['_MESSAGEFORMAT']._serialized_start=1908
+  _globals['_MESSAGEFORMAT']._serialized_end=1956
+  _globals['_STREAMINGSOURCE']._serialized_start=75
+  _globals['_STREAMINGSOURCE']._serialized_end=271
+  _globals['_KAFKASOURCE']._serialized_start=274
+  _globals['_KAFKASOURCE']._serialized_end=433
+  _globals['_KAFKASOURCEV1']._serialized_start=436
+  _globals['_KAFKASOURCEV1']._serialized_end=1009
+  _globals['_KAFKASOURCEV1_SECRETCONFIGSENTRY']._serialized_start=875
+  _globals['_KAFKASOURCEV1_SECRETCONFIGSENTRY']._serialized_end=927
+  _globals['_KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY']._serialized_start=929
+  _globals['_KAFKASOURCEV1_PASSTHROUGHCONFIGSENTRY']._serialized_end=986
+  _globals['_AUTHENTICATION']._serialized_start=1012
+  _globals['_AUTHENTICATION']._serialized_end=1228
+  _globals['_SSL']._serialized_start=1230
+  _globals['_SSL']._serialized_end=1235
+  _globals['_PLAIN']._serialized_start=1237
+  _globals['_PLAIN']._serialized_end=1244
+  _globals['_SASL']._serialized_start=1247
+  _globals['_SASL']._serialized_end=1443
+  _globals['_DESERIALIZATION']._serialized_start=1446
+  _globals['_DESERIALIZATION']._serialized_end=1631
+  _globals['_CUSTOMDESERIALIZER']._serialized_start=1633
+  _globals['_CUSTOMDESERIALIZER']._serialized_end=1699
+  _globals['_GENERICDESERIALIZER']._serialized_start=1701
+  _globals['_GENERICDESERIALIZER']._serialized_end=1810
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact_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()
@@ -16,20 +15,13 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n@qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact.proto\x12+qwak.feature.store.v1.common.jfrog_artifact\"\x1d\n\rJfrogArtifact\x12\x0c\n\x04path\x18\x01 \x01(\tBp\n2com.qwak.ai.feature.store.v1.common.jfrog_artifactP\x01Z8qwak/featurestore/v1/common/jfrog_artifact;jfrogartifactb\x06proto3')
 
-
-
-_JFROGARTIFACT = DESCRIPTOR.message_types_by_name['JfrogArtifact']
-JfrogArtifact = _reflection.GeneratedProtocolMessageType('JfrogArtifact', (_message.Message,), {
-  'DESCRIPTOR' : _JFROGARTIFACT,
-  '__module__' : 'qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.v1.common.jfrog_artifact.JfrogArtifact)
-  })
-_sym_db.RegisterMessage(JfrogArtifact)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.v1.common.jfrog_artifact.jfrog_artifact_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n2com.qwak.ai.feature.store.v1.common.jfrog_artifactP\001Z8qwak/featurestore/v1/common/jfrog_artifact;jfrogartifact'
-  _JFROGARTIFACT._serialized_start=113
-  _JFROGARTIFACT._serialized_end=142
+  _globals['_JFROGARTIFACT']._serialized_start=113
+  _globals['_JFROGARTIFACT']._serialized_end=142
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/v1/common/source_code/source_code_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,40 +16,17 @@ from _qwak_proto.qwak.feature_store.v1.common.jfrog_artifact import jfrog_artifa
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n:qwak/feature_store/v1/common/source_code/source_code.proto\x12(qwak.feature.store.v1.common.source_code\x1a@qwak/feature_store/v1/common/jfrog_artifact/jfrog_artifact.proto\"`\n\x0eSourceCodeSpec\x12N\n\x08\x61rtifact\x18\x01 \x01(\x0b\x32<.qwak.feature.store.v1.common.source_code.SourceCodeArtifact\"k\n\x12SourceCodeArtifact\x12M\n\x0czip_artifact\x18\x01 \x01(\x0b\x32\x35.qwak.feature.store.v1.common.source_code.ZipArtifactH\x00\x42\x06\n\x04type\"\x98\x01\n\x0bZipArtifact\x12\x0e\n\x04path\x18\x01 \x01(\tH\x00\x12Y\n\x13jfrog_artifact_path\x18\x03 \x01(\x0b\x32:.qwak.feature.store.v1.common.jfrog_artifact.JfrogArtifactH\x00\x12\x11\n\tmain_file\x18\x02 \x01(\tB\x0b\n\tpath_typeBg\n/com.qwak.ai.feature.store.v1.common.source_codeP\x01Z2qwak/featurestore/v1/common/source_code;sourcecodeb\x06proto3')
 
-
-
-_SOURCECODESPEC = DESCRIPTOR.message_types_by_name['SourceCodeSpec']
-_SOURCECODEARTIFACT = DESCRIPTOR.message_types_by_name['SourceCodeArtifact']
-_ZIPARTIFACT = DESCRIPTOR.message_types_by_name['ZipArtifact']
-SourceCodeSpec = _reflection.GeneratedProtocolMessageType('SourceCodeSpec', (_message.Message,), {
-  'DESCRIPTOR' : _SOURCECODESPEC,
-  '__module__' : 'qwak.feature_store.v1.common.source_code.source_code_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.v1.common.source_code.SourceCodeSpec)
-  })
-_sym_db.RegisterMessage(SourceCodeSpec)
-
-SourceCodeArtifact = _reflection.GeneratedProtocolMessageType('SourceCodeArtifact', (_message.Message,), {
-  'DESCRIPTOR' : _SOURCECODEARTIFACT,
-  '__module__' : 'qwak.feature_store.v1.common.source_code.source_code_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.v1.common.source_code.SourceCodeArtifact)
-  })
-_sym_db.RegisterMessage(SourceCodeArtifact)
-
-ZipArtifact = _reflection.GeneratedProtocolMessageType('ZipArtifact', (_message.Message,), {
-  'DESCRIPTOR' : _ZIPARTIFACT,
-  '__module__' : 'qwak.feature_store.v1.common.source_code.source_code_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.v1.common.source_code.ZipArtifact)
-  })
-_sym_db.RegisterMessage(ZipArtifact)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.v1.common.source_code.source_code_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n/com.qwak.ai.feature.store.v1.common.source_codeP\001Z2qwak/featurestore/v1/common/source_code;sourcecode'
-  _SOURCECODESPEC._serialized_start=170
-  _SOURCECODESPEC._serialized_end=266
-  _SOURCECODEARTIFACT._serialized_start=268
-  _SOURCECODEARTIFACT._serialized_end=375
-  _ZIPARTIFACT._serialized_start=378
-  _ZIPARTIFACT._serialized_end=530
+  _globals['_SOURCECODESPEC']._serialized_start=170
+  _globals['_SOURCECODESPEC']._serialized_end=266
+  _globals['_SOURCECODEARTIFACT']._serialized_start=268
+  _globals['_SOURCECODEARTIFACT']._serialized_end=375
+  _globals['_ZIPARTIFACT']._serialized_start=378
+  _globals['_ZIPARTIFACT']._serialized_end=530
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/v1/internal/data_source/data_source_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()
@@ -16,66 +15,25 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nDqwak/feature_store/v1/internal/data_source/data_source_service.proto\x12\x31qwak.feature.store.management.v1.internal.sources\"E\n)GetDataSourceSourceCodeDownloadURLRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"B\n*GetDataSourceSourceCodeDownloadURLResponse\x12\x14\n\x0c\x64ownload_url\x18\x01 \x01(\t\"q\n$GetDataSourceSourceCodeUploadRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\x12\x19\n\x0frepository_name\x18\x02 \x01(\tH\x00\x42\x14\n\x12repository_details\"\xf4\x01\n%GetDataSourceSourceCodeUploadResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t\x12\x81\x01\n\rextra_headers\x18\x02 \x03(\x0b\x32j.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse.ExtraHeadersEntry\x1a\x33\n\x11\x45xtraHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x32\xcf\x03\n\x11\x44\x61taSourceService\x12\xd5\x01\n GetDataSourceSourceCodeUploadURL\x12W.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadRequest\x1aX.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse\x12\xe1\x01\n\"GetDataSourceSourceCodeDownloadURL\x12\\.qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLRequest\x1a].qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLResponseBw\n<com.qwak.ai.feature.store.management.v1.api.internal.sourcesP\x01Z5qwak/featurestore/sources;featurestoreinternalsourcesb\x06proto3')
 
-
-
-_GETDATASOURCESOURCECODEDOWNLOADURLREQUEST = DESCRIPTOR.message_types_by_name['GetDataSourceSourceCodeDownloadURLRequest']
-_GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE = DESCRIPTOR.message_types_by_name['GetDataSourceSourceCodeDownloadURLResponse']
-_GETDATASOURCESOURCECODEUPLOADREQUEST = DESCRIPTOR.message_types_by_name['GetDataSourceSourceCodeUploadRequest']
-_GETDATASOURCESOURCECODEUPLOADRESPONSE = DESCRIPTOR.message_types_by_name['GetDataSourceSourceCodeUploadResponse']
-_GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY = _GETDATASOURCESOURCECODEUPLOADRESPONSE.nested_types_by_name['ExtraHeadersEntry']
-GetDataSourceSourceCodeDownloadURLRequest = _reflection.GeneratedProtocolMessageType('GetDataSourceSourceCodeDownloadURLRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETDATASOURCESOURCECODEDOWNLOADURLREQUEST,
-  '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLRequest)
-  })
-_sym_db.RegisterMessage(GetDataSourceSourceCodeDownloadURLRequest)
-
-GetDataSourceSourceCodeDownloadURLResponse = _reflection.GeneratedProtocolMessageType('GetDataSourceSourceCodeDownloadURLResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE,
-  '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeDownloadURLResponse)
-  })
-_sym_db.RegisterMessage(GetDataSourceSourceCodeDownloadURLResponse)
-
-GetDataSourceSourceCodeUploadRequest = _reflection.GeneratedProtocolMessageType('GetDataSourceSourceCodeUploadRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETDATASOURCESOURCECODEUPLOADREQUEST,
-  '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadRequest)
-  })
-_sym_db.RegisterMessage(GetDataSourceSourceCodeUploadRequest)
-
-GetDataSourceSourceCodeUploadResponse = _reflection.GeneratedProtocolMessageType('GetDataSourceSourceCodeUploadResponse', (_message.Message,), {
-
-  'ExtraHeadersEntry' : _reflection.GeneratedProtocolMessageType('ExtraHeadersEntry', (_message.Message,), {
-    'DESCRIPTOR' : _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY,
-    '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse.ExtraHeadersEntry)
-    })
-  ,
-  'DESCRIPTOR' : _GETDATASOURCESOURCECODEUPLOADRESPONSE,
-  '__module__' : 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.sources.GetDataSourceSourceCodeUploadResponse)
-  })
-_sym_db.RegisterMessage(GetDataSourceSourceCodeUploadResponse)
-_sym_db.RegisterMessage(GetDataSourceSourceCodeUploadResponse.ExtraHeadersEntry)
-
-_DATASOURCESERVICE = DESCRIPTOR.services_by_name['DataSourceService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.v1.internal.data_source.data_source_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n<com.qwak.ai.feature.store.management.v1.api.internal.sourcesP\001Z5qwak/featurestore/sources;featurestoreinternalsources'
   _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._options = None
   _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._serialized_options = b'8\001'
-  _GETDATASOURCESOURCECODEDOWNLOADURLREQUEST._serialized_start=123
-  _GETDATASOURCESOURCECODEDOWNLOADURLREQUEST._serialized_end=192
-  _GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE._serialized_start=194
-  _GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE._serialized_end=260
-  _GETDATASOURCESOURCECODEUPLOADREQUEST._serialized_start=262
-  _GETDATASOURCESOURCECODEUPLOADREQUEST._serialized_end=375
-  _GETDATASOURCESOURCECODEUPLOADRESPONSE._serialized_start=378
-  _GETDATASOURCESOURCECODEUPLOADRESPONSE._serialized_end=622
-  _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._serialized_start=571
-  _GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY._serialized_end=622
-  _DATASOURCESERVICE._serialized_start=625
-  _DATASOURCESERVICE._serialized_end=1088
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLREQUEST']._serialized_start=123
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLREQUEST']._serialized_end=192
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE']._serialized_start=194
+  _globals['_GETDATASOURCESOURCECODEDOWNLOADURLRESPONSE']._serialized_end=260
+  _globals['_GETDATASOURCESOURCECODEUPLOADREQUEST']._serialized_start=262
+  _globals['_GETDATASOURCESOURCECODEUPLOADREQUEST']._serialized_end=375
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE']._serialized_start=378
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE']._serialized_end=622
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY']._serialized_start=571
+  _globals['_GETDATASOURCESOURCECODEUPLOADRESPONSE_EXTRAHEADERSENTRY']._serialized_end=622
+  _globals['_DATASOURCESERVICE']._serialized_start=625
+  _globals['_DATASOURCESERVICE']._serialized_end=1088
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/v1/internal/featureset/featureset_token_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()
@@ -16,33 +15,17 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\nHqwak/feature_store/v1/internal/featureset/featureset_token_service.proto\x12\x34qwak.feature.store.management.v1.internal.featureset\"P\n#GetAuthTokenIdByFeaturesetIdRequest\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\x12\x12\n\naccount_id\x18\x02 \x01(\t\"8\n$GetAuthTokenIdByFeaturesetIdResponse\x12\x10\n\x08token_id\x18\x01 \x01(\t2\xef\x01\n\x15\x46\x65\x61turesetAuthService\x12\xd5\x01\n\x1cGetAuthTokenIdByFeaturesetId\x12Y.qwak.feature.store.management.v1.internal.featureset.GetAuthTokenIdByFeaturesetIdRequest\x1aZ.qwak.feature.store.management.v1.internal.featureset.GetAuthTokenIdByFeaturesetIdResponseBC\n?com.qwak.ai.feature.store.management.v1.api.internal.featuresetP\x01\x62\x06proto3')
 
-
-
-_GETAUTHTOKENIDBYFEATURESETIDREQUEST = DESCRIPTOR.message_types_by_name['GetAuthTokenIdByFeaturesetIdRequest']
-_GETAUTHTOKENIDBYFEATURESETIDRESPONSE = DESCRIPTOR.message_types_by_name['GetAuthTokenIdByFeaturesetIdResponse']
-GetAuthTokenIdByFeaturesetIdRequest = _reflection.GeneratedProtocolMessageType('GetAuthTokenIdByFeaturesetIdRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETAUTHTOKENIDBYFEATURESETIDREQUEST,
-  '__module__' : 'qwak.feature_store.v1.internal.featureset.featureset_token_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.featureset.GetAuthTokenIdByFeaturesetIdRequest)
-  })
-_sym_db.RegisterMessage(GetAuthTokenIdByFeaturesetIdRequest)
-
-GetAuthTokenIdByFeaturesetIdResponse = _reflection.GeneratedProtocolMessageType('GetAuthTokenIdByFeaturesetIdResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETAUTHTOKENIDBYFEATURESETIDRESPONSE,
-  '__module__' : 'qwak.feature_store.v1.internal.featureset.featureset_token_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.management.v1.internal.featureset.GetAuthTokenIdByFeaturesetIdResponse)
-  })
-_sym_db.RegisterMessage(GetAuthTokenIdByFeaturesetIdResponse)
-
-_FEATURESETAUTHSERVICE = DESCRIPTOR.services_by_name['FeaturesetAuthService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.v1.internal.featureset.featureset_token_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n?com.qwak.ai.feature.store.management.v1.api.internal.featuresetP\001'
-  _GETAUTHTOKENIDBYFEATURESETIDREQUEST._serialized_start=130
-  _GETAUTHTOKENIDBYFEATURESETIDREQUEST._serialized_end=210
-  _GETAUTHTOKENIDBYFEATURESETIDRESPONSE._serialized_start=212
-  _GETAUTHTOKENIDBYFEATURESETIDRESPONSE._serialized_end=268
-  _FEATURESETAUTHSERVICE._serialized_start=271
-  _FEATURESETAUTHSERVICE._serialized_end=510
+  _globals['_GETAUTHTOKENIDBYFEATURESETIDREQUEST']._serialized_start=130
+  _globals['_GETAUTHTOKENIDBYFEATURESETIDREQUEST']._serialized_end=210
+  _globals['_GETAUTHTOKENIDBYFEATURESETIDRESPONSE']._serialized_start=212
+  _globals['_GETAUTHTOKENIDBYFEATURESETIDRESPONSE']._serialized_end=268
+  _globals['_FEATURESETAUTHSERVICE']._serialized_start=271
+  _globals['_FEATURESETAUTHSERVICE']._serialized_end=510
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/features_operator/v1/features_operator_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,60 +16,21 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1qwak/features_operator/v1/features_operator.proto\x12\x19qwak.features_operator.v1\x1a\x1fgoogle/protobuf/timestamp.proto\"\xa1\x02\n\x19ValidationSuccessResponse\x12\x0e\n\x06sample\x18\x01 \x01(\t\x12H\n\x12\x63olumn_description\x18\x02 \x03(\x0b\x32,.qwak.features_operator.v1.ColumnDescription\x12U\n\x19pandas_column_description\x18\x03 \x03(\x0b\x32\x32.qwak.features_operator.v1.PandasColumnDescription\x12S\n\x18spark_column_description\x18\x04 \x03(\x0b\x32\x31.qwak.features_operator.v1.SparkColumnDescription\"<\n\x11\x43olumnDescription\x12\x13\n\x0b\x63olumn_name\x18\x01 \x01(\t\x12\x12\n\nspark_type\x18\x02 \x01(\t\"C\n\x17PandasColumnDescription\x12\x13\n\x0b\x63olumn_name\x18\x01 \x01(\t\x12\x13\n\x0bpandas_type\x18\x02 \x01(\t\"A\n\x16SparkColumnDescription\x12\x13\n\x0b\x63olumn_name\x18\x01 \x01(\t\x12\x12\n\nspark_type\x18\x02 \x01(\t\"V\n\x19ValidationFailureResponse\x12\r\n\x05phase\x18\x01 \x01(\t\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\tB(\n$com.qwak.ai.features.operator.api.v1P\x01\x62\x06proto3')
 
-
-
-_VALIDATIONSUCCESSRESPONSE = DESCRIPTOR.message_types_by_name['ValidationSuccessResponse']
-_COLUMNDESCRIPTION = DESCRIPTOR.message_types_by_name['ColumnDescription']
-_PANDASCOLUMNDESCRIPTION = DESCRIPTOR.message_types_by_name['PandasColumnDescription']
-_SPARKCOLUMNDESCRIPTION = DESCRIPTOR.message_types_by_name['SparkColumnDescription']
-_VALIDATIONFAILURERESPONSE = DESCRIPTOR.message_types_by_name['ValidationFailureResponse']
-ValidationSuccessResponse = _reflection.GeneratedProtocolMessageType('ValidationSuccessResponse', (_message.Message,), {
-  'DESCRIPTOR' : _VALIDATIONSUCCESSRESPONSE,
-  '__module__' : 'qwak.features_operator.v1.features_operator_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.features_operator.v1.ValidationSuccessResponse)
-  })
-_sym_db.RegisterMessage(ValidationSuccessResponse)
-
-ColumnDescription = _reflection.GeneratedProtocolMessageType('ColumnDescription', (_message.Message,), {
-  'DESCRIPTOR' : _COLUMNDESCRIPTION,
-  '__module__' : 'qwak.features_operator.v1.features_operator_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.features_operator.v1.ColumnDescription)
-  })
-_sym_db.RegisterMessage(ColumnDescription)
-
-PandasColumnDescription = _reflection.GeneratedProtocolMessageType('PandasColumnDescription', (_message.Message,), {
-  'DESCRIPTOR' : _PANDASCOLUMNDESCRIPTION,
-  '__module__' : 'qwak.features_operator.v1.features_operator_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.features_operator.v1.PandasColumnDescription)
-  })
-_sym_db.RegisterMessage(PandasColumnDescription)
-
-SparkColumnDescription = _reflection.GeneratedProtocolMessageType('SparkColumnDescription', (_message.Message,), {
-  'DESCRIPTOR' : _SPARKCOLUMNDESCRIPTION,
-  '__module__' : 'qwak.features_operator.v1.features_operator_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.features_operator.v1.SparkColumnDescription)
-  })
-_sym_db.RegisterMessage(SparkColumnDescription)
-
-ValidationFailureResponse = _reflection.GeneratedProtocolMessageType('ValidationFailureResponse', (_message.Message,), {
-  'DESCRIPTOR' : _VALIDATIONFAILURERESPONSE,
-  '__module__' : 'qwak.features_operator.v1.features_operator_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.features_operator.v1.ValidationFailureResponse)
-  })
-_sym_db.RegisterMessage(ValidationFailureResponse)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.features_operator.v1.features_operator_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n$com.qwak.ai.features.operator.api.v1P\001'
-  _VALIDATIONSUCCESSRESPONSE._serialized_start=114
-  _VALIDATIONSUCCESSRESPONSE._serialized_end=403
-  _COLUMNDESCRIPTION._serialized_start=405
-  _COLUMNDESCRIPTION._serialized_end=465
-  _PANDASCOLUMNDESCRIPTION._serialized_start=467
-  _PANDASCOLUMNDESCRIPTION._serialized_end=534
-  _SPARKCOLUMNDESCRIPTION._serialized_start=536
-  _SPARKCOLUMNDESCRIPTION._serialized_end=601
-  _VALIDATIONFAILURERESPONSE._serialized_start=603
-  _VALIDATIONFAILURERESPONSE._serialized_end=689
+  _globals['_VALIDATIONSUCCESSRESPONSE']._serialized_start=114
+  _globals['_VALIDATIONSUCCESSRESPONSE']._serialized_end=403
+  _globals['_COLUMNDESCRIPTION']._serialized_start=405
+  _globals['_COLUMNDESCRIPTION']._serialized_end=465
+  _globals['_PANDASCOLUMNDESCRIPTION']._serialized_start=467
+  _globals['_PANDASCOLUMNDESCRIPTION']._serialized_end=534
+  _globals['_SPARKCOLUMNDESCRIPTION']._serialized_start=536
+  _globals['_SPARKCOLUMNDESCRIPTION']._serialized_end=601
+  _globals['_VALIDATIONFAILURERESPONSE']._serialized_start=603
+  _globals['_VALIDATIONFAILURERESPONSE']._serialized_end=689
 # @@protoc_insertion_point(module_scope)