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

_qwak_proto/qwak/feature_store/features/feature_set_service_pb2_grpc.py

@@ -54,6 +54,11 @@ class FeatureSetServiceStub(object):
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsRequest.SerializeToString,
                 response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsResponse.FromString,
                 )
+        self.ListFeaturesetsByAccount = channel.unary_unary(
+                '/qwak.feature.store.features.FeatureSetService/ListFeaturesetsByAccount',
+                request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.SerializeToString,
+                response_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.FromString,
+                )
         self.RunBatchFeatureSet = channel.unary_unary(
                 '/qwak.feature.store.features.FeatureSetService/RunBatchFeatureSet',
                 request_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.RunBatchFeatureSetRequest.SerializeToString,
@@ -180,6 +185,13 @@ class FeatureSetServiceServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def ListFeaturesetsByAccount(self, request, context):
+        """List of registered featuresets
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
     def RunBatchFeatureSet(self, request, context):
         """Run now feature set
         """
@@ -313,6 +325,11 @@ def add_FeatureSetServiceServicer_to_server(servicer, server):
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsRequest.FromString,
                     response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeatureSetsResponse.SerializeToString,
             ),
+            'ListFeaturesetsByAccount': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListFeaturesetsByAccount,
+                    request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.FromString,
+                    response_serializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.SerializeToString,
+            ),
             'RunBatchFeatureSet': grpc.unary_unary_rpc_method_handler(
                     servicer.RunBatchFeatureSet,
                     request_deserializer=qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.RunBatchFeatureSetRequest.FromString,
@@ -524,6 +541,23 @@ class FeatureSetService(object):
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
+    @staticmethod
+    def ListFeaturesetsByAccount(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.features.FeatureSetService/ListFeaturesetsByAccount',
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountRequest.SerializeToString,
+            qwak_dot_feature__store_dot_features_dot_feature__set__service__pb2.ListFeaturesetsByAccountResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
     @staticmethod
     def RunBatchFeatureSet(request,
             target,

_qwak_proto/qwak/feature_store/features/feature_set_state_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/feature_store/features/feature_set_state.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()
@@ -18,186 +16,45 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3qwak/feature_store/features/feature_set_state.proto\x12\x1bqwak.feature.store.features\x1a\x1fgoogle/protobuf/timestamp.proto\"\x9e\x03\n\x0f\x46\x65\x61tureSetState\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x02 \x01(\t\x12R\n\x0fspark_app_state\x18\x03 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionState\x12\x44\n\x0f\x65xecutor_states\x18\x04 \x01(\x0b\x32+.qwak.feature.store.features.ExecutorStates\x12L\n\x14\x66\x65\x61ture_set_job_type\x18\x05 \x01(\x0b\x32..qwak.feature.store.features.FeatureSetJobType\x12\x15\n\rerror_message\x18\x06 \x01(\t\x12\x14\n\x0cspark_app_id\x18\x07 \x01(\t\x12\x34\n\x10last_modified_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0e\n\x06run_id\x18\t \x01(\x03\"\xac\x02\n\x11\x46\x65\x61tureSetJobType\x12;\n\tstreaming\x18\x01 \x01(\x0b\x32&.qwak.feature.store.features.StreamingH\x00\x12\x38\n\x08\x62\x61tch_v1\x18\x02 \x01(\x0b\x32$.qwak.feature.store.features.BatchV1H\x00\x12R\n\x15streaming_aggregation\x18\x03 \x01(\x0b\x32\x31.qwak.feature.store.features.StreamingAggregationH\x00\x12@\n\x0c\x64\x65letion_job\x18\x04 \x01(\x0b\x32(.qwak.feature.store.features.DeletionJobH\x00\x42\n\n\x08job_type\"\x87\x01\n\tStreaming\x12\x35\n\x06online\x18\x01 \x01(\x0b\x32#.qwak.feature.store.features.OnlineH\x00\x12\x37\n\x07offline\x18\x02 \x01(\x0b\x32$.qwak.feature.store.features.OfflineH\x00\x42\n\n\x08sub_type\"\x08\n\x06Online\"\t\n\x07Offline\"\t\n\x07\x42\x61tchV1\"\xdf\x01\n\x14StreamingAggregation\x12:\n\trow_level\x18\x01 \x01(\x0b\x32%.qwak.feature.store.features.RowLevelH\x00\x12@\n\x08\x62\x61\x63kfill\x18\x02 \x01(\x0b\x32,.qwak.feature.store.features.BackfillSubTypeH\x00\x12=\n\ncompaction\x18\x03 \x01(\x0b\x32\'.qwak.feature.store.features.CompactionH\x00\x42\n\n\x08sub_type\"\n\n\x08RowLevel\"\x0c\n\nCompaction\"\x11\n\x0f\x42\x61\x63kfillSubType\"\xbe\x02\n\x0b\x44\x65letionJob\x12S\n\x11\x62\x61tch_v1_deletion\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.features.BatchV1DeletionJobSubTypeH\x00\x12[\n\x15streaming_v1_deletion\x18\x02 \x01(\x0b\x32:.qwak.feature.store.features.StreamingV1DeletionJobSubTypeH\x00\x12m\n\x1estreaming_aggregation_deletion\x18\x03 \x01(\x0b\x32\x43.qwak.feature.store.features.StreamingAggregationDeletionJobSubTypeH\x00\x42\x0e\n\x0cjob_sub_type\"\x1b\n\x19\x42\x61tchV1DeletionJobSubType\"\x1f\n\x1dStreamingV1DeletionJobSubType\"(\n&StreamingAggregationDeletionJobSubType\"U\n\x0e\x45xecutorStates\x12\x43\n\x0f\x65xecutor_states\x18\x01 \x03(\x0b\x32*.qwak.feature.store.features.ExecutorState\"p\n\rExecutorState\x12\x15\n\rexecutor_name\x18\x01 \x01(\t\x12H\n\x05state\x18\x02 \x01(\x0e\x32\x39.qwak.feature.store.features.FeatureSetLastExecutionState*\xd5\x01\n\x1c\x46\x65\x61tureSetLastExecutionState\x12\x0b\n\x07UNKNOWN\x10\x00\x12\n\n\x06\x46\x41ILED\x10\x01\x12\x0b\n\x07SUCCESS\x10\x02\x12\x10\n\x0cINVALIDATING\x10\x03\x12\r\n\tSUBMITTED\x10\x04\x12\x0b\n\x07RUNNING\x10\x05\x12\r\n\tCOMPLETED\x10\x06\x12\x15\n\x11SUBMISSION_FAILED\x10\x07\x12\x11\n\rPENDING_RERUN\x10\x08\x12\x0e\n\nSUCCEEDING\x10\t\x12\x0b\n\x07\x46\x41ILING\x10\n\x12\x0b\n\x07PENDING\x10\x0b\x42[\n&com.qwak.ai.feature.store.features.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
 
-_FEATURESETLASTEXECUTIONSTATE = DESCRIPTOR.enum_types_by_name['FeatureSetLastExecutionState']
-FeatureSetLastExecutionState = enum_type_wrapper.EnumTypeWrapper(_FEATURESETLASTEXECUTIONSTATE)
-UNKNOWN = 0
-FAILED = 1
-SUCCESS = 2
-INVALIDATING = 3
-SUBMITTED = 4
-RUNNING = 5
-COMPLETED = 6
-SUBMISSION_FAILED = 7
-PENDING_RERUN = 8
-SUCCEEDING = 9
-FAILING = 10
-PENDING = 11
-
-
-_FEATURESETSTATE = DESCRIPTOR.message_types_by_name['FeatureSetState']
-_FEATURESETJOBTYPE = DESCRIPTOR.message_types_by_name['FeatureSetJobType']
-_STREAMING = DESCRIPTOR.message_types_by_name['Streaming']
-_ONLINE = DESCRIPTOR.message_types_by_name['Online']
-_OFFLINE = DESCRIPTOR.message_types_by_name['Offline']
-_BATCHV1 = DESCRIPTOR.message_types_by_name['BatchV1']
-_STREAMINGAGGREGATION = DESCRIPTOR.message_types_by_name['StreamingAggregation']
-_ROWLEVEL = DESCRIPTOR.message_types_by_name['RowLevel']
-_COMPACTION = DESCRIPTOR.message_types_by_name['Compaction']
-_BACKFILLSUBTYPE = DESCRIPTOR.message_types_by_name['BackfillSubType']
-_DELETIONJOB = DESCRIPTOR.message_types_by_name['DeletionJob']
-_BATCHV1DELETIONJOBSUBTYPE = DESCRIPTOR.message_types_by_name['BatchV1DeletionJobSubType']
-_STREAMINGV1DELETIONJOBSUBTYPE = DESCRIPTOR.message_types_by_name['StreamingV1DeletionJobSubType']
-_STREAMINGAGGREGATIONDELETIONJOBSUBTYPE = DESCRIPTOR.message_types_by_name['StreamingAggregationDeletionJobSubType']
-_EXECUTORSTATES = DESCRIPTOR.message_types_by_name['ExecutorStates']
-_EXECUTORSTATE = DESCRIPTOR.message_types_by_name['ExecutorState']
-FeatureSetState = _reflection.GeneratedProtocolMessageType('FeatureSetState', (_message.Message,), {
-  'DESCRIPTOR' : _FEATURESETSTATE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.FeatureSetState)
-  })
-_sym_db.RegisterMessage(FeatureSetState)
-
-FeatureSetJobType = _reflection.GeneratedProtocolMessageType('FeatureSetJobType', (_message.Message,), {
-  'DESCRIPTOR' : _FEATURESETJOBTYPE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.FeatureSetJobType)
-  })
-_sym_db.RegisterMessage(FeatureSetJobType)
-
-Streaming = _reflection.GeneratedProtocolMessageType('Streaming', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMING,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.Streaming)
-  })
-_sym_db.RegisterMessage(Streaming)
-
-Online = _reflection.GeneratedProtocolMessageType('Online', (_message.Message,), {
-  'DESCRIPTOR' : _ONLINE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.Online)
-  })
-_sym_db.RegisterMessage(Online)
-
-Offline = _reflection.GeneratedProtocolMessageType('Offline', (_message.Message,), {
-  'DESCRIPTOR' : _OFFLINE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.Offline)
-  })
-_sym_db.RegisterMessage(Offline)
-
-BatchV1 = _reflection.GeneratedProtocolMessageType('BatchV1', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHV1,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.BatchV1)
-  })
-_sym_db.RegisterMessage(BatchV1)
-
-StreamingAggregation = _reflection.GeneratedProtocolMessageType('StreamingAggregation', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGAGGREGATION,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.StreamingAggregation)
-  })
-_sym_db.RegisterMessage(StreamingAggregation)
-
-RowLevel = _reflection.GeneratedProtocolMessageType('RowLevel', (_message.Message,), {
-  'DESCRIPTOR' : _ROWLEVEL,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.RowLevel)
-  })
-_sym_db.RegisterMessage(RowLevel)
-
-Compaction = _reflection.GeneratedProtocolMessageType('Compaction', (_message.Message,), {
-  'DESCRIPTOR' : _COMPACTION,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.Compaction)
-  })
-_sym_db.RegisterMessage(Compaction)
-
-BackfillSubType = _reflection.GeneratedProtocolMessageType('BackfillSubType', (_message.Message,), {
-  'DESCRIPTOR' : _BACKFILLSUBTYPE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.BackfillSubType)
-  })
-_sym_db.RegisterMessage(BackfillSubType)
-
-DeletionJob = _reflection.GeneratedProtocolMessageType('DeletionJob', (_message.Message,), {
-  'DESCRIPTOR' : _DELETIONJOB,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.DeletionJob)
-  })
-_sym_db.RegisterMessage(DeletionJob)
-
-BatchV1DeletionJobSubType = _reflection.GeneratedProtocolMessageType('BatchV1DeletionJobSubType', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHV1DELETIONJOBSUBTYPE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.BatchV1DeletionJobSubType)
-  })
-_sym_db.RegisterMessage(BatchV1DeletionJobSubType)
-
-StreamingV1DeletionJobSubType = _reflection.GeneratedProtocolMessageType('StreamingV1DeletionJobSubType', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGV1DELETIONJOBSUBTYPE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.StreamingV1DeletionJobSubType)
-  })
-_sym_db.RegisterMessage(StreamingV1DeletionJobSubType)
-
-StreamingAggregationDeletionJobSubType = _reflection.GeneratedProtocolMessageType('StreamingAggregationDeletionJobSubType', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGAGGREGATIONDELETIONJOBSUBTYPE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.StreamingAggregationDeletionJobSubType)
-  })
-_sym_db.RegisterMessage(StreamingAggregationDeletionJobSubType)
-
-ExecutorStates = _reflection.GeneratedProtocolMessageType('ExecutorStates', (_message.Message,), {
-  'DESCRIPTOR' : _EXECUTORSTATES,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.ExecutorStates)
-  })
-_sym_db.RegisterMessage(ExecutorStates)
-
-ExecutorState = _reflection.GeneratedProtocolMessageType('ExecutorState', (_message.Message,), {
-  'DESCRIPTOR' : _EXECUTORSTATE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.ExecutorState)
-  })
-_sym_db.RegisterMessage(ExecutorState)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.features.feature_set_state_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.feature.store.features.apiP\001Z/qwak/featurestore/features;featurestorefeatures'
-  _FEATURESETLASTEXECUTIONSTATE._serialized_start=1905
-  _FEATURESETLASTEXECUTIONSTATE._serialized_end=2118
-  _FEATURESETSTATE._serialized_start=118
-  _FEATURESETSTATE._serialized_end=532
-  _FEATURESETJOBTYPE._serialized_start=535
-  _FEATURESETJOBTYPE._serialized_end=835
-  _STREAMING._serialized_start=838
-  _STREAMING._serialized_end=973
-  _ONLINE._serialized_start=975
-  _ONLINE._serialized_end=983
-  _OFFLINE._serialized_start=985
-  _OFFLINE._serialized_end=994
-  _BATCHV1._serialized_start=996
-  _BATCHV1._serialized_end=1005
-  _STREAMINGAGGREGATION._serialized_start=1008
-  _STREAMINGAGGREGATION._serialized_end=1231
-  _ROWLEVEL._serialized_start=1233
-  _ROWLEVEL._serialized_end=1243
-  _COMPACTION._serialized_start=1245
-  _COMPACTION._serialized_end=1257
-  _BACKFILLSUBTYPE._serialized_start=1259
-  _BACKFILLSUBTYPE._serialized_end=1276
-  _DELETIONJOB._serialized_start=1279
-  _DELETIONJOB._serialized_end=1597
-  _BATCHV1DELETIONJOBSUBTYPE._serialized_start=1599
-  _BATCHV1DELETIONJOBSUBTYPE._serialized_end=1626
-  _STREAMINGV1DELETIONJOBSUBTYPE._serialized_start=1628
-  _STREAMINGV1DELETIONJOBSUBTYPE._serialized_end=1659
-  _STREAMINGAGGREGATIONDELETIONJOBSUBTYPE._serialized_start=1661
-  _STREAMINGAGGREGATIONDELETIONJOBSUBTYPE._serialized_end=1701
-  _EXECUTORSTATES._serialized_start=1703
-  _EXECUTORSTATES._serialized_end=1788
-  _EXECUTORSTATE._serialized_start=1790
-  _EXECUTORSTATE._serialized_end=1902
+  _globals['_FEATURESETLASTEXECUTIONSTATE']._serialized_start=1905
+  _globals['_FEATURESETLASTEXECUTIONSTATE']._serialized_end=2118
+  _globals['_FEATURESETSTATE']._serialized_start=118
+  _globals['_FEATURESETSTATE']._serialized_end=532
+  _globals['_FEATURESETJOBTYPE']._serialized_start=535
+  _globals['_FEATURESETJOBTYPE']._serialized_end=835
+  _globals['_STREAMING']._serialized_start=838
+  _globals['_STREAMING']._serialized_end=973
+  _globals['_ONLINE']._serialized_start=975
+  _globals['_ONLINE']._serialized_end=983
+  _globals['_OFFLINE']._serialized_start=985
+  _globals['_OFFLINE']._serialized_end=994
+  _globals['_BATCHV1']._serialized_start=996
+  _globals['_BATCHV1']._serialized_end=1005
+  _globals['_STREAMINGAGGREGATION']._serialized_start=1008
+  _globals['_STREAMINGAGGREGATION']._serialized_end=1231
+  _globals['_ROWLEVEL']._serialized_start=1233
+  _globals['_ROWLEVEL']._serialized_end=1243
+  _globals['_COMPACTION']._serialized_start=1245
+  _globals['_COMPACTION']._serialized_end=1257
+  _globals['_BACKFILLSUBTYPE']._serialized_start=1259
+  _globals['_BACKFILLSUBTYPE']._serialized_end=1276
+  _globals['_DELETIONJOB']._serialized_start=1279
+  _globals['_DELETIONJOB']._serialized_end=1597
+  _globals['_BATCHV1DELETIONJOBSUBTYPE']._serialized_start=1599
+  _globals['_BATCHV1DELETIONJOBSUBTYPE']._serialized_end=1626
+  _globals['_STREAMINGV1DELETIONJOBSUBTYPE']._serialized_start=1628
+  _globals['_STREAMINGV1DELETIONJOBSUBTYPE']._serialized_end=1659
+  _globals['_STREAMINGAGGREGATIONDELETIONJOBSUBTYPE']._serialized_start=1661
+  _globals['_STREAMINGAGGREGATIONDELETIONJOBSUBTYPE']._serialized_end=1701
+  _globals['_EXECUTORSTATES']._serialized_start=1703
+  _globals['_EXECUTORSTATES']._serialized_end=1788
+  _globals['_EXECUTORSTATE']._serialized_start=1790
+  _globals['_EXECUTORSTATE']._serialized_end=1902
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/features/feature_set_state_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,113 +16,33 @@ from _qwak_proto.qwak.feature_store.features import feature_set_state_pb2 as qwa
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n;qwak/feature_store/features/feature_set_state_service.proto\x12\x1bqwak.feature.store.features\x1a\x33qwak/feature_store/features/feature_set_state.proto\"a\n\x1fGetFeatureSetStateByIdV1Request\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x0e\n\x06run_id\x18\x02 \x01(\x03\x12\x16\n\x0e\x65nvironment_id\x18\x03 \x01(\t\"k\n GetFeatureSetStateByIdV1Response\x12G\n\x11\x66\x65\x61ture_set_state\x18\x01 \x01(\x0b\x32,.qwak.feature.store.features.FeatureSetState\"e\n!GetFeatureSetStateByNameV1Request\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\t\x12\x0e\n\x06run_id\x18\x02 \x01(\x03\x12\x16\n\x0e\x65nvironment_id\x18\x03 \x01(\t\"m\n\"GetFeatureSetStateByNameV1Response\x12G\n\x11\x66\x65\x61ture_set_state\x18\x01 \x01(\x0b\x32,.qwak.feature.store.features.FeatureSetState\"\"\n GetFeatureSetStatesListV1Request\"l\n!GetFeatureSetStatesListV1Response\x12G\n\x11\x66\x65\x61ture_set_state\x18\x01 \x03(\x0b\x32,.qwak.feature.store.features.FeatureSetState\"\x81\x01\n\x1eUpdateFeatureSetStateV1Request\x12G\n\x11\x66\x65\x61ture_set_state\x18\x01 \x03(\x0b\x32,.qwak.feature.store.features.FeatureSetState\x12\x16\n\x0e\x65nvironment_id\x18\x02 \x01(\t\"!\n\x1fUpdateFeatureSetStateV1Response\"!\n\x1f\x44\x65leteFeatureSetStateV1Response\"8\n\x1e\x44\x65leteFeatureSetStateV1Request\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t2\x93\x06\n\x16\x46\x65\x61tureSetStateService\x12\x95\x01\n\x16GetFeatureSetStateById\x12<.qwak.feature.store.features.GetFeatureSetStateByIdV1Request\x1a=.qwak.feature.store.features.GetFeatureSetStateByIdV1Response\x12\x9b\x01\n\x18GetFeatureSetStateByName\x12>.qwak.feature.store.features.GetFeatureSetStateByNameV1Request\x1a?.qwak.feature.store.features.GetFeatureSetStateByNameV1Response\x12\x98\x01\n\x17GetFeatureSetStatesList\x12=.qwak.feature.store.features.GetFeatureSetStatesListV1Request\x1a>.qwak.feature.store.features.GetFeatureSetStatesListV1Response\x12\x92\x01\n\x15UpdateFeatureSetState\x12;.qwak.feature.store.features.UpdateFeatureSetStateV1Request\x1a<.qwak.feature.store.features.UpdateFeatureSetStateV1Response\x12\x92\x01\n\x15\x44\x65leteFeatureSetState\x12;.qwak.feature.store.features.DeleteFeatureSetStateV1Request\x1a<.qwak.feature.store.features.DeleteFeatureSetStateV1ResponseB]\n(com.qwak.ai.feature.store.management.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
 
-
-
-_GETFEATURESETSTATEBYIDV1REQUEST = DESCRIPTOR.message_types_by_name['GetFeatureSetStateByIdV1Request']
-_GETFEATURESETSTATEBYIDV1RESPONSE = DESCRIPTOR.message_types_by_name['GetFeatureSetStateByIdV1Response']
-_GETFEATURESETSTATEBYNAMEV1REQUEST = DESCRIPTOR.message_types_by_name['GetFeatureSetStateByNameV1Request']
-_GETFEATURESETSTATEBYNAMEV1RESPONSE = DESCRIPTOR.message_types_by_name['GetFeatureSetStateByNameV1Response']
-_GETFEATURESETSTATESLISTV1REQUEST = DESCRIPTOR.message_types_by_name['GetFeatureSetStatesListV1Request']
-_GETFEATURESETSTATESLISTV1RESPONSE = DESCRIPTOR.message_types_by_name['GetFeatureSetStatesListV1Response']
-_UPDATEFEATURESETSTATEV1REQUEST = DESCRIPTOR.message_types_by_name['UpdateFeatureSetStateV1Request']
-_UPDATEFEATURESETSTATEV1RESPONSE = DESCRIPTOR.message_types_by_name['UpdateFeatureSetStateV1Response']
-_DELETEFEATURESETSTATEV1RESPONSE = DESCRIPTOR.message_types_by_name['DeleteFeatureSetStateV1Response']
-_DELETEFEATURESETSTATEV1REQUEST = DESCRIPTOR.message_types_by_name['DeleteFeatureSetStateV1Request']
-GetFeatureSetStateByIdV1Request = _reflection.GeneratedProtocolMessageType('GetFeatureSetStateByIdV1Request', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYIDV1REQUEST,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.GetFeatureSetStateByIdV1Request)
-  })
-_sym_db.RegisterMessage(GetFeatureSetStateByIdV1Request)
-
-GetFeatureSetStateByIdV1Response = _reflection.GeneratedProtocolMessageType('GetFeatureSetStateByIdV1Response', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYIDV1RESPONSE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.GetFeatureSetStateByIdV1Response)
-  })
-_sym_db.RegisterMessage(GetFeatureSetStateByIdV1Response)
-
-GetFeatureSetStateByNameV1Request = _reflection.GeneratedProtocolMessageType('GetFeatureSetStateByNameV1Request', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYNAMEV1REQUEST,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.GetFeatureSetStateByNameV1Request)
-  })
-_sym_db.RegisterMessage(GetFeatureSetStateByNameV1Request)
-
-GetFeatureSetStateByNameV1Response = _reflection.GeneratedProtocolMessageType('GetFeatureSetStateByNameV1Response', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYNAMEV1RESPONSE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.GetFeatureSetStateByNameV1Response)
-  })
-_sym_db.RegisterMessage(GetFeatureSetStateByNameV1Response)
-
-GetFeatureSetStatesListV1Request = _reflection.GeneratedProtocolMessageType('GetFeatureSetStatesListV1Request', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATESLISTV1REQUEST,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.GetFeatureSetStatesListV1Request)
-  })
-_sym_db.RegisterMessage(GetFeatureSetStatesListV1Request)
-
-GetFeatureSetStatesListV1Response = _reflection.GeneratedProtocolMessageType('GetFeatureSetStatesListV1Response', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATESLISTV1RESPONSE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.GetFeatureSetStatesListV1Response)
-  })
-_sym_db.RegisterMessage(GetFeatureSetStatesListV1Response)
-
-UpdateFeatureSetStateV1Request = _reflection.GeneratedProtocolMessageType('UpdateFeatureSetStateV1Request', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEFEATURESETSTATEV1REQUEST,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.UpdateFeatureSetStateV1Request)
-  })
-_sym_db.RegisterMessage(UpdateFeatureSetStateV1Request)
-
-UpdateFeatureSetStateV1Response = _reflection.GeneratedProtocolMessageType('UpdateFeatureSetStateV1Response', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEFEATURESETSTATEV1RESPONSE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.UpdateFeatureSetStateV1Response)
-  })
-_sym_db.RegisterMessage(UpdateFeatureSetStateV1Response)
-
-DeleteFeatureSetStateV1Response = _reflection.GeneratedProtocolMessageType('DeleteFeatureSetStateV1Response', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEFEATURESETSTATEV1RESPONSE,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.DeleteFeatureSetStateV1Response)
-  })
-_sym_db.RegisterMessage(DeleteFeatureSetStateV1Response)
-
-DeleteFeatureSetStateV1Request = _reflection.GeneratedProtocolMessageType('DeleteFeatureSetStateV1Request', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEFEATURESETSTATEV1REQUEST,
-  '__module__' : 'qwak.feature_store.features.feature_set_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.DeleteFeatureSetStateV1Request)
-  })
-_sym_db.RegisterMessage(DeleteFeatureSetStateV1Request)
-
-_FEATURESETSTATESERVICE = DESCRIPTOR.services_by_name['FeatureSetStateService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.features.feature_set_state_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n(com.qwak.ai.feature.store.management.apiP\001Z/qwak/featurestore/features;featurestorefeatures'
-  _GETFEATURESETSTATEBYIDV1REQUEST._serialized_start=145
-  _GETFEATURESETSTATEBYIDV1REQUEST._serialized_end=242
-  _GETFEATURESETSTATEBYIDV1RESPONSE._serialized_start=244
-  _GETFEATURESETSTATEBYIDV1RESPONSE._serialized_end=351
-  _GETFEATURESETSTATEBYNAMEV1REQUEST._serialized_start=353
-  _GETFEATURESETSTATEBYNAMEV1REQUEST._serialized_end=454
-  _GETFEATURESETSTATEBYNAMEV1RESPONSE._serialized_start=456
-  _GETFEATURESETSTATEBYNAMEV1RESPONSE._serialized_end=565
-  _GETFEATURESETSTATESLISTV1REQUEST._serialized_start=567
-  _GETFEATURESETSTATESLISTV1REQUEST._serialized_end=601
-  _GETFEATURESETSTATESLISTV1RESPONSE._serialized_start=603
-  _GETFEATURESETSTATESLISTV1RESPONSE._serialized_end=711
-  _UPDATEFEATURESETSTATEV1REQUEST._serialized_start=714
-  _UPDATEFEATURESETSTATEV1REQUEST._serialized_end=843
-  _UPDATEFEATURESETSTATEV1RESPONSE._serialized_start=845
-  _UPDATEFEATURESETSTATEV1RESPONSE._serialized_end=878
-  _DELETEFEATURESETSTATEV1RESPONSE._serialized_start=880
-  _DELETEFEATURESETSTATEV1RESPONSE._serialized_end=913
-  _DELETEFEATURESETSTATEV1REQUEST._serialized_start=915
-  _DELETEFEATURESETSTATEV1REQUEST._serialized_end=971
-  _FEATURESETSTATESERVICE._serialized_start=974
-  _FEATURESETSTATESERVICE._serialized_end=1761
+  _globals['_GETFEATURESETSTATEBYIDV1REQUEST']._serialized_start=145
+  _globals['_GETFEATURESETSTATEBYIDV1REQUEST']._serialized_end=242
+  _globals['_GETFEATURESETSTATEBYIDV1RESPONSE']._serialized_start=244
+  _globals['_GETFEATURESETSTATEBYIDV1RESPONSE']._serialized_end=351
+  _globals['_GETFEATURESETSTATEBYNAMEV1REQUEST']._serialized_start=353
+  _globals['_GETFEATURESETSTATEBYNAMEV1REQUEST']._serialized_end=454
+  _globals['_GETFEATURESETSTATEBYNAMEV1RESPONSE']._serialized_start=456
+  _globals['_GETFEATURESETSTATEBYNAMEV1RESPONSE']._serialized_end=565
+  _globals['_GETFEATURESETSTATESLISTV1REQUEST']._serialized_start=567
+  _globals['_GETFEATURESETSTATESLISTV1REQUEST']._serialized_end=601
+  _globals['_GETFEATURESETSTATESLISTV1RESPONSE']._serialized_start=603
+  _globals['_GETFEATURESETSTATESLISTV1RESPONSE']._serialized_end=711
+  _globals['_UPDATEFEATURESETSTATEV1REQUEST']._serialized_start=714
+  _globals['_UPDATEFEATURESETSTATEV1REQUEST']._serialized_end=843
+  _globals['_UPDATEFEATURESETSTATEV1RESPONSE']._serialized_start=845
+  _globals['_UPDATEFEATURESETSTATEV1RESPONSE']._serialized_end=878
+  _globals['_DELETEFEATURESETSTATEV1RESPONSE']._serialized_start=880
+  _globals['_DELETEFEATURESETSTATEV1RESPONSE']._serialized_end=913
+  _globals['_DELETEFEATURESETSTATEV1REQUEST']._serialized_start=915
+  _globals['_DELETEFEATURESETSTATEV1REQUEST']._serialized_end=971
+  _globals['_FEATURESETSTATESERVICE']._serialized_start=974
+  _globals['_FEATURESETSTATESERVICE']._serialized_end=1761
 # @@protoc_insertion_point(module_scope)