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

_qwak_proto/qwak/feature_store/sources/batch_pb2.pyi

@@ -32,6 +32,7 @@ class BatchSource(google.protobuf.message.Message):
     ELASTICSEARCHSOURCE_FIELD_NUMBER: builtins.int
     CLICKHOUSESOURCE_FIELD_NUMBER: builtins.int
     FEATURESETOFFLINESOURCE_FIELD_NUMBER: builtins.int
+    CATALOGSOURCE_FIELD_NUMBER: builtins.int
     name: builtins.str
     """Source given name (by the user)"""
     description: builtins.str
@@ -60,6 +61,8 @@ class BatchSource(google.protobuf.message.Message):
     def clickhouseSource(self) -> global___ClickhouseSource: ...
     @property
     def featuresetOfflineSource(self) -> global___FeaturesetOfflineSource: ...
+    @property
+    def catalogSource(self) -> global___CatalogSource: ...
     def __init__(
         self,
         *,
@@ -77,10 +80,11 @@ class BatchSource(google.protobuf.message.Message):
         elasticsearchSource: global___ElasticsearchSource | None = ...,
         clickhouseSource: global___ClickhouseSource | None = ...,
         featuresetOfflineSource: global___FeaturesetOfflineSource | None = ...,
+        catalogSource: global___CatalogSource | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["athenaSource", b"athenaSource", "bigquerySource", b"bigquerySource", "clickhouseSource", b"clickhouseSource", "csvSource", b"csvSource", "elasticsearchSource", b"elasticsearchSource", "featuresetOfflineSource", b"featuresetOfflineSource", "jdbcSource", b"jdbcSource", "mongoSource", b"mongoSource", "parquetSource", b"parquetSource", "snowflakeSource", b"snowflakeSource", "type", b"type", "verticaSource", b"verticaSource"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["athenaSource", b"athenaSource", "bigquerySource", b"bigquerySource", "clickhouseSource", b"clickhouseSource", "csvSource", b"csvSource", "date_created_column", b"date_created_column", "description", b"description", "elasticsearchSource", b"elasticsearchSource", "featuresetOfflineSource", b"featuresetOfflineSource", "jdbcSource", b"jdbcSource", "mongoSource", b"mongoSource", "name", b"name", "parquetSource", b"parquetSource", "snowflakeSource", b"snowflakeSource", "type", b"type", "verticaSource", b"verticaSource"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["athenaSource", "mongoSource", "csvSource", "snowflakeSource", "parquetSource", "jdbcSource", "verticaSource", "bigquerySource", "elasticsearchSource", "clickhouseSource", "featuresetOfflineSource"] | None: ...
+    def HasField(self, field_name: typing_extensions.Literal["athenaSource", b"athenaSource", "bigquerySource", b"bigquerySource", "catalogSource", b"catalogSource", "clickhouseSource", b"clickhouseSource", "csvSource", b"csvSource", "elasticsearchSource", b"elasticsearchSource", "featuresetOfflineSource", b"featuresetOfflineSource", "jdbcSource", b"jdbcSource", "mongoSource", b"mongoSource", "parquetSource", b"parquetSource", "snowflakeSource", b"snowflakeSource", "type", b"type", "verticaSource", b"verticaSource"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["athenaSource", b"athenaSource", "bigquerySource", b"bigquerySource", "catalogSource", b"catalogSource", "clickhouseSource", b"clickhouseSource", "csvSource", b"csvSource", "date_created_column", b"date_created_column", "description", b"description", "elasticsearchSource", b"elasticsearchSource", "featuresetOfflineSource", b"featuresetOfflineSource", "jdbcSource", b"jdbcSource", "mongoSource", b"mongoSource", "name", b"name", "parquetSource", b"parquetSource", "snowflakeSource", b"snowflakeSource", "type", b"type", "verticaSource", b"verticaSource"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["athenaSource", "mongoSource", "csvSource", "snowflakeSource", "parquetSource", "jdbcSource", "verticaSource", "bigquerySource", "elasticsearchSource", "clickhouseSource", "featuresetOfflineSource", "catalogSource"] | None: ...
 
 global___BatchSource = BatchSource
 
@@ -617,7 +621,7 @@ class MongoSource(google.protobuf.message.Message):
     username_secret_name: builtins.str
     """Mongo username"""
     password_secret_name: builtins.str
-    """Mongo password kubernetes secret name"""
+    """Mongo password secret name"""
     database: builtins.str
     """Mongo data base name"""
     collection: builtins.str
@@ -649,6 +653,7 @@ class SnowflakeSource(google.protobuf.message.Message):
     HOST_FIELD_NUMBER: builtins.int
     USERNAME_SECRET_NAME_FIELD_NUMBER: builtins.int
     PASSWORD_SECRET_NAME_FIELD_NUMBER: builtins.int
+    PEM_PRIVATE_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
     DATABASE_FIELD_NUMBER: builtins.int
     SCHEMA_FIELD_NUMBER: builtins.int
     WAREHOUSE_FIELD_NUMBER: builtins.int
@@ -657,9 +662,11 @@ class SnowflakeSource(google.protobuf.message.Message):
     host: builtins.str
     """SF host address <account_identifier>.snowflakecomputing.com]"""
     username_secret_name: builtins.str
-    """SF username kubernetes secret name"""
+    """SF username secret name"""
     password_secret_name: builtins.str
-    """SF password kubernetes secret name"""
+    """SF password secret name"""
+    pem_private_key_secret_name: builtins.str
+    """SF pem formatted private key secret name"""
     database: builtins.str
     """SF database name"""
     schema: builtins.str
@@ -676,13 +683,16 @@ class SnowflakeSource(google.protobuf.message.Message):
         host: builtins.str = ...,
         username_secret_name: builtins.str = ...,
         password_secret_name: builtins.str = ...,
+        pem_private_key_secret_name: builtins.str = ...,
         database: builtins.str = ...,
         schema: builtins.str = ...,
         warehouse: builtins.str = ...,
         table: builtins.str = ...,
         query: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["database", b"database", "host", b"host", "password_secret_name", b"password_secret_name", "query", b"query", "schema", b"schema", "table", b"table", "username_secret_name", b"username_secret_name", "warehouse", b"warehouse"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["auth_secret", b"auth_secret", "password_secret_name", b"password_secret_name", "pem_private_key_secret_name", b"pem_private_key_secret_name"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auth_secret", b"auth_secret", "database", b"database", "host", b"host", "password_secret_name", b"password_secret_name", "pem_private_key_secret_name", b"pem_private_key_secret_name", "query", b"query", "schema", b"schema", "table", b"table", "username_secret_name", b"username_secret_name", "warehouse", b"warehouse"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["auth_secret", b"auth_secret"]) -> typing_extensions.Literal["password_secret_name", "pem_private_key_secret_name"] | None: ...
 
 global___SnowflakeSource = SnowflakeSource
 
@@ -751,6 +761,8 @@ class VerticaSource(google.protobuf.message.Message):
 global___VerticaSource = VerticaSource
 
 class ElasticsearchSource(google.protobuf.message.Message):
+    """Elasticsearch batch source"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     NODES_FIELD_NUMBER: builtins.int
@@ -793,6 +805,86 @@ class ElasticsearchSource(google.protobuf.message.Message):
 
 global___ElasticsearchSource = ElasticsearchSource
 
+class CatalogSource(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    URL_FIELD_NUMBER: builtins.int
+    TABLE_FIELD_NUMBER: builtins.int
+    QUERY_FIELD_NUMBER: builtins.int
+    UNITYCATALOGSOURCE_FIELD_NUMBER: builtins.int
+    url: builtins.str
+    """Connection url i.e https://<workspace-instance-name>.<domain-name>"""
+    table: builtins.str
+    """Catalog table including schema"""
+    query: builtins.str
+    """Catalog query"""
+    @property
+    def unityCatalogSource(self) -> global___UnityCatalogSource:
+        """Unity Catalog source"""
+    def __init__(
+        self,
+        *,
+        url: builtins.str = ...,
+        table: builtins.str = ...,
+        query: builtins.str = ...,
+        unityCatalogSource: global___UnityCatalogSource | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["query", b"query", "select_type", b"select_type", "table", b"table", "type", b"type", "unityCatalogSource", b"unityCatalogSource"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["query", b"query", "select_type", b"select_type", "table", b"table", "type", b"type", "unityCatalogSource", b"unityCatalogSource", "url", b"url"]) -> None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["select_type", b"select_type"]) -> typing_extensions.Literal["table", "query"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["unityCatalogSource"] | None: ...
+
+global___CatalogSource = CatalogSource
+
+class UnityCatalogSource(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CATALOG_FIELD_NUMBER: builtins.int
+    SCHEMA_FIELD_NUMBER: builtins.int
+    PERSONAL_ACCESS_TOKEN_SECRET_NAME_FIELD_NUMBER: builtins.int
+    OAUTH2_FIELD_NUMBER: builtins.int
+    catalog: builtins.str
+    """Unity Catalog catalog name"""
+    schema: builtins.str
+    """Unity Catalog schema name"""
+    personal_access_token_secret_name: builtins.str
+    @property
+    def oauth2(self) -> global___Oauth2: ...
+    def __init__(
+        self,
+        *,
+        catalog: builtins.str = ...,
+        schema: builtins.str = ...,
+        personal_access_token_secret_name: builtins.str = ...,
+        oauth2: global___Oauth2 | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "oauth2", b"oauth2", "personal_access_token_secret_name", b"personal_access_token_secret_name"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["auth_type", b"auth_type", "catalog", b"catalog", "oauth2", b"oauth2", "personal_access_token_secret_name", b"personal_access_token_secret_name", "schema", b"schema"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["auth_type", b"auth_type"]) -> typing_extensions.Literal["personal_access_token_secret_name", "oauth2"] | None: ...
+
+global___UnityCatalogSource = UnityCatalogSource
+
+class Oauth2(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CLIENT_ID_SECRET_NAME_FIELD_NUMBER: builtins.int
+    CLIENT_SECRET_SECRET_NAME_FIELD_NUMBER: builtins.int
+    client_id_secret_name: builtins.str
+    """client id"""
+    client_secret_secret_name: builtins.str
+    """client secret"""
+    def __init__(
+        self,
+        *,
+        client_id_secret_name: builtins.str = ...,
+        client_secret_secret_name: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["client_id_secret_name", b"client_id_secret_name", "client_secret_secret_name", b"client_secret_secret_name"]) -> None: ...
+
+global___Oauth2 = Oauth2
+
 class FeaturesetOfflineSource(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

_qwak_proto/qwak/feature_store/sources/data_source_attribute_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,20 +16,13 @@ from _qwak_proto.qwak.feature_store.v1.common.source_code import source_code_pb2
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n6qwak/feature_store/sources/data_source_attribute.proto\x12\x1aqwak.feature.store.sources\x1a:qwak/feature_store/v1/common/source_code/source_code.proto\"j\n\x14\x44\x61taSourceAttributes\x12R\n\x10source_code_spec\x18\x01 \x01(\x0b\x32\x38.qwak.feature.store.v1.common.source_code.SourceCodeSpecBX\n%com.qwak.ai.feature.store.sources.apiP\x01Z-qwak/featurestore/sources;featurestoresourcesb\x06proto3')
 
-
-
-_DATASOURCEATTRIBUTES = DESCRIPTOR.message_types_by_name['DataSourceAttributes']
-DataSourceAttributes = _reflection.GeneratedProtocolMessageType('DataSourceAttributes', (_message.Message,), {
-  'DESCRIPTOR' : _DATASOURCEATTRIBUTES,
-  '__module__' : 'qwak.feature_store.sources.data_source_attribute_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.feature.store.sources.DataSourceAttributes)
-  })
-_sym_db.RegisterMessage(DataSourceAttributes)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.feature_store.sources.data_source_attribute_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'
-  _DATASOURCEATTRIBUTES._serialized_start=146
-  _DATASOURCEATTRIBUTES._serialized_end=252
+  _globals['_DATASOURCEATTRIBUTES']._serialized_start=146
+  _globals['_DATASOURCEATTRIBUTES']._serialized_end=252
 # @@protoc_insertion_point(module_scope)

_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()
@@ -19,50 +18,11 @@ from _qwak_proto.qwak.feature_store.sources import data_source_attribute_pb2 as
 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\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\"\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\"\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')
-
-
-
-_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
@@ -71,14 +31,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _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'
-  _DATASOURCE._serialized_start=310
-  _DATASOURCE._serialized_end=537
-  _DATASOURCEDEFINITION._serialized_start=539
-  _DATASOURCEDEFINITION._serialized_end=655
-  _DATASOURCEMETADATA._serialized_start=658
-  _DATASOURCEMETADATA._serialized_end=826
-  _DATASOURCESPEC._serialized_start=829
-  _DATASOURCESPEC._serialized_end=1312
-  _FEATURESETBRIEF._serialized_start=1314
-  _FEATURESETBRIEF._serialized_end=1407
+  _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

@@ -77,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"""
@@ -87,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,
         *,
@@ -94,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
 

_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,157 +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\xd5\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\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')
+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'
   _DATASOURCESERVICE.methods_by_name['CreateDataSourceUploadURL']._options = None
   _DATASOURCESERVICE.methods_by_name['CreateDataSourceUploadURL']._serialized_options = b'\210\002\001'
-  _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=2260
+  _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