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

_qwak_proto/qwak/execution/v1/streaming_pb2.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/execution/v1/streaming.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!qwak/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')
+
+_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'
+  _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/execution/v1/streaming_pb2.pyi

@@ -0,0 +1,79 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class OnlineStreamingIngestion(google.protobuf.message.Message):
+    """Online Ingestion"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    featureset_name: builtins.str
+    def __init__(
+        self,
+        *,
+        featureset_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_name", b"featureset_name"]) -> None: ...
+
+global___OnlineStreamingIngestion = OnlineStreamingIngestion
+
+class OfflineStreamingIngestion(google.protobuf.message.Message):
+    """Offline Ingestion"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    SCHEDULED_TRIGGER_FIELD_NUMBER: builtins.int
+    MANUAL_TRIGGER_FIELD_NUMBER: builtins.int
+    featureset_name: builtins.str
+    @property
+    def scheduled_trigger(self) -> global___ScheduledOfflineStreamingTrigger: ...
+    @property
+    def manual_trigger(self) -> global___ManualOfflineStreamingTrigger: ...
+    def __init__(
+        self,
+        *,
+        featureset_name: builtins.str = ...,
+        scheduled_trigger: global___ScheduledOfflineStreamingTrigger | None = ...,
+        manual_trigger: global___ManualOfflineStreamingTrigger | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["manual_trigger", b"manual_trigger", "scheduled_trigger", b"scheduled_trigger", "trigger_type", b"trigger_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["featureset_name", b"featureset_name", "manual_trigger", b"manual_trigger", "scheduled_trigger", b"scheduled_trigger", "trigger_type", b"trigger_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["trigger_type", b"trigger_type"]) -> typing_extensions.Literal["scheduled_trigger", "manual_trigger"] | None: ...
+
+global___OfflineStreamingIngestion = OfflineStreamingIngestion
+
+class ScheduledOfflineStreamingTrigger(google.protobuf.message.Message):
+    """Scheduled offline ingestion (e.g., AirFlow)"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ScheduledOfflineStreamingTrigger = ScheduledOfflineStreamingTrigger
+
+class ManualOfflineStreamingTrigger(google.protobuf.message.Message):
+    """Manual Offline ingestion (that is, someone manually running from the UI)"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ManualOfflineStreamingTrigger = ManualOfflineStreamingTrigger

_qwak_proto/qwak/execution/v1/streaming_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_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)

_qwak_proto/qwak/feature_store/entities/entity_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,133 +16,37 @@ from _qwak_proto.qwak.feature_store.entities import entity_pb2 as qwak_dot_featu
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n0qwak/feature_store/entities/entity_service.proto\x12\x1bqwak.feature.store.entities\x1a(qwak/feature_store/entities/entity.proto\"S\n\x13\x43reateEntityRequest\x12<\n\x0b\x65ntity_spec\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.entities.EntitySpec\"f\n\x13UpdateEntityRequest\x12\x11\n\tentity_id\x18\x01 \x01(\t\x12<\n\x0b\x65ntity_spec\x18\x02 \x01(\x0b\x32\'.qwak.feature.store.entities.EntitySpec\"K\n\x14UpdateEntityResponse\x12\x33\n\x06\x65ntity\x18\x01 \x01(\x0b\x32#.qwak.feature.store.entities.Entity\"K\n\x14\x43reateEntityResponse\x12\x33\n\x06\x65ntity\x18\x01 \x01(\x0b\x32#.qwak.feature.store.entities.Entity\")\n\x14GetEntityByIdRequest\x12\x11\n\tentity_id\x18\x01 \x01(\t\"L\n\x15GetEntityByIdResponse\x12\x33\n\x06\x65ntity\x18\x01 \x01(\x0b\x32#.qwak.feature.store.entities.Entity\"-\n\x16GetEntityByNameRequest\x12\x13\n\x0b\x65ntity_name\x18\x01 \x01(\t\"N\n\x17GetEntityByNameResponse\x12\x33\n\x06\x65ntity\x18\x01 \x01(\x0b\x32#.qwak.feature.store.entities.Entity\"(\n\x13\x44\x65leteEntityRequest\x12\x11\n\tentity_id\x18\x01 \x01(\t\"\x16\n\x14\x44\x65leteEntityResponse\"\x15\n\x13ListEntitiesRequest\"M\n\x14ListEntitiesResponse\x12\x35\n\x08\x65ntities\x18\x01 \x03(\x0b\x32#.qwak.feature.store.entities.Entity2\xd9\x05\n\rEntityService\x12s\n\x0c\x43reateEntity\x12\x30.qwak.feature.store.entities.CreateEntityRequest\x1a\x31.qwak.feature.store.entities.CreateEntityResponse\x12s\n\x0cUpdateEntity\x12\x30.qwak.feature.store.entities.UpdateEntityRequest\x1a\x31.qwak.feature.store.entities.UpdateEntityResponse\x12s\n\x0c\x44\x65leteEntity\x12\x30.qwak.feature.store.entities.DeleteEntityRequest\x1a\x31.qwak.feature.store.entities.DeleteEntityResponse\x12v\n\rGetEntityById\x12\x31.qwak.feature.store.entities.GetEntityByIdRequest\x1a\x32.qwak.feature.store.entities.GetEntityByIdResponse\x12|\n\x0fGetEntityByName\x12\x33.qwak.feature.store.entities.GetEntityByNameRequest\x1a\x34.qwak.feature.store.entities.GetEntityByNameResponse\x12s\n\x0cListEntities\x12\x30.qwak.feature.store.entities.ListEntitiesRequest\x1a\x31.qwak.feature.store.entities.ListEntitiesResponseB]\n(com.qwak.ai.feature.store.management.apiP\x01Z/qwak/featurestore/entities;featurestoreentitiesb\x06proto3')
 
-
-
-_CREATEENTITYREQUEST = DESCRIPTOR.message_types_by_name['CreateEntityRequest']
-_UPDATEENTITYREQUEST = DESCRIPTOR.message_types_by_name['UpdateEntityRequest']
-_UPDATEENTITYRESPONSE = DESCRIPTOR.message_types_by_name['UpdateEntityResponse']
-_CREATEENTITYRESPONSE = DESCRIPTOR.message_types_by_name['CreateEntityResponse']
-_GETENTITYBYIDREQUEST = DESCRIPTOR.message_types_by_name['GetEntityByIdRequest']
-_GETENTITYBYIDRESPONSE = DESCRIPTOR.message_types_by_name['GetEntityByIdResponse']
-_GETENTITYBYNAMEREQUEST = DESCRIPTOR.message_types_by_name['GetEntityByNameRequest']
-_GETENTITYBYNAMERESPONSE = DESCRIPTOR.message_types_by_name['GetEntityByNameResponse']
-_DELETEENTITYREQUEST = DESCRIPTOR.message_types_by_name['DeleteEntityRequest']
-_DELETEENTITYRESPONSE = DESCRIPTOR.message_types_by_name['DeleteEntityResponse']
-_LISTENTITIESREQUEST = DESCRIPTOR.message_types_by_name['ListEntitiesRequest']
-_LISTENTITIESRESPONSE = DESCRIPTOR.message_types_by_name['ListEntitiesResponse']
-CreateEntityRequest = _reflection.GeneratedProtocolMessageType('CreateEntityRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEENTITYREQUEST,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.CreateEntityRequest)
-  })
-_sym_db.RegisterMessage(CreateEntityRequest)
-
-UpdateEntityRequest = _reflection.GeneratedProtocolMessageType('UpdateEntityRequest', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEENTITYREQUEST,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.UpdateEntityRequest)
-  })
-_sym_db.RegisterMessage(UpdateEntityRequest)
-
-UpdateEntityResponse = _reflection.GeneratedProtocolMessageType('UpdateEntityResponse', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEENTITYRESPONSE,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.UpdateEntityResponse)
-  })
-_sym_db.RegisterMessage(UpdateEntityResponse)
-
-CreateEntityResponse = _reflection.GeneratedProtocolMessageType('CreateEntityResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEENTITYRESPONSE,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.CreateEntityResponse)
-  })
-_sym_db.RegisterMessage(CreateEntityResponse)
-
-GetEntityByIdRequest = _reflection.GeneratedProtocolMessageType('GetEntityByIdRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETENTITYBYIDREQUEST,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.GetEntityByIdRequest)
-  })
-_sym_db.RegisterMessage(GetEntityByIdRequest)
-
-GetEntityByIdResponse = _reflection.GeneratedProtocolMessageType('GetEntityByIdResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETENTITYBYIDRESPONSE,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.GetEntityByIdResponse)
-  })
-_sym_db.RegisterMessage(GetEntityByIdResponse)
-
-GetEntityByNameRequest = _reflection.GeneratedProtocolMessageType('GetEntityByNameRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETENTITYBYNAMEREQUEST,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.GetEntityByNameRequest)
-  })
-_sym_db.RegisterMessage(GetEntityByNameRequest)
-
-GetEntityByNameResponse = _reflection.GeneratedProtocolMessageType('GetEntityByNameResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETENTITYBYNAMERESPONSE,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.GetEntityByNameResponse)
-  })
-_sym_db.RegisterMessage(GetEntityByNameResponse)
-
-DeleteEntityRequest = _reflection.GeneratedProtocolMessageType('DeleteEntityRequest', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEENTITYREQUEST,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.DeleteEntityRequest)
-  })
-_sym_db.RegisterMessage(DeleteEntityRequest)
-
-DeleteEntityResponse = _reflection.GeneratedProtocolMessageType('DeleteEntityResponse', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEENTITYRESPONSE,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.DeleteEntityResponse)
-  })
-_sym_db.RegisterMessage(DeleteEntityResponse)
-
-ListEntitiesRequest = _reflection.GeneratedProtocolMessageType('ListEntitiesRequest', (_message.Message,), {
-  'DESCRIPTOR' : _LISTENTITIESREQUEST,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.ListEntitiesRequest)
-  })
-_sym_db.RegisterMessage(ListEntitiesRequest)
-
-ListEntitiesResponse = _reflection.GeneratedProtocolMessageType('ListEntitiesResponse', (_message.Message,), {
-  'DESCRIPTOR' : _LISTENTITIESRESPONSE,
-  '__module__' : 'qwak.feature_store.entities.entity_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.entities.ListEntitiesResponse)
-  })
-_sym_db.RegisterMessage(ListEntitiesResponse)
-
-_ENTITYSERVICE = DESCRIPTOR.services_by_name['EntityService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.entities.entity_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/entities;featurestoreentities'
-  _CREATEENTITYREQUEST._serialized_start=123
-  _CREATEENTITYREQUEST._serialized_end=206
-  _UPDATEENTITYREQUEST._serialized_start=208
-  _UPDATEENTITYREQUEST._serialized_end=310
-  _UPDATEENTITYRESPONSE._serialized_start=312
-  _UPDATEENTITYRESPONSE._serialized_end=387
-  _CREATEENTITYRESPONSE._serialized_start=389
-  _CREATEENTITYRESPONSE._serialized_end=464
-  _GETENTITYBYIDREQUEST._serialized_start=466
-  _GETENTITYBYIDREQUEST._serialized_end=507
-  _GETENTITYBYIDRESPONSE._serialized_start=509
-  _GETENTITYBYIDRESPONSE._serialized_end=585
-  _GETENTITYBYNAMEREQUEST._serialized_start=587
-  _GETENTITYBYNAMEREQUEST._serialized_end=632
-  _GETENTITYBYNAMERESPONSE._serialized_start=634
-  _GETENTITYBYNAMERESPONSE._serialized_end=712
-  _DELETEENTITYREQUEST._serialized_start=714
-  _DELETEENTITYREQUEST._serialized_end=754
-  _DELETEENTITYRESPONSE._serialized_start=756
-  _DELETEENTITYRESPONSE._serialized_end=778
-  _LISTENTITIESREQUEST._serialized_start=780
-  _LISTENTITIESREQUEST._serialized_end=801
-  _LISTENTITIESRESPONSE._serialized_start=803
-  _LISTENTITIESRESPONSE._serialized_end=880
-  _ENTITYSERVICE._serialized_start=883
-  _ENTITYSERVICE._serialized_end=1612
+  _globals['_CREATEENTITYREQUEST']._serialized_start=123
+  _globals['_CREATEENTITYREQUEST']._serialized_end=206
+  _globals['_UPDATEENTITYREQUEST']._serialized_start=208
+  _globals['_UPDATEENTITYREQUEST']._serialized_end=310
+  _globals['_UPDATEENTITYRESPONSE']._serialized_start=312
+  _globals['_UPDATEENTITYRESPONSE']._serialized_end=387
+  _globals['_CREATEENTITYRESPONSE']._serialized_start=389
+  _globals['_CREATEENTITYRESPONSE']._serialized_end=464
+  _globals['_GETENTITYBYIDREQUEST']._serialized_start=466
+  _globals['_GETENTITYBYIDREQUEST']._serialized_end=507
+  _globals['_GETENTITYBYIDRESPONSE']._serialized_start=509
+  _globals['_GETENTITYBYIDRESPONSE']._serialized_end=585
+  _globals['_GETENTITYBYNAMEREQUEST']._serialized_start=587
+  _globals['_GETENTITYBYNAMEREQUEST']._serialized_end=632
+  _globals['_GETENTITYBYNAMERESPONSE']._serialized_start=634
+  _globals['_GETENTITYBYNAMERESPONSE']._serialized_end=712
+  _globals['_DELETEENTITYREQUEST']._serialized_start=714
+  _globals['_DELETEENTITYREQUEST']._serialized_end=754
+  _globals['_DELETEENTITYRESPONSE']._serialized_start=756
+  _globals['_DELETEENTITYRESPONSE']._serialized_end=778
+  _globals['_LISTENTITIESREQUEST']._serialized_start=780
+  _globals['_LISTENTITIESREQUEST']._serialized_end=801
+  _globals['_LISTENTITIESRESPONSE']._serialized_start=803
+  _globals['_LISTENTITIESRESPONSE']._serialized_end=880
+  _globals['_ENTITYSERVICE']._serialized_start=883
+  _globals['_ENTITYSERVICE']._serialized_end=1612
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/features/aggregation_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/feature_store/features/aggregation.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,180 +15,45 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n-qwak/feature_store/features/aggregation.proto\x12\'qwak.feature.store.features.aggregation\"\x85\t\n\x10\x41ggregationField\x12I\n\x0ctime_windows\x18\x01 \x03(\x0b\x32\x33.qwak.feature.store.features.aggregation.TimeWindow\x12\x19\n\x11\x66ield_name_prefix\x18\x02 \x01(\t\x12?\n\x05\x63ount\x18\x03 \x01(\x0b\x32..qwak.feature.store.features.aggregation.CountH\x00\x12;\n\x03sum\x18\x04 \x01(\x0b\x32,.qwak.feature.store.features.aggregation.SumH\x00\x12;\n\x03max\x18\x05 \x01(\x0b\x32,.qwak.feature.store.features.aggregation.MaxH\x00\x12;\n\x03min\x18\x06 \x01(\x0b\x32,.qwak.feature.store.features.aggregation.MinH\x00\x12;\n\x03\x61vg\x18\x07 \x01(\x0b\x32,.qwak.feature.store.features.aggregation.AvgH\x00\x12R\n\x0fsample_variance\x18\x08 \x01(\x0b\x32\x37.qwak.feature.store.features.aggregation.SampleVarianceH\x00\x12Z\n\x13population_variance\x18\t \x01(\x0b\x32;.qwak.feature.store.features.aggregation.PopulationVarianceH\x00\x12L\n\x0csample_stdev\x18\n \x01(\x0b\x32\x34.qwak.feature.store.features.aggregation.SampleSTDEVH\x00\x12T\n\x10population_stdev\x18\x0b \x01(\x0b\x32\x38.qwak.feature.store.features.aggregation.PopulationSTDEVH\x00\x12J\n\x0b\x62oolean_and\x18\x0c \x01(\x0b\x32\x33.qwak.feature.store.features.aggregation.BooleanAndH\x00\x12H\n\nboolean_or\x18\r \x01(\x0b\x32\x32.qwak.feature.store.features.aggregation.BooleanOrH\x00\x12Q\n\x0flast_distinct_n\x18\x0e \x01(\x0b\x32\x36.qwak.feature.store.features.aggregation.LastDistinctNH\x00\x12@\n\x06last_n\x18\x0f \x01(\x0b\x32..qwak.feature.store.features.aggregation.LastNH\x00\x12I\n\npercentile\x18\x10 \x01(\x0b\x32\x33.qwak.feature.store.features.aggregation.PercentileH\x00\x42\x0c\n\naggregator\"b\n\nTimeWindow\x12\x44\n\ttime_unit\x18\x01 \x01(\x0e\x32\x31.qwak.feature.store.features.aggregation.TimeUnit\x12\x0e\n\x06length\x18\x02 \x01(\x03\"\x1b\n\x05\x43ount\x12\x12\n\nfield_name\x18\x01 \x01(\t\"\x19\n\x03Sum\x12\x12\n\nfield_name\x18\x01 \x01(\t\"\x19\n\x03Max\x12\x12\n\nfield_name\x18\x01 \x01(\t\"\x19\n\x03Min\x12\x12\n\nfield_name\x18\x01 \x01(\t\"\x19\n\x03\x41vg\x12\x12\n\nfield_name\x18\x01 \x01(\t\"$\n\x0eSampleVariance\x12\x12\n\nfield_name\x18\x01 \x01(\t\"(\n\x12PopulationVariance\x12\x12\n\nfield_name\x18\x01 \x01(\t\"!\n\x0bSampleSTDEV\x12\x12\n\nfield_name\x18\x01 \x01(\t\"%\n\x0fPopulationSTDEV\x12\x12\n\nfield_name\x18\x01 \x01(\t\" \n\nBooleanAnd\x12\x12\n\nfield_name\x18\x01 \x01(\t\"\x1f\n\tBooleanOr\x12\x12\n\nfield_name\x18\x01 \x01(\t\".\n\rLastDistinctN\x12\x12\n\nfield_name\x18\x01 \x01(\t\x12\t\n\x01n\x18\x02 \x01(\x05\"&\n\x05LastN\x12\x12\n\nfield_name\x18\x01 \x01(\t\x12\t\n\x01n\x18\x02 \x01(\x05\"+\n\nPercentile\x12\x12\n\nfield_name\x18\x01 \x01(\t\x12\t\n\x01p\x18\x02 \x01(\x05*\x8d\x01\n\x08TimeUnit\x12\x15\n\x11TIME_UNIT_INVALID\x10\x00\x12\x13\n\x0fTIME_UNIT_WEEKS\x10\x03\x12\x12\n\x0eTIME_UNIT_DAYS\x10\x04\x12\x13\n\x0fTIME_UNIT_HOURS\x10\x05\x12\x15\n\x11TIME_UNIT_MINUTES\x10\x06\x12\x15\n\x11TIME_UNIT_SECONDS\x10\x07\x42^\n)com.qwak.ai.feature.store.aggregation.apiP\x01Z/qwak/featurestore/features;featurestorefeaturesb\x06proto3')
 
-_TIMEUNIT = DESCRIPTOR.enum_types_by_name['TimeUnit']
-TimeUnit = enum_type_wrapper.EnumTypeWrapper(_TIMEUNIT)
-TIME_UNIT_INVALID = 0
-TIME_UNIT_WEEKS = 3
-TIME_UNIT_DAYS = 4
-TIME_UNIT_HOURS = 5
-TIME_UNIT_MINUTES = 6
-TIME_UNIT_SECONDS = 7
-
-
-_AGGREGATIONFIELD = DESCRIPTOR.message_types_by_name['AggregationField']
-_TIMEWINDOW = DESCRIPTOR.message_types_by_name['TimeWindow']
-_COUNT = DESCRIPTOR.message_types_by_name['Count']
-_SUM = DESCRIPTOR.message_types_by_name['Sum']
-_MAX = DESCRIPTOR.message_types_by_name['Max']
-_MIN = DESCRIPTOR.message_types_by_name['Min']
-_AVG = DESCRIPTOR.message_types_by_name['Avg']
-_SAMPLEVARIANCE = DESCRIPTOR.message_types_by_name['SampleVariance']
-_POPULATIONVARIANCE = DESCRIPTOR.message_types_by_name['PopulationVariance']
-_SAMPLESTDEV = DESCRIPTOR.message_types_by_name['SampleSTDEV']
-_POPULATIONSTDEV = DESCRIPTOR.message_types_by_name['PopulationSTDEV']
-_BOOLEANAND = DESCRIPTOR.message_types_by_name['BooleanAnd']
-_BOOLEANOR = DESCRIPTOR.message_types_by_name['BooleanOr']
-_LASTDISTINCTN = DESCRIPTOR.message_types_by_name['LastDistinctN']
-_LASTN = DESCRIPTOR.message_types_by_name['LastN']
-_PERCENTILE = DESCRIPTOR.message_types_by_name['Percentile']
-AggregationField = _reflection.GeneratedProtocolMessageType('AggregationField', (_message.Message,), {
-  'DESCRIPTOR' : _AGGREGATIONFIELD,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.AggregationField)
-  })
-_sym_db.RegisterMessage(AggregationField)
-
-TimeWindow = _reflection.GeneratedProtocolMessageType('TimeWindow', (_message.Message,), {
-  'DESCRIPTOR' : _TIMEWINDOW,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.TimeWindow)
-  })
-_sym_db.RegisterMessage(TimeWindow)
-
-Count = _reflection.GeneratedProtocolMessageType('Count', (_message.Message,), {
-  'DESCRIPTOR' : _COUNT,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.Count)
-  })
-_sym_db.RegisterMessage(Count)
-
-Sum = _reflection.GeneratedProtocolMessageType('Sum', (_message.Message,), {
-  'DESCRIPTOR' : _SUM,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.Sum)
-  })
-_sym_db.RegisterMessage(Sum)
-
-Max = _reflection.GeneratedProtocolMessageType('Max', (_message.Message,), {
-  'DESCRIPTOR' : _MAX,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.Max)
-  })
-_sym_db.RegisterMessage(Max)
-
-Min = _reflection.GeneratedProtocolMessageType('Min', (_message.Message,), {
-  'DESCRIPTOR' : _MIN,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.Min)
-  })
-_sym_db.RegisterMessage(Min)
-
-Avg = _reflection.GeneratedProtocolMessageType('Avg', (_message.Message,), {
-  'DESCRIPTOR' : _AVG,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.Avg)
-  })
-_sym_db.RegisterMessage(Avg)
-
-SampleVariance = _reflection.GeneratedProtocolMessageType('SampleVariance', (_message.Message,), {
-  'DESCRIPTOR' : _SAMPLEVARIANCE,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.SampleVariance)
-  })
-_sym_db.RegisterMessage(SampleVariance)
-
-PopulationVariance = _reflection.GeneratedProtocolMessageType('PopulationVariance', (_message.Message,), {
-  'DESCRIPTOR' : _POPULATIONVARIANCE,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.PopulationVariance)
-  })
-_sym_db.RegisterMessage(PopulationVariance)
-
-SampleSTDEV = _reflection.GeneratedProtocolMessageType('SampleSTDEV', (_message.Message,), {
-  'DESCRIPTOR' : _SAMPLESTDEV,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.SampleSTDEV)
-  })
-_sym_db.RegisterMessage(SampleSTDEV)
-
-PopulationSTDEV = _reflection.GeneratedProtocolMessageType('PopulationSTDEV', (_message.Message,), {
-  'DESCRIPTOR' : _POPULATIONSTDEV,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.PopulationSTDEV)
-  })
-_sym_db.RegisterMessage(PopulationSTDEV)
-
-BooleanAnd = _reflection.GeneratedProtocolMessageType('BooleanAnd', (_message.Message,), {
-  'DESCRIPTOR' : _BOOLEANAND,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.BooleanAnd)
-  })
-_sym_db.RegisterMessage(BooleanAnd)
-
-BooleanOr = _reflection.GeneratedProtocolMessageType('BooleanOr', (_message.Message,), {
-  'DESCRIPTOR' : _BOOLEANOR,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.BooleanOr)
-  })
-_sym_db.RegisterMessage(BooleanOr)
-
-LastDistinctN = _reflection.GeneratedProtocolMessageType('LastDistinctN', (_message.Message,), {
-  'DESCRIPTOR' : _LASTDISTINCTN,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.LastDistinctN)
-  })
-_sym_db.RegisterMessage(LastDistinctN)
-
-LastN = _reflection.GeneratedProtocolMessageType('LastN', (_message.Message,), {
-  'DESCRIPTOR' : _LASTN,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.LastN)
-  })
-_sym_db.RegisterMessage(LastN)
-
-Percentile = _reflection.GeneratedProtocolMessageType('Percentile', (_message.Message,), {
-  'DESCRIPTOR' : _PERCENTILE,
-  '__module__' : 'qwak.feature_store.features.aggregation_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.features.aggregation.Percentile)
-  })
-_sym_db.RegisterMessage(Percentile)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.features.aggregation_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n)com.qwak.ai.feature.store.aggregation.apiP\001Z/qwak/featurestore/features;featurestorefeatures'
-  _TIMEUNIT._serialized_start=1842
-  _TIMEUNIT._serialized_end=1983
-  _AGGREGATIONFIELD._serialized_start=91
-  _AGGREGATIONFIELD._serialized_end=1248
-  _TIMEWINDOW._serialized_start=1250
-  _TIMEWINDOW._serialized_end=1348
-  _COUNT._serialized_start=1350
-  _COUNT._serialized_end=1377
-  _SUM._serialized_start=1379
-  _SUM._serialized_end=1404
-  _MAX._serialized_start=1406
-  _MAX._serialized_end=1431
-  _MIN._serialized_start=1433
-  _MIN._serialized_end=1458
-  _AVG._serialized_start=1460
-  _AVG._serialized_end=1485
-  _SAMPLEVARIANCE._serialized_start=1487
-  _SAMPLEVARIANCE._serialized_end=1523
-  _POPULATIONVARIANCE._serialized_start=1525
-  _POPULATIONVARIANCE._serialized_end=1565
-  _SAMPLESTDEV._serialized_start=1567
-  _SAMPLESTDEV._serialized_end=1600
-  _POPULATIONSTDEV._serialized_start=1602
-  _POPULATIONSTDEV._serialized_end=1639
-  _BOOLEANAND._serialized_start=1641
-  _BOOLEANAND._serialized_end=1673
-  _BOOLEANOR._serialized_start=1675
-  _BOOLEANOR._serialized_end=1706
-  _LASTDISTINCTN._serialized_start=1708
-  _LASTDISTINCTN._serialized_end=1754
-  _LASTN._serialized_start=1756
-  _LASTN._serialized_end=1794
-  _PERCENTILE._serialized_start=1796
-  _PERCENTILE._serialized_end=1839
+  _globals['_TIMEUNIT']._serialized_start=1842
+  _globals['_TIMEUNIT']._serialized_end=1983
+  _globals['_AGGREGATIONFIELD']._serialized_start=91
+  _globals['_AGGREGATIONFIELD']._serialized_end=1248
+  _globals['_TIMEWINDOW']._serialized_start=1250
+  _globals['_TIMEWINDOW']._serialized_end=1348
+  _globals['_COUNT']._serialized_start=1350
+  _globals['_COUNT']._serialized_end=1377
+  _globals['_SUM']._serialized_start=1379
+  _globals['_SUM']._serialized_end=1404
+  _globals['_MAX']._serialized_start=1406
+  _globals['_MAX']._serialized_end=1431
+  _globals['_MIN']._serialized_start=1433
+  _globals['_MIN']._serialized_end=1458
+  _globals['_AVG']._serialized_start=1460
+  _globals['_AVG']._serialized_end=1485
+  _globals['_SAMPLEVARIANCE']._serialized_start=1487
+  _globals['_SAMPLEVARIANCE']._serialized_end=1523
+  _globals['_POPULATIONVARIANCE']._serialized_start=1525
+  _globals['_POPULATIONVARIANCE']._serialized_end=1565
+  _globals['_SAMPLESTDEV']._serialized_start=1567
+  _globals['_SAMPLESTDEV']._serialized_end=1600
+  _globals['_POPULATIONSTDEV']._serialized_start=1602
+  _globals['_POPULATIONSTDEV']._serialized_end=1639
+  _globals['_BOOLEANAND']._serialized_start=1641
+  _globals['_BOOLEANAND']._serialized_end=1673
+  _globals['_BOOLEANOR']._serialized_start=1675
+  _globals['_BOOLEANOR']._serialized_end=1706
+  _globals['_LASTDISTINCTN']._serialized_start=1708
+  _globals['_LASTDISTINCTN']._serialized_end=1754
+  _globals['_LASTN']._serialized_start=1756
+  _globals['_LASTN']._serialized_end=1794
+  _globals['_PERCENTILE']._serialized_start=1796
+  _globals['_PERCENTILE']._serialized_end=1839
 # @@protoc_insertion_point(module_scope)