chalkpy 2.94.6__tar.gz → 2.94.8__tar.gz

{chalkpy-2.94.6 → chalkpy-2.94.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.94.6
+Version: 2.94.8
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

chalkpy-2.94.8/chalk/_gen/chalk/protosql/v1/sql_service_pb2.py

@@ -0,0 +1,95 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/protosql/v1/sql_service.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from 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\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()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.protosql.v1.sql_service_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\025com.chalk.protosql.v1B\017SqlServiceProtoP\001\242\002\003CPX\252\002\021Chalk.Protosql.V1\312\002\021Chalk\\Protosql\\V1\342\002\035Chalk\\Protosql\\V1\\GPBMetadata\352\002\023Chalk::Protosql::V1"
+    _globals["_SQLSERVICE"].methods_by_name["ExecuteSqlQuery"]._options = None
+    _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["_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)

{chalkpy-2.94.6 → chalkpy-2.94.8}/chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi

@@ -1,5 +1,6 @@
 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
@@ -24,13 +25,30 @@ 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__ = ()
-    def __init__(self) -> None: ...
+    __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",)
@@ -226,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: ...

{chalkpy-2.94.6 → chalkpy-2.94.8}/chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py

@@ -25,6 +25,11 @@ class SqlServiceStub(object):
             request_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryRequest.SerializeToString,
             response_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryResponse.FromString,
         )
+        self.PollSqlQuery = channel.unary_unary(
+            "/chalk.protosql.v1.SqlService/PollSqlQuery",
+            request_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryRequest.SerializeToString,
+            response_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryResponse.FromString,
+        )
         self.GetDbCatalogs = channel.unary_unary(
             "/chalk.protosql.v1.SqlService/GetDbCatalogs",
             request_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.GetDbCatalogsRequest.SerializeToString,
@@ -57,6 +62,12 @@ class SqlServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def PollSqlQuery(self, request, context):
+        """Poll for the status and results of an asynchronous SQL query"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
     def GetDbCatalogs(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -88,6 +99,11 @@ def add_SqlServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryRequest.FromString,
             response_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PlanSqlQueryResponse.SerializeToString,
         ),
+        "PollSqlQuery": grpc.unary_unary_rpc_method_handler(
+            servicer.PollSqlQuery,
+            request_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryRequest.FromString,
+            response_serializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryResponse.SerializeToString,
+        ),
         "GetDbCatalogs": grpc.unary_unary_rpc_method_handler(
             servicer.GetDbCatalogs,
             request_deserializer=chalk_dot_protosql_dot_v1_dot_sql__service__pb2.GetDbCatalogsRequest.FromString,
@@ -170,6 +186,35 @@ class SqlService(object):
             metadata,
         )
 
+    @staticmethod
+    def PollSqlQuery(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.protosql.v1.SqlService/PollSqlQuery",
+            chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryRequest.SerializeToString,
+            chalk_dot_protosql_dot_v1_dot_sql__service__pb2.PollSqlQueryResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
     @staticmethod
     def GetDbCatalogs(
         request,

{chalkpy-2.94.6 → chalkpy-2.94.8}/chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi

@@ -18,6 +18,8 @@ from chalk._gen.chalk.protosql.v1.sql_service_pb2 import (
     GetTablesResponse,
     PlanSqlQueryRequest,
     PlanSqlQueryResponse,
+    PollSqlQueryRequest,
+    PollSqlQueryResponse,
 )
 from grpc import (
     Channel,
@@ -36,6 +38,11 @@ class SqlServiceStub:
         PlanSqlQueryRequest,
         PlanSqlQueryResponse,
     ]
+    PollSqlQuery: UnaryUnaryMultiCallable[
+        PollSqlQueryRequest,
+        PollSqlQueryResponse,
+    ]
+    """Poll for the status and results of an asynchronous SQL query"""
     GetDbCatalogs: UnaryUnaryMultiCallable[
         GetDbCatalogsRequest,
         GetDbCatalogsResponse,
@@ -63,6 +70,13 @@ class SqlServiceServicer(metaclass=ABCMeta):
         context: ServicerContext,
     ) -> PlanSqlQueryResponse: ...
     @abstractmethod
+    def PollSqlQuery(
+        self,
+        request: PollSqlQueryRequest,
+        context: ServicerContext,
+    ) -> PollSqlQueryResponse:
+        """Poll for the status and results of an asynchronous SQL query"""
+    @abstractmethod
     def GetDbCatalogs(
         self,
         request: GetDbCatalogsRequest,

{chalkpy-2.94.6 → chalkpy-2.94.8}/chalk/_gen/chalk/server/v1/environment_pb2.py

@@ -18,7 +18,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n!chalk/server/v1/environment.proto\x12\x0f\x63halk.server.v1\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"{\n\x13\x41WSCloudWatchConfig\x12)\n\x0elog_group_path\x18\x01 \x01(\tH\x00R\x0clogGroupPath\x88\x01\x01\x12&\n\x0flog_group_paths\x18\x02 \x03(\tR\rlogGroupPathsB\x11\n\x0f_log_group_path"\xab\x02\n\x16\x41WSSecretManagerConfig\x12)\n\x0esecret_kms_arn\x18\x01 \x01(\tH\x00R\x0csecretKmsArn\x88\x01\x01\x12X\n\x0bsecret_tags\x18\x02 \x03(\x0b\x32\x37.chalk.server.v1.AWSSecretManagerConfig.SecretTagsEntryR\nsecretTags\x12(\n\rsecret_prefix\x18\x03 \x01(\tH\x01R\x0csecretPrefix\x88\x01\x01\x1a=\n\x0fSecretTagsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_secret_kms_arnB\x10\n\x0e_secret_prefix"9\n\x1bGCPSecretReplicationReplica\x12\x1a\n\x08location\x18\x01 \x01(\tR\x08location"D\n\x0fGCPRegionConfig\x12"\n\nscope_type\x18\x01 \x01(\tH\x00R\tscopeType\x88\x01\x01\x42\r\n\x0b_scope_type"\x9e\x01\n\x16GCPSecretManagerConfig\x12(\n\rsecret_region\x18\x01 \x01(\tH\x00R\x0csecretRegion\x88\x01\x01\x12H\n\x08replicas\x18\x02 \x03(\x0b\x32,.chalk.server.v1.GCPSecretReplicationReplicaR\x08replicasB\x10\n\x0e_secret_region"\xad\x01\n\x13GCPWorkloadIdentity\x12,\n\x12gcp_project_number\x18\x01 \x01(\tR\x10gcpProjectNumber\x12.\n\x13gcp_service_account\x18\x02 \x01(\tR\x11gcpServiceAccount\x12\x17\n\x07pool_id\x18\x03 \x01(\tR\x06poolId\x12\x1f\n\x0bprovider_id\x18\x04 \x01(\tR\nproviderId"\x88\x02\n\x11\x44ockerBuildConfig\x12\x18\n\x07\x62uilder\x18\x01 \x01(\tR\x07\x62uilder\x12,\n\x12push_registry_type\x18\x02 \x01(\tR\x10pushRegistryType\x12\x37\n\x18push_registry_tag_prefix\x18\x03 \x01(\tR\x15pushRegistryTagPrefix\x12\x43\n\x1eregistry_credentials_secret_id\x18\x04 \x01(\tR\x1bregistryCredentialsSecretId\x12-\n\x12notification_topic\x18\x05 \x01(\tR\x11notificationTopic"l\n\x16\x45lasticsearchLogConfig\x12\x1a\n\x08username\x18\x01 \x01(\tR\x08username\x12\x1a\n\x08password\x18\x02 \x01(\tR\x08password\x12\x1a\n\x08\x65ndpoint\x18\x03 \x01(\tR\x08\x65ndpoint"\xa7\x07\n\x0e\x41WSCloudConfig\x12\x1d\n\naccount_id\x18\x01 \x01(\tR\taccountId\x12.\n\x13management_role_arn\x18\x02 \x01(\tR\x11managementRoleArn\x12\x16\n\x06region\x18\x03 \x01(\tR\x06region\x12$\n\x0b\x65xternal_id\x18\x04 \x01(\tH\x00R\nexternalId\x88\x01\x01\x12k\n\x1d\x64\x65precated_cloud_watch_config\x18\x05 \x01(\x0b\x32$.chalk.server.v1.AWSCloudWatchConfigB\x02\x18\x01R\x1a\x64\x65precatedCloudWatchConfig\x12t\n deprecated_secret_manager_config\x18\x06 \x01(\x0b\x32\'.chalk.server.v1.AWSSecretManagerConfigB\x02\x18\x01R\x1d\x64\x65precatedSecretManagerConfig\x12U\n\x11workload_identity\x18\x07 \x01(\x0b\x32$.chalk.server.v1.GCPWorkloadIdentityB\x02\x18\x01R\x10workloadIdentity\x12R\n\x13\x64ocker_build_config\x18\x08 \x01(\x0b\x32".chalk.server.v1.DockerBuildConfigR\x11\x64ockerBuildConfig\x12\x61\n\x18\x65lasticsearch_log_config\x18\t \x01(\x0b\x32\'.chalk.server.v1.ElasticsearchLogConfigR\x16\x65lasticsearchLogConfig\x12Q\n\x11\x63loudwatch_config\x18\n \x01(\x0b\x32$.chalk.server.v1.AWSCloudWatchConfigR\x10\x63loudwatchConfig\x12Z\n\x14secretmanager_config\x18\x0b \x01(\x0b\x32\'.chalk.server.v1.AWSSecretManagerConfigR\x13secretmanagerConfig\x12X\n\x15gcp_workload_identity\x18\x0c \x01(\x0b\x32$.chalk.server.v1.GCPWorkloadIdentityR\x13gcpWorkloadIdentityB\x0e\n\x0c_external_id"\xa0\x03\n\x0eGCPCloudConfig\x12\x1d\n\nproject_id\x18\x01 \x01(\tR\tprojectId\x12\x16\n\x06region\x18\x02 \x01(\tR\x06region\x12\x41\n\x1amanagement_service_account\x18\x03 \x01(\tH\x00R\x18managementServiceAccount\x88\x01\x01\x12R\n\x13\x64ocker_build_config\x18\x04 \x01(\x0b\x32".chalk.server.v1.DockerBuildConfigR\x11\x64ockerBuildConfig\x12Z\n\x14secretmanager_config\x18\x05 \x01(\x0b\x32\'.chalk.server.v1.GCPSecretManagerConfigR\x13secretmanagerConfig\x12\x45\n\rregion_config\x18\x06 \x01(\x0b\x32 .chalk.server.v1.GCPRegionConfigR\x0cregionConfigB\x1d\n\x1b_management_service_account"Z\n\x1c\x41zureContainerRegistryConfig\x12(\n\rregistry_name\x18\x01 \x01(\tH\x00R\x0cregistryName\x88\x01\x01\x42\x10\n\x0e_registry_name"H\n\x13\x41zureKeyVaultConfig\x12"\n\nvault_name\x18\x01 \x01(\tH\x00R\tvaultName\x88\x01\x01\x42\r\n\x0b_vault_name"\x80\x04\n\x10\x41zureCloudConfig\x12\'\n\x0fsubscription_id\x18\x01 \x01(\tR\x0esubscriptionId\x12\x1b\n\ttenant_id\x18\x02 \x01(\tR\x08tenantId\x12\x16\n\x06region\x18\x03 \x01(\tR\x06region\x12%\n\x0eresource_group\x18\x04 \x01(\tR\rresourceGroup\x12R\n\x13\x64ocker_build_config\x18\x06 \x01(\x0b\x32".chalk.server.v1.DockerBuildConfigR\x11\x64ockerBuildConfig\x12i\n\x19\x63ontainer_registry_config\x18\x07 \x01(\x0b\x32-.chalk.server.v1.AzureContainerRegistryConfigR\x17\x63ontainerRegistryConfig\x12N\n\x10key_vault_config\x18\x08 \x01(\x0b\x32$.chalk.server.v1.AzureKeyVaultConfigR\x0ekeyVaultConfig\x12X\n\x15gcp_workload_identity\x18\t \x01(\x0b\x32$.chalk.server.v1.GCPWorkloadIdentityR\x13gcpWorkloadIdentity"\xbc\x01\n\x0b\x43loudConfig\x12\x33\n\x03\x61ws\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.AWSCloudConfigH\x00R\x03\x61ws\x12\x33\n\x03gcp\x18\x02 \x01(\x0b\x32\x1f.chalk.server.v1.GCPCloudConfigH\x00R\x03gcp\x12\x39\n\x05\x61zure\x18\x03 \x01(\x0b\x32!.chalk.server.v1.AzureCloudConfigH\x00R\x05\x61zureB\x08\n\x06\x63onfig"\xdb\x01\n\x1e\x45nvironmentObjectStorageConfig\x12%\n\x0e\x64\x61taset_bucket\x18\x01 \x01(\tR\rdatasetBucket\x12,\n\x12plan_stages_bucket\x18\x02 \x01(\tR\x10planStagesBucket\x12\x30\n\x14source_bundle_bucket\x18\x03 \x01(\tR\x12sourceBundleBucket\x12\x32\n\x15model_registry_bucket\x18\x04 \x01(\tR\x13modelRegistryBucket"\xdd%\n\x0b\x45nvironment\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x1d\n\nproject_id\x18\x02 \x01(\tR\tprojectId\x12\x0e\n\x02id\x18\x03 \x01(\tR\x02id\x12\x17\n\x07team_id\x18\x04 \x01(\tR\x06teamId\x12\x35\n\x14\x61\x63tive_deployment_id\x18\x05 \x01(\tH\x00R\x12\x61\x63tiveDeploymentId\x88\x01\x01\x12"\n\nworker_url\x18\x06 \x01(\tH\x01R\tworkerUrl\x88\x01\x01\x12$\n\x0bservice_url\x18\x07 \x01(\tH\x02R\nserviceUrl\x88\x01\x01\x12"\n\nbranch_url\x18\x08 \x01(\tH\x03R\tbranchUrl\x88\x01\x01\x12\x35\n\x14offline_store_secret\x18\t \x01(\tH\x04R\x12offlineStoreSecret\x88\x01\x01\x12\x33\n\x13online_store_secret\x18\n \x01(\tH\x05R\x11onlineStoreSecret\x88\x01\x01\x12\x35\n\x14\x66\x65\x61ture_store_secret\x18\x0b \x01(\tH\x06R\x12\x66\x65\x61tureStoreSecret\x88\x01\x01\x12,\n\x0fpostgres_secret\x18\x0c \x01(\tH\x07R\x0epostgresSecret\x88\x01\x01\x12/\n\x11online_store_kind\x18\r \x01(\tH\x08R\x0fonlineStoreKind\x88\x01\x01\x12\x1c\n\x07\x65mq_uri\x18\x0e \x01(\tH\tR\x06\x65mqUri\x88\x01\x01\x12\x31\n\x12vpc_connector_name\x18\x0f \x01(\tH\nR\x10vpcConnectorName\x88\x01\x01\x12/\n\x11kube_cluster_name\x18\x10 \x01(\tH\x0bR\x0fkubeClusterName\x88\x01\x01\x12<\n\x18\x62ranch_kube_cluster_name\x18\x11 \x01(\tH\x0cR\x15\x62ranchKubeClusterName\x88\x01\x01\x12<\n\x18\x65ngine_kube_cluster_name\x18\x12 \x01(\tH\rR\x15\x65ngineKubeClusterName\x88\x01\x01\x12I\n\x1fshadow_engine_kube_cluster_name\x18\x13 \x01(\tH\x0eR\x1bshadowEngineKubeClusterName\x88\x01\x01\x12\x31\n\x12kube_job_namespace\x18\x14 \x01(\tH\x0fR\x10kubeJobNamespace\x88\x01\x01\x12\x39\n\x16kube_preview_namespace\x18\x15 \x01(\tH\x10R\x14kubePreviewNamespace\x88\x01\x01\x12>\n\x19kube_service_account_name\x18\x16 \x01(\tH\x11R\x16kubeServiceAccountName\x88\x01\x01\x12\x42\n\x1bstreaming_query_service_uri\x18\x17 \x01(\tH\x12R\x18streamingQueryServiceUri\x88\x01\x01\x12`\n.skip_offline_writes_for_online_cached_features\x18\x18 \x01(\x08R(skipOfflineWritesForOnlineCachedFeatures\x12-\n\x10result_bus_topic\x18\x19 \x01(\tH\x13R\x0eresultBusTopic\x88\x01\x01\x12;\n\x17online_persistence_mode\x18\x1a \x01(\tH\x14R\x15onlinePersistenceMode\x88\x01\x01\x12/\n\x11metrics_bus_topic\x18\x1b \x01(\tH\x15R\x0fmetricsBusTopic\x88\x01\x01\x12\x39\n\x16\x62igtable_instance_name\x18\x1c \x01(\tH\x16R\x14\x62igtableInstanceName\x88\x01\x01\x12\x33\n\x13\x62igtable_table_name\x18\x1d \x01(\tH\x17R\x11\x62igtableTableName\x88\x01\x01\x12\x37\n\x15\x63loud_account_locator\x18\x1e \x01(\tH\x18R\x13\x63loudAccountLocator\x88\x01\x01\x12&\n\x0c\x63loud_region\x18\x1f \x01(\tH\x19R\x0b\x63loudRegion\x88\x01\x01\x12-\n\x10\x63loud_tenancy_id\x18  \x01(\tH\x1aR\x0e\x63loudTenancyId\x88\x01\x01\x12\x35\n\x14source_bundle_bucket\x18! \x01(\tH\x1bR\x12sourceBundleBucket\x88\x01\x01\x12\x42\n\x1b\x65ngine_docker_registry_path\x18" \x01(\tH\x1cR\x18\x65ngineDockerRegistryPath\x88\x01\x01\x12,\n\x0f\x64\x65\x66\x61ult_planner\x18# \x01(\tH\x1dR\x0e\x64\x65\x66\x61ultPlanner\x88\x01\x01\x12\x63\n\x13\x61\x64\x64itional_env_vars\x18$ \x03(\x0b\x32\x33.chalk.server.v1.Environment.AdditionalEnvVarsEntryR\x11\x61\x64\x64itionalEnvVars\x12p\n\x18\x61\x64\x64itional_cron_env_vars\x18% \x03(\x0b\x32\x37.chalk.server.v1.Environment.AdditionalCronEnvVarsEntryR\x15\x61\x64\x64itionalCronEnvVars\x12=\n\x18private_pip_repositories\x18& \x01(\tH\x1eR\x16privatePipRepositories\x88\x01\x01\x12\x1d\n\nis_sandbox\x18\' \x01(\x08R\tisSandbox\x12I\n\x0e\x63loud_provider\x18) \x01(\x0e\x32".chalk.server.v1.CloudProviderKindR\rcloudProvider\x12\x44\n\x0c\x63loud_config\x18* \x01(\x0b\x32\x1c.chalk.server.v1.CloudConfigH\x1fR\x0b\x63loudConfig\x88\x01\x01\x12Z\n\x10spec_config_json\x18( \x03(\x0b\x32\x30.chalk.server.v1.Environment.SpecConfigJsonEntryR\x0especConfigJson\x12@\n\x0b\x61rchived_at\x18+ \x01(\x0b\x32\x1a.google.protobuf.TimestampH R\narchivedAt\x88\x01\x01\x12S\n$metadata_server_metrics_store_secret\x18, \x01(\tH!R metadataServerMetricsStoreSecret\x88\x01\x01\x12M\n!query_server_metrics_store_secret\x18- \x01(\tH"R\x1dqueryServerMetricsStoreSecret\x88\x01\x01\x12/\n\x11pinned_base_image\x18. \x01(\tH#R\x0fpinnedBaseImage\x88\x01\x01\x12\x31\n\x12\x63luster_gateway_id\x18/ \x01(\tH$R\x10\x63lusterGatewayId\x88\x01\x01\x12\x39\n\x16\x63luster_timescaledb_id\x18\x30 \x01(\tH%R\x14\x63lusterTimescaledbId\x88\x01\x01\x12T\n$background_persistence_deployment_id\x18\x31 \x01(\tH&R!backgroundPersistenceDeploymentId\x88\x01\x01\x12\x65\n\x13\x65nvironment_buckets\x18\x32 \x01(\x0b\x32/.chalk.server.v1.EnvironmentObjectStorageConfigH\'R\x12\x65nvironmentBuckets\x88\x01\x01\x12\x41\n\x1a\x63luster_timescaledb_secret\x18\x33 \x01(\tH(R\x18\x63lusterTimescaledbSecret\x88\x01\x01\x12+\n\x0fgrpc_engine_url\x18\x34 \x01(\tH)R\rgrpcEngineUrl\x88\x01\x01\x12/\n\x11kube_cluster_mode\x18\x35 \x01(\tH*R\x0fkubeClusterMode\x88\x01\x01\x12(\n\rdashboard_url\x18\x36 \x01(\tH+R\x0c\x64\x61shboardUrl\x88\x01\x01\x12+\n\x0fkube_cluster_id\x18\x37 \x01(\tH,R\rkubeClusterId\x88\x01\x01\x12\x1d\n\x07managed\x18\x38 \x01(\x08H-R\x07managed\x88\x01\x01\x12;\n\x17telemetry_deployment_id\x18\x39 \x01(\tH.R\x15telemetryDeploymentId\x88\x01\x01\x12\x42\n\x0csuspended_at\x18: \x01(\x0b\x32\x1a.google.protobuf.TimestampH/R\x0bsuspendedAt\x88\x01\x01\x12`\n\x15\x64\x65\x66\x61ult_build_profile\x18; \x01(\x0e\x32\'.chalk.server.v1.DeploymentBuildProfileH0R\x13\x64\x65\x66\x61ultBuildProfile\x88\x01\x01\x1a\x44\n\x16\x41\x64\x64itionalEnvVarsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1aH\n\x1a\x41\x64\x64itionalCronEnvVarsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1aY\n\x13SpecConfigJsonEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\x17\n\x15_active_deployment_idB\r\n\x0b_worker_urlB\x0e\n\x0c_service_urlB\r\n\x0b_branch_urlB\x17\n\x15_offline_store_secretB\x16\n\x14_online_store_secretB\x17\n\x15_feature_store_secretB\x12\n\x10_postgres_secretB\x14\n\x12_online_store_kindB\n\n\x08_emq_uriB\x15\n\x13_vpc_connector_nameB\x14\n\x12_kube_cluster_nameB\x1b\n\x19_branch_kube_cluster_nameB\x1b\n\x19_engine_kube_cluster_nameB"\n _shadow_engine_kube_cluster_nameB\x15\n\x13_kube_job_namespaceB\x19\n\x17_kube_preview_namespaceB\x1c\n\x1a_kube_service_account_nameB\x1e\n\x1c_streaming_query_service_uriB\x13\n\x11_result_bus_topicB\x1a\n\x18_online_persistence_modeB\x14\n\x12_metrics_bus_topicB\x19\n\x17_bigtable_instance_nameB\x16\n\x14_bigtable_table_nameB\x18\n\x16_cloud_account_locatorB\x0f\n\r_cloud_regionB\x13\n\x11_cloud_tenancy_idB\x17\n\x15_source_bundle_bucketB\x1e\n\x1c_engine_docker_registry_pathB\x12\n\x10_default_plannerB\x1b\n\x19_private_pip_repositoriesB\x0f\n\r_cloud_configB\x0e\n\x0c_archived_atB\'\n%_metadata_server_metrics_store_secretB$\n"_query_server_metrics_store_secretB\x14\n\x12_pinned_base_imageB\x15\n\x13_cluster_gateway_idB\x19\n\x17_cluster_timescaledb_idB\'\n%_background_persistence_deployment_idB\x16\n\x14_environment_bucketsB\x1d\n\x1b_cluster_timescaledb_secretB\x12\n\x10_grpc_engine_urlB\x14\n\x12_kube_cluster_modeB\x10\n\x0e_dashboard_urlB\x12\n\x10_kube_cluster_idB\n\n\x08_managedB\x1a\n\x18_telemetry_deployment_idB\x0f\n\r_suspended_atB\x18\n\x16_default_build_profile*\xb2\x01\n\x11\x43loudProviderKind\x12#\n\x1f\x43LOUD_PROVIDER_KIND_UNSPECIFIED\x10\x00\x12\x1f\n\x1b\x43LOUD_PROVIDER_KIND_UNKNOWN\x10\x01\x12\x1b\n\x17\x43LOUD_PROVIDER_KIND_GCP\x10\x02\x12\x1b\n\x17\x43LOUD_PROVIDER_KIND_AWS\x10\x03\x12\x1d\n\x19\x43LOUD_PROVIDER_KIND_AZURE\x10\x04*\xf4\x01\n\x16\x44\x65ploymentBuildProfile\x12(\n$DEPLOYMENT_BUILD_PROFILE_UNSPECIFIED\x10\x00\x12,\n(DEPLOYMENT_BUILD_PROFILE_O3_NO_PROFILING\x10\x01\x12)\n%DEPLOYMENT_BUILD_PROFILE_O3_PROFILING\x10\x02\x12,\n(DEPLOYMENT_BUILD_PROFILE_O2_NO_PROFILING\x10\x03\x12)\n%DEPLOYMENT_BUILD_PROFILE_O2_PROFILING\x10\x04\x42\x99\x01\n\x13\x63om.chalk.server.v1B\x10\x45nvironmentProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
+    b'\n!chalk/server/v1/environment.proto\x12\x0f\x63halk.server.v1\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"{\n\x13\x41WSCloudWatchConfig\x12)\n\x0elog_group_path\x18\x01 \x01(\tH\x00R\x0clogGroupPath\x88\x01\x01\x12&\n\x0flog_group_paths\x18\x02 \x03(\tR\rlogGroupPathsB\x11\n\x0f_log_group_path"\xab\x02\n\x16\x41WSSecretManagerConfig\x12)\n\x0esecret_kms_arn\x18\x01 \x01(\tH\x00R\x0csecretKmsArn\x88\x01\x01\x12X\n\x0bsecret_tags\x18\x02 \x03(\x0b\x32\x37.chalk.server.v1.AWSSecretManagerConfig.SecretTagsEntryR\nsecretTags\x12(\n\rsecret_prefix\x18\x03 \x01(\tH\x01R\x0csecretPrefix\x88\x01\x01\x1a=\n\x0fSecretTagsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_secret_kms_arnB\x10\n\x0e_secret_prefix"9\n\x1bGCPSecretReplicationReplica\x12\x1a\n\x08location\x18\x01 \x01(\tR\x08location"D\n\x0fGCPRegionConfig\x12"\n\nscope_type\x18\x01 \x01(\tH\x00R\tscopeType\x88\x01\x01\x42\r\n\x0b_scope_type"\x9e\x01\n\x16GCPSecretManagerConfig\x12(\n\rsecret_region\x18\x01 \x01(\tH\x00R\x0csecretRegion\x88\x01\x01\x12H\n\x08replicas\x18\x02 \x03(\x0b\x32,.chalk.server.v1.GCPSecretReplicationReplicaR\x08replicasB\x10\n\x0e_secret_region"\xad\x01\n\x13GCPWorkloadIdentity\x12,\n\x12gcp_project_number\x18\x01 \x01(\tR\x10gcpProjectNumber\x12.\n\x13gcp_service_account\x18\x02 \x01(\tR\x11gcpServiceAccount\x12\x17\n\x07pool_id\x18\x03 \x01(\tR\x06poolId\x12\x1f\n\x0bprovider_id\x18\x04 \x01(\tR\nproviderId"\x88\x02\n\x11\x44ockerBuildConfig\x12\x18\n\x07\x62uilder\x18\x01 \x01(\tR\x07\x62uilder\x12,\n\x12push_registry_type\x18\x02 \x01(\tR\x10pushRegistryType\x12\x37\n\x18push_registry_tag_prefix\x18\x03 \x01(\tR\x15pushRegistryTagPrefix\x12\x43\n\x1eregistry_credentials_secret_id\x18\x04 \x01(\tR\x1bregistryCredentialsSecretId\x12-\n\x12notification_topic\x18\x05 \x01(\tR\x11notificationTopic"l\n\x16\x45lasticsearchLogConfig\x12\x1a\n\x08username\x18\x01 \x01(\tR\x08username\x12\x1a\n\x08password\x18\x02 \x01(\tR\x08password\x12\x1a\n\x08\x65ndpoint\x18\x03 \x01(\tR\x08\x65ndpoint"\xa7\x07\n\x0e\x41WSCloudConfig\x12\x1d\n\naccount_id\x18\x01 \x01(\tR\taccountId\x12.\n\x13management_role_arn\x18\x02 \x01(\tR\x11managementRoleArn\x12\x16\n\x06region\x18\x03 \x01(\tR\x06region\x12$\n\x0b\x65xternal_id\x18\x04 \x01(\tH\x00R\nexternalId\x88\x01\x01\x12k\n\x1d\x64\x65precated_cloud_watch_config\x18\x05 \x01(\x0b\x32$.chalk.server.v1.AWSCloudWatchConfigB\x02\x18\x01R\x1a\x64\x65precatedCloudWatchConfig\x12t\n deprecated_secret_manager_config\x18\x06 \x01(\x0b\x32\'.chalk.server.v1.AWSSecretManagerConfigB\x02\x18\x01R\x1d\x64\x65precatedSecretManagerConfig\x12U\n\x11workload_identity\x18\x07 \x01(\x0b\x32$.chalk.server.v1.GCPWorkloadIdentityB\x02\x18\x01R\x10workloadIdentity\x12R\n\x13\x64ocker_build_config\x18\x08 \x01(\x0b\x32".chalk.server.v1.DockerBuildConfigR\x11\x64ockerBuildConfig\x12\x61\n\x18\x65lasticsearch_log_config\x18\t \x01(\x0b\x32\'.chalk.server.v1.ElasticsearchLogConfigR\x16\x65lasticsearchLogConfig\x12Q\n\x11\x63loudwatch_config\x18\n \x01(\x0b\x32$.chalk.server.v1.AWSCloudWatchConfigR\x10\x63loudwatchConfig\x12Z\n\x14secretmanager_config\x18\x0b \x01(\x0b\x32\'.chalk.server.v1.AWSSecretManagerConfigR\x13secretmanagerConfig\x12X\n\x15gcp_workload_identity\x18\x0c \x01(\x0b\x32$.chalk.server.v1.GCPWorkloadIdentityR\x13gcpWorkloadIdentityB\x0e\n\x0c_external_id"\xa0\x03\n\x0eGCPCloudConfig\x12\x1d\n\nproject_id\x18\x01 \x01(\tR\tprojectId\x12\x16\n\x06region\x18\x02 \x01(\tR\x06region\x12\x41\n\x1amanagement_service_account\x18\x03 \x01(\tH\x00R\x18managementServiceAccount\x88\x01\x01\x12R\n\x13\x64ocker_build_config\x18\x04 \x01(\x0b\x32".chalk.server.v1.DockerBuildConfigR\x11\x64ockerBuildConfig\x12Z\n\x14secretmanager_config\x18\x05 \x01(\x0b\x32\'.chalk.server.v1.GCPSecretManagerConfigR\x13secretmanagerConfig\x12\x45\n\rregion_config\x18\x06 \x01(\x0b\x32 .chalk.server.v1.GCPRegionConfigR\x0cregionConfigB\x1d\n\x1b_management_service_account"Z\n\x1c\x41zureContainerRegistryConfig\x12(\n\rregistry_name\x18\x01 \x01(\tH\x00R\x0cregistryName\x88\x01\x01\x42\x10\n\x0e_registry_name"H\n\x13\x41zureKeyVaultConfig\x12"\n\nvault_name\x18\x01 \x01(\tH\x00R\tvaultName\x88\x01\x01\x42\r\n\x0b_vault_name"\x80\x04\n\x10\x41zureCloudConfig\x12\'\n\x0fsubscription_id\x18\x01 \x01(\tR\x0esubscriptionId\x12\x1b\n\ttenant_id\x18\x02 \x01(\tR\x08tenantId\x12\x16\n\x06region\x18\x03 \x01(\tR\x06region\x12%\n\x0eresource_group\x18\x04 \x01(\tR\rresourceGroup\x12R\n\x13\x64ocker_build_config\x18\x06 \x01(\x0b\x32".chalk.server.v1.DockerBuildConfigR\x11\x64ockerBuildConfig\x12i\n\x19\x63ontainer_registry_config\x18\x07 \x01(\x0b\x32-.chalk.server.v1.AzureContainerRegistryConfigR\x17\x63ontainerRegistryConfig\x12N\n\x10key_vault_config\x18\x08 \x01(\x0b\x32$.chalk.server.v1.AzureKeyVaultConfigR\x0ekeyVaultConfig\x12X\n\x15gcp_workload_identity\x18\t \x01(\x0b\x32$.chalk.server.v1.GCPWorkloadIdentityR\x13gcpWorkloadIdentity"\xbc\x01\n\x0b\x43loudConfig\x12\x33\n\x03\x61ws\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.AWSCloudConfigH\x00R\x03\x61ws\x12\x33\n\x03gcp\x18\x02 \x01(\x0b\x32\x1f.chalk.server.v1.GCPCloudConfigH\x00R\x03gcp\x12\x39\n\x05\x61zure\x18\x03 \x01(\x0b\x32!.chalk.server.v1.AzureCloudConfigH\x00R\x05\x61zureB\x08\n\x06\x63onfig"\xdb\x01\n\x1e\x45nvironmentObjectStorageConfig\x12%\n\x0e\x64\x61taset_bucket\x18\x01 \x01(\tR\rdatasetBucket\x12,\n\x12plan_stages_bucket\x18\x02 \x01(\tR\x10planStagesBucket\x12\x30\n\x14source_bundle_bucket\x18\x03 \x01(\tR\x12sourceBundleBucket\x12\x32\n\x15model_registry_bucket\x18\x04 \x01(\tR\x13modelRegistryBucket"\xfe&\n\x0b\x45nvironment\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x1d\n\nproject_id\x18\x02 \x01(\tR\tprojectId\x12\x0e\n\x02id\x18\x03 \x01(\tR\x02id\x12\x17\n\x07team_id\x18\x04 \x01(\tR\x06teamId\x12\x35\n\x14\x61\x63tive_deployment_id\x18\x05 \x01(\tH\x00R\x12\x61\x63tiveDeploymentId\x88\x01\x01\x12"\n\nworker_url\x18\x06 \x01(\tH\x01R\tworkerUrl\x88\x01\x01\x12$\n\x0bservice_url\x18\x07 \x01(\tH\x02R\nserviceUrl\x88\x01\x01\x12"\n\nbranch_url\x18\x08 \x01(\tH\x03R\tbranchUrl\x88\x01\x01\x12\x35\n\x14offline_store_secret\x18\t \x01(\tH\x04R\x12offlineStoreSecret\x88\x01\x01\x12\x33\n\x13online_store_secret\x18\n \x01(\tH\x05R\x11onlineStoreSecret\x88\x01\x01\x12\x35\n\x14\x66\x65\x61ture_store_secret\x18\x0b \x01(\tH\x06R\x12\x66\x65\x61tureStoreSecret\x88\x01\x01\x12,\n\x0fpostgres_secret\x18\x0c \x01(\tH\x07R\x0epostgresSecret\x88\x01\x01\x12/\n\x11online_store_kind\x18\r \x01(\tH\x08R\x0fonlineStoreKind\x88\x01\x01\x12\x1c\n\x07\x65mq_uri\x18\x0e \x01(\tH\tR\x06\x65mqUri\x88\x01\x01\x12\x31\n\x12vpc_connector_name\x18\x0f \x01(\tH\nR\x10vpcConnectorName\x88\x01\x01\x12/\n\x11kube_cluster_name\x18\x10 \x01(\tH\x0bR\x0fkubeClusterName\x88\x01\x01\x12<\n\x18\x62ranch_kube_cluster_name\x18\x11 \x01(\tH\x0cR\x15\x62ranchKubeClusterName\x88\x01\x01\x12<\n\x18\x65ngine_kube_cluster_name\x18\x12 \x01(\tH\rR\x15\x65ngineKubeClusterName\x88\x01\x01\x12I\n\x1fshadow_engine_kube_cluster_name\x18\x13 \x01(\tH\x0eR\x1bshadowEngineKubeClusterName\x88\x01\x01\x12\x31\n\x12kube_job_namespace\x18\x14 \x01(\tH\x0fR\x10kubeJobNamespace\x88\x01\x01\x12\x39\n\x16kube_preview_namespace\x18\x15 \x01(\tH\x10R\x14kubePreviewNamespace\x88\x01\x01\x12>\n\x19kube_service_account_name\x18\x16 \x01(\tH\x11R\x16kubeServiceAccountName\x88\x01\x01\x12\x42\n\x1bstreaming_query_service_uri\x18\x17 \x01(\tH\x12R\x18streamingQueryServiceUri\x88\x01\x01\x12`\n.skip_offline_writes_for_online_cached_features\x18\x18 \x01(\x08R(skipOfflineWritesForOnlineCachedFeatures\x12-\n\x10result_bus_topic\x18\x19 \x01(\tH\x13R\x0eresultBusTopic\x88\x01\x01\x12;\n\x17online_persistence_mode\x18\x1a \x01(\tH\x14R\x15onlinePersistenceMode\x88\x01\x01\x12/\n\x11metrics_bus_topic\x18\x1b \x01(\tH\x15R\x0fmetricsBusTopic\x88\x01\x01\x12\x39\n\x16\x62igtable_instance_name\x18\x1c \x01(\tH\x16R\x14\x62igtableInstanceName\x88\x01\x01\x12\x33\n\x13\x62igtable_table_name\x18\x1d \x01(\tH\x17R\x11\x62igtableTableName\x88\x01\x01\x12\x37\n\x15\x63loud_account_locator\x18\x1e \x01(\tH\x18R\x13\x63loudAccountLocator\x88\x01\x01\x12&\n\x0c\x63loud_region\x18\x1f \x01(\tH\x19R\x0b\x63loudRegion\x88\x01\x01\x12-\n\x10\x63loud_tenancy_id\x18  \x01(\tH\x1aR\x0e\x63loudTenancyId\x88\x01\x01\x12\x35\n\x14source_bundle_bucket\x18! \x01(\tH\x1bR\x12sourceBundleBucket\x88\x01\x01\x12\x42\n\x1b\x65ngine_docker_registry_path\x18" \x01(\tH\x1cR\x18\x65ngineDockerRegistryPath\x88\x01\x01\x12,\n\x0f\x64\x65\x66\x61ult_planner\x18# \x01(\tH\x1dR\x0e\x64\x65\x66\x61ultPlanner\x88\x01\x01\x12\x63\n\x13\x61\x64\x64itional_env_vars\x18$ \x03(\x0b\x32\x33.chalk.server.v1.Environment.AdditionalEnvVarsEntryR\x11\x61\x64\x64itionalEnvVars\x12p\n\x18\x61\x64\x64itional_cron_env_vars\x18% \x03(\x0b\x32\x37.chalk.server.v1.Environment.AdditionalCronEnvVarsEntryR\x15\x61\x64\x64itionalCronEnvVars\x12=\n\x18private_pip_repositories\x18& \x01(\tH\x1eR\x16privatePipRepositories\x88\x01\x01\x12\x1d\n\nis_sandbox\x18\' \x01(\x08R\tisSandbox\x12I\n\x0e\x63loud_provider\x18) \x01(\x0e\x32".chalk.server.v1.CloudProviderKindR\rcloudProvider\x12\x44\n\x0c\x63loud_config\x18* \x01(\x0b\x32\x1c.chalk.server.v1.CloudConfigH\x1fR\x0b\x63loudConfig\x88\x01\x01\x12Z\n\x10spec_config_json\x18( \x03(\x0b\x32\x30.chalk.server.v1.Environment.SpecConfigJsonEntryR\x0especConfigJson\x12@\n\x0b\x61rchived_at\x18+ \x01(\x0b\x32\x1a.google.protobuf.TimestampH R\narchivedAt\x88\x01\x01\x12S\n$metadata_server_metrics_store_secret\x18, \x01(\tH!R metadataServerMetricsStoreSecret\x88\x01\x01\x12M\n!query_server_metrics_store_secret\x18- \x01(\tH"R\x1dqueryServerMetricsStoreSecret\x88\x01\x01\x12/\n\x11pinned_base_image\x18. \x01(\tH#R\x0fpinnedBaseImage\x88\x01\x01\x12\x31\n\x12\x63luster_gateway_id\x18/ \x01(\tH$R\x10\x63lusterGatewayId\x88\x01\x01\x12\x39\n\x16\x63luster_timescaledb_id\x18\x30 \x01(\tH%R\x14\x63lusterTimescaledbId\x88\x01\x01\x12T\n$background_persistence_deployment_id\x18\x31 \x01(\tH&R!backgroundPersistenceDeploymentId\x88\x01\x01\x12\x65\n\x13\x65nvironment_buckets\x18\x32 \x01(\x0b\x32/.chalk.server.v1.EnvironmentObjectStorageConfigH\'R\x12\x65nvironmentBuckets\x88\x01\x01\x12\x41\n\x1a\x63luster_timescaledb_secret\x18\x33 \x01(\tH(R\x18\x63lusterTimescaledbSecret\x88\x01\x01\x12+\n\x0fgrpc_engine_url\x18\x34 \x01(\tH)R\rgrpcEngineUrl\x88\x01\x01\x12/\n\x11kube_cluster_mode\x18\x35 \x01(\tH*R\x0fkubeClusterMode\x88\x01\x01\x12(\n\rdashboard_url\x18\x36 \x01(\tH+R\x0c\x64\x61shboardUrl\x88\x01\x01\x12+\n\x0fkube_cluster_id\x18\x37 \x01(\tH,R\rkubeClusterId\x88\x01\x01\x12\x1d\n\x07managed\x18\x38 \x01(\x08H-R\x07managed\x88\x01\x01\x12;\n\x17telemetry_deployment_id\x18\x39 \x01(\tH.R\x15telemetryDeploymentId\x88\x01\x01\x12\x42\n\x0csuspended_at\x18: \x01(\x0b\x32\x1a.google.protobuf.TimestampH/R\x0bsuspendedAt\x88\x01\x01\x12`\n\x15\x64\x65\x66\x61ult_build_profile\x18; \x01(\x0e\x32\'.chalk.server.v1.DeploymentBuildProfileH0R\x13\x64\x65\x66\x61ultBuildProfile\x88\x01\x01\x12H\n\x0evector_db_kind\x18< \x01(\x0e\x32\x1d.chalk.server.v1.VectorDBKindH1R\x0cvectorDbKind\x88\x01\x01\x12-\n\x10vector_db_secret\x18= \x01(\tH2R\x0evectorDbSecret\x88\x01\x01\x1a\x44\n\x16\x41\x64\x64itionalEnvVarsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1aH\n\x1a\x41\x64\x64itionalCronEnvVarsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1aY\n\x13SpecConfigJsonEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\x17\n\x15_active_deployment_idB\r\n\x0b_worker_urlB\x0e\n\x0c_service_urlB\r\n\x0b_branch_urlB\x17\n\x15_offline_store_secretB\x16\n\x14_online_store_secretB\x17\n\x15_feature_store_secretB\x12\n\x10_postgres_secretB\x14\n\x12_online_store_kindB\n\n\x08_emq_uriB\x15\n\x13_vpc_connector_nameB\x14\n\x12_kube_cluster_nameB\x1b\n\x19_branch_kube_cluster_nameB\x1b\n\x19_engine_kube_cluster_nameB"\n _shadow_engine_kube_cluster_nameB\x15\n\x13_kube_job_namespaceB\x19\n\x17_kube_preview_namespaceB\x1c\n\x1a_kube_service_account_nameB\x1e\n\x1c_streaming_query_service_uriB\x13\n\x11_result_bus_topicB\x1a\n\x18_online_persistence_modeB\x14\n\x12_metrics_bus_topicB\x19\n\x17_bigtable_instance_nameB\x16\n\x14_bigtable_table_nameB\x18\n\x16_cloud_account_locatorB\x0f\n\r_cloud_regionB\x13\n\x11_cloud_tenancy_idB\x17\n\x15_source_bundle_bucketB\x1e\n\x1c_engine_docker_registry_pathB\x12\n\x10_default_plannerB\x1b\n\x19_private_pip_repositoriesB\x0f\n\r_cloud_configB\x0e\n\x0c_archived_atB\'\n%_metadata_server_metrics_store_secretB$\n"_query_server_metrics_store_secretB\x14\n\x12_pinned_base_imageB\x15\n\x13_cluster_gateway_idB\x19\n\x17_cluster_timescaledb_idB\'\n%_background_persistence_deployment_idB\x16\n\x14_environment_bucketsB\x1d\n\x1b_cluster_timescaledb_secretB\x12\n\x10_grpc_engine_urlB\x14\n\x12_kube_cluster_modeB\x10\n\x0e_dashboard_urlB\x12\n\x10_kube_cluster_idB\n\n\x08_managedB\x1a\n\x18_telemetry_deployment_idB\x0f\n\r_suspended_atB\x18\n\x16_default_build_profileB\x11\n\x0f_vector_db_kindB\x13\n\x11_vector_db_secret*\xb2\x01\n\x11\x43loudProviderKind\x12#\n\x1f\x43LOUD_PROVIDER_KIND_UNSPECIFIED\x10\x00\x12\x1f\n\x1b\x43LOUD_PROVIDER_KIND_UNKNOWN\x10\x01\x12\x1b\n\x17\x43LOUD_PROVIDER_KIND_GCP\x10\x02\x12\x1b\n\x17\x43LOUD_PROVIDER_KIND_AWS\x10\x03\x12\x1d\n\x19\x43LOUD_PROVIDER_KIND_AZURE\x10\x04*\x85\x01\n\x0cVectorDBKind\x12\x1e\n\x1aVECTOR_DB_KIND_UNSPECIFIED\x10\x00\x12\x1d\n\x19VECTOR_DB_KIND_OPENSEARCH\x10\x01\x12\x1b\n\x17VECTOR_DB_KIND_PGVECTOR\x10\x02\x12\x19\n\x15VECTOR_DB_KIND_MILVUS\x10\x03*\xf4\x01\n\x16\x44\x65ploymentBuildProfile\x12(\n$DEPLOYMENT_BUILD_PROFILE_UNSPECIFIED\x10\x00\x12,\n(DEPLOYMENT_BUILD_PROFILE_O3_NO_PROFILING\x10\x01\x12)\n%DEPLOYMENT_BUILD_PROFILE_O3_PROFILING\x10\x02\x12,\n(DEPLOYMENT_BUILD_PROFILE_O2_NO_PROFILING\x10\x03\x12)\n%DEPLOYMENT_BUILD_PROFILE_O2_PROFILING\x10\x04\x42\x99\x01\n\x13\x63om.chalk.server.v1B\x10\x45nvironmentProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -43,10 +43,12 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_ENVIRONMENT_ADDITIONALCRONENVVARSENTRY"]._serialized_options = b"8\001"
     _globals["_ENVIRONMENT_SPECCONFIGJSONENTRY"]._options = None
     _globals["_ENVIRONMENT_SPECCONFIGJSONENTRY"]._serialized_options = b"8\001"
-    _globals["_CLOUDPROVIDERKIND"]._serialized_start = 8671
-    _globals["_CLOUDPROVIDERKIND"]._serialized_end = 8849
-    _globals["_DEPLOYMENTBUILDPROFILE"]._serialized_start = 8852
-    _globals["_DEPLOYMENTBUILDPROFILE"]._serialized_end = 9096
+    _globals["_CLOUDPROVIDERKIND"]._serialized_start = 8832
+    _globals["_CLOUDPROVIDERKIND"]._serialized_end = 9010
+    _globals["_VECTORDBKIND"]._serialized_start = 9013
+    _globals["_VECTORDBKIND"]._serialized_end = 9146
+    _globals["_DEPLOYMENTBUILDPROFILE"]._serialized_start = 9149
+    _globals["_DEPLOYMENTBUILDPROFILE"]._serialized_end = 9393
     _globals["_AWSCLOUDWATCHCONFIG"]._serialized_start = 117
     _globals["_AWSCLOUDWATCHCONFIG"]._serialized_end = 240
     _globals["_AWSSECRETMANAGERCONFIG"]._serialized_start = 243
@@ -80,11 +82,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_ENVIRONMENTOBJECTSTORAGECONFIG"]._serialized_start = 3617
     _globals["_ENVIRONMENTOBJECTSTORAGECONFIG"]._serialized_end = 3836
     _globals["_ENVIRONMENT"]._serialized_start = 3839
-    _globals["_ENVIRONMENT"]._serialized_end = 8668
-    _globals["_ENVIRONMENT_ADDITIONALENVVARSENTRY"]._serialized_start = 7250
-    _globals["_ENVIRONMENT_ADDITIONALENVVARSENTRY"]._serialized_end = 7318
-    _globals["_ENVIRONMENT_ADDITIONALCRONENVVARSENTRY"]._serialized_start = 7320
-    _globals["_ENVIRONMENT_ADDITIONALCRONENVVARSENTRY"]._serialized_end = 7392
-    _globals["_ENVIRONMENT_SPECCONFIGJSONENTRY"]._serialized_start = 7394
-    _globals["_ENVIRONMENT_SPECCONFIGJSONENTRY"]._serialized_end = 7483
+    _globals["_ENVIRONMENT"]._serialized_end = 8829
+    _globals["_ENVIRONMENT_ADDITIONALENVVARSENTRY"]._serialized_start = 7371
+    _globals["_ENVIRONMENT_ADDITIONALENVVARSENTRY"]._serialized_end = 7439
+    _globals["_ENVIRONMENT_ADDITIONALCRONENVVARSENTRY"]._serialized_start = 7441
+    _globals["_ENVIRONMENT_ADDITIONALCRONENVVARSENTRY"]._serialized_end = 7513
+    _globals["_ENVIRONMENT_SPECCONFIGJSONENTRY"]._serialized_start = 7515
+    _globals["_ENVIRONMENT_SPECCONFIGJSONENTRY"]._serialized_end = 7604
 # @@protoc_insertion_point(module_scope)

{chalkpy-2.94.6 → chalkpy-2.94.8}/chalk/_gen/chalk/server/v1/environment_pb2.pyi

@@ -22,6 +22,13 @@ class CloudProviderKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     CLOUD_PROVIDER_KIND_AWS: _ClassVar[CloudProviderKind]
     CLOUD_PROVIDER_KIND_AZURE: _ClassVar[CloudProviderKind]
 
+class VectorDBKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    VECTOR_DB_KIND_UNSPECIFIED: _ClassVar[VectorDBKind]
+    VECTOR_DB_KIND_OPENSEARCH: _ClassVar[VectorDBKind]
+    VECTOR_DB_KIND_PGVECTOR: _ClassVar[VectorDBKind]
+    VECTOR_DB_KIND_MILVUS: _ClassVar[VectorDBKind]
+
 class DeploymentBuildProfile(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
     DEPLOYMENT_BUILD_PROFILE_UNSPECIFIED: _ClassVar[DeploymentBuildProfile]
@@ -35,6 +42,10 @@ CLOUD_PROVIDER_KIND_UNKNOWN: CloudProviderKind
 CLOUD_PROVIDER_KIND_GCP: CloudProviderKind
 CLOUD_PROVIDER_KIND_AWS: CloudProviderKind
 CLOUD_PROVIDER_KIND_AZURE: CloudProviderKind
+VECTOR_DB_KIND_UNSPECIFIED: VectorDBKind
+VECTOR_DB_KIND_OPENSEARCH: VectorDBKind
+VECTOR_DB_KIND_PGVECTOR: VectorDBKind
+VECTOR_DB_KIND_MILVUS: VectorDBKind
 DEPLOYMENT_BUILD_PROFILE_UNSPECIFIED: DeploymentBuildProfile
 DEPLOYMENT_BUILD_PROFILE_O3_NO_PROFILING: DeploymentBuildProfile
 DEPLOYMENT_BUILD_PROFILE_O3_PROFILING: DeploymentBuildProfile
@@ -386,6 +397,8 @@ class Environment(_message.Message):
         "telemetry_deployment_id",
         "suspended_at",
         "default_build_profile",
+        "vector_db_kind",
+        "vector_db_secret",
     )
     class AdditionalEnvVarsEntry(_message.Message):
         __slots__ = ("key", "value")
@@ -472,6 +485,8 @@ class Environment(_message.Message):
     TELEMETRY_DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
     SUSPENDED_AT_FIELD_NUMBER: _ClassVar[int]
     DEFAULT_BUILD_PROFILE_FIELD_NUMBER: _ClassVar[int]
+    VECTOR_DB_KIND_FIELD_NUMBER: _ClassVar[int]
+    VECTOR_DB_SECRET_FIELD_NUMBER: _ClassVar[int]
     name: str
     project_id: str
     id: str
@@ -531,6 +546,8 @@ class Environment(_message.Message):
     telemetry_deployment_id: str
     suspended_at: _timestamp_pb2.Timestamp
     default_build_profile: DeploymentBuildProfile
+    vector_db_kind: VectorDBKind
+    vector_db_secret: str
     def __init__(
         self,
         name: _Optional[str] = ...,
@@ -592,4 +609,6 @@ class Environment(_message.Message):
         telemetry_deployment_id: _Optional[str] = ...,
         suspended_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
         default_build_profile: _Optional[_Union[DeploymentBuildProfile, str]] = ...,
+        vector_db_kind: _Optional[_Union[VectorDBKind, str]] = ...,
+        vector_db_secret: _Optional[str] = ...,
     ) -> None: ...

{chalkpy-2.94.6 → chalkpy-2.94.8}/chalk/_gen/chalk/server/v1/scheduler_pb2.py

@@ -21,7 +21,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1f\x63halk/server/v1/scheduler.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1b\x63halk/server/v1/batch.proto\x1a)chalk/server/v1/scheduled_query_run.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xfe\x05\n\x0f\x43ronResolverRun\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12!\n\x0cresolver_fqn\x18\x03 \x01(\tR\x0bresolverFqn\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12+\n\x11schedule_readable\x18\x05 \x01(\tR\x10scheduleReadable\x12\x36\n\x17schedule_resolver_value\x18\x06 \x01(\tR\x15scheduleResolverValue\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12(\n\rdeployment_id\x18\x08 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x31\n\x03\x65nd\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x03\x65nd\x88\x01\x01\x12:\n\x05\x62\x61tch\x18\n \x01(\x0b\x32\x1f.chalk.server.v1.BatchOperationH\x02R\x05\x62\x61tch\x88\x01\x01\x12\x46\n\x0ctrigger_kind\x18\x0b \x01(\x0e\x32#.chalk.server.v1.CronRunTriggerKindR\x0btriggerKind\x12@\n\x0blower_bound\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\nupperBound\x88\x01\x01\x12$\n\x0bmax_samples\x18\x0e \x01(\x03H\x05R\nmaxSamples\x88\x01\x01\x42\x10\n\x0e_deployment_idB\x06\n\x04_endB\x08\n\x06_batchB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x0e\n\x0c_max_samples"\x9d\x03\n ManualTriggerCronResolverRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12$\n\x0bmax_samples\x18\x02 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12@\n\x0blower_bound\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nupperBound\x88\x01\x01\x12\x30\n\x11timestamping_mode\x18\x06 \x01(\tH\x03R\x10timestampingMode\x88\x01\x01\x12$\n\x0bjob_options\x18\x07 \x01(\tH\x04R\njobOptions\x88\x01\x01\x42\x0e\n\x0c_max_samplesB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x14\n\x12_timestamping_modeB\x0e\n\x0c_job_options"q\n!ManualTriggerCronResolverResponse\x12L\n\x11\x63ron_resolver_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x0f\x63ronResolverRun"\xf0\x04\n"ManualTriggerScheduledQueryRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x03R\x0b\x63ronQueryId\x12p\n\x0fplanner_options\x18\x02 \x03(\x0b\x32G.chalk.server.v1.ManualTriggerScheduledQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\x33\n\x15incremental_resolvers\x18\x03 \x03(\tR\x14incrementalResolvers\x12$\n\x0bmax_samples\x18\x04 \x01(\x03H\x01R\nmaxSamples\x88\x01\x01\x12j\n\renv_overrides\x18\x05 \x03(\x0b\x32\x45.chalk.server.v1.ManualTriggerScheduledQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12(\n\x0f\x63ron_query_name\x18\x06 \x01(\tH\x00R\rcronQueryName\x1aY\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x17\n\x15\x63ron_query_identifierB\x0e\n\x0c_max_samples"y\n#ManualTriggerScheduledQueryResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun"0\n\x1eGetScheduledResolverRunRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"U\n\x1fGetScheduledResolverRunResponse\x12\x32\n\x03run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x03run"\xa7\x03\n ListScheduledResolverRunsRequest\x12\x37\n\x06\x63ursor\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x06\x63ursor\x88\x01\x01\x12,\n\x0fresolver_filter\x18\x02 \x01(\tH\x01R\x0eresolverFilter\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x12\x35\n\x05start\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\x05start\x88\x01\x01\x12\x31\n\x03\x65nd\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\x03\x65nd\x88\x01\x01\x12J\n\rstatus_filter\x18\x06 \x01(\x0e\x32 .chalk.server.v1.OperationStatusH\x05R\x0cstatusFilter\x88\x01\x01\x42\t\n\x07_cursorB\x12\n\x10_resolver_filterB\x08\n\x06_limitB\x08\n\x06_startB\x06\n\x04_endB\x10\n\x0e_status_filter"Y\n!ListScheduledResolverRunsResponse\x12\x34\n\x04runs\x18\x01 \x03(\x0b\x32 .chalk.server.v1.CronResolverRunR\x04runs*\x9c\x01\n\x12\x43ronRunTriggerKind\x12%\n!CRON_RUN_TRIGGER_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x43RON_RUN_TRIGGER_KIND_MANUAL\x10\x01\x12\x1d\n\x19\x43RON_RUN_TRIGGER_KIND_API\x10\x02\x12\x1e\n\x1a\x43RON_RUN_TRIGGER_KIND_CRON\x10\x03\x32\xba\x04\n\x10SchedulerService\x12\x87\x01\n\x19ManualTriggerCronResolver\x12\x31.chalk.server.v1.ManualTriggerCronResolverRequest\x1a\x32.chalk.server.v1.ManualTriggerCronResolverResponse"\x03\x80}\x11\x12\x8d\x01\n\x1bManualTriggerScheduledQuery\x12\x33.chalk.server.v1.ManualTriggerScheduledQueryRequest\x1a\x34.chalk.server.v1.ManualTriggerScheduledQueryResponse"\x03\x80}\x11\x12\x81\x01\n\x17GetScheduledResolverRun\x12/.chalk.server.v1.GetScheduledResolverRunRequest\x1a\x30.chalk.server.v1.GetScheduledResolverRunResponse"\x03\x80}\x10\x12\x87\x01\n\x19ListScheduledResolverRuns\x12\x31.chalk.server.v1.ListScheduledResolverRunsRequest\x1a\x32.chalk.server.v1.ListScheduledResolverRunsResponse"\x03\x80}\x10\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0eSchedulerProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
+    b'\n\x1f\x63halk/server/v1/scheduler.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1b\x63halk/server/v1/batch.proto\x1a)chalk/server/v1/scheduled_query_run.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xfe\x05\n\x0f\x43ronResolverRun\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12!\n\x0cresolver_fqn\x18\x03 \x01(\tR\x0bresolverFqn\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12+\n\x11schedule_readable\x18\x05 \x01(\tR\x10scheduleReadable\x12\x36\n\x17schedule_resolver_value\x18\x06 \x01(\tR\x15scheduleResolverValue\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12(\n\rdeployment_id\x18\x08 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x31\n\x03\x65nd\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x03\x65nd\x88\x01\x01\x12:\n\x05\x62\x61tch\x18\n \x01(\x0b\x32\x1f.chalk.server.v1.BatchOperationH\x02R\x05\x62\x61tch\x88\x01\x01\x12\x46\n\x0ctrigger_kind\x18\x0b \x01(\x0e\x32#.chalk.server.v1.CronRunTriggerKindR\x0btriggerKind\x12@\n\x0blower_bound\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\nupperBound\x88\x01\x01\x12$\n\x0bmax_samples\x18\x0e \x01(\x03H\x05R\nmaxSamples\x88\x01\x01\x42\x10\n\x0e_deployment_idB\x06\n\x04_endB\x08\n\x06_batchB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x0e\n\x0c_max_samples"\x9d\x03\n ManualTriggerCronResolverRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12$\n\x0bmax_samples\x18\x02 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12@\n\x0blower_bound\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nupperBound\x88\x01\x01\x12\x30\n\x11timestamping_mode\x18\x06 \x01(\tH\x03R\x10timestampingMode\x88\x01\x01\x12$\n\x0bjob_options\x18\x07 \x01(\tH\x04R\njobOptions\x88\x01\x01\x42\x0e\n\x0c_max_samplesB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x14\n\x12_timestamping_modeB\x0e\n\x0c_job_options"q\n!ManualTriggerCronResolverResponse\x12L\n\x11\x63ron_resolver_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x0f\x63ronResolverRun"\x9c\x05\n"ManualTriggerScheduledQueryRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x03R\x0b\x63ronQueryId\x12p\n\x0fplanner_options\x18\x02 \x03(\x0b\x32G.chalk.server.v1.ManualTriggerScheduledQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\x33\n\x15incremental_resolvers\x18\x03 \x03(\tR\x14incrementalResolvers\x12$\n\x0bmax_samples\x18\x04 \x01(\x03H\x01R\nmaxSamples\x88\x01\x01\x12j\n\renv_overrides\x18\x05 \x03(\x0b\x32\x45.chalk.server.v1.ManualTriggerScheduledQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12(\n\x0f\x63ron_query_name\x18\x06 \x01(\tH\x00R\rcronQueryName\x12*\n\x11store_plan_stages\x18\x07 \x01(\x08R\x0fstorePlanStages\x1aY\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x17\n\x15\x63ron_query_identifierB\x0e\n\x0c_max_samples"y\n#ManualTriggerScheduledQueryResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun"0\n\x1eGetScheduledResolverRunRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"U\n\x1fGetScheduledResolverRunResponse\x12\x32\n\x03run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x03run"\xa7\x03\n ListScheduledResolverRunsRequest\x12\x37\n\x06\x63ursor\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x06\x63ursor\x88\x01\x01\x12,\n\x0fresolver_filter\x18\x02 \x01(\tH\x01R\x0eresolverFilter\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x12\x35\n\x05start\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\x05start\x88\x01\x01\x12\x31\n\x03\x65nd\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\x03\x65nd\x88\x01\x01\x12J\n\rstatus_filter\x18\x06 \x01(\x0e\x32 .chalk.server.v1.OperationStatusH\x05R\x0cstatusFilter\x88\x01\x01\x42\t\n\x07_cursorB\x12\n\x10_resolver_filterB\x08\n\x06_limitB\x08\n\x06_startB\x06\n\x04_endB\x10\n\x0e_status_filter"Y\n!ListScheduledResolverRunsResponse\x12\x34\n\x04runs\x18\x01 \x03(\x0b\x32 .chalk.server.v1.CronResolverRunR\x04runs*\x9c\x01\n\x12\x43ronRunTriggerKind\x12%\n!CRON_RUN_TRIGGER_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x43RON_RUN_TRIGGER_KIND_MANUAL\x10\x01\x12\x1d\n\x19\x43RON_RUN_TRIGGER_KIND_API\x10\x02\x12\x1e\n\x1a\x43RON_RUN_TRIGGER_KIND_CRON\x10\x03\x32\xba\x04\n\x10SchedulerService\x12\x87\x01\n\x19ManualTriggerCronResolver\x12\x31.chalk.server.v1.ManualTriggerCronResolverRequest\x1a\x32.chalk.server.v1.ManualTriggerCronResolverResponse"\x03\x80}\x11\x12\x8d\x01\n\x1bManualTriggerScheduledQuery\x12\x33.chalk.server.v1.ManualTriggerScheduledQueryRequest\x1a\x34.chalk.server.v1.ManualTriggerScheduledQueryResponse"\x03\x80}\x11\x12\x81\x01\n\x17GetScheduledResolverRun\x12/.chalk.server.v1.GetScheduledResolverRunRequest\x1a\x30.chalk.server.v1.GetScheduledResolverRunResponse"\x03\x80}\x10\x12\x87\x01\n\x19ListScheduledResolverRuns\x12\x31.chalk.server.v1.ListScheduledResolverRunsRequest\x1a\x32.chalk.server.v1.ListScheduledResolverRunsResponse"\x03\x80}\x10\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0eSchedulerProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -44,8 +44,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SCHEDULERSERVICE"].methods_by_name["GetScheduledResolverRun"]._serialized_options = b"\200}\020"
     _globals["_SCHEDULERSERVICE"].methods_by_name["ListScheduledResolverRuns"]._options = None
     _globals["_SCHEDULERSERVICE"].methods_by_name["ListScheduledResolverRuns"]._serialized_options = b"\200}\020"
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 2925
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 3081
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 2969
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 3125
     _globals["_CRONRESOLVERRUN"]._serialized_start = 221
     _globals["_CRONRESOLVERRUN"]._serialized_end = 987
     _globals["_MANUALTRIGGERCRONRESOLVERREQUEST"]._serialized_start = 990
@@ -53,21 +53,21 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_MANUALTRIGGERCRONRESOLVERRESPONSE"]._serialized_start = 1405
     _globals["_MANUALTRIGGERCRONRESOLVERRESPONSE"]._serialized_end = 1518
     _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST"]._serialized_start = 1521
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST"]._serialized_end = 2145
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_start = 1950
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_end = 2039
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 2041
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 2104
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_start = 2147
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_end = 2268
-    _globals["_GETSCHEDULEDRESOLVERRUNREQUEST"]._serialized_start = 2270
-    _globals["_GETSCHEDULEDRESOLVERRUNREQUEST"]._serialized_end = 2318
-    _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_start = 2320
-    _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_end = 2405
-    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_start = 2408
-    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_end = 2831
-    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_start = 2833
-    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_end = 2922
-    _globals["_SCHEDULERSERVICE"]._serialized_start = 3084
-    _globals["_SCHEDULERSERVICE"]._serialized_end = 3654
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST"]._serialized_end = 2189
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_start = 1994
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_end = 2083
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 2085
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 2148
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_start = 2191
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_end = 2312
+    _globals["_GETSCHEDULEDRESOLVERRUNREQUEST"]._serialized_start = 2314
+    _globals["_GETSCHEDULEDRESOLVERRUNREQUEST"]._serialized_end = 2362
+    _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_start = 2364
+    _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_end = 2449
+    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_start = 2452
+    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_end = 2875
+    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_start = 2877
+    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_end = 2966
+    _globals["_SCHEDULERSERVICE"]._serialized_start = 3128
+    _globals["_SCHEDULERSERVICE"]._serialized_end = 3698
 # @@protoc_insertion_point(module_scope)

{chalkpy-2.94.6 → chalkpy-2.94.8}/chalk/_gen/chalk/server/v1/scheduler_pb2.pyi

@@ -130,6 +130,7 @@ class ManualTriggerScheduledQueryRequest(_message.Message):
         "max_samples",
         "env_overrides",
         "cron_query_name",
+        "store_plan_stages",
     )
     class PlannerOptionsEntry(_message.Message):
         __slots__ = ("key", "value")
@@ -155,12 +156,14 @@ class ManualTriggerScheduledQueryRequest(_message.Message):
     MAX_SAMPLES_FIELD_NUMBER: _ClassVar[int]
     ENV_OVERRIDES_FIELD_NUMBER: _ClassVar[int]
     CRON_QUERY_NAME_FIELD_NUMBER: _ClassVar[int]
+    STORE_PLAN_STAGES_FIELD_NUMBER: _ClassVar[int]
     cron_query_id: int
     planner_options: _containers.MessageMap[str, _struct_pb2.Value]
     incremental_resolvers: _containers.RepeatedScalarFieldContainer[str]
     max_samples: int
     env_overrides: _containers.ScalarMap[str, str]
     cron_query_name: str
+    store_plan_stages: bool
     def __init__(
         self,
         cron_query_id: _Optional[int] = ...,
@@ -169,6 +172,7 @@ class ManualTriggerScheduledQueryRequest(_message.Message):
         max_samples: _Optional[int] = ...,
         env_overrides: _Optional[_Mapping[str, str]] = ...,
         cron_query_name: _Optional[str] = ...,
+        store_plan_stages: bool = ...,
     ) -> None: ...
 
 class ManualTriggerScheduledQueryResponse(_message.Message):