chalkpy 2.94.5__py3-none-any.whl → 2.94.6__py3-none-any.whl

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

@@ -18,7 +18,7 @@ from chalk._gen.chalk.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v
 
 
 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!ExecuteSqlSyncQueryRequestOptions"$\n"ExecuteSqlAsyncQueryRequestOptions"\xb1\x02\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\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"O\n"ExecuteSqlSyncQueryResponsePayload\x12)\n\x10parquet_response\x18\x01 \x01(\x0cR\x0fparquetResponse"\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*\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\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"#\n!ExecuteSqlSyncQueryRequestOptions"$\n"ExecuteSqlAsyncQueryRequestOptions"?\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*\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\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'
 )
 
 _globals = globals()
@@ -39,40 +39,44 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _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 = 2431
-    _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_end = 2604
+    _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_start = 2742
+    _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_end = 2915
     _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_start = 126
     _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_end = 161
     _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_start = 163
     _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_end = 199
-    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 202
-    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 507
-    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 509
-    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 588
-    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 591
-    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 751
-    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 754
-    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 1085
-    _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 1087
-    _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 1130
-    _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 1132
-    _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 1242
-    _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 1244
-    _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 1266
-    _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 1268
-    _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 1381
-    _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 1384
-    _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 1592
-    _globals["_DBSCHEMAINFO"]._serialized_start = 1594
-    _globals["_DBSCHEMAINFO"]._serialized_end = 1681
-    _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 1684
-    _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 1818
-    _globals["_GETTABLESREQUEST"]._serialized_start = 1821
-    _globals["_GETTABLESREQUEST"]._serialized_end = 2108
-    _globals["_TABLEINFO"]._serialized_start = 2111
-    _globals["_TABLEINFO"]._serialized_end = 2300
-    _globals["_GETTABLESRESPONSE"]._serialized_start = 2302
-    _globals["_GETTABLESRESPONSE"]._serialized_end = 2428
-    _globals["_SQLSERVICE"]._serialized_start = 2607
-    _globals["_SQLSERVICE"]._serialized_end = 3132
+    _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_start = 201
+    _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_end = 264
+    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 267
+    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 679
+    _globals["_SIGNEDOUTPUTURIS"]._serialized_start = 681
+    _globals["_SIGNEDOUTPUTURIS"]._serialized_end = 719
+    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 722
+    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 899
+    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 902
+    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 1062
+    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 1065
+    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 1396
+    _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 1398
+    _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 1441
+    _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 1443
+    _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 1553
+    _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 1555
+    _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 1577
+    _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 1579
+    _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 1692
+    _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 1695
+    _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 1903
+    _globals["_DBSCHEMAINFO"]._serialized_start = 1905
+    _globals["_DBSCHEMAINFO"]._serialized_end = 1992
+    _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 1995
+    _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 2129
+    _globals["_GETTABLESREQUEST"]._serialized_start = 2132
+    _globals["_GETTABLESREQUEST"]._serialized_end = 2419
+    _globals["_TABLEINFO"]._serialized_start = 2422
+    _globals["_TABLEINFO"]._serialized_end = 2611
+    _globals["_GETTABLESRESPONSE"]._serialized_start = 2613
+    _globals["_GETTABLESRESPONSE"]._serialized_end = 2739
+    _globals["_SQLSERVICE"]._serialized_start = 2918
+    _globals["_SQLSERVICE"]._serialized_end = 3443
 # @@protoc_insertion_point(module_scope)

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

@@ -32,29 +32,50 @@ class ExecuteSqlAsyncQueryRequestOptions(_message.Message):
     __slots__ = ()
     def __init__(self) -> 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", "correlation_id", "sync_options", "async_options")
+    __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
     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",)
+    __slots__ = ("parquet_response", "signed_output_uris")
     PARQUET_RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    SIGNED_OUTPUT_URIS_FIELD_NUMBER: _ClassVar[int]
     parquet_response: bytes
-    def __init__(self, parquet_response: _Optional[bytes] = ...) -> None: ...
+    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")

chalk/_gen/chalk/server/v1/benchmark_pb2.py

@@ -16,10 +16,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 online_query_pb2 as chalk_dot_common_dot_v1_dot_online__query__pb2
 from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1f\x63halk/server/v1/benchmark.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a"chalk/common/v1/online_query.proto\x1a\x1egoogle/protobuf/duration.proto"n\n\x1cSimpleOnlineQueryBulkRequest\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures\x12\'\n\x0foutput_features\x18\x02 \x03(\tR\x0eoutputFeatures"\xfc\x02\n\x16\x43reateBenchmarkRequest\x12\x1d\n\nwarmup_qps\x18\x01 \x01(\x03R\twarmupQps\x12\x42\n\x0fwarmup_duration\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0ewarmupDuration\x12\x10\n\x03qps\x18\x03 \x01(\x03R\x03qps\x12\x35\n\x08\x64uration\x18\x04 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12H\n\nquery_bulk\x18\x05 \x01(\x0b\x32\'.chalk.common.v1.OnlineQueryBulkRequestH\x00R\tqueryBulk\x12[\n\x11simple_query_bulk\x18\x06 \x01(\x0b\x32-.chalk.server.v1.SimpleOnlineQueryBulkRequestH\x00R\x0fsimpleQueryBulkB\x0f\n\rquery_request"S\n\x17\x43reateBenchmarkResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32 .chalk.server.v1.BenchmarkStatusR\x06status*\xe8\x01\n\x0f\x42\x65nchmarkStatus\x12 \n\x1c\x42\x45NCHMARK_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_QUEUED\x10\x01\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_WORKING\x10\x02\x12\x1e\n\x1a\x42\x45NCHMARK_STATUS_COMPLETED\x10\x03\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_FAILED\x10\x04\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_SKIPPED\x10\x05\x12\x1d\n\x19\x42\x45NCHMARK_STATUS_CANCELED\x10\x06\x32}\n\x10\x42\x65nchmarkService\x12i\n\x0f\x43reateBenchmark\x12\'.chalk.server.v1.CreateBenchmarkRequest\x1a(.chalk.server.v1.CreateBenchmarkResponse"\x03\x80}\x03\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0e\x42\x65nchmarkProtoP\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/benchmark.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a"chalk/common/v1/online_query.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"6\n\rInputFeatures\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures"\xef\x01\n\x1cSimpleOnlineQueryBulkRequest\x12)\n\x0einput_features\x18\x01 \x03(\tB\x02\x18\x01R\rinputFeatures\x12P\n\x13input_features_list\x18\x03 \x01(\x0b\x32\x1e.chalk.server.v1.InputFeaturesH\x00R\x11inputFeaturesList\x12\x1f\n\ninput_file\x18\x04 \x01(\tH\x00R\tinputFile\x12\'\n\x0foutput_features\x18\x02 \x03(\tR\x0eoutputFeaturesB\x08\n\x06inputs"\xfc\x02\n\x16\x43reateBenchmarkRequest\x12\x1d\n\nwarmup_qps\x18\x01 \x01(\x03R\twarmupQps\x12\x42\n\x0fwarmup_duration\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0ewarmupDuration\x12\x10\n\x03qps\x18\x03 \x01(\x03R\x03qps\x12\x35\n\x08\x64uration\x18\x04 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12H\n\nquery_bulk\x18\x05 \x01(\x0b\x32\'.chalk.common.v1.OnlineQueryBulkRequestH\x00R\tqueryBulk\x12[\n\x11simple_query_bulk\x18\x06 \x01(\x0b\x32-.chalk.server.v1.SimpleOnlineQueryBulkRequestH\x00R\x0fsimpleQueryBulkB\x0f\n\rquery_request"v\n\x17\x43reateBenchmarkResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32 .chalk.server.v1.BenchmarkStatusR\x06status\x12!\n\x0c\x62\x65nchmark_id\x18\x02 \x01(\tR\x0b\x62\x65nchmarkId"\x99\x01\n\x12\x42\x65nchmarkInputFile\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x17\n\x04size\x18\x02 \x01(\x03H\x00R\x04size\x88\x01\x01\x12>\n\nupdated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tupdatedAt\x88\x01\x01\x42\x07\n\x05_sizeB\r\n\x0b_updated_at"7\n\x1dGetAvailableInputFilesRequest\x12\x16\n\x06prefix\x18\x01 \x01(\tR\x06prefix"f\n\x1eGetAvailableInputFilesResponse\x12\x44\n\x0binput_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\ninputFiles"e\n\x1dGetInputFileUploadUrlsRequest\x12\x44\n\x0binput_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\ninputFiles"G\n\x1eGetInputFileUploadUrlsResponse\x12%\n\x0epresigned_urls\x18\x01 \x03(\tR\rpresignedUrls"8\n\x1eGetAvailableResultFilesRequest\x12\x16\n\x06prefix\x18\x01 \x01(\tR\x06prefix"i\n\x1fGetAvailableResultFilesResponse\x12\x46\n\x0cresult_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\x0bresultFiles"9\n\x18GetResultFileUrlsRequest\x12\x1d\n\nfile_names\x18\x01 \x03(\tR\tfileNames"B\n\x19GetResultFileUrlsResponse\x12%\n\x0epresigned_urls\x18\x01 \x03(\tR\rpresignedUrls*\xe8\x01\n\x0f\x42\x65nchmarkStatus\x12 \n\x1c\x42\x45NCHMARK_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_QUEUED\x10\x01\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_WORKING\x10\x02\x12\x1e\n\x1a\x42\x45NCHMARK_STATUS_COMPLETED\x10\x03\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_FAILED\x10\x04\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_SKIPPED\x10\x05\x12\x1d\n\x19\x42\x45NCHMARK_STATUS_CANCELED\x10\x06\x32\xf2\x04\n\x10\x42\x65nchmarkService\x12i\n\x0f\x43reateBenchmark\x12\'.chalk.server.v1.CreateBenchmarkRequest\x1a(.chalk.server.v1.CreateBenchmarkResponse"\x03\x80}\x03\x12~\n\x16GetAvailableInputFiles\x12..chalk.server.v1.GetAvailableInputFilesRequest\x1a/.chalk.server.v1.GetAvailableInputFilesResponse"\x03\x80}\x03\x12~\n\x16GetInputFileUploadUrls\x12..chalk.server.v1.GetInputFileUploadUrlsRequest\x1a/.chalk.server.v1.GetInputFileUploadUrlsResponse"\x03\x80}\x03\x12\x81\x01\n\x17GetAvailableResultFiles\x12/.chalk.server.v1.GetAvailableResultFilesRequest\x1a\x30.chalk.server.v1.GetAvailableResultFilesResponse"\x03\x80}\x03\x12o\n\x11GetResultFileUrls\x12).chalk.server.v1.GetResultFileUrlsRequest\x1a*.chalk.server.v1.GetResultFileUrlsResponse"\x03\x80}\x03\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0e\x42\x65nchmarkProtoP\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()
@@ -30,16 +31,46 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals[
         "DESCRIPTOR"
     ]._serialized_options = b"\n\023com.chalk.server.v1B\016BenchmarkProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"].fields_by_name["input_features"]._options = None
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"].fields_by_name["input_features"]._serialized_options = b"\030\001"
     _globals["_BENCHMARKSERVICE"].methods_by_name["CreateBenchmark"]._options = None
     _globals["_BENCHMARKSERVICE"].methods_by_name["CreateBenchmark"]._serialized_options = b"\200}\003"
-    _globals["_BENCHMARKSTATUS"]._serialized_start = 734
-    _globals["_BENCHMARKSTATUS"]._serialized_end = 966
-    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_start = 153
-    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_end = 263
-    _globals["_CREATEBENCHMARKREQUEST"]._serialized_start = 266
-    _globals["_CREATEBENCHMARKREQUEST"]._serialized_end = 646
-    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_start = 648
-    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_end = 731
-    _globals["_BENCHMARKSERVICE"]._serialized_start = 968
-    _globals["_BENCHMARKSERVICE"]._serialized_end = 1093
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableInputFiles"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableInputFiles"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetInputFileUploadUrls"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetInputFileUploadUrls"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableResultFiles"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableResultFiles"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._options = None
+    _globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._serialized_options = b"\200}\003"
+    _globals["_BENCHMARKSTATUS"]._serialized_start = 1773
+    _globals["_BENCHMARKSTATUS"]._serialized_end = 2005
+    _globals["_INPUTFEATURES"]._serialized_start = 186
+    _globals["_INPUTFEATURES"]._serialized_end = 240
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_start = 243
+    _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_end = 482
+    _globals["_CREATEBENCHMARKREQUEST"]._serialized_start = 485
+    _globals["_CREATEBENCHMARKREQUEST"]._serialized_end = 865
+    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_start = 867
+    _globals["_CREATEBENCHMARKRESPONSE"]._serialized_end = 985
+    _globals["_BENCHMARKINPUTFILE"]._serialized_start = 988
+    _globals["_BENCHMARKINPUTFILE"]._serialized_end = 1141
+    _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_start = 1143
+    _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_end = 1198
+    _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_start = 1200
+    _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_end = 1302
+    _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_start = 1304
+    _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_end = 1405
+    _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_start = 1407
+    _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_end = 1478
+    _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_start = 1480
+    _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_end = 1536
+    _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_start = 1538
+    _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_end = 1643
+    _globals["_GETRESULTFILEURLSREQUEST"]._serialized_start = 1645
+    _globals["_GETRESULTFILEURLSREQUEST"]._serialized_end = 1702
+    _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_start = 1704
+    _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_end = 1770
+    _globals["_BENCHMARKSERVICE"]._serialized_start = 2008
+    _globals["_BENCHMARKSERVICE"]._serialized_end = 2634
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/benchmark_pb2.pyi

@@ -1,6 +1,7 @@
 from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
 from chalk._gen.chalk.common.v1 import online_query_pb2 as _online_query_pb2
 from google.protobuf import duration_pb2 as _duration_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
@@ -33,14 +34,28 @@ BENCHMARK_STATUS_FAILED: BenchmarkStatus
 BENCHMARK_STATUS_SKIPPED: BenchmarkStatus
 BENCHMARK_STATUS_CANCELED: BenchmarkStatus
 
+class InputFeatures(_message.Message):
+    __slots__ = ("input_features",)
+    INPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    input_features: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, input_features: _Optional[_Iterable[str]] = ...) -> None: ...
+
 class SimpleOnlineQueryBulkRequest(_message.Message):
-    __slots__ = ("input_features", "output_features")
+    __slots__ = ("input_features", "input_features_list", "input_file", "output_features")
     INPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    INPUT_FEATURES_LIST_FIELD_NUMBER: _ClassVar[int]
+    INPUT_FILE_FIELD_NUMBER: _ClassVar[int]
     OUTPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
     input_features: _containers.RepeatedScalarFieldContainer[str]
+    input_features_list: InputFeatures
+    input_file: str
     output_features: _containers.RepeatedScalarFieldContainer[str]
     def __init__(
-        self, input_features: _Optional[_Iterable[str]] = ..., output_features: _Optional[_Iterable[str]] = ...
+        self,
+        input_features: _Optional[_Iterable[str]] = ...,
+        input_features_list: _Optional[_Union[InputFeatures, _Mapping]] = ...,
+        input_file: _Optional[str] = ...,
+        output_features: _Optional[_Iterable[str]] = ...,
     ) -> None: ...
 
 class CreateBenchmarkRequest(_message.Message):
@@ -68,7 +83,74 @@ class CreateBenchmarkRequest(_message.Message):
     ) -> None: ...
 
 class CreateBenchmarkResponse(_message.Message):
-    __slots__ = ("status",)
+    __slots__ = ("status", "benchmark_id")
     STATUS_FIELD_NUMBER: _ClassVar[int]
+    BENCHMARK_ID_FIELD_NUMBER: _ClassVar[int]
     status: BenchmarkStatus
-    def __init__(self, status: _Optional[_Union[BenchmarkStatus, str]] = ...) -> None: ...
+    benchmark_id: str
+    def __init__(
+        self, status: _Optional[_Union[BenchmarkStatus, str]] = ..., benchmark_id: _Optional[str] = ...
+    ) -> None: ...
+
+class BenchmarkInputFile(_message.Message):
+    __slots__ = ("name", "size", "updated_at")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    SIZE_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    size: int
+    updated_at: _timestamp_pb2.Timestamp
+    def __init__(
+        self,
+        name: _Optional[str] = ...,
+        size: _Optional[int] = ...,
+        updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+    ) -> None: ...
+
+class GetAvailableInputFilesRequest(_message.Message):
+    __slots__ = ("prefix",)
+    PREFIX_FIELD_NUMBER: _ClassVar[int]
+    prefix: str
+    def __init__(self, prefix: _Optional[str] = ...) -> None: ...
+
+class GetAvailableInputFilesResponse(_message.Message):
+    __slots__ = ("input_files",)
+    INPUT_FILES_FIELD_NUMBER: _ClassVar[int]
+    input_files: _containers.RepeatedCompositeFieldContainer[BenchmarkInputFile]
+    def __init__(self, input_files: _Optional[_Iterable[_Union[BenchmarkInputFile, _Mapping]]] = ...) -> None: ...
+
+class GetInputFileUploadUrlsRequest(_message.Message):
+    __slots__ = ("input_files",)
+    INPUT_FILES_FIELD_NUMBER: _ClassVar[int]
+    input_files: _containers.RepeatedCompositeFieldContainer[BenchmarkInputFile]
+    def __init__(self, input_files: _Optional[_Iterable[_Union[BenchmarkInputFile, _Mapping]]] = ...) -> None: ...
+
+class GetInputFileUploadUrlsResponse(_message.Message):
+    __slots__ = ("presigned_urls",)
+    PRESIGNED_URLS_FIELD_NUMBER: _ClassVar[int]
+    presigned_urls: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, presigned_urls: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class GetAvailableResultFilesRequest(_message.Message):
+    __slots__ = ("prefix",)
+    PREFIX_FIELD_NUMBER: _ClassVar[int]
+    prefix: str
+    def __init__(self, prefix: _Optional[str] = ...) -> None: ...
+
+class GetAvailableResultFilesResponse(_message.Message):
+    __slots__ = ("result_files",)
+    RESULT_FILES_FIELD_NUMBER: _ClassVar[int]
+    result_files: _containers.RepeatedCompositeFieldContainer[BenchmarkInputFile]
+    def __init__(self, result_files: _Optional[_Iterable[_Union[BenchmarkInputFile, _Mapping]]] = ...) -> None: ...
+
+class GetResultFileUrlsRequest(_message.Message):
+    __slots__ = ("file_names",)
+    FILE_NAMES_FIELD_NUMBER: _ClassVar[int]
+    file_names: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, file_names: _Optional[_Iterable[str]] = ...) -> None: ...
+
+class GetResultFileUrlsResponse(_message.Message):
+    __slots__ = ("presigned_urls",)
+    PRESIGNED_URLS_FIELD_NUMBER: _ClassVar[int]
+    presigned_urls: _containers.RepeatedScalarFieldContainer[str]
+    def __init__(self, presigned_urls: _Optional[_Iterable[str]] = ...) -> None: ...

chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py

@@ -20,6 +20,26 @@ class BenchmarkServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkResponse.FromString,
         )
+        self.GetAvailableInputFiles = channel.unary_unary(
+            "/chalk.server.v1.BenchmarkService/GetAvailableInputFiles",
+            request_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableInputFilesRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableInputFilesResponse.FromString,
+        )
+        self.GetInputFileUploadUrls = channel.unary_unary(
+            "/chalk.server.v1.BenchmarkService/GetInputFileUploadUrls",
+            request_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetInputFileUploadUrlsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetInputFileUploadUrlsResponse.FromString,
+        )
+        self.GetAvailableResultFiles = channel.unary_unary(
+            "/chalk.server.v1.BenchmarkService/GetAvailableResultFiles",
+            request_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableResultFilesRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableResultFilesResponse.FromString,
+        )
+        self.GetResultFileUrls = channel.unary_unary(
+            "/chalk.server.v1.BenchmarkService/GetResultFileUrls",
+            request_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetResultFileUrlsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetResultFileUrlsResponse.FromString,
+        )
 
 
 class BenchmarkServiceServicer(object):
@@ -31,6 +51,30 @@ class BenchmarkServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def GetAvailableInputFiles(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def GetInputFileUploadUrls(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def GetAvailableResultFiles(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def GetResultFileUrls(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
 
 def add_BenchmarkServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -39,6 +83,26 @@ def add_BenchmarkServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.CreateBenchmarkResponse.SerializeToString,
         ),
+        "GetAvailableInputFiles": grpc.unary_unary_rpc_method_handler(
+            servicer.GetAvailableInputFiles,
+            request_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableInputFilesRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableInputFilesResponse.SerializeToString,
+        ),
+        "GetInputFileUploadUrls": grpc.unary_unary_rpc_method_handler(
+            servicer.GetInputFileUploadUrls,
+            request_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetInputFileUploadUrlsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetInputFileUploadUrlsResponse.SerializeToString,
+        ),
+        "GetAvailableResultFiles": grpc.unary_unary_rpc_method_handler(
+            servicer.GetAvailableResultFiles,
+            request_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableResultFilesRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableResultFilesResponse.SerializeToString,
+        ),
+        "GetResultFileUrls": grpc.unary_unary_rpc_method_handler(
+            servicer.GetResultFileUrls,
+            request_deserializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetResultFileUrlsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_benchmark__pb2.GetResultFileUrlsResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.BenchmarkService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -76,3 +140,119 @@ class BenchmarkService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def GetAvailableInputFiles(
+        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.server.v1.BenchmarkService/GetAvailableInputFiles",
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableInputFilesRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableInputFilesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetInputFileUploadUrls(
+        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.server.v1.BenchmarkService/GetInputFileUploadUrls",
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetInputFileUploadUrlsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetInputFileUploadUrlsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetAvailableResultFiles(
+        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.server.v1.BenchmarkService/GetAvailableResultFiles",
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableResultFilesRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetAvailableResultFilesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetResultFileUrls(
+        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.server.v1.BenchmarkService/GetResultFileUrls",
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetResultFileUrlsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_benchmark__pb2.GetResultFileUrlsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi

@@ -10,6 +10,14 @@ from abc import (
 from chalk._gen.chalk.server.v1.benchmark_pb2 import (
     CreateBenchmarkRequest,
     CreateBenchmarkResponse,
+    GetAvailableInputFilesRequest,
+    GetAvailableInputFilesResponse,
+    GetAvailableResultFilesRequest,
+    GetAvailableResultFilesResponse,
+    GetInputFileUploadUrlsRequest,
+    GetInputFileUploadUrlsResponse,
+    GetResultFileUrlsRequest,
+    GetResultFileUrlsResponse,
 )
 from grpc import (
     Channel,
@@ -24,6 +32,22 @@ class BenchmarkServiceStub:
         CreateBenchmarkRequest,
         CreateBenchmarkResponse,
     ]
+    GetAvailableInputFiles: UnaryUnaryMultiCallable[
+        GetAvailableInputFilesRequest,
+        GetAvailableInputFilesResponse,
+    ]
+    GetInputFileUploadUrls: UnaryUnaryMultiCallable[
+        GetInputFileUploadUrlsRequest,
+        GetInputFileUploadUrlsResponse,
+    ]
+    GetAvailableResultFiles: UnaryUnaryMultiCallable[
+        GetAvailableResultFilesRequest,
+        GetAvailableResultFilesResponse,
+    ]
+    GetResultFileUrls: UnaryUnaryMultiCallable[
+        GetResultFileUrlsRequest,
+        GetResultFileUrlsResponse,
+    ]
 
 class BenchmarkServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -32,5 +56,29 @@ class BenchmarkServiceServicer(metaclass=ABCMeta):
         request: CreateBenchmarkRequest,
         context: ServicerContext,
     ) -> CreateBenchmarkResponse: ...
+    @abstractmethod
+    def GetAvailableInputFiles(
+        self,
+        request: GetAvailableInputFilesRequest,
+        context: ServicerContext,
+    ) -> GetAvailableInputFilesResponse: ...
+    @abstractmethod
+    def GetInputFileUploadUrls(
+        self,
+        request: GetInputFileUploadUrlsRequest,
+        context: ServicerContext,
+    ) -> GetInputFileUploadUrlsResponse: ...
+    @abstractmethod
+    def GetAvailableResultFiles(
+        self,
+        request: GetAvailableResultFilesRequest,
+        context: ServicerContext,
+    ) -> GetAvailableResultFilesResponse: ...
+    @abstractmethod
+    def GetResultFileUrls(
+        self,
+        request: GetResultFileUrlsRequest,
+        context: ServicerContext,
+    ) -> GetResultFileUrlsResponse: ...
 
 def add_BenchmarkServiceServicer_to_server(servicer: BenchmarkServiceServicer, server: Server) -> None: ...