chalkpy 2.89.22__py3-none-any.whl → 2.95.3__py3-none-any.whl

chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi

@@ -1,7 +1,14 @@
 from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
-from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
@@ -23,25 +30,206 @@ class KubernetesScaledObjectTargetRef(_message.Message):
         env_source_container_name: _Optional[str] = ...,
     ) -> None: ...
 
+class KubernetesGroupVersionKindResource(_message.Message):
+    __slots__ = ("group", "version", "kind", "resource")
+    GROUP_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_FIELD_NUMBER: _ClassVar[int]
+    group: str
+    version: str
+    kind: str
+    resource: str
+    def __init__(
+        self,
+        group: _Optional[str] = ...,
+        version: _Optional[str] = ...,
+        kind: _Optional[str] = ...,
+        resource: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectCondition(_message.Message):
+    __slots__ = ("type", "status", "reason", "message")
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    REASON_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    type: str
+    status: str
+    reason: str
+    message: str
+    def __init__(
+        self,
+        type: _Optional[str] = ...,
+        status: _Optional[str] = ...,
+        reason: _Optional[str] = ...,
+        message: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectHealthStatus(_message.Message):
+    __slots__ = ("number_of_failures", "status")
+    NUMBER_OF_FAILURES_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    number_of_failures: int
+    status: str
+    def __init__(self, number_of_failures: _Optional[int] = ..., status: _Optional[str] = ...) -> None: ...
+
+class KubernetesScaledObjectAuthenticationRef(_message.Message):
+    __slots__ = ("name", "kind")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    kind: str
+    def __init__(self, name: _Optional[str] = ..., kind: _Optional[str] = ...) -> None: ...
+
+class KubernetesScaledObjectTrigger(_message.Message):
+    __slots__ = ("type", "name", "use_cached_metrics", "metadata", "authentication_ref", "metric_type")
+    class MetadataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    USE_CACHED_METRICS_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    AUTHENTICATION_REF_FIELD_NUMBER: _ClassVar[int]
+    METRIC_TYPE_FIELD_NUMBER: _ClassVar[int]
+    type: str
+    name: str
+    use_cached_metrics: bool
+    metadata: _containers.ScalarMap[str, str]
+    authentication_ref: KubernetesScaledObjectAuthenticationRef
+    metric_type: str
+    def __init__(
+        self,
+        type: _Optional[str] = ...,
+        name: _Optional[str] = ...,
+        use_cached_metrics: bool = ...,
+        metadata: _Optional[_Mapping[str, str]] = ...,
+        authentication_ref: _Optional[_Union[KubernetesScaledObjectAuthenticationRef, _Mapping]] = ...,
+        metric_type: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectFallback(_message.Message):
+    __slots__ = ("failure_threshold", "replicas")
+    FAILURE_THRESHOLD_FIELD_NUMBER: _ClassVar[int]
+    REPLICAS_FIELD_NUMBER: _ClassVar[int]
+    failure_threshold: int
+    replicas: int
+    def __init__(self, failure_threshold: _Optional[int] = ..., replicas: _Optional[int] = ...) -> None: ...
+
+class KubernetesScaledObjectScalingModifiers(_message.Message):
+    __slots__ = ("formula", "target", "activation_target", "metric_type")
+    FORMULA_FIELD_NUMBER: _ClassVar[int]
+    TARGET_FIELD_NUMBER: _ClassVar[int]
+    ACTIVATION_TARGET_FIELD_NUMBER: _ClassVar[int]
+    METRIC_TYPE_FIELD_NUMBER: _ClassVar[int]
+    formula: str
+    target: str
+    activation_target: str
+    metric_type: str
+    def __init__(
+        self,
+        formula: _Optional[str] = ...,
+        target: _Optional[str] = ...,
+        activation_target: _Optional[str] = ...,
+        metric_type: _Optional[str] = ...,
+    ) -> None: ...
+
+class KubernetesScaledObjectHPAConfig(_message.Message):
+    __slots__ = ("name",)
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    def __init__(self, name: _Optional[str] = ...) -> None: ...
+
+class KubernetesScaledObjectAdvancedConfig(_message.Message):
+    __slots__ = ("horizontal_pod_autoscaler_config", "restore_to_original_replica_count", "scaling_modifiers")
+    HORIZONTAL_POD_AUTOSCALER_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    RESTORE_TO_ORIGINAL_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
+    SCALING_MODIFIERS_FIELD_NUMBER: _ClassVar[int]
+    horizontal_pod_autoscaler_config: KubernetesScaledObjectHPAConfig
+    restore_to_original_replica_count: bool
+    scaling_modifiers: KubernetesScaledObjectScalingModifiers
+    def __init__(
+        self,
+        horizontal_pod_autoscaler_config: _Optional[_Union[KubernetesScaledObjectHPAConfig, _Mapping]] = ...,
+        restore_to_original_replica_count: bool = ...,
+        scaling_modifiers: _Optional[_Union[KubernetesScaledObjectScalingModifiers, _Mapping]] = ...,
+    ) -> None: ...
+
 class KubernetesScaledObjectStatus(_message.Message):
-    __slots__ = ("scale_target_kind", "original_replica_count", "last_active_time", "paused_replica_count", "hpa_name")
+    __slots__ = (
+        "scale_target_kind",
+        "scale_target_gvkr",
+        "original_replica_count",
+        "last_active_time",
+        "external_metric_names",
+        "resource_metric_names",
+        "composite_scaler_name",
+        "conditions",
+        "health",
+        "paused_replica_count",
+        "hpa_name",
+        "triggers_types",
+        "authentications_types",
+    )
+    class HealthEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: KubernetesScaledObjectHealthStatus
+        def __init__(
+            self,
+            key: _Optional[str] = ...,
+            value: _Optional[_Union[KubernetesScaledObjectHealthStatus, _Mapping]] = ...,
+        ) -> None: ...
+
     SCALE_TARGET_KIND_FIELD_NUMBER: _ClassVar[int]
+    SCALE_TARGET_GVKR_FIELD_NUMBER: _ClassVar[int]
     ORIGINAL_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     LAST_ACTIVE_TIME_FIELD_NUMBER: _ClassVar[int]
+    EXTERNAL_METRIC_NAMES_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_METRIC_NAMES_FIELD_NUMBER: _ClassVar[int]
+    COMPOSITE_SCALER_NAME_FIELD_NUMBER: _ClassVar[int]
+    CONDITIONS_FIELD_NUMBER: _ClassVar[int]
+    HEALTH_FIELD_NUMBER: _ClassVar[int]
     PAUSED_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     HPA_NAME_FIELD_NUMBER: _ClassVar[int]
+    TRIGGERS_TYPES_FIELD_NUMBER: _ClassVar[int]
+    AUTHENTICATIONS_TYPES_FIELD_NUMBER: _ClassVar[int]
     scale_target_kind: str
+    scale_target_gvkr: KubernetesGroupVersionKindResource
     original_replica_count: int
     last_active_time: _timestamp_pb2.Timestamp
+    external_metric_names: _containers.RepeatedScalarFieldContainer[str]
+    resource_metric_names: _containers.RepeatedScalarFieldContainer[str]
+    composite_scaler_name: str
+    conditions: _containers.RepeatedCompositeFieldContainer[KubernetesScaledObjectCondition]
+    health: _containers.MessageMap[str, KubernetesScaledObjectHealthStatus]
     paused_replica_count: int
     hpa_name: str
+    triggers_types: str
+    authentications_types: str
     def __init__(
         self,
         scale_target_kind: _Optional[str] = ...,
+        scale_target_gvkr: _Optional[_Union[KubernetesGroupVersionKindResource, _Mapping]] = ...,
         original_replica_count: _Optional[int] = ...,
         last_active_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        external_metric_names: _Optional[_Iterable[str]] = ...,
+        resource_metric_names: _Optional[_Iterable[str]] = ...,
+        composite_scaler_name: _Optional[str] = ...,
+        conditions: _Optional[_Iterable[_Union[KubernetesScaledObjectCondition, _Mapping]]] = ...,
+        health: _Optional[_Mapping[str, KubernetesScaledObjectHealthStatus]] = ...,
         paused_replica_count: _Optional[int] = ...,
         hpa_name: _Optional[str] = ...,
+        triggers_types: _Optional[str] = ...,
+        authentications_types: _Optional[str] = ...,
     ) -> None: ...
 
 class KubernetesScaledObjectSpec(_message.Message):
@@ -52,6 +240,10 @@ class KubernetesScaledObjectSpec(_message.Message):
         "cooldown_period",
         "idle_replica_count",
         "max_replica_count",
+        "advanced",
+        "triggers",
+        "fallback",
+        "min_replica_count",
     )
     SCALE_TARGET_REF_FIELD_NUMBER: _ClassVar[int]
     POLLING_INTERVAL_FIELD_NUMBER: _ClassVar[int]
@@ -59,12 +251,20 @@ class KubernetesScaledObjectSpec(_message.Message):
     COOLDOWN_PERIOD_FIELD_NUMBER: _ClassVar[int]
     IDLE_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     MAX_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
+    ADVANCED_FIELD_NUMBER: _ClassVar[int]
+    TRIGGERS_FIELD_NUMBER: _ClassVar[int]
+    FALLBACK_FIELD_NUMBER: _ClassVar[int]
+    MIN_REPLICA_COUNT_FIELD_NUMBER: _ClassVar[int]
     scale_target_ref: KubernetesScaledObjectTargetRef
     polling_interval: int
     initial_cooldown_period: int
     cooldown_period: int
     idle_replica_count: int
     max_replica_count: int
+    advanced: KubernetesScaledObjectAdvancedConfig
+    triggers: _containers.RepeatedCompositeFieldContainer[KubernetesScaledObjectTrigger]
+    fallback: KubernetesScaledObjectFallback
+    min_replica_count: int
     def __init__(
         self,
         scale_target_ref: _Optional[_Union[KubernetesScaledObjectTargetRef, _Mapping]] = ...,
@@ -73,6 +273,10 @@ class KubernetesScaledObjectSpec(_message.Message):
         cooldown_period: _Optional[int] = ...,
         idle_replica_count: _Optional[int] = ...,
         max_replica_count: _Optional[int] = ...,
+        advanced: _Optional[_Union[KubernetesScaledObjectAdvancedConfig, _Mapping]] = ...,
+        triggers: _Optional[_Iterable[_Union[KubernetesScaledObjectTrigger, _Mapping]]] = ...,
+        fallback: _Optional[_Union[KubernetesScaledObjectFallback, _Mapping]] = ...,
+        min_replica_count: _Optional[int] = ...,
     ) -> None: ...
 
 class KubernetesScaledObjectData(_message.Message):
@@ -86,3 +290,49 @@ class KubernetesScaledObjectData(_message.Message):
         spec: _Optional[_Union[KubernetesScaledObjectSpec, _Mapping]] = ...,
         status: _Optional[_Union[KubernetesScaledObjectStatus, _Mapping]] = ...,
     ) -> None: ...
+
+class KubernetesScaledObject(_message.Message):
+    __slots__ = ("name", "namespace", "labels", "annotations", "creation_timestamp", "spec", "status", "cluster_name")
+    class LabelsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+
+    class AnnotationsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    NAMESPACE_FIELD_NUMBER: _ClassVar[int]
+    LABELS_FIELD_NUMBER: _ClassVar[int]
+    ANNOTATIONS_FIELD_NUMBER: _ClassVar[int]
+    CREATION_TIMESTAMP_FIELD_NUMBER: _ClassVar[int]
+    SPEC_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    CLUSTER_NAME_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    namespace: str
+    labels: _containers.ScalarMap[str, str]
+    annotations: _containers.ScalarMap[str, str]
+    creation_timestamp: int
+    spec: KubernetesScaledObjectSpec
+    status: KubernetesScaledObjectStatus
+    cluster_name: str
+    def __init__(
+        self,
+        name: _Optional[str] = ...,
+        namespace: _Optional[str] = ...,
+        labels: _Optional[_Mapping[str, str]] = ...,
+        annotations: _Optional[_Mapping[str, str]] = ...,
+        creation_timestamp: _Optional[int] = ...,
+        spec: _Optional[_Union[KubernetesScaledObjectSpec, _Mapping]] = ...,
+        status: _Optional[_Union[KubernetesScaledObjectStatus, _Mapping]] = ...,
+        cluster_name: _Optional[str] = ...,
+    ) -> None: ...

chalk/_gen/chalk/protosql/v1/sql_service_pb2.py

@@ -15,10 +15,11 @@ _sym_db = _symbol_database.Default()
 
 from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
 from chalk._gen.chalk.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v1_dot_chalk__error__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n#chalk/protosql/v1/sql_service.proto\x12\x11\x63halk.protosql.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto".\n\x16\x45xecuteSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query"\x90\x01\n\x17\x45xecuteSqlQueryResponse\x12\x19\n\x08query_id\x18\x01 \x01(\tR\x07queryId\x12\x1a\n\x07parquet\x18\x02 \x01(\x0cH\x00R\x07parquet\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\t\n\x07payload"+\n\x13PlanSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query"n\n\x14PlanSqlQueryResponse\x12!\n\x0clogical_plan\x18\x01 \x01(\tR\x0blogicalPlan\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x16\n\x14GetDbCatalogsRequest"q\n\x15GetDbCatalogsResponse\x12#\n\rcatalog_names\x18\x01 \x03(\tR\x0c\x63\x61talogNames\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\xd0\x01\n\x13GetDbSchemasRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_pattern"W\n\x0c\x44\x62SchemaInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName"\x86\x01\n\x14GetDbSchemasResponse\x12\x39\n\x07schemas\x18\x01 \x03(\x0b\x32\x1f.chalk.protosql.v1.DbSchemaInfoR\x07schemas\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x9f\x02\n\x10GetTablesRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12>\n\x19table_name_filter_pattern\x18\x03 \x01(\tH\x02R\x16tableNameFilterPattern\x88\x01\x01\x12\'\n\x0finclude_schemas\x18\n \x01(\x08R\x0eincludeSchemasB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_patternB\x1c\n\x1a_table_name_filter_pattern"\xbd\x01\n\tTableInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName\x12\x1d\n\ntable_name\x18\x03 \x01(\tR\ttableName\x12\x31\n\x12table_arrow_schema\x18\x04 \x01(\x0cH\x00R\x10tableArrowSchema\x88\x01\x01\x42\x15\n\x13_table_arrow_schema"~\n\x11GetTablesResponse\x12\x34\n\x06tables\x18\x01 \x03(\x0b\x32\x1c.chalk.protosql.v1.TableInfoR\x06tables\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors2\x8d\x04\n\nSqlService\x12m\n\x0f\x45xecuteSqlQuery\x12).chalk.protosql.v1.ExecuteSqlQueryRequest\x1a*.chalk.protosql.v1.ExecuteSqlQueryResponse"\x03\x80}\x03\x12\x64\n\x0cPlanSqlQuery\x12&.chalk.protosql.v1.PlanSqlQueryRequest\x1a\'.chalk.protosql.v1.PlanSqlQueryResponse"\x03\x80}\x03\x12g\n\rGetDbCatalogs\x12\'.chalk.protosql.v1.GetDbCatalogsRequest\x1a(.chalk.protosql.v1.GetDbCatalogsResponse"\x03\x80}\x0b\x12\x64\n\x0cGetDbSchemas\x12&.chalk.protosql.v1.GetDbSchemasRequest\x1a\'.chalk.protosql.v1.GetDbSchemasResponse"\x03\x80}\x0b\x12[\n\tGetTables\x12#.chalk.protosql.v1.GetTablesRequest\x1a$.chalk.protosql.v1.GetTablesResponse"\x03\x80}\x0b\x42\x8e\x01\n\x15\x63om.chalk.protosql.v1B\x0fSqlServiceProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x11\x43halk.Protosql.V1\xca\x02\x11\x43halk\\Protosql\\V1\xe2\x02\x1d\x43halk\\Protosql\\V1\\GPBMetadata\xea\x02\x13\x43halk::Protosql::V1b\x06proto3'
+    b'\n#chalk/protosql/v1/sql_service.proto\x12\x11\x63halk.protosql.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xd2\x01\n\x0cSqlQueryInfo\x12!\n\x0coperation_id\x18\x01 \x01(\tR\x0boperationId\x12>\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tcreatedAt\x88\x01\x01\x12@\n\x0b\x66inished_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\nfinishedAt\x88\x01\x01\x42\r\n\x0b_created_atB\x0e\n\x0c_finished_at"#\n!ExecuteSqlSyncQueryRequestOptions"|\n"ExecuteSqlAsyncQueryRequestOptions\x12V\n\x0e\x65xecution_mode\x18\x01 \x01(\x0e\x32/.chalk.protosql.v1.ExecuteSqlAsyncExecutionModeR\rexecutionMode"?\n#ExecuteSqlResultPersistenceSettings\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled"\x9c\x03\n\x16\x45xecuteSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query\x12*\n\x0e\x63orrelation_id\x18\x02 \x01(\tH\x01R\rcorrelationId\x88\x01\x01\x12i\n\x14persistence_settings\x18\x03 \x01(\x0b\x32\x36.chalk.protosql.v1.ExecuteSqlResultPersistenceSettingsR\x13persistenceSettings\x12Y\n\x0csync_options\x18\n \x01(\x0b\x32\x34.chalk.protosql.v1.ExecuteSqlSyncQueryRequestOptionsH\x00R\x0bsyncOptions\x12\\\n\rasync_options\x18\x0b \x01(\x0b\x32\x35.chalk.protosql.v1.ExecuteSqlAsyncQueryRequestOptionsH\x00R\x0c\x61syncOptionsB\t\n\x07optionsB\x11\n\x0f_correlation_id"&\n\x10SignedOutputUris\x12\x12\n\x04uris\x18\x01 \x03(\tR\x04uris"\xb1\x01\n"ExecuteSqlSyncQueryResponsePayload\x12+\n\x10parquet_response\x18\x01 \x01(\x0cH\x00R\x0fparquetResponse\x12S\n\x12signed_output_uris\x18\n \x01(\x0b\x32#.chalk.protosql.v1.SignedOutputUrisH\x00R\x10signedOutputUrisB\t\n\x07payload"\xa0\x01\n#ExecuteSqlAsyncQueryResponsePayload\x12!\n\x0coperation_id\x18\x01 \x01(\tR\x0boperationId\x12V\n\x0e\x65xecution_mode\x18\x02 \x01(\x0e\x32/.chalk.protosql.v1.ExecuteSqlAsyncExecutionModeR\rexecutionMode"\xcb\x02\n\x17\x45xecuteSqlQueryResponse\x12\x19\n\x08query_id\x18\x01 \x01(\tR\x07queryId\x12\x1a\n\x07parquet\x18\x02 \x01(\x0cH\x00R\x07parquet\x12Z\n\x0csync_payload\x18\n \x01(\x0b\x32\x35.chalk.protosql.v1.ExecuteSqlSyncQueryResponsePayloadH\x00R\x0bsyncPayload\x12]\n\rasync_payload\x18\x0b \x01(\x0b\x32\x36.chalk.protosql.v1.ExecuteSqlAsyncQueryResponsePayloadH\x00R\x0c\x61syncPayload\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\t\n\x07payload"+\n\x13PlanSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query"n\n\x14PlanSqlQueryResponse\x12!\n\x0clogical_plan\x18\x01 \x01(\tR\x0blogicalPlan\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x16\n\x14GetDbCatalogsRequest"q\n\x15GetDbCatalogsResponse\x12#\n\rcatalog_names\x18\x01 \x03(\tR\x0c\x63\x61talogNames\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\xd0\x01\n\x13GetDbSchemasRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_pattern"W\n\x0c\x44\x62SchemaInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName"\x86\x01\n\x14GetDbSchemasResponse\x12\x39\n\x07schemas\x18\x01 \x03(\x0b\x32\x1f.chalk.protosql.v1.DbSchemaInfoR\x07schemas\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x9f\x02\n\x10GetTablesRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12>\n\x19table_name_filter_pattern\x18\x03 \x01(\tH\x02R\x16tableNameFilterPattern\x88\x01\x01\x12\'\n\x0finclude_schemas\x18\n \x01(\x08R\x0eincludeSchemasB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_patternB\x1c\n\x1a_table_name_filter_pattern"\xbd\x01\n\tTableInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName\x12\x1d\n\ntable_name\x18\x03 \x01(\tR\ttableName\x12\x31\n\x12table_arrow_schema\x18\x04 \x01(\x0cH\x00R\x10tableArrowSchema\x88\x01\x01\x42\x15\n\x13_table_arrow_schema"~\n\x11GetTablesResponse\x12\x34\n\x06tables\x18\x01 \x03(\x0b\x32\x1c.chalk.protosql.v1.TableInfoR\x06tables\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x16\n\x14SqlQueryProgressInfo"I\n\x12SqlQueryFailedInfo\x12\x33\n\x06\x65rrors\x18\x01 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"8\n\x13PollSqlQueryRequest\x12!\n\x0coperation_id\x18\x01 \x01(\tR\x0boperationId"\xb2\x02\n\x14PollSqlQueryResponse\x12\x33\n\x04info\x18\x01 \x01(\x0b\x32\x1f.chalk.protosql.v1.SqlQueryInfoR\x04info\x12\x45\n\x08progress\x18\n \x01(\x0b\x32\'.chalk.protosql.v1.SqlQueryProgressInfoH\x00R\x08progress\x12S\n\x08response\x18\x0b \x01(\x0b\x32\x35.chalk.protosql.v1.ExecuteSqlSyncQueryResponsePayloadH\x00R\x08response\x12?\n\x06\x66\x61iled\x18\x0c \x01(\x0b\x32%.chalk.protosql.v1.SqlQueryFailedInfoH\x00R\x06\x66\x61iledB\x08\n\x06result*\xad\x01\n\x1c\x45xecuteSqlAsyncExecutionMode\x12\x30\n,EXECUTE_SQL_ASYNC_EXECUTION_MODE_UNSPECIFIED\x10\x00\x12/\n+EXECUTE_SQL_ASYNC_EXECUTION_MODE_IN_PROCESS\x10\x01\x12*\n&EXECUTE_SQL_ASYNC_EXECUTION_MODE_ASYNC\x10\x02\x32\xf3\x04\n\nSqlService\x12m\n\x0f\x45xecuteSqlQuery\x12).chalk.protosql.v1.ExecuteSqlQueryRequest\x1a*.chalk.protosql.v1.ExecuteSqlQueryResponse"\x03\x80}\x03\x12\x64\n\x0cPlanSqlQuery\x12&.chalk.protosql.v1.PlanSqlQueryRequest\x1a\'.chalk.protosql.v1.PlanSqlQueryResponse"\x03\x80}\x03\x12\x64\n\x0cPollSqlQuery\x12&.chalk.protosql.v1.PollSqlQueryRequest\x1a\'.chalk.protosql.v1.PollSqlQueryResponse"\x03\x80}\x03\x12g\n\rGetDbCatalogs\x12\'.chalk.protosql.v1.GetDbCatalogsRequest\x1a(.chalk.protosql.v1.GetDbCatalogsResponse"\x03\x80}\x0b\x12\x64\n\x0cGetDbSchemas\x12&.chalk.protosql.v1.GetDbSchemasRequest\x1a\'.chalk.protosql.v1.GetDbSchemasResponse"\x03\x80}\x0b\x12[\n\tGetTables\x12#.chalk.protosql.v1.GetTablesRequest\x1a$.chalk.protosql.v1.GetTablesResponse"\x03\x80}\x0b\x42\x8e\x01\n\x15\x63om.chalk.protosql.v1B\x0fSqlServiceProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x11\x43halk.Protosql.V1\xca\x02\x11\x43halk\\Protosql\\V1\xe2\x02\x1d\x43halk\\Protosql\\V1\\GPBMetadata\xea\x02\x13\x43halk::Protosql::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -33,36 +34,62 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SQLSERVICE"].methods_by_name["ExecuteSqlQuery"]._serialized_options = b"\200}\003"
     _globals["_SQLSERVICE"].methods_by_name["PlanSqlQuery"]._options = None
     _globals["_SQLSERVICE"].methods_by_name["PlanSqlQuery"]._serialized_options = b"\200}\003"
+    _globals["_SQLSERVICE"].methods_by_name["PollSqlQuery"]._options = None
+    _globals["_SQLSERVICE"].methods_by_name["PollSqlQuery"]._serialized_options = b"\200}\003"
     _globals["_SQLSERVICE"].methods_by_name["GetDbCatalogs"]._options = None
     _globals["_SQLSERVICE"].methods_by_name["GetDbCatalogs"]._serialized_options = b"\200}\013"
     _globals["_SQLSERVICE"].methods_by_name["GetDbSchemas"]._options = None
     _globals["_SQLSERVICE"].methods_by_name["GetDbSchemas"]._serialized_options = b"\200}\013"
     _globals["_SQLSERVICE"].methods_by_name["GetTables"]._options = None
     _globals["_SQLSERVICE"].methods_by_name["GetTables"]._serialized_options = b"\200}\013"
-    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 126
-    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 172
-    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 175
-    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 319
-    _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 321
-    _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 364
-    _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 366
-    _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 476
-    _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 478
-    _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 500
-    _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 502
-    _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 615
-    _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 618
-    _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 826
-    _globals["_DBSCHEMAINFO"]._serialized_start = 828
-    _globals["_DBSCHEMAINFO"]._serialized_end = 915
-    _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 918
-    _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 1052
-    _globals["_GETTABLESREQUEST"]._serialized_start = 1055
-    _globals["_GETTABLESREQUEST"]._serialized_end = 1342
-    _globals["_TABLEINFO"]._serialized_start = 1345
-    _globals["_TABLEINFO"]._serialized_end = 1534
-    _globals["_GETTABLESRESPONSE"]._serialized_start = 1536
-    _globals["_GETTABLESRESPONSE"]._serialized_end = 1662
-    _globals["_SQLSERVICE"]._serialized_start = 1665
-    _globals["_SQLSERVICE"]._serialized_end = 2190
+    _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_start = 3542
+    _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_end = 3715
+    _globals["_SQLQUERYINFO"]._serialized_start = 160
+    _globals["_SQLQUERYINFO"]._serialized_end = 370
+    _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_start = 372
+    _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_end = 407
+    _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_start = 409
+    _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_end = 533
+    _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_start = 535
+    _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_end = 598
+    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 601
+    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 1013
+    _globals["_SIGNEDOUTPUTURIS"]._serialized_start = 1015
+    _globals["_SIGNEDOUTPUTURIS"]._serialized_end = 1053
+    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 1056
+    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 1233
+    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 1236
+    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 1396
+    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 1399
+    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 1730
+    _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 1732
+    _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 1775
+    _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 1777
+    _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 1887
+    _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 1889
+    _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 1911
+    _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 1913
+    _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 2026
+    _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 2029
+    _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 2237
+    _globals["_DBSCHEMAINFO"]._serialized_start = 2239
+    _globals["_DBSCHEMAINFO"]._serialized_end = 2326
+    _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 2329
+    _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 2463
+    _globals["_GETTABLESREQUEST"]._serialized_start = 2466
+    _globals["_GETTABLESREQUEST"]._serialized_end = 2753
+    _globals["_TABLEINFO"]._serialized_start = 2756
+    _globals["_TABLEINFO"]._serialized_end = 2945
+    _globals["_GETTABLESRESPONSE"]._serialized_start = 2947
+    _globals["_GETTABLESRESPONSE"]._serialized_end = 3073
+    _globals["_SQLQUERYPROGRESSINFO"]._serialized_start = 3075
+    _globals["_SQLQUERYPROGRESSINFO"]._serialized_end = 3097
+    _globals["_SQLQUERYFAILEDINFO"]._serialized_start = 3099
+    _globals["_SQLQUERYFAILEDINFO"]._serialized_end = 3172
+    _globals["_POLLSQLQUERYREQUEST"]._serialized_start = 3174
+    _globals["_POLLSQLQUERYREQUEST"]._serialized_end = 3230
+    _globals["_POLLSQLQUERYRESPONSE"]._serialized_start = 3233
+    _globals["_POLLSQLQUERYRESPONSE"]._serialized_end = 3539
+    _globals["_SQLSERVICE"]._serialized_start = 3718
+    _globals["_SQLSERVICE"]._serialized_end = 4345
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi

@@ -1,6 +1,8 @@
 from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
 from chalk._gen.chalk.common.v1 import chalk_error_pb2 as _chalk_error_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
 from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
 from typing import (
@@ -13,24 +15,116 @@ from typing import (
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
+class ExecuteSqlAsyncExecutionMode(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    EXECUTE_SQL_ASYNC_EXECUTION_MODE_UNSPECIFIED: _ClassVar[ExecuteSqlAsyncExecutionMode]
+    EXECUTE_SQL_ASYNC_EXECUTION_MODE_IN_PROCESS: _ClassVar[ExecuteSqlAsyncExecutionMode]
+    EXECUTE_SQL_ASYNC_EXECUTION_MODE_ASYNC: _ClassVar[ExecuteSqlAsyncExecutionMode]
+
+EXECUTE_SQL_ASYNC_EXECUTION_MODE_UNSPECIFIED: ExecuteSqlAsyncExecutionMode
+EXECUTE_SQL_ASYNC_EXECUTION_MODE_IN_PROCESS: ExecuteSqlAsyncExecutionMode
+EXECUTE_SQL_ASYNC_EXECUTION_MODE_ASYNC: ExecuteSqlAsyncExecutionMode
+
+class SqlQueryInfo(_message.Message):
+    __slots__ = ("operation_id", "created_at", "finished_at")
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    FINISHED_AT_FIELD_NUMBER: _ClassVar[int]
+    operation_id: str
+    created_at: _timestamp_pb2.Timestamp
+    finished_at: _timestamp_pb2.Timestamp
+    def __init__(
+        self,
+        operation_id: _Optional[str] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        finished_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+    ) -> None: ...
+
+class ExecuteSqlSyncQueryRequestOptions(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class ExecuteSqlAsyncQueryRequestOptions(_message.Message):
+    __slots__ = ("execution_mode",)
+    EXECUTION_MODE_FIELD_NUMBER: _ClassVar[int]
+    execution_mode: ExecuteSqlAsyncExecutionMode
+    def __init__(self, execution_mode: _Optional[_Union[ExecuteSqlAsyncExecutionMode, str]] = ...) -> None: ...
+
+class ExecuteSqlResultPersistenceSettings(_message.Message):
+    __slots__ = ("enabled",)
+    ENABLED_FIELD_NUMBER: _ClassVar[int]
+    enabled: bool
+    def __init__(self, enabled: bool = ...) -> None: ...
+
 class ExecuteSqlQueryRequest(_message.Message):
-    __slots__ = ("query",)
+    __slots__ = ("query", "correlation_id", "persistence_settings", "sync_options", "async_options")
     QUERY_FIELD_NUMBER: _ClassVar[int]
+    CORRELATION_ID_FIELD_NUMBER: _ClassVar[int]
+    PERSISTENCE_SETTINGS_FIELD_NUMBER: _ClassVar[int]
+    SYNC_OPTIONS_FIELD_NUMBER: _ClassVar[int]
+    ASYNC_OPTIONS_FIELD_NUMBER: _ClassVar[int]
     query: str
-    def __init__(self, query: _Optional[str] = ...) -> None: ...
+    correlation_id: str
+    persistence_settings: ExecuteSqlResultPersistenceSettings
+    sync_options: ExecuteSqlSyncQueryRequestOptions
+    async_options: ExecuteSqlAsyncQueryRequestOptions
+    def __init__(
+        self,
+        query: _Optional[str] = ...,
+        correlation_id: _Optional[str] = ...,
+        persistence_settings: _Optional[_Union[ExecuteSqlResultPersistenceSettings, _Mapping]] = ...,
+        sync_options: _Optional[_Union[ExecuteSqlSyncQueryRequestOptions, _Mapping]] = ...,
+        async_options: _Optional[_Union[ExecuteSqlAsyncQueryRequestOptions, _Mapping]] = ...,
+    ) -> None: ...
+
+class SignedOutputUris(_message.Message):
+    __slots__ = ("uris",)
+    URIS_FIELD_NUMBER: _ClassVar[int]
+    uris: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, uris: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class ExecuteSqlSyncQueryResponsePayload(_message.Message):
+    __slots__ = ("parquet_response", "signed_output_uris")
+    PARQUET_RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    SIGNED_OUTPUT_URIS_FIELD_NUMBER: _ClassVar[int]
+    parquet_response: bytes
+    signed_output_uris: SignedOutputUris
+    def __init__(
+        self,
+        parquet_response: _Optional[bytes] = ...,
+        signed_output_uris: _Optional[_Union[SignedOutputUris, _Mapping]] = ...,
+    ) -> None: ...
+
+class ExecuteSqlAsyncQueryResponsePayload(_message.Message):
+    __slots__ = ("operation_id", "execution_mode")
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    EXECUTION_MODE_FIELD_NUMBER: _ClassVar[int]
+    operation_id: str
+    execution_mode: ExecuteSqlAsyncExecutionMode
+    def __init__(
+        self,
+        operation_id: _Optional[str] = ...,
+        execution_mode: _Optional[_Union[ExecuteSqlAsyncExecutionMode, str]] = ...,
+    ) -> None: ...
 
 class ExecuteSqlQueryResponse(_message.Message):
-    __slots__ = ("query_id", "parquet", "errors")
+    __slots__ = ("query_id", "parquet", "sync_payload", "async_payload", "errors")
     QUERY_ID_FIELD_NUMBER: _ClassVar[int]
     PARQUET_FIELD_NUMBER: _ClassVar[int]
+    SYNC_PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    ASYNC_PAYLOAD_FIELD_NUMBER: _ClassVar[int]
     ERRORS_FIELD_NUMBER: _ClassVar[int]
     query_id: str
     parquet: bytes
+    sync_payload: ExecuteSqlSyncQueryResponsePayload
+    async_payload: ExecuteSqlAsyncQueryResponsePayload
     errors: _containers.RepeatedCompositeFieldContainer[_chalk_error_pb2.ChalkError]
     def __init__(
         self,
         query_id: _Optional[str] = ...,
         parquet: _Optional[bytes] = ...,
+        sync_payload: _Optional[_Union[ExecuteSqlSyncQueryResponsePayload, _Mapping]] = ...,
+        async_payload: _Optional[_Union[ExecuteSqlAsyncQueryResponsePayload, _Mapping]] = ...,
         errors: _Optional[_Iterable[_Union[_chalk_error_pb2.ChalkError, _Mapping]]] = ...,
     ) -> None: ...
 
@@ -150,3 +244,37 @@ class GetTablesResponse(_message.Message):
         tables: _Optional[_Iterable[_Union[TableInfo, _Mapping]]] = ...,
         errors: _Optional[_Iterable[_Union[_chalk_error_pb2.ChalkError, _Mapping]]] = ...,
     ) -> None: ...
+
+class SqlQueryProgressInfo(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class SqlQueryFailedInfo(_message.Message):
+    __slots__ = ("errors",)
+    ERRORS_FIELD_NUMBER: _ClassVar[int]
+    errors: _containers.RepeatedCompositeFieldContainer[_chalk_error_pb2.ChalkError]
+    def __init__(self, errors: _Optional[_Iterable[_Union[_chalk_error_pb2.ChalkError, _Mapping]]] = ...) -> None: ...
+
+class PollSqlQueryRequest(_message.Message):
+    __slots__ = ("operation_id",)
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    operation_id: str
+    def __init__(self, operation_id: _Optional[str] = ...) -> None: ...
+
+class PollSqlQueryResponse(_message.Message):
+    __slots__ = ("info", "progress", "response", "failed")
+    INFO_FIELD_NUMBER: _ClassVar[int]
+    PROGRESS_FIELD_NUMBER: _ClassVar[int]
+    RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    FAILED_FIELD_NUMBER: _ClassVar[int]
+    info: SqlQueryInfo
+    progress: SqlQueryProgressInfo
+    response: ExecuteSqlSyncQueryResponsePayload
+    failed: SqlQueryFailedInfo
+    def __init__(
+        self,
+        info: _Optional[_Union[SqlQueryInfo, _Mapping]] = ...,
+        progress: _Optional[_Union[SqlQueryProgressInfo, _Mapping]] = ...,
+        response: _Optional[_Union[ExecuteSqlSyncQueryResponsePayload, _Mapping]] = ...,
+        failed: _Optional[_Union[SqlQueryFailedInfo, _Mapping]] = ...,
+    ) -> None: ...