frogml 1.1.62__py3-none-any.whl → 1.2.19__py3-none-any.whl

frogml/_proto/qwak/feature_store/sources/batch_pb2.pyi

@@ -2,19 +2,15 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.message
-import sys
 import typing
 
-if sys.version_info >= (3, 8):
-    import typing as typing_extensions
-else:
-    import typing_extensions
-
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class BatchSource(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -32,6 +28,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 +57,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,13 +76,15 @@ 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.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.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.Literal["type", b"type"]) -> typing.Literal["athenaSource", "mongoSource", "csvSource", "snowflakeSource", "parquetSource", "jdbcSource", "verticaSource", "bigquerySource", "elasticsearchSource", "clickhouseSource", "featuresetOfflineSource", "catalogSource"] | None: ...
 
 global___BatchSource = BatchSource
 
+@typing.final
 class FileSystemConfiguration(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -115,12 +116,13 @@ class FileSystemConfiguration(google.protobuf.message.Message):
         gcs_user_credentials: global___GcsUserCredentials | None = ...,
         gcs_service_account_impersonation: global___GcsServiceAccountImpersonation | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["aws_s3_anonymous", b"aws_s3_anonymous", "aws_s3_assume_role_configuration", b"aws_s3_assume_role_configuration", "aws_s3_configuration", b"aws_s3_configuration", "gcs_service_account_impersonation", b"gcs_service_account_impersonation", "gcs_unauthenticated", b"gcs_unauthenticated", "gcs_user_credentials", b"gcs_user_credentials", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["aws_s3_anonymous", b"aws_s3_anonymous", "aws_s3_assume_role_configuration", b"aws_s3_assume_role_configuration", "aws_s3_configuration", b"aws_s3_configuration", "gcs_service_account_impersonation", b"gcs_service_account_impersonation", "gcs_unauthenticated", b"gcs_unauthenticated", "gcs_user_credentials", b"gcs_user_credentials", "type", b"type"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["aws_s3_configuration", "aws_s3_anonymous", "aws_s3_assume_role_configuration", "gcs_unauthenticated", "gcs_user_credentials", "gcs_service_account_impersonation"] | None: ...
+    def HasField(self, field_name: typing.Literal["aws_s3_anonymous", b"aws_s3_anonymous", "aws_s3_assume_role_configuration", b"aws_s3_assume_role_configuration", "aws_s3_configuration", b"aws_s3_configuration", "gcs_service_account_impersonation", b"gcs_service_account_impersonation", "gcs_unauthenticated", b"gcs_unauthenticated", "gcs_user_credentials", b"gcs_user_credentials", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["aws_s3_anonymous", b"aws_s3_anonymous", "aws_s3_assume_role_configuration", b"aws_s3_assume_role_configuration", "aws_s3_configuration", b"aws_s3_configuration", "gcs_service_account_impersonation", b"gcs_service_account_impersonation", "gcs_unauthenticated", b"gcs_unauthenticated", "gcs_user_credentials", b"gcs_user_credentials", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["type", b"type"]) -> typing.Literal["aws_s3_configuration", "aws_s3_anonymous", "aws_s3_assume_role_configuration", "gcs_unauthenticated", "gcs_user_credentials", "gcs_service_account_impersonation"] | None: ...
 
 global___FileSystemConfiguration = FileSystemConfiguration
 
+@typing.final
 class AwsS3AssumeRole(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -132,10 +134,11 @@ class AwsS3AssumeRole(google.protobuf.message.Message):
         *,
         role_arn: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["role_arn", b"role_arn"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["role_arn", b"role_arn"]) -> None: ...
 
 global___AwsS3AssumeRole = AwsS3AssumeRole
 
+@typing.final
 class AwsS3FileSystemConfiguration(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -159,10 +162,11 @@ class AwsS3FileSystemConfiguration(google.protobuf.message.Message):
         session_token_secret_name: builtins.str = ...,
         bucket: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["access_key_secret_name", b"access_key_secret_name", "bucket", b"bucket", "secret_key_secret_name", b"secret_key_secret_name", "session_token_secret_name", b"session_token_secret_name"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["access_key_secret_name", b"access_key_secret_name", "bucket", b"bucket", "secret_key_secret_name", b"secret_key_secret_name", "session_token_secret_name", b"session_token_secret_name"]) -> None: ...
 
 global___AwsS3FileSystemConfiguration = AwsS3FileSystemConfiguration
 
+@typing.final
 class AnonymousS3Configuration(google.protobuf.message.Message):
     """Anonymous S3 access"""
 
@@ -174,6 +178,7 @@ class AnonymousS3Configuration(google.protobuf.message.Message):
 
 global___AnonymousS3Configuration = AnonymousS3Configuration
 
+@typing.final
 class JdbcSource(google.protobuf.message.Message):
     """Jdbc Batch source"""
 
@@ -219,12 +224,13 @@ class JdbcSource(google.protobuf.message.Message):
         redshiftSource: global___RedshiftSource | None = ...,
         athenaSource: global___AthenaSourceV1 | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["athenaSource", b"athenaSource", "mysqlSource", b"mysqlSource", "postgresqlSource", b"postgresqlSource", "redshiftSource", b"redshiftSource", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["athenaSource", b"athenaSource", "db_table", b"db_table", "mysqlSource", b"mysqlSource", "password_secret_name", b"password_secret_name", "postgresqlSource", b"postgresqlSource", "query", b"query", "redshiftSource", b"redshiftSource", "type", b"type", "url", b"url", "username_secret_name", b"username_secret_name"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["mysqlSource", "postgresqlSource", "redshiftSource", "athenaSource"] | None: ...
+    def HasField(self, field_name: typing.Literal["athenaSource", b"athenaSource", "mysqlSource", b"mysqlSource", "postgresqlSource", b"postgresqlSource", "redshiftSource", b"redshiftSource", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["athenaSource", b"athenaSource", "db_table", b"db_table", "mysqlSource", b"mysqlSource", "password_secret_name", b"password_secret_name", "postgresqlSource", b"postgresqlSource", "query", b"query", "redshiftSource", b"redshiftSource", "type", b"type", "url", b"url", "username_secret_name", b"username_secret_name"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["type", b"type"]) -> typing.Literal["mysqlSource", "postgresqlSource", "redshiftSource", "athenaSource"] | None: ...
 
 global___JdbcSource = JdbcSource
 
+@typing.final
 class MysqlSource(google.protobuf.message.Message):
     """MySQL source"""
 
@@ -236,6 +242,7 @@ class MysqlSource(google.protobuf.message.Message):
 
 global___MysqlSource = MysqlSource
 
+@typing.final
 class PostgresqlSource(google.protobuf.message.Message):
     """PostgreSQL source"""
 
@@ -247,6 +254,7 @@ class PostgresqlSource(google.protobuf.message.Message):
 
 global___PostgresqlSource = PostgresqlSource
 
+@typing.final
 class ClickhouseSource(google.protobuf.message.Message):
     """ClickHouse source"""
 
@@ -276,12 +284,13 @@ class ClickhouseSource(google.protobuf.message.Message):
         table: builtins.str = ...,
         sql: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["query_type", b"query_type", "sql", b"sql", "table", b"table"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["password_secret_name", b"password_secret_name", "query_type", b"query_type", "sql", b"sql", "table", b"table", "url", b"url", "username_secret_name", b"username_secret_name"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["query_type", b"query_type"]) -> typing_extensions.Literal["table", "sql"] | None: ...
+    def HasField(self, field_name: typing.Literal["query_type", b"query_type", "sql", b"sql", "table", b"table"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["password_secret_name", b"password_secret_name", "query_type", b"query_type", "sql", b"sql", "table", b"table", "url", b"url", "username_secret_name", b"username_secret_name"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["query_type", b"query_type"]) -> typing.Literal["table", "sql"] | None: ...
 
 global___ClickhouseSource = ClickhouseSource
 
+@typing.final
 class RedshiftSource(google.protobuf.message.Message):
     """Redshift batch source"""
 
@@ -307,10 +316,11 @@ class RedshiftSource(google.protobuf.message.Message):
         access_key_id: builtins.str = ...,
         secret_access_key: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["access_key_id", b"access_key_id", "db_user", b"db_user", "iam_role_arn", b"iam_role_arn", "secret_access_key", b"secret_access_key"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["access_key_id", b"access_key_id", "db_user", b"db_user", "iam_role_arn", b"iam_role_arn", "secret_access_key", b"secret_access_key"]) -> None: ...
 
 global___RedshiftSource = RedshiftSource
 
+@typing.final
 class AthenaSourceV1(google.protobuf.message.Message):
     """Athena batch source"""
 
@@ -324,16 +334,17 @@ class AthenaSourceV1(google.protobuf.message.Message):
     WORKGROUP_FIELD_NUMBER: builtins.int
     aws_region: builtins.str
     """aws region"""
+    s3_output_location: builtins.str
+    """S3 Output Location for the query results"""
+    workgroup: builtins.str
     @property
     def aws_authentication(self) -> global___AwsAuthentication:
         """Authentication method"""
-    s3_output_location: builtins.str
-    """S3 Output Location for the query results"""
+
     @property
     def date_partition_columns(self) -> global___DatePartitionColumns: ...
     @property
     def time_fragmented_partition_columns(self) -> global___TimeFragmentedPartitionColumns: ...
-    workgroup: builtins.str
     def __init__(
         self,
         *,
@@ -344,15 +355,16 @@ class AthenaSourceV1(google.protobuf.message.Message):
         time_fragmented_partition_columns: global___TimeFragmentedPartitionColumns | None = ...,
         workgroup: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["aws_authentication", b"aws_authentication", "date_partition_columns", b"date_partition_columns", "optional_workgroup", b"optional_workgroup", "time_fragmented_partition_columns", b"time_fragmented_partition_columns", "time_partition_columns", b"time_partition_columns", "workgroup", b"workgroup"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["aws_authentication", b"aws_authentication", "aws_region", b"aws_region", "date_partition_columns", b"date_partition_columns", "optional_workgroup", b"optional_workgroup", "s3_output_location", b"s3_output_location", "time_fragmented_partition_columns", b"time_fragmented_partition_columns", "time_partition_columns", b"time_partition_columns", "workgroup", b"workgroup"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["aws_authentication", b"aws_authentication", "date_partition_columns", b"date_partition_columns", "optional_workgroup", b"optional_workgroup", "time_fragmented_partition_columns", b"time_fragmented_partition_columns", "time_partition_columns", b"time_partition_columns", "workgroup", b"workgroup"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["aws_authentication", b"aws_authentication", "aws_region", b"aws_region", "date_partition_columns", b"date_partition_columns", "optional_workgroup", b"optional_workgroup", "s3_output_location", b"s3_output_location", "time_fragmented_partition_columns", b"time_fragmented_partition_columns", "time_partition_columns", b"time_partition_columns", "workgroup", b"workgroup"]) -> None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_workgroup", b"optional_workgroup"]) -> typing_extensions.Literal["workgroup"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_workgroup", b"optional_workgroup"]) -> typing.Literal["workgroup"] | None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["time_partition_columns", b"time_partition_columns"]) -> typing_extensions.Literal["date_partition_columns", "time_fragmented_partition_columns"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["time_partition_columns", b"time_partition_columns"]) -> typing.Literal["date_partition_columns", "time_fragmented_partition_columns"] | None: ...
 
 global___AthenaSourceV1 = AthenaSourceV1
 
+@typing.final
 class DatePartitionColumns(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -368,10 +380,11 @@ class DatePartitionColumns(google.protobuf.message.Message):
         date_column_name: builtins.str = ...,
         date_format: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["date_column_name", b"date_column_name", "date_format", b"date_format"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["date_column_name", b"date_column_name", "date_format", b"date_format"]) -> None: ...
 
 global___DatePartitionColumns = DatePartitionColumns
 
+@typing.final
 class TimeFragmentedPartitionColumns(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -381,6 +394,7 @@ class TimeFragmentedPartitionColumns(google.protobuf.message.Message):
     @property
     def year_partition_column(self) -> global___YearFragmentColumn:
         """Year partition column"""
+
     @property
     def month_partition_column(self) -> global___MonthFragmentColumn: ...
     @property
@@ -392,15 +406,16 @@ class TimeFragmentedPartitionColumns(google.protobuf.message.Message):
         month_partition_column: global___MonthFragmentColumn | None = ...,
         day_partition_column: global___DayFragmentColumn | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["day_partition", b"day_partition", "day_partition_column", b"day_partition_column", "month_partition", b"month_partition", "month_partition_column", b"month_partition_column", "year_partition_column", b"year_partition_column"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["day_partition", b"day_partition", "day_partition_column", b"day_partition_column", "month_partition", b"month_partition", "month_partition_column", b"month_partition_column", "year_partition_column", b"year_partition_column"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["day_partition", b"day_partition", "day_partition_column", b"day_partition_column", "month_partition", b"month_partition", "month_partition_column", b"month_partition_column", "year_partition_column", b"year_partition_column"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["day_partition", b"day_partition", "day_partition_column", b"day_partition_column", "month_partition", b"month_partition", "month_partition_column", b"month_partition_column", "year_partition_column", b"year_partition_column"]) -> None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["day_partition", b"day_partition"]) -> typing_extensions.Literal["day_partition_column"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["day_partition", b"day_partition"]) -> typing.Literal["day_partition_column"] | None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["month_partition", b"month_partition"]) -> typing_extensions.Literal["month_partition_column"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["month_partition", b"month_partition"]) -> typing.Literal["month_partition_column"] | None: ...
 
 global___TimeFragmentedPartitionColumns = TimeFragmentedPartitionColumns
 
+@typing.final
 class YearFragmentColumn(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -411,18 +426,20 @@ class YearFragmentColumn(google.protobuf.message.Message):
     @property
     def numeric_column_representation(self) -> global___NumericColumnRepresentation:
         """Numeric year column representation supports: '2022' | 2022 for year 2022"""
+
     def __init__(
         self,
         *,
         column_name: builtins.str = ...,
         numeric_column_representation: global___NumericColumnRepresentation | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["column_name", b"column_name", "column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["column_representation", b"column_representation"]) -> typing_extensions.Literal["numeric_column_representation"] | None: ...
+    def HasField(self, field_name: typing.Literal["column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["column_name", b"column_name", "column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["column_representation", b"column_representation"]) -> typing.Literal["numeric_column_representation"] | None: ...
 
 global___YearFragmentColumn = YearFragmentColumn
 
+@typing.final
 class MonthFragmentColumn(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -434,9 +451,11 @@ class MonthFragmentColumn(google.protobuf.message.Message):
     @property
     def numeric_column_representation(self) -> global___NumericColumnRepresentation:
         """Numeric month column representation supports: '1' | '01' | 1   for january"""
+
     @property
     def textual_column_representation(self) -> global___TextualColumnRepresentation:
         """Textual month column representation supports: 'Jan' | 'January' and is not case sensative"""
+
     def __init__(
         self,
         *,
@@ -444,12 +463,13 @@ class MonthFragmentColumn(google.protobuf.message.Message):
         numeric_column_representation: global___NumericColumnRepresentation | None = ...,
         textual_column_representation: global___TextualColumnRepresentation | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation", "textual_column_representation", b"textual_column_representation"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["column_name", b"column_name", "column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation", "textual_column_representation", b"textual_column_representation"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["column_representation", b"column_representation"]) -> typing_extensions.Literal["numeric_column_representation", "textual_column_representation"] | None: ...
+    def HasField(self, field_name: typing.Literal["column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation", "textual_column_representation", b"textual_column_representation"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["column_name", b"column_name", "column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation", "textual_column_representation", b"textual_column_representation"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["column_representation", b"column_representation"]) -> typing.Literal["numeric_column_representation", "textual_column_representation"] | None: ...
 
 global___MonthFragmentColumn = MonthFragmentColumn
 
+@typing.final
 class DayFragmentColumn(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -460,18 +480,20 @@ class DayFragmentColumn(google.protobuf.message.Message):
     @property
     def numeric_column_representation(self) -> global___NumericColumnRepresentation:
         """Numeric day column representation supports: '1' | '01' | 1 for 1st day of the month"""
+
     def __init__(
         self,
         *,
         column_name: builtins.str = ...,
         numeric_column_representation: global___NumericColumnRepresentation | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["column_name", b"column_name", "column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["column_representation", b"column_representation"]) -> typing_extensions.Literal["numeric_column_representation"] | None: ...
+    def HasField(self, field_name: typing.Literal["column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["column_name", b"column_name", "column_representation", b"column_representation", "numeric_column_representation", b"numeric_column_representation"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["column_representation", b"column_representation"]) -> typing.Literal["numeric_column_representation"] | None: ...
 
 global___DayFragmentColumn = DayFragmentColumn
 
+@typing.final
 class NumericColumnRepresentation(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -481,6 +503,7 @@ class NumericColumnRepresentation(google.protobuf.message.Message):
 
 global___NumericColumnRepresentation = NumericColumnRepresentation
 
+@typing.final
 class TextualColumnRepresentation(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -490,6 +513,7 @@ class TextualColumnRepresentation(google.protobuf.message.Message):
 
 global___TextualColumnRepresentation = TextualColumnRepresentation
 
+@typing.final
 class AthenaSource(google.protobuf.message.Message):
     """Athena batch source"""
 
@@ -507,10 +531,11 @@ class AthenaSource(google.protobuf.message.Message):
         database: builtins.str = ...,
         table: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["database", b"database", "table", b"table"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["database", b"database", "table", b"table"]) -> None: ...
 
 global___AthenaSource = AthenaSource
 
+@typing.final
 class BigquerySource(google.protobuf.message.Message):
     """Bigquery batch source"""
 
@@ -565,10 +590,11 @@ class BigquerySource(google.protobuf.message.Message):
         materialization_project: builtins.str = ...,
         materialization_expiration_time_in_minutes: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["credentials_secret_name", b"credentials_secret_name", "dataset", b"dataset", "materialization_dataset", b"materialization_dataset", "materialization_expiration_time_in_minutes", b"materialization_expiration_time_in_minutes", "materialization_project", b"materialization_project", "parent_project", b"parent_project", "project", b"project", "sql", b"sql", "table", b"table", "views_enabled", b"views_enabled"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["credentials_secret_name", b"credentials_secret_name", "dataset", b"dataset", "materialization_dataset", b"materialization_dataset", "materialization_expiration_time_in_minutes", b"materialization_expiration_time_in_minutes", "materialization_project", b"materialization_project", "parent_project", b"parent_project", "project", b"project", "sql", b"sql", "table", b"table", "views_enabled", b"views_enabled"]) -> None: ...
 
 global___BigquerySource = BigquerySource
 
+@typing.final
 class CsvSource(google.protobuf.message.Message):
     """Csv batch source"""
 
@@ -587,6 +613,7 @@ class CsvSource(google.protobuf.message.Message):
     @property
     def filesystem_configuration(self) -> global___FileSystemConfiguration:
         """File system configuration"""
+
     def __init__(
         self,
         *,
@@ -595,11 +622,12 @@ class CsvSource(google.protobuf.message.Message):
         escape_character: builtins.str = ...,
         filesystem_configuration: global___FileSystemConfiguration | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["filesystem_configuration", b"filesystem_configuration"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["escape_character", b"escape_character", "filesystem_configuration", b"filesystem_configuration", "path", b"path", "quote_character", b"quote_character"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["filesystem_configuration", b"filesystem_configuration"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["escape_character", b"escape_character", "filesystem_configuration", b"filesystem_configuration", "path", b"path", "quote_character", b"quote_character"]) -> None: ...
 
 global___CsvSource = CsvSource
 
+@typing.final
 class MongoSource(google.protobuf.message.Message):
     """Mongo batch source"""
 
@@ -637,10 +665,11 @@ class MongoSource(google.protobuf.message.Message):
         connection_params: builtins.str = ...,
         protocol: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["collection", b"collection", "connection_params", b"connection_params", "database", b"database", "hosts", b"hosts", "password_secret_name", b"password_secret_name", "protocol", b"protocol", "username_secret_name", b"username_secret_name"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["collection", b"collection", "connection_params", b"connection_params", "database", b"database", "hosts", b"hosts", "password_secret_name", b"password_secret_name", "protocol", b"protocol", "username_secret_name", b"username_secret_name"]) -> None: ...
 
 global___MongoSource = MongoSource
 
+@typing.final
 class SnowflakeSource(google.protobuf.message.Message):
     """Snowflake batch source"""
 
@@ -686,12 +715,13 @@ class SnowflakeSource(google.protobuf.message.Message):
         table: builtins.str = ...,
         query: builtins.str = ...,
     ) -> 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: ...
+    def HasField(self, field_name: typing.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.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.Literal["auth_secret", b"auth_secret"]) -> typing.Literal["password_secret_name", "pem_private_key_secret_name"] | None: ...
 
 global___SnowflakeSource = SnowflakeSource
 
+@typing.final
 class ParquetSource(google.protobuf.message.Message):
     """Parquet batch source"""
 
@@ -704,17 +734,19 @@ class ParquetSource(google.protobuf.message.Message):
     @property
     def filesystem_configuration(self) -> global___FileSystemConfiguration:
         """File system configuration"""
+
     def __init__(
         self,
         *,
         path: builtins.str = ...,
         filesystem_configuration: global___FileSystemConfiguration | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["filesystem_configuration", b"filesystem_configuration"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["filesystem_configuration", b"filesystem_configuration", "path", b"path"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["filesystem_configuration", b"filesystem_configuration"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["filesystem_configuration", b"filesystem_configuration", "path", b"path"]) -> None: ...
 
 global___ParquetSource = ParquetSource
 
+@typing.final
 class VerticaSource(google.protobuf.message.Message):
     """Vertica batch source"""
 
@@ -752,11 +784,14 @@ class VerticaSource(google.protobuf.message.Message):
         username_secret_name: builtins.str = ...,
         password_secret_name: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["database", b"database", "host", b"host", "password_secret_name", b"password_secret_name", "port", b"port", "schema", b"schema", "table", b"table", "username_secret_name", b"username_secret_name"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["database", b"database", "host", b"host", "password_secret_name", b"password_secret_name", "port", b"port", "schema", b"schema", "table", b"table", "username_secret_name", b"username_secret_name"]) -> None: ...
 
 global___VerticaSource = VerticaSource
 
+@typing.final
 class ElasticsearchSource(google.protobuf.message.Message):
+    """Elasticsearch batch source"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     NODES_FIELD_NUMBER: builtins.int
@@ -795,10 +830,95 @@ class ElasticsearchSource(google.protobuf.message.Message):
         username_secret_name: builtins.str = ...,
         password_secret_name: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["exclude_fields", b"exclude_fields", "nodes", b"nodes", "parse_dates", b"parse_dates", "password_secret_name", b"password_secret_name", "port", b"port", "query", b"query", "resource", b"resource", "username_secret_name", b"username_secret_name"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["exclude_fields", b"exclude_fields", "nodes", b"nodes", "parse_dates", b"parse_dates", "password_secret_name", b"password_secret_name", "port", b"port", "query", b"query", "resource", b"resource", "username_secret_name", b"username_secret_name"]) -> None: ...
 
 global___ElasticsearchSource = ElasticsearchSource
 
+@typing.final
+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.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.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.Literal["select_type", b"select_type"]) -> typing.Literal["table", "query"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["type", b"type"]) -> typing.Literal["unityCatalogSource"] | None: ...
+
+global___CatalogSource = CatalogSource
+
+@typing.final
+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.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.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.Literal["auth_type", b"auth_type"]) -> typing.Literal["personal_access_token_secret_name", "oauth2"] | None: ...
+
+global___UnityCatalogSource = UnityCatalogSource
+
+@typing.final
+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.Literal["client_id_secret_name", b"client_id_secret_name", "client_secret_secret_name", b"client_secret_secret_name"]) -> None: ...
+
+global___Oauth2 = Oauth2
+
+@typing.final
 class FeaturesetOfflineSource(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -810,10 +930,11 @@ class FeaturesetOfflineSource(google.protobuf.message.Message):
         *,
         alias: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["alias", b"alias"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["alias", b"alias"]) -> None: ...
 
 global___FeaturesetOfflineSource = FeaturesetOfflineSource
 
+@typing.final
 class AwsAuthentication(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -829,12 +950,13 @@ class AwsAuthentication(google.protobuf.message.Message):
         aws_assume_role_authentication: global___AwsAssumeRoleAuthentication | None = ...,
         aws_static_credentials_authentication: global___AwsCredentialsAuthentication | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["aws_assume_role_authentication", b"aws_assume_role_authentication", "aws_static_credentials_authentication", b"aws_static_credentials_authentication", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["aws_assume_role_authentication", b"aws_assume_role_authentication", "aws_static_credentials_authentication", b"aws_static_credentials_authentication", "type", b"type"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["aws_assume_role_authentication", "aws_static_credentials_authentication"] | None: ...
+    def HasField(self, field_name: typing.Literal["aws_assume_role_authentication", b"aws_assume_role_authentication", "aws_static_credentials_authentication", b"aws_static_credentials_authentication", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["aws_assume_role_authentication", b"aws_assume_role_authentication", "aws_static_credentials_authentication", b"aws_static_credentials_authentication", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["type", b"type"]) -> typing.Literal["aws_assume_role_authentication", "aws_static_credentials_authentication"] | None: ...
 
 global___AwsAuthentication = AwsAuthentication
 
+@typing.final
 class AwsAssumeRoleAuthentication(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -846,10 +968,11 @@ class AwsAssumeRoleAuthentication(google.protobuf.message.Message):
         *,
         role_arn: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["role_arn", b"role_arn"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["role_arn", b"role_arn"]) -> None: ...
 
 global___AwsAssumeRoleAuthentication = AwsAssumeRoleAuthentication
 
+@typing.final
 class AwsCredentialsAuthentication(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -865,10 +988,11 @@ class AwsCredentialsAuthentication(google.protobuf.message.Message):
         access_key_secret_name: builtins.str = ...,
         secret_key_secret_name: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["access_key_secret_name", b"access_key_secret_name", "secret_key_secret_name", b"secret_key_secret_name"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["access_key_secret_name", b"access_key_secret_name", "secret_key_secret_name", b"secret_key_secret_name"]) -> None: ...
 
 global___AwsCredentialsAuthentication = AwsCredentialsAuthentication
 
+@typing.final
 class GcsUnauthenticated(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -878,6 +1002,7 @@ class GcsUnauthenticated(google.protobuf.message.Message):
 
 global___GcsUnauthenticated = GcsUnauthenticated
 
+@typing.final
 class GcsUserCredentials(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -897,10 +1022,11 @@ class GcsUserCredentials(google.protobuf.message.Message):
         client_secret_secret_name: builtins.str = ...,
         refresh_token_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", "refresh_token_secret_name", b"refresh_token_secret_name"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["client_id_secret_name", b"client_id_secret_name", "client_secret_secret_name", b"client_secret_secret_name", "refresh_token_secret_name", b"refresh_token_secret_name"]) -> None: ...
 
 global___GcsUserCredentials = GcsUserCredentials
 
+@typing.final
 class GcsServiceAccountImpersonation(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -912,6 +1038,6 @@ class GcsServiceAccountImpersonation(google.protobuf.message.Message):
         *,
         service_account_user: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["service_account_user", b"service_account_user"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["service_account_user", b"service_account_user"]) -> None: ...
 
 global___GcsServiceAccountImpersonation = GcsServiceAccountImpersonation