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

_qwak_proto/qwak/feature_store/sources/data_source_attribute_pb2.pyi

@@ -0,0 +1,33 @@
+"""
+@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.v1.common.source_code.source_code_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class DataSourceAttributes(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SOURCE_CODE_SPEC_FIELD_NUMBER: builtins.int
+    @property
+    def source_code_spec(self) -> qwak.feature_store.v1.common.source_code.source_code_pb2.SourceCodeSpec:
+        """Data source - source code"""
+    def __init__(
+        self,
+        *,
+        source_code_spec: qwak.feature_store.v1.common.source_code.source_code_pb2.SourceCodeSpec | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["source_code_spec", b"source_code_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["source_code_spec", b"source_code_spec"]) -> None: ...
+
+global___DataSourceAttributes = DataSourceAttributes

_qwak_proto/qwak/feature_store/sources/data_source_attribute_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/sources/data_source_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()
@@ -15,64 +14,31 @@ _sym_db = _symbol_database.Default()
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 from _qwak_proto.qwak.feature_store.sources import batch_pb2 as qwak_dot_feature__store_dot_sources_dot_batch__pb2
 from _qwak_proto.qwak.feature_store.sources import streaming_pb2 as qwak_dot_feature__store_dot_sources_dot_streaming__pb2
+from _qwak_proto.qwak.feature_store.sources import data_source_attribute_pb2 as qwak_dot_feature__store_dot_sources_dot_data__source__attribute__pb2
+from _qwak_proto.qwak.feature_store.v1.common.source_code import source_code_pb2 as qwak_dot_feature__store_dot_v1_dot_common_dot_source__code_dot_source__code__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/feature_store/sources/data_source.proto\x12\x1aqwak.feature.store.sources\x1a\x1fgoogle/protobuf/timestamp.proto\x1a&qwak/feature_store/sources/batch.proto\x1a*qwak/feature_store/sources/streaming.proto\"\xe3\x01\n\nDataSource\x12P\n\x16\x64\x61ta_source_definition\x18\x01 \x01(\x0b\x32\x30.qwak.feature.store.sources.DataSourceDefinition\x12@\n\x08metadata\x18\x02 \x01(\x0b\x32..qwak.feature.store.sources.DataSourceMetadata\x12\x41\n\x0c\x66\x65\x61ture_sets\x18\x03 \x03(\x0b\x32+.qwak.feature.store.sources.FeatureSetBrief\"t\n\x14\x44\x61taSourceDefinition\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"\xa8\x01\n\x12\x44\x61taSourceMetadata\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\"\x9f\x01\n\x0e\x44\x61taSourceSpec\x12?\n\x0c\x62\x61tch_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSourceH\x00\x12\x44\n\rstream_source\x18\x02 \x01(\x0b\x32+.qwak.feature.store.sources.StreamingSourceH\x00\x42\x06\n\x04type\"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(\tBX\n%com.qwak.ai.feature.store.sources.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
-
-
-
-_DATASOURCE = DESCRIPTOR.message_types_by_name['DataSource']
-_DATASOURCEDEFINITION = DESCRIPTOR.message_types_by_name['DataSourceDefinition']
-_DATASOURCEMETADATA = DESCRIPTOR.message_types_by_name['DataSourceMetadata']
-_DATASOURCESPEC = DESCRIPTOR.message_types_by_name['DataSourceSpec']
-_FEATURESETBRIEF = DESCRIPTOR.message_types_by_name['FeatureSetBrief']
-DataSource = _reflection.GeneratedProtocolMessageType('DataSource', (_message.Message,), {
-  'DESCRIPTOR' : _DATASOURCE,
-  '__module__' : 'qwak.feature_store.sources.data_source_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.DataSource)
-  })
-_sym_db.RegisterMessage(DataSource)
-
-DataSourceDefinition = _reflection.GeneratedProtocolMessageType('DataSourceDefinition', (_message.Message,), {
-  'DESCRIPTOR' : _DATASOURCEDEFINITION,
-  '__module__' : 'qwak.feature_store.sources.data_source_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.DataSourceDefinition)
-  })
-_sym_db.RegisterMessage(DataSourceDefinition)
-
-DataSourceMetadata = _reflection.GeneratedProtocolMessageType('DataSourceMetadata', (_message.Message,), {
-  'DESCRIPTOR' : _DATASOURCEMETADATA,
-  '__module__' : 'qwak.feature_store.sources.data_source_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.DataSourceMetadata)
-  })
-_sym_db.RegisterMessage(DataSourceMetadata)
-
-DataSourceSpec = _reflection.GeneratedProtocolMessageType('DataSourceSpec', (_message.Message,), {
-  'DESCRIPTOR' : _DATASOURCESPEC,
-  '__module__' : 'qwak.feature_store.sources.data_source_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.DataSourceSpec)
-  })
-_sym_db.RegisterMessage(DataSourceSpec)
-
-FeatureSetBrief = _reflection.GeneratedProtocolMessageType('FeatureSetBrief', (_message.Message,), {
-  'DESCRIPTOR' : _FEATURESETBRIEF,
-  '__module__' : 'qwak.feature_store.sources.data_source_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.FeatureSetBrief)
-  })
-_sym_db.RegisterMessage(FeatureSetBrief)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n,qwak/feature_store/sources/data_source.proto\x12\x1aqwak.feature.store.sources\x1a\x1fgoogle/protobuf/timestamp.proto\x1a&qwak/feature_store/sources/batch.proto\x1a*qwak/feature_store/sources/streaming.proto\x1a\x36qwak/feature_store/sources/data_source_attribute.proto\x1a:qwak/feature_store/v1/common/source_code/source_code.proto\"\xe3\x01\n\nDataSource\x12P\n\x16\x64\x61ta_source_definition\x18\x01 \x01(\x0b\x32\x30.qwak.feature.store.sources.DataSourceDefinition\x12@\n\x08metadata\x18\x02 \x01(\x0b\x32..qwak.feature.store.sources.DataSourceMetadata\x12\x41\n\x0c\x66\x65\x61ture_sets\x18\x03 \x03(\x0b\x32+.qwak.feature.store.sources.FeatureSetBrief\"t\n\x14\x44\x61taSourceDefinition\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"\xc0\x01\n\x12\x44\x61taSourceMetadata\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\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\"\xe3\x03\n\x0e\x44\x61taSourceSpec\x12?\n\x0c\x62\x61tch_source\x18\x01 \x01(\x0b\x32\'.qwak.feature.store.sources.BatchSourceH\x00\x12\x44\n\rstream_source\x18\x02 \x01(\x0b\x32+.qwak.feature.store.sources.StreamingSourceH\x00\x12V\n\x10source_code_spec\x18\x03 \x01(\x0b\x32\x38.qwak.feature.store.v1.common.source_code.SourceCodeSpecB\x02\x18\x01\x12P\n\x16\x64\x61ta_source_attributes\x18\x04 \x01(\x0b\x32\x30.qwak.feature.store.sources.DataSourceAttributes\x12(\n\x1a\x66\x65\x61tureset_repository_name\x18\x05 \x01(\tB\x02\x18\x01H\x01\x12%\n\x1b\x64\x61ta_source_repository_name\x18\x06 \x01(\tH\x02\x42\x06\n\x04typeB\"\n featureset_repository_identifierB#\n!data_source_repository_identifier\"]\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(\t\x12\x18\n\x10\x66\x65\x61ture_set_type\x18\x03 \x01(\x05\x42X\n%com.qwak.ai.feature.store.sources.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.sources.data_source_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n%com.qwak.ai.feature.store.sources.apiP\001Z-qwak/featurestore/sources;featurestoresources'
-  _DATASOURCE._serialized_start=194
-  _DATASOURCE._serialized_end=421
-  _DATASOURCEDEFINITION._serialized_start=423
-  _DATASOURCEDEFINITION._serialized_end=539
-  _DATASOURCEMETADATA._serialized_start=542
-  _DATASOURCEMETADATA._serialized_end=710
-  _DATASOURCESPEC._serialized_start=713
-  _DATASOURCESPEC._serialized_end=872
-  _FEATURESETBRIEF._serialized_start=874
-  _FEATURESETBRIEF._serialized_end=941
+  _DATASOURCESPEC.fields_by_name['source_code_spec']._options = None
+  _DATASOURCESPEC.fields_by_name['source_code_spec']._serialized_options = b'\030\001'
+  _DATASOURCESPEC.fields_by_name['featureset_repository_name']._options = None
+  _DATASOURCESPEC.fields_by_name['featureset_repository_name']._serialized_options = b'\030\001'
+  _globals['_DATASOURCE']._serialized_start=310
+  _globals['_DATASOURCE']._serialized_end=537
+  _globals['_DATASOURCEDEFINITION']._serialized_start=539
+  _globals['_DATASOURCEDEFINITION']._serialized_end=655
+  _globals['_DATASOURCEMETADATA']._serialized_start=658
+  _globals['_DATASOURCEMETADATA']._serialized_end=850
+  _globals['_DATASOURCESPEC']._serialized_start=853
+  _globals['_DATASOURCESPEC']._serialized_end=1336
+  _globals['_FEATURESETBRIEF']._serialized_start=1338
+  _globals['_FEATURESETBRIEF']._serialized_end=1431
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/sources/data_source_pb2.pyi

@@ -9,8 +9,11 @@ import google.protobuf.internal.containers
 import google.protobuf.message
 import google.protobuf.timestamp_pb2
 import qwak.feature_store.sources.batch_pb2
+import qwak.feature_store.sources.data_source_attribute_pb2
 import qwak.feature_store.sources.streaming_pb2
+import qwak.feature_store.v1.common.source_code.source_code_pb2
 import sys
+import typing
 
 if sys.version_info >= (3, 8):
     import typing as typing_extensions
@@ -74,6 +77,7 @@ class DataSourceMetadata(google.protobuf.message.Message):
     CREATED_BY_FIELD_NUMBER: builtins.int
     LAST_MODIFIED_AT_FIELD_NUMBER: builtins.int
     LAST_MODIFIED_BY_FIELD_NUMBER: builtins.int
+    ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
     @property
     def created_at(self) -> google.protobuf.timestamp_pb2.Timestamp:
         """When the data source was created"""
@@ -84,6 +88,8 @@ class DataSourceMetadata(google.protobuf.message.Message):
         """Last modified"""
     last_modified_by: builtins.str
     """Last modified by"""
+    environment_id: builtins.str
+    """Environment id"""
     def __init__(
         self,
         *,
@@ -91,9 +97,10 @@ class DataSourceMetadata(google.protobuf.message.Message):
         created_by: builtins.str = ...,
         last_modified_at: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         last_modified_by: builtins.str = ...,
+        environment_id: builtins.str = ...,
     ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "last_modified_at", b"last_modified_at"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["created_at", b"created_at", "created_by", b"created_by", "environment_id", b"environment_id", "last_modified_at", b"last_modified_at", "last_modified_by", b"last_modified_by"]) -> None: ...
 
 global___DataSourceMetadata = DataSourceMetadata
 
@@ -102,18 +109,42 @@ class DataSourceSpec(google.protobuf.message.Message):
 
     BATCH_SOURCE_FIELD_NUMBER: builtins.int
     STREAM_SOURCE_FIELD_NUMBER: builtins.int
+    SOURCE_CODE_SPEC_FIELD_NUMBER: builtins.int
+    DATA_SOURCE_ATTRIBUTES_FIELD_NUMBER: builtins.int
+    FEATURESET_REPOSITORY_NAME_FIELD_NUMBER: builtins.int
+    DATA_SOURCE_REPOSITORY_NAME_FIELD_NUMBER: builtins.int
     @property
     def batch_source(self) -> qwak.feature_store.sources.batch_pb2.BatchSource: ...
     @property
     def stream_source(self) -> qwak.feature_store.sources.streaming_pb2.StreamingSource: ...
+    @property
+    def source_code_spec(self) -> qwak.feature_store.v1.common.source_code.source_code_pb2.SourceCodeSpec: ...
+    @property
+    def data_source_attributes(self) -> qwak.feature_store.sources.data_source_attribute_pb2.DataSourceAttributes:
+        """Data source attributes"""
+    featureset_repository_name: builtins.str
+    """This field is marked as deprecated
+    Typo in the name
+    """
+    data_source_repository_name: builtins.str
+    """Data source repository name"""
     def __init__(
         self,
         *,
         batch_source: qwak.feature_store.sources.batch_pb2.BatchSource | None = ...,
         stream_source: qwak.feature_store.sources.streaming_pb2.StreamingSource | None = ...,
+        source_code_spec: qwak.feature_store.v1.common.source_code.source_code_pb2.SourceCodeSpec | None = ...,
+        data_source_attributes: qwak.feature_store.sources.data_source_attribute_pb2.DataSourceAttributes | None = ...,
+        featureset_repository_name: builtins.str = ...,
+        data_source_repository_name: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["batch_source", b"batch_source", "stream_source", b"stream_source", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["batch_source", b"batch_source", "stream_source", b"stream_source", "type", b"type"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["batch_source", b"batch_source", "data_source_attributes", b"data_source_attributes", "data_source_repository_identifier", b"data_source_repository_identifier", "data_source_repository_name", b"data_source_repository_name", "featureset_repository_identifier", b"featureset_repository_identifier", "featureset_repository_name", b"featureset_repository_name", "source_code_spec", b"source_code_spec", "stream_source", b"stream_source", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_source", b"batch_source", "data_source_attributes", b"data_source_attributes", "data_source_repository_identifier", b"data_source_repository_identifier", "data_source_repository_name", b"data_source_repository_name", "featureset_repository_identifier", b"featureset_repository_identifier", "featureset_repository_name", b"featureset_repository_name", "source_code_spec", b"source_code_spec", "stream_source", b"stream_source", "type", b"type"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["data_source_repository_identifier", b"data_source_repository_identifier"]) -> typing_extensions.Literal["data_source_repository_name"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["featureset_repository_identifier", b"featureset_repository_identifier"]) -> typing_extensions.Literal["featureset_repository_name"] | None: ...
+    @typing.overload
     def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["batch_source", "stream_source"] | None: ...
 
 global___DataSourceSpec = DataSourceSpec
@@ -123,16 +154,20 @@ class FeatureSetBrief(google.protobuf.message.Message):
 
     FEATURE_SET_ID_FIELD_NUMBER: builtins.int
     FEATURE_SET_NAME_FIELD_NUMBER: builtins.int
+    FEATURE_SET_TYPE_FIELD_NUMBER: builtins.int
     feature_set_id: builtins.str
     """Feature set id"""
     feature_set_name: builtins.str
     """Feature set name"""
+    feature_set_type: builtins.int
+    """Feature set type"""
     def __init__(
         self,
         *,
         feature_set_id: builtins.str = ...,
         feature_set_name: builtins.str = ...,
+        feature_set_type: builtins.int = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["feature_set_id", b"feature_set_id", "feature_set_name", b"feature_set_name"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["feature_set_id", b"feature_set_id", "feature_set_name", b"feature_set_name", "feature_set_type", b"feature_set_type"]) -> None: ...
 
 global___FeatureSetBrief = FeatureSetBrief

_qwak_proto/qwak/feature_store/sources/data_source_service_pb2.py

@@ -4,9 +4,8 @@
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -15,155 +14,49 @@ _sym_db = _symbol_database.Default()
 from _qwak_proto.qwak.feature_store.sources import data_source_pb2 as qwak_dot_feature__store_dot_sources_dot_data__source__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4qwak/feature_store/sources/data_source_service.proto\x12\x1aqwak.feature.store.sources\x1a,qwak/feature_store/sources/data_source.proto\"_\n\x17\x43reateDataSourceRequest\x12\x44\n\x10\x64\x61ta_source_spec\x18\x01 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18\x43reateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"w\n\x17UpdateDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18UpdateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"1\n\x17\x44\x65leteDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"\x1a\n\x18\x44\x65leteDataSourceResponse\"2\n\x18GetDataSourceByIdRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"X\n\x19GetDataSourceByIdResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"6\n\x1aGetDataSourceByNameRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"Z\n\x1bGetDataSourceByNameResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"\x18\n\x16ListDataSourcesRequest\"W\n\x17ListDataSourcesResponse\x12<\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32&.qwak.feature.store.sources.DataSource\"Q\n CreateDataSourceUploadURLRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\x12\x13\n\x0bobject_name\x18\x02 \x01(\t\"7\n!CreateDataSourceUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t2\xd0\x08\n\x11\x44\x61taSourceService\x12}\n\x10\x43reateDataSource\x12\x33.qwak.feature.store.sources.CreateDataSourceRequest\x1a\x34.qwak.feature.store.sources.CreateDataSourceResponse\x12}\n\x10UpdateDataSource\x12\x33.qwak.feature.store.sources.UpdateDataSourceRequest\x1a\x34.qwak.feature.store.sources.UpdateDataSourceResponse\x12}\n\x10\x44\x65leteDataSource\x12\x33.qwak.feature.store.sources.DeleteDataSourceRequest\x1a\x34.qwak.feature.store.sources.DeleteDataSourceResponse\x12\x80\x01\n\x11GetDataSourceById\x12\x34.qwak.feature.store.sources.GetDataSourceByIdRequest\x1a\x35.qwak.feature.store.sources.GetDataSourceByIdResponse\x12\x86\x01\n\x13GetDataSourceByName\x12\x36.qwak.feature.store.sources.GetDataSourceByNameRequest\x1a\x37.qwak.feature.store.sources.GetDataSourceByNameResponse\x12z\n\x0fListDataSources\x12\x32.qwak.feature.store.sources.ListDataSourcesRequest\x1a\x33.qwak.feature.store.sources.ListDataSourcesResponse\x12\x98\x01\n\x19\x43reateDataSourceUploadURL\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponse\x12\x9a\x01\n\x1b\x43reateDataSourceUploadURLV1\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponseB[\n(com.qwak.ai.feature.store.management.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4qwak/feature_store/sources/data_source_service.proto\x12\x1aqwak.feature.store.sources\x1a,qwak/feature_store/sources/data_source.proto\"_\n\x17\x43reateDataSourceRequest\x12\x44\n\x10\x64\x61ta_source_spec\x18\x01 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18\x43reateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"w\n\x17UpdateDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\x12\x44\n\x10\x64\x61ta_source_spec\x18\x02 \x01(\x0b\x32*.qwak.feature.store.sources.DataSourceSpec\"W\n\x18UpdateDataSourceResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"1\n\x17\x44\x65leteDataSourceRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"\x1a\n\x18\x44\x65leteDataSourceResponse\"2\n\x18GetDataSourceByIdRequest\x12\x16\n\x0e\x64\x61ta_source_id\x18\x01 \x01(\t\"X\n\x19GetDataSourceByIdResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"6\n\x1aGetDataSourceByNameRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\"Z\n\x1bGetDataSourceByNameResponse\x12;\n\x0b\x64\x61ta_source\x18\x01 \x01(\x0b\x32&.qwak.feature.store.sources.DataSource\"\x18\n\x16ListDataSourcesRequest\"W\n\x17ListDataSourcesResponse\x12<\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32&.qwak.feature.store.sources.DataSource\"!\n\x1fListDataSourcesByAccountRequest\"`\n ListDataSourcesByAccountResponse\x12<\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32&.qwak.feature.store.sources.DataSource\"Q\n CreateDataSourceUploadURLRequest\x12\x18\n\x10\x64\x61ta_source_name\x18\x01 \x01(\t\x12\x13\n\x0bobject_name\x18\x02 \x01(\t\"7\n!CreateDataSourceUploadURLResponse\x12\x12\n\nupload_url\x18\x01 \x01(\t2\xed\t\n\x11\x44\x61taSourceService\x12}\n\x10\x43reateDataSource\x12\x33.qwak.feature.store.sources.CreateDataSourceRequest\x1a\x34.qwak.feature.store.sources.CreateDataSourceResponse\x12}\n\x10UpdateDataSource\x12\x33.qwak.feature.store.sources.UpdateDataSourceRequest\x1a\x34.qwak.feature.store.sources.UpdateDataSourceResponse\x12}\n\x10\x44\x65leteDataSource\x12\x33.qwak.feature.store.sources.DeleteDataSourceRequest\x1a\x34.qwak.feature.store.sources.DeleteDataSourceResponse\x12\x80\x01\n\x11GetDataSourceById\x12\x34.qwak.feature.store.sources.GetDataSourceByIdRequest\x1a\x35.qwak.feature.store.sources.GetDataSourceByIdResponse\x12\x86\x01\n\x13GetDataSourceByName\x12\x36.qwak.feature.store.sources.GetDataSourceByNameRequest\x1a\x37.qwak.feature.store.sources.GetDataSourceByNameResponse\x12z\n\x0fListDataSources\x12\x32.qwak.feature.store.sources.ListDataSourcesRequest\x1a\x33.qwak.feature.store.sources.ListDataSourcesResponse\x12\x95\x01\n\x18ListDataSourcesByAccount\x12;.qwak.feature.store.sources.ListDataSourcesByAccountRequest\x1a<.qwak.feature.store.sources.ListDataSourcesByAccountResponse\x12\x9d\x01\n\x19\x43reateDataSourceUploadURL\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponse\"\x03\x88\x02\x01\x12\x9a\x01\n\x1b\x43reateDataSourceUploadURLV1\x12<.qwak.feature.store.sources.CreateDataSourceUploadURLRequest\x1a=.qwak.feature.store.sources.CreateDataSourceUploadURLResponseB[\n(com.qwak.ai.feature.store.management.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
 
-
-
-_CREATEDATASOURCEREQUEST = DESCRIPTOR.message_types_by_name['CreateDataSourceRequest']
-_CREATEDATASOURCERESPONSE = DESCRIPTOR.message_types_by_name['CreateDataSourceResponse']
-_UPDATEDATASOURCEREQUEST = DESCRIPTOR.message_types_by_name['UpdateDataSourceRequest']
-_UPDATEDATASOURCERESPONSE = DESCRIPTOR.message_types_by_name['UpdateDataSourceResponse']
-_DELETEDATASOURCEREQUEST = DESCRIPTOR.message_types_by_name['DeleteDataSourceRequest']
-_DELETEDATASOURCERESPONSE = DESCRIPTOR.message_types_by_name['DeleteDataSourceResponse']
-_GETDATASOURCEBYIDREQUEST = DESCRIPTOR.message_types_by_name['GetDataSourceByIdRequest']
-_GETDATASOURCEBYIDRESPONSE = DESCRIPTOR.message_types_by_name['GetDataSourceByIdResponse']
-_GETDATASOURCEBYNAMEREQUEST = DESCRIPTOR.message_types_by_name['GetDataSourceByNameRequest']
-_GETDATASOURCEBYNAMERESPONSE = DESCRIPTOR.message_types_by_name['GetDataSourceByNameResponse']
-_LISTDATASOURCESREQUEST = DESCRIPTOR.message_types_by_name['ListDataSourcesRequest']
-_LISTDATASOURCESRESPONSE = DESCRIPTOR.message_types_by_name['ListDataSourcesResponse']
-_CREATEDATASOURCEUPLOADURLREQUEST = DESCRIPTOR.message_types_by_name['CreateDataSourceUploadURLRequest']
-_CREATEDATASOURCEUPLOADURLRESPONSE = DESCRIPTOR.message_types_by_name['CreateDataSourceUploadURLResponse']
-CreateDataSourceRequest = _reflection.GeneratedProtocolMessageType('CreateDataSourceRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEDATASOURCEREQUEST,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.CreateDataSourceRequest)
-  })
-_sym_db.RegisterMessage(CreateDataSourceRequest)
-
-CreateDataSourceResponse = _reflection.GeneratedProtocolMessageType('CreateDataSourceResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEDATASOURCERESPONSE,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.CreateDataSourceResponse)
-  })
-_sym_db.RegisterMessage(CreateDataSourceResponse)
-
-UpdateDataSourceRequest = _reflection.GeneratedProtocolMessageType('UpdateDataSourceRequest', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEDATASOURCEREQUEST,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.UpdateDataSourceRequest)
-  })
-_sym_db.RegisterMessage(UpdateDataSourceRequest)
-
-UpdateDataSourceResponse = _reflection.GeneratedProtocolMessageType('UpdateDataSourceResponse', (_message.Message,), {
-  'DESCRIPTOR' : _UPDATEDATASOURCERESPONSE,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.UpdateDataSourceResponse)
-  })
-_sym_db.RegisterMessage(UpdateDataSourceResponse)
-
-DeleteDataSourceRequest = _reflection.GeneratedProtocolMessageType('DeleteDataSourceRequest', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEDATASOURCEREQUEST,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.DeleteDataSourceRequest)
-  })
-_sym_db.RegisterMessage(DeleteDataSourceRequest)
-
-DeleteDataSourceResponse = _reflection.GeneratedProtocolMessageType('DeleteDataSourceResponse', (_message.Message,), {
-  'DESCRIPTOR' : _DELETEDATASOURCERESPONSE,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.DeleteDataSourceResponse)
-  })
-_sym_db.RegisterMessage(DeleteDataSourceResponse)
-
-GetDataSourceByIdRequest = _reflection.GeneratedProtocolMessageType('GetDataSourceByIdRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETDATASOURCEBYIDREQUEST,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.GetDataSourceByIdRequest)
-  })
-_sym_db.RegisterMessage(GetDataSourceByIdRequest)
-
-GetDataSourceByIdResponse = _reflection.GeneratedProtocolMessageType('GetDataSourceByIdResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETDATASOURCEBYIDRESPONSE,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.GetDataSourceByIdResponse)
-  })
-_sym_db.RegisterMessage(GetDataSourceByIdResponse)
-
-GetDataSourceByNameRequest = _reflection.GeneratedProtocolMessageType('GetDataSourceByNameRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETDATASOURCEBYNAMEREQUEST,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.GetDataSourceByNameRequest)
-  })
-_sym_db.RegisterMessage(GetDataSourceByNameRequest)
-
-GetDataSourceByNameResponse = _reflection.GeneratedProtocolMessageType('GetDataSourceByNameResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETDATASOURCEBYNAMERESPONSE,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.GetDataSourceByNameResponse)
-  })
-_sym_db.RegisterMessage(GetDataSourceByNameResponse)
-
-ListDataSourcesRequest = _reflection.GeneratedProtocolMessageType('ListDataSourcesRequest', (_message.Message,), {
-  'DESCRIPTOR' : _LISTDATASOURCESREQUEST,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.ListDataSourcesRequest)
-  })
-_sym_db.RegisterMessage(ListDataSourcesRequest)
-
-ListDataSourcesResponse = _reflection.GeneratedProtocolMessageType('ListDataSourcesResponse', (_message.Message,), {
-  'DESCRIPTOR' : _LISTDATASOURCESRESPONSE,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.ListDataSourcesResponse)
-  })
-_sym_db.RegisterMessage(ListDataSourcesResponse)
-
-CreateDataSourceUploadURLRequest = _reflection.GeneratedProtocolMessageType('CreateDataSourceUploadURLRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEDATASOURCEUPLOADURLREQUEST,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.CreateDataSourceUploadURLRequest)
-  })
-_sym_db.RegisterMessage(CreateDataSourceUploadURLRequest)
-
-CreateDataSourceUploadURLResponse = _reflection.GeneratedProtocolMessageType('CreateDataSourceUploadURLResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CREATEDATASOURCEUPLOADURLRESPONSE,
-  '__module__' : 'qwak.feature_store.sources.data_source_service_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.CreateDataSourceUploadURLResponse)
-  })
-_sym_db.RegisterMessage(CreateDataSourceUploadURLResponse)
-
-_DATASOURCESERVICE = DESCRIPTOR.services_by_name['DataSourceService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.sources.data_source_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n(com.qwak.ai.feature.store.management.apiP\001Z-qwak/featurestore/sources;featurestoresources'
-  _CREATEDATASOURCEREQUEST._serialized_start=130
-  _CREATEDATASOURCEREQUEST._serialized_end=225
-  _CREATEDATASOURCERESPONSE._serialized_start=227
-  _CREATEDATASOURCERESPONSE._serialized_end=314
-  _UPDATEDATASOURCEREQUEST._serialized_start=316
-  _UPDATEDATASOURCEREQUEST._serialized_end=435
-  _UPDATEDATASOURCERESPONSE._serialized_start=437
-  _UPDATEDATASOURCERESPONSE._serialized_end=524
-  _DELETEDATASOURCEREQUEST._serialized_start=526
-  _DELETEDATASOURCEREQUEST._serialized_end=575
-  _DELETEDATASOURCERESPONSE._serialized_start=577
-  _DELETEDATASOURCERESPONSE._serialized_end=603
-  _GETDATASOURCEBYIDREQUEST._serialized_start=605
-  _GETDATASOURCEBYIDREQUEST._serialized_end=655
-  _GETDATASOURCEBYIDRESPONSE._serialized_start=657
-  _GETDATASOURCEBYIDRESPONSE._serialized_end=745
-  _GETDATASOURCEBYNAMEREQUEST._serialized_start=747
-  _GETDATASOURCEBYNAMEREQUEST._serialized_end=801
-  _GETDATASOURCEBYNAMERESPONSE._serialized_start=803
-  _GETDATASOURCEBYNAMERESPONSE._serialized_end=893
-  _LISTDATASOURCESREQUEST._serialized_start=895
-  _LISTDATASOURCESREQUEST._serialized_end=919
-  _LISTDATASOURCESRESPONSE._serialized_start=921
-  _LISTDATASOURCESRESPONSE._serialized_end=1008
-  _CREATEDATASOURCEUPLOADURLREQUEST._serialized_start=1010
-  _CREATEDATASOURCEUPLOADURLREQUEST._serialized_end=1091
-  _CREATEDATASOURCEUPLOADURLRESPONSE._serialized_start=1093
-  _CREATEDATASOURCEUPLOADURLRESPONSE._serialized_end=1148
-  _DATASOURCESERVICE._serialized_start=1151
-  _DATASOURCESERVICE._serialized_end=2255
+  _DATASOURCESERVICE.methods_by_name['CreateDataSourceUploadURL']._options = None
+  _DATASOURCESERVICE.methods_by_name['CreateDataSourceUploadURL']._serialized_options = b'\210\002\001'
+  _globals['_CREATEDATASOURCEREQUEST']._serialized_start=130
+  _globals['_CREATEDATASOURCEREQUEST']._serialized_end=225
+  _globals['_CREATEDATASOURCERESPONSE']._serialized_start=227
+  _globals['_CREATEDATASOURCERESPONSE']._serialized_end=314
+  _globals['_UPDATEDATASOURCEREQUEST']._serialized_start=316
+  _globals['_UPDATEDATASOURCEREQUEST']._serialized_end=435
+  _globals['_UPDATEDATASOURCERESPONSE']._serialized_start=437
+  _globals['_UPDATEDATASOURCERESPONSE']._serialized_end=524
+  _globals['_DELETEDATASOURCEREQUEST']._serialized_start=526
+  _globals['_DELETEDATASOURCEREQUEST']._serialized_end=575
+  _globals['_DELETEDATASOURCERESPONSE']._serialized_start=577
+  _globals['_DELETEDATASOURCERESPONSE']._serialized_end=603
+  _globals['_GETDATASOURCEBYIDREQUEST']._serialized_start=605
+  _globals['_GETDATASOURCEBYIDREQUEST']._serialized_end=655
+  _globals['_GETDATASOURCEBYIDRESPONSE']._serialized_start=657
+  _globals['_GETDATASOURCEBYIDRESPONSE']._serialized_end=745
+  _globals['_GETDATASOURCEBYNAMEREQUEST']._serialized_start=747
+  _globals['_GETDATASOURCEBYNAMEREQUEST']._serialized_end=801
+  _globals['_GETDATASOURCEBYNAMERESPONSE']._serialized_start=803
+  _globals['_GETDATASOURCEBYNAMERESPONSE']._serialized_end=893
+  _globals['_LISTDATASOURCESREQUEST']._serialized_start=895
+  _globals['_LISTDATASOURCESREQUEST']._serialized_end=919
+  _globals['_LISTDATASOURCESRESPONSE']._serialized_start=921
+  _globals['_LISTDATASOURCESRESPONSE']._serialized_end=1008
+  _globals['_LISTDATASOURCESBYACCOUNTREQUEST']._serialized_start=1010
+  _globals['_LISTDATASOURCESBYACCOUNTREQUEST']._serialized_end=1043
+  _globals['_LISTDATASOURCESBYACCOUNTRESPONSE']._serialized_start=1045
+  _globals['_LISTDATASOURCESBYACCOUNTRESPONSE']._serialized_end=1141
+  _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_start=1143
+  _globals['_CREATEDATASOURCEUPLOADURLREQUEST']._serialized_end=1224
+  _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_start=1226
+  _globals['_CREATEDATASOURCEUPLOADURLRESPONSE']._serialized_end=1281
+  _globals['_DATASOURCESERVICE']._serialized_start=1284
+  _globals['_DATASOURCESERVICE']._serialized_end=2545
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/feature_store/sources/data_source_service_pb2.pyi

@@ -202,6 +202,31 @@ class ListDataSourcesResponse(google.protobuf.message.Message):
 
 global___ListDataSourcesResponse = ListDataSourcesResponse
 
+class ListDataSourcesByAccountRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___ListDataSourcesByAccountRequest = ListDataSourcesByAccountRequest
+
+class ListDataSourcesByAccountResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DATA_SOURCES_FIELD_NUMBER: builtins.int
+    @property
+    def data_sources(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.feature_store.sources.data_source_pb2.DataSource]:
+        """List of registered data sources"""
+    def __init__(
+        self,
+        *,
+        data_sources: collections.abc.Iterable[qwak.feature_store.sources.data_source_pb2.DataSource] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["data_sources", b"data_sources"]) -> None: ...
+
+global___ListDataSourcesByAccountResponse = ListDataSourcesByAccountResponse
+
 class CreateDataSourceUploadURLRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

_qwak_proto/qwak/feature_store/sources/data_source_service_pb2_grpc.py

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