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

_qwak_proto/qwak/feature_store/serving/metadata_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,42 +15,21 @@ from _qwak_proto.qwak.feature_store.features import feature_set_types_pb2 as qwa
 from _qwak_proto.qwak.feature_store.entities import entity_pb2 as qwak_dot_feature__store_dot_entities_dot_entity__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/feature_store/serving/metadata.proto\x12*com.qwak.ai.feature.store.serving.metadata\x1a\x33qwak/feature_store/features/feature_set_types.proto\x1a(qwak/feature_store/entities/entity.proto\"\xb0\x02\n&StreamingAggregationFeaturesetMetadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x65\n streaming_aggregation_featureset\x18\x04 \x01(\x0b\x32;.qwak.feature.store.features.StreamingAggregationFeatureSet\x12=\n\x06\x65ntity\x18\x05 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\x12\x18\n\x10\x61vro_schema_json\x18\x06 \x01(\t\"\x8e\x02\n\x19\x42\x61tchFeaturesetV1Metadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x1d\n\x15qwak_protocol_version\x18\x04 \x01(\x05\x12K\n\x13\x62\x61tch_featureset_v1\x18\x05 \x01(\x0b\x32..qwak.feature.store.features.BatchFeatureSetV1\x12=\n\x06\x65ntity\x18\x06 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\"\x9a\x02\n\x1dStreamingFeaturesetV1Metadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x1d\n\x15qwak_protocol_version\x18\x04 \x01(\x05\x12S\n\x17streaming_featureset_v1\x18\x05 \x01(\x0b\x32\x32.qwak.feature.store.features.StreamingFeatureSetV1\x12=\n\x06\x65ntity\x18\x06 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinitionB,\n*com.qwak.ai.feature.store.serving.metadatab\x06proto3')
-
-
-
-_STREAMINGAGGREGATIONFEATURESETMETADATA = DESCRIPTOR.message_types_by_name['StreamingAggregationFeaturesetMetadata']
-_BATCHFEATURESETV1METADATA = DESCRIPTOR.message_types_by_name['BatchFeaturesetV1Metadata']
-_STREAMINGFEATURESETV1METADATA = DESCRIPTOR.message_types_by_name['StreamingFeaturesetV1Metadata']
-StreamingAggregationFeaturesetMetadata = _reflection.GeneratedProtocolMessageType('StreamingAggregationFeaturesetMetadata', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGAGGREGATIONFEATURESETMETADATA,
-  '__module__' : 'qwak.feature_store.serving.metadata_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.metadata.StreamingAggregationFeaturesetMetadata)
-  })
-_sym_db.RegisterMessage(StreamingAggregationFeaturesetMetadata)
-
-BatchFeaturesetV1Metadata = _reflection.GeneratedProtocolMessageType('BatchFeaturesetV1Metadata', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHFEATURESETV1METADATA,
-  '__module__' : 'qwak.feature_store.serving.metadata_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.metadata.BatchFeaturesetV1Metadata)
-  })
-_sym_db.RegisterMessage(BatchFeaturesetV1Metadata)
-
-StreamingFeaturesetV1Metadata = _reflection.GeneratedProtocolMessageType('StreamingFeaturesetV1Metadata', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGFEATURESETV1METADATA,
-  '__module__' : 'qwak.feature_store.serving.metadata_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.metadata.StreamingFeaturesetV1Metadata)
-  })
-_sym_db.RegisterMessage(StreamingFeaturesetV1Metadata)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/feature_store/serving/metadata.proto\x12*com.qwak.ai.feature.store.serving.metadata\x1a\x33qwak/feature_store/features/feature_set_types.proto\x1a(qwak/feature_store/entities/entity.proto\"\xb0\x02\n&StreamingAggregationFeaturesetMetadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x65\n streaming_aggregation_featureset\x18\x04 \x01(\x0b\x32;.qwak.feature.store.features.StreamingAggregationFeatureSet\x12=\n\x06\x65ntity\x18\x05 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\x12\x18\n\x10\x61vro_schema_json\x18\x06 \x01(\t\"\x8e\x02\n\x19\x42\x61tchFeaturesetV1Metadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x1d\n\x15qwak_protocol_version\x18\x04 \x01(\x05\x12K\n\x13\x62\x61tch_featureset_v1\x18\x05 \x01(\x0b\x32..qwak.feature.store.features.BatchFeatureSetV1\x12=\n\x06\x65ntity\x18\x06 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\"\x9a\x02\n\x1dStreamingFeaturesetV1Metadata\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x02 \x01(\t\x12\x15\n\rfeatureset_id\x18\x03 \x01(\t\x12\x1d\n\x15qwak_protocol_version\x18\x04 \x01(\x05\x12S\n\x17streaming_featureset_v1\x18\x05 \x01(\x0b\x32\x32.qwak.feature.store.features.StreamingFeatureSetV1\x12=\n\x06\x65ntity\x18\x06 \x01(\x0b\x32-.qwak.feature.store.entities.EntityDefinition\"\x7f\n\x1a\x46\x65\x61turesetMemoryUsageStats\x12\x16\n\x0e\x65nvironment_id\x18\x01 \x01(\t\x12\x15\n\rfeatureset_id\x18\x02 \x01(\t\x12\x17\n\x0f\x66\x65\x61tureset_name\x18\x03 \x01(\t\x12\x19\n\x11online_used_bytes\x18\x04 \x01(\x03\x42,\n*com.qwak.ai.feature.store.serving.metadatab\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.serving.metadata_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n*com.qwak.ai.feature.store.serving.metadata'
-  _STREAMINGAGGREGATIONFEATURESETMETADATA._serialized_start=185
-  _STREAMINGAGGREGATIONFEATURESETMETADATA._serialized_end=489
-  _BATCHFEATURESETV1METADATA._serialized_start=492
-  _BATCHFEATURESETV1METADATA._serialized_end=762
-  _STREAMINGFEATURESETV1METADATA._serialized_start=765
-  _STREAMINGFEATURESETV1METADATA._serialized_end=1047
+  _globals['_STREAMINGAGGREGATIONFEATURESETMETADATA']._serialized_start=185
+  _globals['_STREAMINGAGGREGATIONFEATURESETMETADATA']._serialized_end=489
+  _globals['_BATCHFEATURESETV1METADATA']._serialized_start=492
+  _globals['_BATCHFEATURESETV1METADATA']._serialized_end=762
+  _globals['_STREAMINGFEATURESETV1METADATA']._serialized_start=765
+  _globals['_STREAMINGFEATURESETV1METADATA']._serialized_end=1047
+  _globals['_FEATURESETMEMORYUSAGESTATS']._serialized_start=1049
+  _globals['_FEATURESETMEMORYUSAGESTATS']._serialized_end=1176
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/serving/metadata_pb2.pyi

@@ -129,3 +129,30 @@ class StreamingFeaturesetV1Metadata(google.protobuf.message.Message):
     def ClearField(self, field_name: typing_extensions.Literal["entity", b"entity", "environment_id", b"environment_id", "featureset_id", b"featureset_id", "featureset_name", b"featureset_name", "qwak_protocol_version", b"qwak_protocol_version", "streaming_featureset_v1", b"streaming_featureset_v1"]) -> None: ...
 
 global___StreamingFeaturesetV1Metadata = StreamingFeaturesetV1Metadata
+
+class FeaturesetMemoryUsageStats(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    FEATURESET_ID_FIELD_NUMBER: builtins.int
+    FEATURESET_NAME_FIELD_NUMBER: builtins.int
+    ONLINE_USED_BYTES_FIELD_NUMBER: builtins.int
+    environment_id: builtins.str
+    """environment id"""
+    featureset_id: builtins.str
+    """featureset id"""
+    featureset_name: builtins.str
+    """featureset name"""
+    online_used_bytes: builtins.int
+    """number of bytes used in the online store"""
+    def __init__(
+        self,
+        *,
+        environment_id: builtins.str = ...,
+        featureset_id: builtins.str = ...,
+        featureset_name: builtins.str = ...,
+        online_used_bytes: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["environment_id", b"environment_id", "featureset_id", b"featureset_id", "featureset_name", b"featureset_name", "online_used_bytes", b"online_used_bytes"]) -> None: ...
+
+global___FeaturesetMemoryUsageStats = FeaturesetMemoryUsageStats

_qwak_proto/qwak/feature_store/serving/serving_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,180 +16,9 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(qwak/feature_store/serving/serving.proto\x12%com.qwak.ai.feature.store.serving.api\x1a\x1fgoogle/protobuf/timestamp.proto\"\xa8\x02\n\x1eRequestedEntitiesMatrixRequest\x12U\n\x14\x65ntities_to_features\x18\x01 \x03(\x0b\x32\x37.com.qwak.ai.feature.store.serving.api.EntityToFeatures\x12\\\n\x14\x65ntity_values_matrix\x18\x02 \x01(\x0b\x32>.com.qwak.ai.feature.store.serving.api.RequestedEntitiesMatrix\x12Q\n\x11request_meta_data\x18\x03 \x01(\x0b\x32\x36.com.qwak.ai.feature.store.serving.api.RequestMetaData\"R\n\x0fRequestMetaData\x12\x12\n\nmodel_name\x18\x01 \x01(\t\x12\x10\n\x08num_keys\x18\x02 \x01(\x05\x12\x19\n\x11\x66\x65\x61ture_set_names\x18\x03 \x03(\t\"\xa5\x01\n\x17RequestedEntitiesMatrix\x12\x45\n\x06header\x18\x01 \x01(\x0b\x32\x35.com.qwak.ai.feature.store.serving.api.EntitiesHeader\x12\x43\n\x04rows\x18\x02 \x03(\x0b\x32\x35.com.qwak.ai.feature.store.serving.api.EntityValueRow\"&\n\x0e\x45ntitiesHeader\x12\x14\n\x0c\x65ntity_names\x18\x01 \x03(\t\"6\n\x0e\x45ntityValueRow\x12\r\n\x05index\x18\x01 \x01(\x05\x12\x15\n\rentity_values\x18\x02 \x03(\t\"i\n\x10\x45ntityToFeatures\x12@\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32..com.qwak.ai.feature.store.serving.api.Feature\x12\x13\n\x0b\x65ntity_name\x18\x02 \x01(\t\"\xd3\x01\n\x14\x46\x65\x61tureValuesRequest\x12@\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32..com.qwak.ai.feature.store.serving.api.Feature\x12R\n\x06\x65ntity\x18\x02 \x01(\x0b\x32\x42.com.qwak.ai.feature.store.serving.api.FeatureValuesRequest.Entity\x1a%\n\x06\x45ntity\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\"\xa2\x04\n\x07\x46\x65\x61ture\x12L\n\rbatch_feature\x18\x01 \x01(\x0b\x32\x33.com.qwak.ai.feature.store.serving.api.BatchFeatureH\x00\x12X\n\x12on_the_fly_feature\x18\x02 \x01(\x0b\x32\x36.com.qwak.ai.feature.store.serving.api.OnTheFlyFeatureB\x02\x18\x01H\x00\x12T\n\x11streaming_feature\x18\x03 \x01(\x0b\x32\x37.com.qwak.ai.feature.store.serving.api.StreamingFeatureH\x00\x12\x63\n\x15streaming_aggregation\x18\x04 \x01(\x0b\x32\x42.com.qwak.ai.feature.store.serving.api.StreamingAggregationFeatureH\x00\x12Q\n\x10\x62\x61tch_v1_feature\x18\x05 \x01(\x0b\x32\x35.com.qwak.ai.feature.store.serving.api.BatchV1FeatureH\x00\x12Y\n\x14streaming_v1_feature\x18\x06 \x01(\x0b\x32\x39.com.qwak.ai.feature.store.serving.api.StreamingV1FeatureH\x00\x42\x06\n\x04type\"\x1c\n\x0c\x42\x61tchFeature\x12\x0c\n\x04name\x18\x01 \x01(\t\"\x1e\n\x0e\x42\x61tchV1Feature\x12\x0c\n\x04name\x18\x01 \x01(\t\" \n\x10StreamingFeature\x12\x0c\n\x04name\x18\x01 \x01(\t\"\"\n\x12StreamingV1Feature\x12\x0c\n\x04name\x18\x01 \x01(\t\"+\n\x1bStreamingAggregationFeature\x12\x0c\n\x04name\x18\x01 \x01(\t\"u\n\x0fOnTheFlyFeature\x12\x10\n\x04name\x18\x01 \x01(\tB\x02\x18\x01\x12P\n\x0fon_the_fly_data\x18\x02 \x01(\x0b\x32\x33.com.qwak.ai.feature.store.serving.api.OnTheFlyDataB\x02\x18\x01\"O\n\x1aMultiFeatureValuesResponse\x12\x19\n\x11pandas_df_as_json\x18\x01 \x01(\t\x12\x16\n\x0e\x66\x65\x61ture_values\x18\x02 \x01(\t\"\xf4\x02\n\x15\x46\x65\x61tureValuesResponse\x12\x61\n\x0e\x66\x65\x61ture_values\x18\x01 \x03(\x0b\x32I.com.qwak.ai.feature.store.serving.api.FeatureValuesResponse.FeatureValue\x1a\xf7\x01\n\x0c\x46\x65\x61tureValue\x12\x14\n\x0c\x66\x65\x61ture_name\x18\x01 \x01(\t\x12\x43\n\rfeature_value\x18\x02 \x01(\x0b\x32,.com.qwak.ai.feature.store.serving.api.Value\x12\x65\n\x06status\x18\x03 \x01(\x0e\x32U.com.qwak.ai.feature.store.serving.api.FeatureValuesResponse.FeatureValue.ValueStatus\"%\n\x0bValueStatus\x12\x0b\n\x07INVALID\x10\x00\x12\t\n\x05VALID\x10\x01\"J\n\x0cOnTheFlyData\x12\x17\n\tuser_data\x18\x01 \x01(\tB\x02\x18\x01H\x00\x12\x19\n\x0bschema_data\x18\x02 \x01(\tB\x02\x18\x01H\x00\x42\x06\n\x04\x64\x61ta\"\x9a\x02\n\x05Value\x12\x13\n\tbytes_val\x18\x01 \x01(\x0cH\x00\x12\x14\n\nstring_val\x18\x02 \x01(\tH\x00\x12\x13\n\tint32_val\x18\x03 \x01(\x05H\x00\x12\x13\n\tint64_val\x18\x04 \x01(\x03H\x00\x12\x14\n\ndouble_val\x18\x05 \x01(\x01H\x00\x12\x13\n\tfloat_val\x18\x06 \x01(\x02H\x00\x12\x12\n\x08\x62ool_val\x18\x07 \x01(\x08H\x00\x12\x33\n\rtimestamp_val\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00\x12\x41\n\tarray_val\x18\t \x01(\x0b\x32,.com.qwak.ai.feature.store.serving.api.ArrayH\x00\x42\x05\n\x03val\"H\n\x05\x41rray\x12?\n\tarray_val\x18\x01 \x03(\x0b\x32,.com.qwak.ai.feature.store.serving.api.Value2\xba\x02\n\x0eServingService\x12\x88\x01\n\x0bGetFeatures\x12;.com.qwak.ai.feature.store.serving.api.FeatureValuesRequest\x1a<.com.qwak.ai.feature.store.serving.api.FeatureValuesResponse\x12\x9c\x01\n\x10GetMultiFeatures\x12\x45.com.qwak.ai.feature.store.serving.api.RequestedEntitiesMatrixRequest\x1a\x41.com.qwak.ai.feature.store.serving.api.MultiFeatureValuesResponseB\'\n%com.qwak.ai.feature.store.serving.apib\x06proto3')
 
-
-
-_REQUESTEDENTITIESMATRIXREQUEST = DESCRIPTOR.message_types_by_name['RequestedEntitiesMatrixRequest']
-_REQUESTMETADATA = DESCRIPTOR.message_types_by_name['RequestMetaData']
-_REQUESTEDENTITIESMATRIX = DESCRIPTOR.message_types_by_name['RequestedEntitiesMatrix']
-_ENTITIESHEADER = DESCRIPTOR.message_types_by_name['EntitiesHeader']
-_ENTITYVALUEROW = DESCRIPTOR.message_types_by_name['EntityValueRow']
-_ENTITYTOFEATURES = DESCRIPTOR.message_types_by_name['EntityToFeatures']
-_FEATUREVALUESREQUEST = DESCRIPTOR.message_types_by_name['FeatureValuesRequest']
-_FEATUREVALUESREQUEST_ENTITY = _FEATUREVALUESREQUEST.nested_types_by_name['Entity']
-_FEATURE = DESCRIPTOR.message_types_by_name['Feature']
-_BATCHFEATURE = DESCRIPTOR.message_types_by_name['BatchFeature']
-_BATCHV1FEATURE = DESCRIPTOR.message_types_by_name['BatchV1Feature']
-_STREAMINGFEATURE = DESCRIPTOR.message_types_by_name['StreamingFeature']
-_STREAMINGV1FEATURE = DESCRIPTOR.message_types_by_name['StreamingV1Feature']
-_STREAMINGAGGREGATIONFEATURE = DESCRIPTOR.message_types_by_name['StreamingAggregationFeature']
-_ONTHEFLYFEATURE = DESCRIPTOR.message_types_by_name['OnTheFlyFeature']
-_MULTIFEATUREVALUESRESPONSE = DESCRIPTOR.message_types_by_name['MultiFeatureValuesResponse']
-_FEATUREVALUESRESPONSE = DESCRIPTOR.message_types_by_name['FeatureValuesResponse']
-_FEATUREVALUESRESPONSE_FEATUREVALUE = _FEATUREVALUESRESPONSE.nested_types_by_name['FeatureValue']
-_ONTHEFLYDATA = DESCRIPTOR.message_types_by_name['OnTheFlyData']
-_VALUE = DESCRIPTOR.message_types_by_name['Value']
-_ARRAY = DESCRIPTOR.message_types_by_name['Array']
-_FEATUREVALUESRESPONSE_FEATUREVALUE_VALUESTATUS = _FEATUREVALUESRESPONSE_FEATUREVALUE.enum_types_by_name['ValueStatus']
-RequestedEntitiesMatrixRequest = _reflection.GeneratedProtocolMessageType('RequestedEntitiesMatrixRequest', (_message.Message,), {
-  'DESCRIPTOR' : _REQUESTEDENTITIESMATRIXREQUEST,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.RequestedEntitiesMatrixRequest)
-  })
-_sym_db.RegisterMessage(RequestedEntitiesMatrixRequest)
-
-RequestMetaData = _reflection.GeneratedProtocolMessageType('RequestMetaData', (_message.Message,), {
-  'DESCRIPTOR' : _REQUESTMETADATA,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.RequestMetaData)
-  })
-_sym_db.RegisterMessage(RequestMetaData)
-
-RequestedEntitiesMatrix = _reflection.GeneratedProtocolMessageType('RequestedEntitiesMatrix', (_message.Message,), {
-  'DESCRIPTOR' : _REQUESTEDENTITIESMATRIX,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.RequestedEntitiesMatrix)
-  })
-_sym_db.RegisterMessage(RequestedEntitiesMatrix)
-
-EntitiesHeader = _reflection.GeneratedProtocolMessageType('EntitiesHeader', (_message.Message,), {
-  'DESCRIPTOR' : _ENTITIESHEADER,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.EntitiesHeader)
-  })
-_sym_db.RegisterMessage(EntitiesHeader)
-
-EntityValueRow = _reflection.GeneratedProtocolMessageType('EntityValueRow', (_message.Message,), {
-  'DESCRIPTOR' : _ENTITYVALUEROW,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.EntityValueRow)
-  })
-_sym_db.RegisterMessage(EntityValueRow)
-
-EntityToFeatures = _reflection.GeneratedProtocolMessageType('EntityToFeatures', (_message.Message,), {
-  'DESCRIPTOR' : _ENTITYTOFEATURES,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.EntityToFeatures)
-  })
-_sym_db.RegisterMessage(EntityToFeatures)
-
-FeatureValuesRequest = _reflection.GeneratedProtocolMessageType('FeatureValuesRequest', (_message.Message,), {
-
-  'Entity' : _reflection.GeneratedProtocolMessageType('Entity', (_message.Message,), {
-    'DESCRIPTOR' : _FEATUREVALUESREQUEST_ENTITY,
-    '__module__' : 'qwak.feature_store.serving.serving_pb2'
-    # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.FeatureValuesRequest.Entity)
-    })
-  ,
-  'DESCRIPTOR' : _FEATUREVALUESREQUEST,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.FeatureValuesRequest)
-  })
-_sym_db.RegisterMessage(FeatureValuesRequest)
-_sym_db.RegisterMessage(FeatureValuesRequest.Entity)
-
-Feature = _reflection.GeneratedProtocolMessageType('Feature', (_message.Message,), {
-  'DESCRIPTOR' : _FEATURE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.Feature)
-  })
-_sym_db.RegisterMessage(Feature)
-
-BatchFeature = _reflection.GeneratedProtocolMessageType('BatchFeature', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHFEATURE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.BatchFeature)
-  })
-_sym_db.RegisterMessage(BatchFeature)
-
-BatchV1Feature = _reflection.GeneratedProtocolMessageType('BatchV1Feature', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHV1FEATURE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.BatchV1Feature)
-  })
-_sym_db.RegisterMessage(BatchV1Feature)
-
-StreamingFeature = _reflection.GeneratedProtocolMessageType('StreamingFeature', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGFEATURE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.StreamingFeature)
-  })
-_sym_db.RegisterMessage(StreamingFeature)
-
-StreamingV1Feature = _reflection.GeneratedProtocolMessageType('StreamingV1Feature', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGV1FEATURE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.StreamingV1Feature)
-  })
-_sym_db.RegisterMessage(StreamingV1Feature)
-
-StreamingAggregationFeature = _reflection.GeneratedProtocolMessageType('StreamingAggregationFeature', (_message.Message,), {
-  'DESCRIPTOR' : _STREAMINGAGGREGATIONFEATURE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.StreamingAggregationFeature)
-  })
-_sym_db.RegisterMessage(StreamingAggregationFeature)
-
-OnTheFlyFeature = _reflection.GeneratedProtocolMessageType('OnTheFlyFeature', (_message.Message,), {
-  'DESCRIPTOR' : _ONTHEFLYFEATURE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.OnTheFlyFeature)
-  })
-_sym_db.RegisterMessage(OnTheFlyFeature)
-
-MultiFeatureValuesResponse = _reflection.GeneratedProtocolMessageType('MultiFeatureValuesResponse', (_message.Message,), {
-  'DESCRIPTOR' : _MULTIFEATUREVALUESRESPONSE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.MultiFeatureValuesResponse)
-  })
-_sym_db.RegisterMessage(MultiFeatureValuesResponse)
-
-FeatureValuesResponse = _reflection.GeneratedProtocolMessageType('FeatureValuesResponse', (_message.Message,), {
-
-  'FeatureValue' : _reflection.GeneratedProtocolMessageType('FeatureValue', (_message.Message,), {
-    'DESCRIPTOR' : _FEATUREVALUESRESPONSE_FEATUREVALUE,
-    '__module__' : 'qwak.feature_store.serving.serving_pb2'
-    # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.FeatureValuesResponse.FeatureValue)
-    })
-  ,
-  'DESCRIPTOR' : _FEATUREVALUESRESPONSE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.FeatureValuesResponse)
-  })
-_sym_db.RegisterMessage(FeatureValuesResponse)
-_sym_db.RegisterMessage(FeatureValuesResponse.FeatureValue)
-
-OnTheFlyData = _reflection.GeneratedProtocolMessageType('OnTheFlyData', (_message.Message,), {
-  'DESCRIPTOR' : _ONTHEFLYDATA,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.OnTheFlyData)
-  })
-_sym_db.RegisterMessage(OnTheFlyData)
-
-Value = _reflection.GeneratedProtocolMessageType('Value', (_message.Message,), {
-  'DESCRIPTOR' : _VALUE,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.Value)
-  })
-_sym_db.RegisterMessage(Value)
-
-Array = _reflection.GeneratedProtocolMessageType('Array', (_message.Message,), {
-  'DESCRIPTOR' : _ARRAY,
-  '__module__' : 'qwak.feature_store.serving.serving_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.api.Array)
-  })
-_sym_db.RegisterMessage(Array)
-
-_SERVINGSERVICE = DESCRIPTOR.services_by_name['ServingService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.serving.serving_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -205,50 +33,50 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _ONTHEFLYDATA.fields_by_name['user_data']._serialized_options = b'\030\001'
   _ONTHEFLYDATA.fields_by_name['schema_data']._options = None
   _ONTHEFLYDATA.fields_by_name['schema_data']._serialized_options = b'\030\001'
-  _REQUESTEDENTITIESMATRIXREQUEST._serialized_start=117
-  _REQUESTEDENTITIESMATRIXREQUEST._serialized_end=413
-  _REQUESTMETADATA._serialized_start=415
-  _REQUESTMETADATA._serialized_end=497
-  _REQUESTEDENTITIESMATRIX._serialized_start=500
-  _REQUESTEDENTITIESMATRIX._serialized_end=665
-  _ENTITIESHEADER._serialized_start=667
-  _ENTITIESHEADER._serialized_end=705
-  _ENTITYVALUEROW._serialized_start=707
-  _ENTITYVALUEROW._serialized_end=761
-  _ENTITYTOFEATURES._serialized_start=763
-  _ENTITYTOFEATURES._serialized_end=868
-  _FEATUREVALUESREQUEST._serialized_start=871
-  _FEATUREVALUESREQUEST._serialized_end=1082
-  _FEATUREVALUESREQUEST_ENTITY._serialized_start=1045
-  _FEATUREVALUESREQUEST_ENTITY._serialized_end=1082
-  _FEATURE._serialized_start=1085
-  _FEATURE._serialized_end=1631
-  _BATCHFEATURE._serialized_start=1633
-  _BATCHFEATURE._serialized_end=1661
-  _BATCHV1FEATURE._serialized_start=1663
-  _BATCHV1FEATURE._serialized_end=1693
-  _STREAMINGFEATURE._serialized_start=1695
-  _STREAMINGFEATURE._serialized_end=1727
-  _STREAMINGV1FEATURE._serialized_start=1729
-  _STREAMINGV1FEATURE._serialized_end=1763
-  _STREAMINGAGGREGATIONFEATURE._serialized_start=1765
-  _STREAMINGAGGREGATIONFEATURE._serialized_end=1808
-  _ONTHEFLYFEATURE._serialized_start=1810
-  _ONTHEFLYFEATURE._serialized_end=1927
-  _MULTIFEATUREVALUESRESPONSE._serialized_start=1929
-  _MULTIFEATUREVALUESRESPONSE._serialized_end=2008
-  _FEATUREVALUESRESPONSE._serialized_start=2011
-  _FEATUREVALUESRESPONSE._serialized_end=2383
-  _FEATUREVALUESRESPONSE_FEATUREVALUE._serialized_start=2136
-  _FEATUREVALUESRESPONSE_FEATUREVALUE._serialized_end=2383
-  _FEATUREVALUESRESPONSE_FEATUREVALUE_VALUESTATUS._serialized_start=2346
-  _FEATUREVALUESRESPONSE_FEATUREVALUE_VALUESTATUS._serialized_end=2383
-  _ONTHEFLYDATA._serialized_start=2385
-  _ONTHEFLYDATA._serialized_end=2459
-  _VALUE._serialized_start=2462
-  _VALUE._serialized_end=2744
-  _ARRAY._serialized_start=2746
-  _ARRAY._serialized_end=2818
-  _SERVINGSERVICE._serialized_start=2821
-  _SERVINGSERVICE._serialized_end=3135
+  _globals['_REQUESTEDENTITIESMATRIXREQUEST']._serialized_start=117
+  _globals['_REQUESTEDENTITIESMATRIXREQUEST']._serialized_end=413
+  _globals['_REQUESTMETADATA']._serialized_start=415
+  _globals['_REQUESTMETADATA']._serialized_end=497
+  _globals['_REQUESTEDENTITIESMATRIX']._serialized_start=500
+  _globals['_REQUESTEDENTITIESMATRIX']._serialized_end=665
+  _globals['_ENTITIESHEADER']._serialized_start=667
+  _globals['_ENTITIESHEADER']._serialized_end=705
+  _globals['_ENTITYVALUEROW']._serialized_start=707
+  _globals['_ENTITYVALUEROW']._serialized_end=761
+  _globals['_ENTITYTOFEATURES']._serialized_start=763
+  _globals['_ENTITYTOFEATURES']._serialized_end=868
+  _globals['_FEATUREVALUESREQUEST']._serialized_start=871
+  _globals['_FEATUREVALUESREQUEST']._serialized_end=1082
+  _globals['_FEATUREVALUESREQUEST_ENTITY']._serialized_start=1045
+  _globals['_FEATUREVALUESREQUEST_ENTITY']._serialized_end=1082
+  _globals['_FEATURE']._serialized_start=1085
+  _globals['_FEATURE']._serialized_end=1631
+  _globals['_BATCHFEATURE']._serialized_start=1633
+  _globals['_BATCHFEATURE']._serialized_end=1661
+  _globals['_BATCHV1FEATURE']._serialized_start=1663
+  _globals['_BATCHV1FEATURE']._serialized_end=1693
+  _globals['_STREAMINGFEATURE']._serialized_start=1695
+  _globals['_STREAMINGFEATURE']._serialized_end=1727
+  _globals['_STREAMINGV1FEATURE']._serialized_start=1729
+  _globals['_STREAMINGV1FEATURE']._serialized_end=1763
+  _globals['_STREAMINGAGGREGATIONFEATURE']._serialized_start=1765
+  _globals['_STREAMINGAGGREGATIONFEATURE']._serialized_end=1808
+  _globals['_ONTHEFLYFEATURE']._serialized_start=1810
+  _globals['_ONTHEFLYFEATURE']._serialized_end=1927
+  _globals['_MULTIFEATUREVALUESRESPONSE']._serialized_start=1929
+  _globals['_MULTIFEATUREVALUESRESPONSE']._serialized_end=2008
+  _globals['_FEATUREVALUESRESPONSE']._serialized_start=2011
+  _globals['_FEATUREVALUESRESPONSE']._serialized_end=2383
+  _globals['_FEATUREVALUESRESPONSE_FEATUREVALUE']._serialized_start=2136
+  _globals['_FEATUREVALUESRESPONSE_FEATUREVALUE']._serialized_end=2383
+  _globals['_FEATUREVALUESRESPONSE_FEATUREVALUE_VALUESTATUS']._serialized_start=2346
+  _globals['_FEATUREVALUESRESPONSE_FEATUREVALUE_VALUESTATUS']._serialized_end=2383
+  _globals['_ONTHEFLYDATA']._serialized_start=2385
+  _globals['_ONTHEFLYDATA']._serialized_end=2459
+  _globals['_VALUE']._serialized_start=2462
+  _globals['_VALUE']._serialized_end=2744
+  _globals['_ARRAY']._serialized_start=2746
+  _globals['_ARRAY']._serialized_end=2818
+  _globals['_SERVINGSERVICE']._serialized_start=2821
+  _globals['_SERVINGSERVICE']._serialized_end=3135
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/serving/v1/value_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,30 +16,15 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)qwak/feature_store/serving/v1/value.proto\x12$com.qwak.ai.feature.store.serving.v1\x1a\x1fgoogle/protobuf/timestamp.proto\"\xad\x02\n\x07ValueV1\x12\x0e\n\x06is_set\x18\x01 \x01(\x08\x12\x13\n\tbytes_val\x18\x02 \x01(\x0cH\x00\x12\x14\n\nstring_val\x18\x03 \x01(\tH\x00\x12\x13\n\tint32_val\x18\x04 \x01(\x05H\x00\x12\x13\n\tint64_val\x18\x05 \x01(\x03H\x00\x12\x14\n\ndouble_val\x18\x06 \x01(\x01H\x00\x12\x13\n\tfloat_val\x18\x07 \x01(\x02H\x00\x12\x12\n\x08\x62ool_val\x18\x08 \x01(\x08H\x00\x12\x33\n\rtimestamp_val\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00\x12\x42\n\tarray_val\x18\n \x01(\x0b\x32-.com.qwak.ai.feature.store.serving.v1.ArrayV1H\x00\x42\x05\n\x03val\"K\n\x07\x41rrayV1\x12@\n\tarray_val\x18\x01 \x03(\x0b\x32-.com.qwak.ai.feature.store.serving.v1.ValueV1B&\n$com.qwak.ai.feature.store.serving.v1b\x06proto3')
 
-
-
-_VALUEV1 = DESCRIPTOR.message_types_by_name['ValueV1']
-_ARRAYV1 = DESCRIPTOR.message_types_by_name['ArrayV1']
-ValueV1 = _reflection.GeneratedProtocolMessageType('ValueV1', (_message.Message,), {
-  'DESCRIPTOR' : _VALUEV1,
-  '__module__' : 'qwak.feature_store.serving.v1.value_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.v1.ValueV1)
-  })
-_sym_db.RegisterMessage(ValueV1)
-
-ArrayV1 = _reflection.GeneratedProtocolMessageType('ArrayV1', (_message.Message,), {
-  'DESCRIPTOR' : _ARRAYV1,
-  '__module__' : 'qwak.feature_store.serving.v1.value_pb2'
-  # @@protoc_insertion_point(class_scope:com.qwak.ai.feature.store.serving.v1.ArrayV1)
-  })
-_sym_db.RegisterMessage(ArrayV1)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.serving.v1.value_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n$com.qwak.ai.feature.store.serving.v1'
-  _VALUEV1._serialized_start=117
-  _VALUEV1._serialized_end=418
-  _ARRAYV1._serialized_start=420
-  _ARRAYV1._serialized_end=495
+  _globals['_VALUEV1']._serialized_start=117
+  _globals['_VALUEV1']._serialized_end=418
+  _globals['_ARRAYV1']._serialized_start=420
+  _globals['_ARRAYV1']._serialized_end=495
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/sinks/sink_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: qwak/feature_store/sinks/sink.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()
+
+
+from _qwak_proto.qwak.feature_store.sources import streaming_pb2 as qwak_dot_feature__store_dot_sources_dot_streaming__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#qwak/feature_store/sinks/sink.proto\x12\x18qwak.feature.store.sinks\x1a*qwak/feature_store/sources/streaming.proto\"\xb3\x01\n\rStreamingSink\x12L\n\x10\x61ttachment_point\x18\x01 \x01(\x0b\x32\x32.qwak.feature.store.sinks.StreamingAttachmentPoint\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x39\n\nkafka_sink\x18\x02 \x01(\x0b\x32#.qwak.feature.store.sinks.KafkaSinkH\x00\x42\x0b\n\tsink_type\"\xb9\x01\n\tKafkaSink\x12\r\n\x05topic\x18\x01 \x01(\t\x12\x19\n\x11\x62ootstrap_servers\x18\x02 \x01(\t\x12\x41\n\x0emessage_format\x18\x03 \x01(\x0e\x32).qwak.feature.store.sources.MessageFormat\x12?\n\x0b\x61uth_config\x18\x04 \x01(\x0b\x32*.qwak.feature.store.sources.Authentication\" \n\x1eOnlineStreamingAttachmentPoint\"!\n\x1fOfflineStreamingAttachmentPoint\"\x83\x02\n\x18StreamingAttachmentPoint\x12\x65\n!online_streaming_attachment_point\x18\x01 \x01(\x0b\x32\x38.qwak.feature.store.sinks.OnlineStreamingAttachmentPointH\x00\x12g\n\"offline_streaming_attachment_point\x18\x02 \x01(\x0b\x32\x39.qwak.feature.store.sinks.OfflineStreamingAttachmentPointH\x00\x42\x17\n\x15\x61ttachment_point_typeBR\n#com.qwak.ai.feature.store.sinks.apiP\x01Z)qwak/featurestore/sinks;featurestoresinksb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.sinks.sink_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n#com.qwak.ai.feature.store.sinks.apiP\001Z)qwak/featurestore/sinks;featurestoresinks'
+  _globals['_STREAMINGSINK']._serialized_start=110
+  _globals['_STREAMINGSINK']._serialized_end=289
+  _globals['_KAFKASINK']._serialized_start=292
+  _globals['_KAFKASINK']._serialized_end=477
+  _globals['_ONLINESTREAMINGATTACHMENTPOINT']._serialized_start=479
+  _globals['_ONLINESTREAMINGATTACHMENTPOINT']._serialized_end=511
+  _globals['_OFFLINESTREAMINGATTACHMENTPOINT']._serialized_start=513
+  _globals['_OFFLINESTREAMINGATTACHMENTPOINT']._serialized_end=546
+  _globals['_STREAMINGATTACHMENTPOINT']._serialized_start=549
+  _globals['_STREAMINGATTACHMENTPOINT']._serialized_end=808
+# @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/sinks/sink_pb2.pyi

@@ -0,0 +1,137 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import qwak.feature_store.sources.streaming_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class StreamingSink(google.protobuf.message.Message):
+    """This section defines the general sink definitions based on the FS type, i.e.-
+    a type for streaming featuresets, a type for batch featuresets etc.
+    this is necessary for considerations such as attachment points etc. that are
+    inherent to the featureset type and not the concrete sink type
+
+
+    a sink for a streaming featureset (streaming here means the sink is aware of the online/offline
+    jobs and defines which of the jobs it attaches to)
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ATTACHMENT_POINT_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    KAFKA_SINK_FIELD_NUMBER: builtins.int
+    @property
+    def attachment_point(self) -> global___StreamingAttachmentPoint:
+        """The attachment point (i.e., which job is responsible for writing to this sink)"""
+    name: builtins.str
+    """sink name"""
+    @property
+    def kafka_sink(self) -> global___KafkaSink:
+        """A kafka sink
+        NOTE: DO NOT USE field number 3, it's already used above
+        """
+    def __init__(
+        self,
+        *,
+        attachment_point: global___StreamingAttachmentPoint | None = ...,
+        name: builtins.str = ...,
+        kafka_sink: global___KafkaSink | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["attachment_point", b"attachment_point", "kafka_sink", b"kafka_sink", "sink_type", b"sink_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["attachment_point", b"attachment_point", "kafka_sink", b"kafka_sink", "name", b"name", "sink_type", b"sink_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["sink_type", b"sink_type"]) -> typing_extensions.Literal["kafka_sink"] | None: ...
+
+global___StreamingSink = StreamingSink
+
+class KafkaSink(google.protobuf.message.Message):
+    """######################### Concrete  Sink Types #########################
+    these are concrete sink types (e.g., Kafka, Kinesis)
+    the relation of sink-by-fs-type:conrete-sink-type is a many:many relation
+
+    A featureset sink that writes to kafka
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TOPIC_FIELD_NUMBER: builtins.int
+    BOOTSTRAP_SERVERS_FIELD_NUMBER: builtins.int
+    MESSAGE_FORMAT_FIELD_NUMBER: builtins.int
+    AUTH_CONFIG_FIELD_NUMBER: builtins.int
+    topic: builtins.str
+    """Target topic"""
+    bootstrap_servers: builtins.str
+    """bootstrap servers, separated by a comma"""
+    message_format: qwak.feature_store.sources.streaming_pb2.MessageFormat.ValueType
+    """The format of the messages written to the sink"""
+    @property
+    def auth_config(self) -> qwak.feature_store.sources.streaming_pb2.Authentication:
+        """Auth configuration"""
+    def __init__(
+        self,
+        *,
+        topic: builtins.str = ...,
+        bootstrap_servers: builtins.str = ...,
+        message_format: qwak.feature_store.sources.streaming_pb2.MessageFormat.ValueType = ...,
+        auth_config: qwak.feature_store.sources.streaming_pb2.Authentication | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["auth_config", b"auth_config"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auth_config", b"auth_config", "bootstrap_servers", b"bootstrap_servers", "message_format", b"message_format", "topic", b"topic"]) -> None: ...
+
+global___KafkaSink = KafkaSink
+
+class OnlineStreamingAttachmentPoint(google.protobuf.message.Message):
+    """######################### Attachment Points #########################
+
+    Attachment Point on the online job of streaming featureset
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___OnlineStreamingAttachmentPoint = OnlineStreamingAttachmentPoint
+
+class OfflineStreamingAttachmentPoint(google.protobuf.message.Message):
+    """Attachment Point on the offline job of a streaming featureset"""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___OfflineStreamingAttachmentPoint = OfflineStreamingAttachmentPoint
+
+class StreamingAttachmentPoint(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ONLINE_STREAMING_ATTACHMENT_POINT_FIELD_NUMBER: builtins.int
+    OFFLINE_STREAMING_ATTACHMENT_POINT_FIELD_NUMBER: builtins.int
+    @property
+    def online_streaming_attachment_point(self) -> global___OnlineStreamingAttachmentPoint: ...
+    @property
+    def offline_streaming_attachment_point(self) -> global___OfflineStreamingAttachmentPoint: ...
+    def __init__(
+        self,
+        *,
+        online_streaming_attachment_point: global___OnlineStreamingAttachmentPoint | None = ...,
+        offline_streaming_attachment_point: global___OfflineStreamingAttachmentPoint | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["attachment_point_type", b"attachment_point_type", "offline_streaming_attachment_point", b"offline_streaming_attachment_point", "online_streaming_attachment_point", b"online_streaming_attachment_point"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["attachment_point_type", b"attachment_point_type", "offline_streaming_attachment_point", b"offline_streaming_attachment_point", "online_streaming_attachment_point", b"online_streaming_attachment_point"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["attachment_point_type", b"attachment_point_type"]) -> typing_extensions.Literal["online_streaming_attachment_point", "offline_streaming_attachment_point"] | None: ...
+
+global___StreamingAttachmentPoint = StreamingAttachmentPoint

_qwak_proto/qwak/feature_store/sinks/sink_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
+