qwak-core 0.4.284__py3-none-any.whl → 0.4.285__py3-none-any.whl

_qwak_proto/qwak/execution/v1/state/execution_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()
@@ -18,133 +17,37 @@ from _qwak_proto.qwak.execution.v1.state import featureset_state_pb2 as qwak_dot
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n5qwak/execution/v1/state/execution_state_service.proto\x12%qwak.feature.store.execution.v1.state\x1a-qwak/execution/v1/state/execution_state.proto\x1a.qwak/execution/v1/state/featureset_state.proto\"6\n\x1dGetFeaturesetStateByIdRequest\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\"r\n\x1eGetFeaturesetStateByIdResponse\x12P\n\x10\x66\x65\x61tureset_state\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\":\n\x1fGetFeaturesetStateByNameRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"t\n GetFeaturesetStateByNameResponse\x12P\n\x10\x66\x65\x61tureset_state\x18\x01 \x01(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\"=\n#GetMultiFeaturesetStateByIdsRequest\x12\x16\n\x0e\x66\x65\x61tureset_ids\x18\x01 \x03(\t\"y\n$GetMultiFeaturesetStateByIdsResponse\x12Q\n\x11\x66\x65\x61tureset_states\x18\x01 \x03(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\"?\n$GetMultiFeaturesetStateByNameRequest\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x03(\t\"y\n%GetMultiFeaturesetStateByNameResponse\x12P\n\x10\x66\x65\x61tureset_state\x18\x01 \x03(\x0b\x32\x36.qwak.feature.store.execution.v1.state.FeaturesetState\"r\n UpdateMultiExecutionStateRequest\x12N\n\x0f\x65xecution_state\x18\x01 \x03(\x0b\x32\x35.qwak.feature.store.execution.v1.state.ExecutionState\"#\n!UpdateMultiExecutionStateResponse\"9\n DeleteFeaturesetStateByIdRequest\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\"@\n!DeleteFeaturesetStateByIdResponse\x12\x1b\n\x13num_deleted_records\x18\x01 \x01(\x03\x32\xd2\x08\n!FeatureStoreExecutionStateService\x12\xa5\x01\n\x16GetFeaturesetStateById\x12\x44.qwak.feature.store.execution.v1.state.GetFeaturesetStateByIdRequest\x1a\x45.qwak.feature.store.execution.v1.state.GetFeaturesetStateByIdResponse\x12\xab\x01\n\x18GetFeaturesetStateByName\x12\x46.qwak.feature.store.execution.v1.state.GetFeaturesetStateByNameRequest\x1aG.qwak.feature.store.execution.v1.state.GetFeaturesetStateByNameResponse\x12\xb7\x01\n\x1cGetMultiFeaturesetStateByIds\x12J.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByIdsRequest\x1aK.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByIdsResponse\x12\xba\x01\n\x1dGetMultiFeaturesetStateByName\x12K.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByNameRequest\x1aL.qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByNameResponse\x12\xae\x01\n\x19UpdateMultiExecutionState\x12G.qwak.feature.store.execution.v1.state.UpdateMultiExecutionStateRequest\x1aH.qwak.feature.store.execution.v1.state.UpdateMultiExecutionStateResponse\x12\xae\x01\n\x19\x44\x65leteFeaturesetStateById\x12G.qwak.feature.store.execution.v1.state.DeleteFeaturesetStateByIdRequest\x1aH.qwak.feature.store.execution.v1.state.DeleteFeaturesetStateByIdResponseB`\n+com.qwak.ai.features.execution.api.v1.stateP\x01Z/qwak/execution_management/state;execution_stateb\x06proto3')
 
-
-
-_GETFEATURESETSTATEBYIDREQUEST = DESCRIPTOR.message_types_by_name['GetFeaturesetStateByIdRequest']
-_GETFEATURESETSTATEBYIDRESPONSE = DESCRIPTOR.message_types_by_name['GetFeaturesetStateByIdResponse']
-_GETFEATURESETSTATEBYNAMEREQUEST = DESCRIPTOR.message_types_by_name['GetFeaturesetStateByNameRequest']
-_GETFEATURESETSTATEBYNAMERESPONSE = DESCRIPTOR.message_types_by_name['GetFeaturesetStateByNameResponse']
-_GETMULTIFEATURESETSTATEBYIDSREQUEST = DESCRIPTOR.message_types_by_name['GetMultiFeaturesetStateByIdsRequest']
-_GETMULTIFEATURESETSTATEBYIDSRESPONSE = DESCRIPTOR.message_types_by_name['GetMultiFeaturesetStateByIdsResponse']
-_GETMULTIFEATURESETSTATEBYNAMEREQUEST = DESCRIPTOR.message_types_by_name['GetMultiFeaturesetStateByNameRequest']
-_GETMULTIFEATURESETSTATEBYNAMERESPONSE = DESCRIPTOR.message_types_by_name['GetMultiFeaturesetStateByNameResponse']
-_UPDATEMULTIEXECUTIONSTATEREQUEST = DESCRIPTOR.message_types_by_name['UpdateMultiExecutionStateRequest']
-_UPDATEMULTIEXECUTIONSTATERESPONSE = DESCRIPTOR.message_types_by_name['UpdateMultiExecutionStateResponse']
-_DELETEFEATURESETSTATEBYIDREQUEST = DESCRIPTOR.message_types_by_name['DeleteFeaturesetStateByIdRequest']
-_DELETEFEATURESETSTATEBYIDRESPONSE = DESCRIPTOR.message_types_by_name['DeleteFeaturesetStateByIdResponse']
-GetFeaturesetStateByIdRequest = _reflection.GeneratedProtocolMessageType('GetFeaturesetStateByIdRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYIDREQUEST,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetFeaturesetStateByIdRequest)
-  })
-_sym_db.RegisterMessage(GetFeaturesetStateByIdRequest)
-
-GetFeaturesetStateByIdResponse = _reflection.GeneratedProtocolMessageType('GetFeaturesetStateByIdResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYIDRESPONSE,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetFeaturesetStateByIdResponse)
-  })
-_sym_db.RegisterMessage(GetFeaturesetStateByIdResponse)
-
-GetFeaturesetStateByNameRequest = _reflection.GeneratedProtocolMessageType('GetFeaturesetStateByNameRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYNAMEREQUEST,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetFeaturesetStateByNameRequest)
-  })
-_sym_db.RegisterMessage(GetFeaturesetStateByNameRequest)
-
-GetFeaturesetStateByNameResponse = _reflection.GeneratedProtocolMessageType('GetFeaturesetStateByNameResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETFEATURESETSTATEBYNAMERESPONSE,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetFeaturesetStateByNameResponse)
-  })
-_sym_db.RegisterMessage(GetFeaturesetStateByNameResponse)
-
-GetMultiFeaturesetStateByIdsRequest = _reflection.GeneratedProtocolMessageType('GetMultiFeaturesetStateByIdsRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETMULTIFEATURESETSTATEBYIDSREQUEST,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByIdsRequest)
-  })
-_sym_db.RegisterMessage(GetMultiFeaturesetStateByIdsRequest)
-
-GetMultiFeaturesetStateByIdsResponse = _reflection.GeneratedProtocolMessageType('GetMultiFeaturesetStateByIdsResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETMULTIFEATURESETSTATEBYIDSRESPONSE,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByIdsResponse)
-  })
-_sym_db.RegisterMessage(GetMultiFeaturesetStateByIdsResponse)
-
-GetMultiFeaturesetStateByNameRequest = _reflection.GeneratedProtocolMessageType('GetMultiFeaturesetStateByNameRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETMULTIFEATURESETSTATEBYNAMEREQUEST,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByNameRequest)
-  })
-_sym_db.RegisterMessage(GetMultiFeaturesetStateByNameRequest)
-
-GetMultiFeaturesetStateByNameResponse = _reflection.GeneratedProtocolMessageType('GetMultiFeaturesetStateByNameResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETMULTIFEATURESETSTATEBYNAMERESPONSE,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.GetMultiFeaturesetStateByNameResponse)
-  })
-_sym_db.RegisterMessage(GetMultiFeaturesetStateByNameResponse)
-
-UpdateMultiExecutionStateRequest = _reflection.GeneratedProtocolMessageType('UpdateMultiExecutionStateRequest', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEMULTIEXECUTIONSTATEREQUEST,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.UpdateMultiExecutionStateRequest)
-  })
-_sym_db.RegisterMessage(UpdateMultiExecutionStateRequest)
-
-UpdateMultiExecutionStateResponse = _reflection.GeneratedProtocolMessageType('UpdateMultiExecutionStateResponse', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEMULTIEXECUTIONSTATERESPONSE,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.UpdateMultiExecutionStateResponse)
-  })
-_sym_db.RegisterMessage(UpdateMultiExecutionStateResponse)
-
-DeleteFeaturesetStateByIdRequest = _reflection.GeneratedProtocolMessageType('DeleteFeaturesetStateByIdRequest', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEFEATURESETSTATEBYIDREQUEST,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.DeleteFeaturesetStateByIdRequest)
-  })
-_sym_db.RegisterMessage(DeleteFeaturesetStateByIdRequest)
-
-DeleteFeaturesetStateByIdResponse = _reflection.GeneratedProtocolMessageType('DeleteFeaturesetStateByIdResponse', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEFEATURESETSTATEBYIDRESPONSE,
-  '__module__' : 'qwak.execution.v1.state.execution_state_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.DeleteFeaturesetStateByIdResponse)
-  })
-_sym_db.RegisterMessage(DeleteFeaturesetStateByIdResponse)
-
-_FEATURESTOREEXECUTIONSTATESERVICE = DESCRIPTOR.services_by_name['FeatureStoreExecutionStateService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.execution.v1.state.execution_state_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.features.execution.api.v1.stateP\001Z/qwak/execution_management/state;execution_state'
-  _GETFEATURESETSTATEBYIDREQUEST._serialized_start=191
-  _GETFEATURESETSTATEBYIDREQUEST._serialized_end=245
-  _GETFEATURESETSTATEBYIDRESPONSE._serialized_start=247
-  _GETFEATURESETSTATEBYIDRESPONSE._serialized_end=361
-  _GETFEATURESETSTATEBYNAMEREQUEST._serialized_start=363
-  _GETFEATURESETSTATEBYNAMEREQUEST._serialized_end=421
-  _GETFEATURESETSTATEBYNAMERESPONSE._serialized_start=423
-  _GETFEATURESETSTATEBYNAMERESPONSE._serialized_end=539
-  _GETMULTIFEATURESETSTATEBYIDSREQUEST._serialized_start=541
-  _GETMULTIFEATURESETSTATEBYIDSREQUEST._serialized_end=602
-  _GETMULTIFEATURESETSTATEBYIDSRESPONSE._serialized_start=604
-  _GETMULTIFEATURESETSTATEBYIDSRESPONSE._serialized_end=725
-  _GETMULTIFEATURESETSTATEBYNAMEREQUEST._serialized_start=727
-  _GETMULTIFEATURESETSTATEBYNAMEREQUEST._serialized_end=790
-  _GETMULTIFEATURESETSTATEBYNAMERESPONSE._serialized_start=792
-  _GETMULTIFEATURESETSTATEBYNAMERESPONSE._serialized_end=913
-  _UPDATEMULTIEXECUTIONSTATEREQUEST._serialized_start=915
-  _UPDATEMULTIEXECUTIONSTATEREQUEST._serialized_end=1029
-  _UPDATEMULTIEXECUTIONSTATERESPONSE._serialized_start=1031
-  _UPDATEMULTIEXECUTIONSTATERESPONSE._serialized_end=1066
-  _DELETEFEATURESETSTATEBYIDREQUEST._serialized_start=1068
-  _DELETEFEATURESETSTATEBYIDREQUEST._serialized_end=1125
-  _DELETEFEATURESETSTATEBYIDRESPONSE._serialized_start=1127
-  _DELETEFEATURESETSTATEBYIDRESPONSE._serialized_end=1191
-  _FEATURESTOREEXECUTIONSTATESERVICE._serialized_start=1194
-  _FEATURESTOREEXECUTIONSTATESERVICE._serialized_end=2300
+  _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_start=191
+  _globals['_GETFEATURESETSTATEBYIDREQUEST']._serialized_end=245
+  _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_start=247
+  _globals['_GETFEATURESETSTATEBYIDRESPONSE']._serialized_end=361
+  _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_start=363
+  _globals['_GETFEATURESETSTATEBYNAMEREQUEST']._serialized_end=421
+  _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_start=423
+  _globals['_GETFEATURESETSTATEBYNAMERESPONSE']._serialized_end=539
+  _globals['_GETMULTIFEATURESETSTATEBYIDSREQUEST']._serialized_start=541
+  _globals['_GETMULTIFEATURESETSTATEBYIDSREQUEST']._serialized_end=602
+  _globals['_GETMULTIFEATURESETSTATEBYIDSRESPONSE']._serialized_start=604
+  _globals['_GETMULTIFEATURESETSTATEBYIDSRESPONSE']._serialized_end=725
+  _globals['_GETMULTIFEATURESETSTATEBYNAMEREQUEST']._serialized_start=727
+  _globals['_GETMULTIFEATURESETSTATEBYNAMEREQUEST']._serialized_end=790
+  _globals['_GETMULTIFEATURESETSTATEBYNAMERESPONSE']._serialized_start=792
+  _globals['_GETMULTIFEATURESETSTATEBYNAMERESPONSE']._serialized_end=913
+  _globals['_UPDATEMULTIEXECUTIONSTATEREQUEST']._serialized_start=915
+  _globals['_UPDATEMULTIEXECUTIONSTATEREQUEST']._serialized_end=1029
+  _globals['_UPDATEMULTIEXECUTIONSTATERESPONSE']._serialized_start=1031
+  _globals['_UPDATEMULTIEXECUTIONSTATERESPONSE']._serialized_end=1066
+  _globals['_DELETEFEATURESETSTATEBYIDREQUEST']._serialized_start=1068
+  _globals['_DELETEFEATURESETSTATEBYIDREQUEST']._serialized_end=1125
+  _globals['_DELETEFEATURESETSTATEBYIDRESPONSE']._serialized_start=1127
+  _globals['_DELETEFEATURESETSTATEBYIDRESPONSE']._serialized_end=1191
+  _globals['_FEATURESTOREEXECUTIONSTATESERVICE']._serialized_start=1194
+  _globals['_FEATURESTOREEXECUTIONSTATESERVICE']._serialized_end=2300
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/execution/v1/state/featureset_state_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/execution/v1/state/featureset_state.proto\x12%qwak.feature.store.execution.v1.state\"A\n\x0f\x46\x65\x61turesetState\x12\x15\n\rfeatureset_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\tB`\n+com.qwak.ai.features.execution.api.v1.stateP\x01Z/qwak/execution_management/state;execution_stateb\x06proto3')
 
-
-
-_FEATURESETSTATE = DESCRIPTOR.message_types_by_name['FeaturesetState']
-FeaturesetState = _reflection.GeneratedProtocolMessageType('FeaturesetState', (_message.Message,), {
-  'DESCRIPTOR' : _FEATURESETSTATE,
-  '__module__' : 'qwak.execution.v1.state.featureset_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.FeaturesetState)
-  })
-_sym_db.RegisterMessage(FeaturesetState)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.execution.v1.state.featureset_state_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.features.execution.api.v1.stateP\001Z/qwak/execution_management/state;execution_state'
-  _FEATURESETSTATE._serialized_start=89
-  _FEATURESETSTATE._serialized_end=154
+  _globals['_FEATURESETSTATE']._serialized_start=89
+  _globals['_FEATURESETSTATE']._serialized_end=154
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/execution/v1/state/spark_execution_state_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/execution/v1/state/spark_execution_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()
@@ -17,75 +15,23 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n3qwak/execution/v1/state/spark_execution_state.proto\x12%qwak.feature.store.execution.v1.state\"\x8f\x03\n\x13SparkExecutionState\x12\x16\n\x0e\x61pplication_id\x18\x01 \x01(\t\x12S\n\x16spark_application_type\x18\x02 \x01(\x0e\x32\x33.qwak.feature.store.execution.v1.state.SparkAppType\x12O\n\x0fspark_app_state\x18\x03 \x01(\x0b\x32\x34.qwak.feature.store.execution.v1.state.SparkAppStateH\x00\x12l\n\x11\x61pplicationLabels\x18\x04 \x03(\x0b\x32Q.qwak.feature.store.execution.v1.state.SparkExecutionState.ApplicationLabelsEntry\x1a\x38\n\x16\x41pplicationLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x12\n\x10state_descriptor\"t\n\rSparkAppState\x12L\n\nstate_type\x18\x01 \x01(\x0e\x32\x38.qwak.feature.store.execution.v1.state.SparkAppStateType\x12\x15\n\rerror_message\x18\x02 \x01(\t*\xdf\x03\n\x0cSparkAppType\x12\x1a\n\x16SPARK_APP_TYPE_INVALID\x10\x00\x12\"\n\x1eSPARK_APP_TYPE_BATCH_INGESTION\x10\x01\x12!\n\x1dSPARK_APP_TYPE_BATCH_BACKFILL\x10\x02\x12!\n\x1dSPARK_APP_TYPE_BATCH_DELETION\x10\x03\x12#\n\x1fSPARK_APP_TYPE_STREAMING_ONLINE\x10\x04\x12$\n SPARK_APP_TYPE_STREAMING_OFFLINE\x10\x05\x12%\n!SPARK_APP_TYPE_STREAMING_BACKFILL\x10\x06\x12%\n!SPARK_APP_TYPE_STREAMING_DELETION\x10\x07\x12*\n&SPARK_APP_TYPE_STREAMING_AGGR_ROWLEVEL\x10\x08\x12,\n(SPARK_APP_TYPE_STREAMING_AGGR_COMPACTION\x10\t\x12*\n&SPARK_APP_TYPE_STREAMING_AGGR_BACKFILL\x10\n\x12*\n&SPARK_APP_TYPE_STREAMING_AGGR_DELETION\x10\x0b*\x88\x03\n\x11SparkAppStateType\x12\x1b\n\x17SPARK_APP_STATE_INVALID\x10\x00\x12\x1d\n\x19SPARK_APP_STATE_COMPLETED\x10\x01\x12\x1a\n\x16SPARK_APP_STATE_FAILED\x10\x02\x12%\n!SPARK_APP_STATE_SUBMISSION_FAILED\x10\x03\x12\x1b\n\x17SPARK_APP_STATE_FAILING\x10\x04\x12 \n\x1cSPARK_APP_STATE_INVALIDATING\x10\x05\x12\x19\n\x15SPARK_APP_STATE_EMPTY\x10\x06\x12!\n\x1dSPARK_APP_STATE_PENDING_RERUN\x10\x07\x12\x1b\n\x17SPARK_APP_STATE_RUNNING\x10\x08\x12\x1d\n\x19SPARK_APP_STATE_SUBMITTED\x10\t\x12\x1e\n\x1aSPARK_APP_STATE_SUCCEEDING\x10\n\x12\x1b\n\x17SPARK_APP_STATE_UNKNOWN\x10\x0b\x42`\n+com.qwak.ai.features.execution.api.v1.stateP\x01Z/qwak/execution_management/state;execution_stateb\x06proto3')
 
-_SPARKAPPTYPE = DESCRIPTOR.enum_types_by_name['SparkAppType']
-SparkAppType = enum_type_wrapper.EnumTypeWrapper(_SPARKAPPTYPE)
-_SPARKAPPSTATETYPE = DESCRIPTOR.enum_types_by_name['SparkAppStateType']
-SparkAppStateType = enum_type_wrapper.EnumTypeWrapper(_SPARKAPPSTATETYPE)
-SPARK_APP_TYPE_INVALID = 0
-SPARK_APP_TYPE_BATCH_INGESTION = 1
-SPARK_APP_TYPE_BATCH_BACKFILL = 2
-SPARK_APP_TYPE_BATCH_DELETION = 3
-SPARK_APP_TYPE_STREAMING_ONLINE = 4
-SPARK_APP_TYPE_STREAMING_OFFLINE = 5
-SPARK_APP_TYPE_STREAMING_BACKFILL = 6
-SPARK_APP_TYPE_STREAMING_DELETION = 7
-SPARK_APP_TYPE_STREAMING_AGGR_ROWLEVEL = 8
-SPARK_APP_TYPE_STREAMING_AGGR_COMPACTION = 9
-SPARK_APP_TYPE_STREAMING_AGGR_BACKFILL = 10
-SPARK_APP_TYPE_STREAMING_AGGR_DELETION = 11
-SPARK_APP_STATE_INVALID = 0
-SPARK_APP_STATE_COMPLETED = 1
-SPARK_APP_STATE_FAILED = 2
-SPARK_APP_STATE_SUBMISSION_FAILED = 3
-SPARK_APP_STATE_FAILING = 4
-SPARK_APP_STATE_INVALIDATING = 5
-SPARK_APP_STATE_EMPTY = 6
-SPARK_APP_STATE_PENDING_RERUN = 7
-SPARK_APP_STATE_RUNNING = 8
-SPARK_APP_STATE_SUBMITTED = 9
-SPARK_APP_STATE_SUCCEEDING = 10
-SPARK_APP_STATE_UNKNOWN = 11
-
-
-_SPARKEXECUTIONSTATE = DESCRIPTOR.message_types_by_name['SparkExecutionState']
-_SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY = _SPARKEXECUTIONSTATE.nested_types_by_name['ApplicationLabelsEntry']
-_SPARKAPPSTATE = DESCRIPTOR.message_types_by_name['SparkAppState']
-SparkExecutionState = _reflection.GeneratedProtocolMessageType('SparkExecutionState', (_message.Message,), {
-
-  'ApplicationLabelsEntry' : _reflection.GeneratedProtocolMessageType('ApplicationLabelsEntry', (_message.Message,), {
-    'DESCRIPTOR' : _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY,
-    '__module__' : 'qwak.execution.v1.state.spark_execution_state_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.SparkExecutionState.ApplicationLabelsEntry)
-    })
-  ,
-  'DESCRIPTOR' : _SPARKEXECUTIONSTATE,
-  '__module__' : 'qwak.execution.v1.state.spark_execution_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.SparkExecutionState)
-  })
-_sym_db.RegisterMessage(SparkExecutionState)
-_sym_db.RegisterMessage(SparkExecutionState.ApplicationLabelsEntry)
-
-SparkAppState = _reflection.GeneratedProtocolMessageType('SparkAppState', (_message.Message,), {
-  'DESCRIPTOR' : _SPARKAPPSTATE,
-  '__module__' : 'qwak.execution.v1.state.spark_execution_state_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.state.SparkAppState)
-  })
-_sym_db.RegisterMessage(SparkAppState)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.execution.v1.state.spark_execution_state_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.features.execution.api.v1.stateP\001Z/qwak/execution_management/state;execution_state'
   _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._options = None
   _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._serialized_options = b'8\001'
-  _SPARKAPPTYPE._serialized_start=615
-  _SPARKAPPTYPE._serialized_end=1094
-  _SPARKAPPSTATETYPE._serialized_start=1097
-  _SPARKAPPSTATETYPE._serialized_end=1489
-  _SPARKEXECUTIONSTATE._serialized_start=95
-  _SPARKEXECUTIONSTATE._serialized_end=494
-  _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._serialized_start=418
-  _SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY._serialized_end=474
-  _SPARKAPPSTATE._serialized_start=496
-  _SPARKAPPSTATE._serialized_end=612
+  _globals['_SPARKAPPTYPE']._serialized_start=615
+  _globals['_SPARKAPPTYPE']._serialized_end=1094
+  _globals['_SPARKAPPSTATETYPE']._serialized_start=1097
+  _globals['_SPARKAPPSTATETYPE']._serialized_end=1489
+  _globals['_SPARKEXECUTIONSTATE']._serialized_start=95
+  _globals['_SPARKEXECUTIONSTATE']._serialized_end=494
+  _globals['_SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY']._serialized_start=418
+  _globals['_SPARKEXECUTIONSTATE_APPLICATIONLABELSENTRY']._serialized_end=474
+  _globals['_SPARKAPPSTATE']._serialized_start=496
+  _globals['_SPARKAPPSTATE']._serialized_end=612
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/execution/v1/streaming_aggregation_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,60 +15,21 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/execution/v1/streaming_aggregation.proto\x12\x1fqwak.feature.store.execution.v1\"@\n%RowLevelStreamingAggregationIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"\xf8\x01\n\'CompactionStreamingAggregationIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12T\n\x14scheduled_compaction\x18\x02 \x01(\x0b\x32\x34.qwak.feature.store.execution.v1.ScheduledCompactionH\x00\x12N\n\x11manual_compaction\x18\x03 \x01(\x0b\x32\x31.qwak.feature.store.execution.v1.ManualCompactionH\x00\x42\x0e\n\x0ctrigger_type\"\x15\n\x13ScheduledCompaction\"\x12\n\x10ManualCompaction\"@\n%StreamingAggregationBackfillIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\tB)\n%com.qwak.ai.features.execution.api.v1P\x01\x62\x06proto3')
 
-
-
-_ROWLEVELSTREAMINGAGGREGATIONINGESTION = DESCRIPTOR.message_types_by_name['RowLevelStreamingAggregationIngestion']
-_COMPACTIONSTREAMINGAGGREGATIONINGESTION = DESCRIPTOR.message_types_by_name['CompactionStreamingAggregationIngestion']
-_SCHEDULEDCOMPACTION = DESCRIPTOR.message_types_by_name['ScheduledCompaction']
-_MANUALCOMPACTION = DESCRIPTOR.message_types_by_name['ManualCompaction']
-_STREAMINGAGGREGATIONBACKFILLINGESTION = DESCRIPTOR.message_types_by_name['StreamingAggregationBackfillIngestion']
-RowLevelStreamingAggregationIngestion = _reflection.GeneratedProtocolMessageType('RowLevelStreamingAggregationIngestion', (_message.Message,), {
-  'DESCRIPTOR' : _ROWLEVELSTREAMINGAGGREGATIONINGESTION,
-  '__module__' : 'qwak.execution.v1.streaming_aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.RowLevelStreamingAggregationIngestion)
-  })
-_sym_db.RegisterMessage(RowLevelStreamingAggregationIngestion)
-
-CompactionStreamingAggregationIngestion = _reflection.GeneratedProtocolMessageType('CompactionStreamingAggregationIngestion', (_message.Message,), {
-  'DESCRIPTOR' : _COMPACTIONSTREAMINGAGGREGATIONINGESTION,
-  '__module__' : 'qwak.execution.v1.streaming_aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.CompactionStreamingAggregationIngestion)
-  })
-_sym_db.RegisterMessage(CompactionStreamingAggregationIngestion)
-
-ScheduledCompaction = _reflection.GeneratedProtocolMessageType('ScheduledCompaction', (_message.Message,), {
-  'DESCRIPTOR' : _SCHEDULEDCOMPACTION,
-  '__module__' : 'qwak.execution.v1.streaming_aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.ScheduledCompaction)
-  })
-_sym_db.RegisterMessage(ScheduledCompaction)
-
-ManualCompaction = _reflection.GeneratedProtocolMessageType('ManualCompaction', (_message.Message,), {
-  'DESCRIPTOR' : _MANUALCOMPACTION,
-  '__module__' : 'qwak.execution.v1.streaming_aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.ManualCompaction)
-  })
-_sym_db.RegisterMessage(ManualCompaction)
-
-StreamingAggregationBackfillIngestion = _reflection.GeneratedProtocolMessageType('StreamingAggregationBackfillIngestion', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGAGGREGATIONBACKFILLINGESTION,
-  '__module__' : 'qwak.execution.v1.streaming_aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.StreamingAggregationBackfillIngestion)
-  })
-_sym_db.RegisterMessage(StreamingAggregationBackfillIngestion)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.execution.v1.streaming_aggregation_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.features.execution.api.v1P\001'
-  _ROWLEVELSTREAMINGAGGREGATIONINGESTION._serialized_start=82
-  _ROWLEVELSTREAMINGAGGREGATIONINGESTION._serialized_end=146
-  _COMPACTIONSTREAMINGAGGREGATIONINGESTION._serialized_start=149
-  _COMPACTIONSTREAMINGAGGREGATIONINGESTION._serialized_end=397
-  _SCHEDULEDCOMPACTION._serialized_start=399
-  _SCHEDULEDCOMPACTION._serialized_end=420
-  _MANUALCOMPACTION._serialized_start=422
-  _MANUALCOMPACTION._serialized_end=440
-  _STREAMINGAGGREGATIONBACKFILLINGESTION._serialized_start=442
-  _STREAMINGAGGREGATIONBACKFILLINGESTION._serialized_end=506
+  _globals['_ROWLEVELSTREAMINGAGGREGATIONINGESTION']._serialized_start=82
+  _globals['_ROWLEVELSTREAMINGAGGREGATIONINGESTION']._serialized_end=146
+  _globals['_COMPACTIONSTREAMINGAGGREGATIONINGESTION']._serialized_start=149
+  _globals['_COMPACTIONSTREAMINGAGGREGATIONINGESTION']._serialized_end=397
+  _globals['_SCHEDULEDCOMPACTION']._serialized_start=399
+  _globals['_SCHEDULEDCOMPACTION']._serialized_end=420
+  _globals['_MANUALCOMPACTION']._serialized_start=422
+  _globals['_MANUALCOMPACTION']._serialized_end=440
+  _globals['_STREAMINGAGGREGATIONBACKFILLINGESTION']._serialized_start=442
+  _globals['_STREAMINGAGGREGATIONBACKFILLINGESTION']._serialized_end=506
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/execution/v1/streaming_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,50 +15,19 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!qwak/execution/v1/streaming.proto\x12\x1fqwak.feature.store.execution.v1\"3\n\x18OnlineStreamingIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\"\xfe\x01\n\x19OfflineStreamingIngestion\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x01 \x01(\t\x12^\n\x11scheduled_trigger\x18\x02 \x01(\x0b\x32\x41.qwak.feature.store.execution.v1.ScheduledOfflineStreamingTriggerH\x00\x12X\n\x0emanual_trigger\x18\x03 \x01(\x0b\x32>.qwak.feature.store.execution.v1.ManualOfflineStreamingTriggerH\x00\x42\x0e\n\x0ctrigger_type\"\"\n ScheduledOfflineStreamingTrigger\"\x1f\n\x1dManualOfflineStreamingTriggerB)\n%com.qwak.ai.features.execution.api.v1P\x01\x62\x06proto3')
 
-
-
-_ONLINESTREAMINGINGESTION = DESCRIPTOR.message_types_by_name['OnlineStreamingIngestion']
-_OFFLINESTREAMINGINGESTION = DESCRIPTOR.message_types_by_name['OfflineStreamingIngestion']
-_SCHEDULEDOFFLINESTREAMINGTRIGGER = DESCRIPTOR.message_types_by_name['ScheduledOfflineStreamingTrigger']
-_MANUALOFFLINESTREAMINGTRIGGER = DESCRIPTOR.message_types_by_name['ManualOfflineStreamingTrigger']
-OnlineStreamingIngestion = _reflection.GeneratedProtocolMessageType('OnlineStreamingIngestion', (_message.Message,), {
-  'DESCRIPTOR' : _ONLINESTREAMINGINGESTION,
-  '__module__' : 'qwak.execution.v1.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.OnlineStreamingIngestion)
-  })
-_sym_db.RegisterMessage(OnlineStreamingIngestion)
-
-OfflineStreamingIngestion = _reflection.GeneratedProtocolMessageType('OfflineStreamingIngestion', (_message.Message,), {
-  'DESCRIPTOR' : _OFFLINESTREAMINGINGESTION,
-  '__module__' : 'qwak.execution.v1.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.OfflineStreamingIngestion)
-  })
-_sym_db.RegisterMessage(OfflineStreamingIngestion)
-
-ScheduledOfflineStreamingTrigger = _reflection.GeneratedProtocolMessageType('ScheduledOfflineStreamingTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _SCHEDULEDOFFLINESTREAMINGTRIGGER,
-  '__module__' : 'qwak.execution.v1.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.ScheduledOfflineStreamingTrigger)
-  })
-_sym_db.RegisterMessage(ScheduledOfflineStreamingTrigger)
-
-ManualOfflineStreamingTrigger = _reflection.GeneratedProtocolMessageType('ManualOfflineStreamingTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _MANUALOFFLINESTREAMINGTRIGGER,
-  '__module__' : 'qwak.execution.v1.streaming_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.execution.v1.ManualOfflineStreamingTrigger)
-  })
-_sym_db.RegisterMessage(ManualOfflineStreamingTrigger)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.execution.v1.streaming_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.features.execution.api.v1P\001'
-  _ONLINESTREAMINGINGESTION._serialized_start=70
-  _ONLINESTREAMINGINGESTION._serialized_end=121
-  _OFFLINESTREAMINGINGESTION._serialized_start=124
-  _OFFLINESTREAMINGINGESTION._serialized_end=378
-  _SCHEDULEDOFFLINESTREAMINGTRIGGER._serialized_start=380
-  _SCHEDULEDOFFLINESTREAMINGTRIGGER._serialized_end=414
-  _MANUALOFFLINESTREAMINGTRIGGER._serialized_start=416
-  _MANUALOFFLINESTREAMINGTRIGGER._serialized_end=447
+  _globals['_ONLINESTREAMINGINGESTION']._serialized_start=70
+  _globals['_ONLINESTREAMINGINGESTION']._serialized_end=121
+  _globals['_OFFLINESTREAMINGINGESTION']._serialized_start=124
+  _globals['_OFFLINESTREAMINGINGESTION']._serialized_end=378
+  _globals['_SCHEDULEDOFFLINESTREAMINGTRIGGER']._serialized_start=380
+  _globals['_SCHEDULEDOFFLINESTREAMINGTRIGGER']._serialized_end=414
+  _globals['_MANUALOFFLINESTREAMINGTRIGGER']._serialized_start=416
+  _globals['_MANUALOFFLINESTREAMINGTRIGGER']._serialized_end=447
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/entities/entity_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,63 +16,23 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(qwak/feature_store/entities/entity.proto\x12\x1bqwak.feature.store.entities\x1a\x1fgoogle/protobuf/timestamp.proto\"\xd5\x01\n\x06\x45ntity\x12H\n\x11\x65ntity_definition\x18\x01 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\x12=\n\x08metadata\x18\x02 \x01(\x0b\x32+.qwak.feature.store.entities.EntityMetadata\x12\x42\n\x0c\x66\x65\x61ture_sets\x18\x03 \x03(\x0b\x32,.qwak.feature.store.entities.FeatureSetBrief\"c\n\x10\x45ntityDefinition\x12\x11\n\tentity_id\x18\x01 \x01(\t\x12<\n\x0b\x65ntity_spec\x18\x02 \x01(\x0b\x32\'.qwak.feature.store.entities.EntitySpec\"\xa4\x01\n\x0e\x45ntityMetadata\x12.\n\ncreated_at\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x12\n\ncreated_by\x18\x02 \x01(\t\x12\x34\n\x10last_modified_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x18\n\x10last_modified_by\x18\x04 \x01(\t\"\xb3\x01\n\nEntitySpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04keys\x18\x02 \x03(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x45\n\nvalue_type\x18\x04 \x01(\x0e\x32\x31.qwak.feature.store.entities.EntitySpec.ValueType\"-\n\tValueType\x12\x0b\n\x07INVALID\x10\x00\x12\n\n\x06STRING\x10\x01\x12\x07\n\x03INT\x10\x02\"C\n\x0f\x46\x65\x61tureSetBrief\x12\x16\n\x0e\x66\x65\x61ture_set_id\x18\x01 \x01(\t\x12\x18\n\x10\x66\x65\x61ture_set_name\x18\x02 \x01(\tB[\n&com.qwak.ai.feature.store.entities.apiP\x01Z/qwak/featurestore/entities;featurestoreentitiesb\x06proto3')
 
-
-
-_ENTITY = DESCRIPTOR.message_types_by_name['Entity']
-_ENTITYDEFINITION = DESCRIPTOR.message_types_by_name['EntityDefinition']
-_ENTITYMETADATA = DESCRIPTOR.message_types_by_name['EntityMetadata']
-_ENTITYSPEC = DESCRIPTOR.message_types_by_name['EntitySpec']
-_FEATURESETBRIEF = DESCRIPTOR.message_types_by_name['FeatureSetBrief']
-_ENTITYSPEC_VALUETYPE = _ENTITYSPEC.enum_types_by_name['ValueType']
-Entity = _reflection.GeneratedProtocolMessageType('Entity', (_message.Message,), {
-  'DESCRIPTOR' : _ENTITY,
-  '__module__' : 'qwak.feature_store.entities.entity_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.Entity)
-  })
-_sym_db.RegisterMessage(Entity)
-
-EntityDefinition = _reflection.GeneratedProtocolMessageType('EntityDefinition', (_message.Message,), {
-  'DESCRIPTOR' : _ENTITYDEFINITION,
-  '__module__' : 'qwak.feature_store.entities.entity_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.EntityDefinition)
-  })
-_sym_db.RegisterMessage(EntityDefinition)
-
-EntityMetadata = _reflection.GeneratedProtocolMessageType('EntityMetadata', (_message.Message,), {
-  'DESCRIPTOR' : _ENTITYMETADATA,
-  '__module__' : 'qwak.feature_store.entities.entity_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.EntityMetadata)
-  })
-_sym_db.RegisterMessage(EntityMetadata)
-
-EntitySpec = _reflection.GeneratedProtocolMessageType('EntitySpec', (_message.Message,), {
-  'DESCRIPTOR' : _ENTITYSPEC,
-  '__module__' : 'qwak.feature_store.entities.entity_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.EntitySpec)
-  })
-_sym_db.RegisterMessage(EntitySpec)
-
-FeatureSetBrief = _reflection.GeneratedProtocolMessageType('FeatureSetBrief', (_message.Message,), {
-  'DESCRIPTOR' : _FEATURESETBRIEF,
-  '__module__' : 'qwak.feature_store.entities.entity_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.FeatureSetBrief)
-  })
-_sym_db.RegisterMessage(FeatureSetBrief)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.entities.entity_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n&com.qwak.ai.feature.store.entities.apiP\001Z/qwak/featurestore/entities;featurestoreentities'
-  _ENTITY._serialized_start=107
-  _ENTITY._serialized_end=320
-  _ENTITYDEFINITION._serialized_start=322
-  _ENTITYDEFINITION._serialized_end=421
-  _ENTITYMETADATA._serialized_start=424
-  _ENTITYMETADATA._serialized_end=588
-  _ENTITYSPEC._serialized_start=591
-  _ENTITYSPEC._serialized_end=770
-  _ENTITYSPEC_VALUETYPE._serialized_start=725
-  _ENTITYSPEC_VALUETYPE._serialized_end=770
-  _FEATURESETBRIEF._serialized_start=772
-  _FEATURESETBRIEF._serialized_end=839
+  _globals['_ENTITY']._serialized_start=107
+  _globals['_ENTITY']._serialized_end=320
+  _globals['_ENTITYDEFINITION']._serialized_start=322
+  _globals['_ENTITYDEFINITION']._serialized_end=421
+  _globals['_ENTITYMETADATA']._serialized_start=424
+  _globals['_ENTITYMETADATA']._serialized_end=588
+  _globals['_ENTITYSPEC']._serialized_start=591
+  _globals['_ENTITYSPEC']._serialized_end=770
+  _globals['_ENTITYSPEC_VALUETYPE']._serialized_start=725
+  _globals['_ENTITYSPEC_VALUETYPE']._serialized_end=770
+  _globals['_FEATURESETBRIEF']._serialized_start=772
+  _globals['_FEATURESETBRIEF']._serialized_end=839
 # @@protoc_insertion_point(module_scope)