chalkpy 2.95.6__py3-none-any.whl → 2.95.8__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (28) hide show
  1. chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +47 -47
  2. chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +12 -3
  3. chalk/_gen/chalk/server/v1/benchmark_pb2.py +34 -31
  4. chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +37 -1
  5. chalk/_gen/chalk/server/v1/branches_pb2_grpc.py +78 -0
  6. chalk/_gen/chalk/server/v1/builder_pb2.py +310 -310
  7. chalk/_gen/chalk/server/v1/builder_pb2.pyi +4 -0
  8. chalk/_gen/chalk/server/v1/deploy_pb2.py +34 -34
  9. chalk/_gen/chalk/server/v1/deploy_pb2.pyi +10 -1
  10. chalk/_gen/chalk/server/v1/deployment_pb2.py +10 -10
  11. chalk/_gen/chalk/server/v1/deployment_pb2.pyi +6 -2
  12. chalk/_gen/chalk/server/v1/monitoring_pb2.py +83 -65
  13. chalk/_gen/chalk/server/v1/monitoring_pb2.pyi +97 -0
  14. chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py +90 -0
  15. chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi +24 -0
  16. chalk/_gen/chalk/server/v1/scheduler_pb2.py +32 -26
  17. chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +12 -0
  18. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +45 -0
  19. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +12 -0
  20. chalk/_gen/chalk/server/v1/sql_queries_pb2.py +10 -10
  21. chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +8 -0
  22. chalk/_version.py +1 -1
  23. chalk/client/client_impl.py +5 -3
  24. {chalkpy-2.95.6.dist-info → chalkpy-2.95.8.dist-info}/METADATA +1 -1
  25. {chalkpy-2.95.6.dist-info → chalkpy-2.95.8.dist-info}/RECORD +28 -27
  26. {chalkpy-2.95.6.dist-info → chalkpy-2.95.8.dist-info}/WHEEL +0 -0
  27. {chalkpy-2.95.6.dist-info → chalkpy-2.95.8.dist-info}/entry_points.txt +0 -0
  28. {chalkpy-2.95.6.dist-info → chalkpy-2.95.8.dist-info}/top_level.txt +0 -0
@@ -19,7 +19,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
19
19
 
20
20
 
21
21
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
22
- 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'
22
+ 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"\xbb\x01\n"ExecuteSqlAsyncQueryRequestOptions\x12V\n\x0e\x65xecution_mode\x18\x01 \x01(\x0e\x32/.chalk.protosql.v1.ExecuteSqlAsyncExecutionModeR\rexecutionMode\x12*\n\x0eresource_group\x18\x02 \x01(\tH\x00R\rresourceGroup\x88\x01\x01\x42\x11\n\x0f_resource_group"?\n#ExecuteSqlResultPersistenceSettings\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled"\xe0\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\x12-\n\x10max_memory_bytes\x18\x04 \x01(\x03H\x02R\x0emaxMemoryBytes\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_idB\x13\n\x11_max_memory_bytes"&\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'
23
23
  )
24
24
 
25
25
  _globals = globals()
@@ -42,54 +42,54 @@ if _descriptor._USE_C_DESCRIPTORS == False:
42
42
  _globals["_SQLSERVICE"].methods_by_name["GetDbSchemas"]._serialized_options = b"\200}\013"
43
43
  _globals["_SQLSERVICE"].methods_by_name["GetTables"]._options = None
44
44
  _globals["_SQLSERVICE"].methods_by_name["GetTables"]._serialized_options = b"\200}\013"
45
- _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_start = 3542
46
- _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_end = 3715
45
+ _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_start = 3674
46
+ _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_end = 3847
47
47
  _globals["_SQLQUERYINFO"]._serialized_start = 160
48
48
  _globals["_SQLQUERYINFO"]._serialized_end = 370
49
49
  _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_start = 372
50
50
  _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_end = 407
51
- _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_start = 409
52
- _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_end = 533
53
- _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_start = 535
54
- _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_end = 598
55
- _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 601
56
- _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 1013
57
- _globals["_SIGNEDOUTPUTURIS"]._serialized_start = 1015
58
- _globals["_SIGNEDOUTPUTURIS"]._serialized_end = 1053
59
- _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 1056
60
- _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 1233
61
- _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 1236
62
- _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 1396
63
- _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 1399
64
- _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 1730
65
- _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 1732
66
- _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 1775
67
- _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 1777
68
- _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 1887
69
- _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 1889
70
- _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 1911
71
- _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 1913
72
- _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 2026
73
- _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 2029
74
- _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 2237
75
- _globals["_DBSCHEMAINFO"]._serialized_start = 2239
76
- _globals["_DBSCHEMAINFO"]._serialized_end = 2326
77
- _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 2329
78
- _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 2463
79
- _globals["_GETTABLESREQUEST"]._serialized_start = 2466
80
- _globals["_GETTABLESREQUEST"]._serialized_end = 2753
81
- _globals["_TABLEINFO"]._serialized_start = 2756
82
- _globals["_TABLEINFO"]._serialized_end = 2945
83
- _globals["_GETTABLESRESPONSE"]._serialized_start = 2947
84
- _globals["_GETTABLESRESPONSE"]._serialized_end = 3073
85
- _globals["_SQLQUERYPROGRESSINFO"]._serialized_start = 3075
86
- _globals["_SQLQUERYPROGRESSINFO"]._serialized_end = 3097
87
- _globals["_SQLQUERYFAILEDINFO"]._serialized_start = 3099
88
- _globals["_SQLQUERYFAILEDINFO"]._serialized_end = 3172
89
- _globals["_POLLSQLQUERYREQUEST"]._serialized_start = 3174
90
- _globals["_POLLSQLQUERYREQUEST"]._serialized_end = 3230
91
- _globals["_POLLSQLQUERYRESPONSE"]._serialized_start = 3233
92
- _globals["_POLLSQLQUERYRESPONSE"]._serialized_end = 3539
93
- _globals["_SQLSERVICE"]._serialized_start = 3718
94
- _globals["_SQLSERVICE"]._serialized_end = 4345
51
+ _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_start = 410
52
+ _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_end = 597
53
+ _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_start = 599
54
+ _globals["_EXECUTESQLRESULTPERSISTENCESETTINGS"]._serialized_end = 662
55
+ _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 665
56
+ _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 1145
57
+ _globals["_SIGNEDOUTPUTURIS"]._serialized_start = 1147
58
+ _globals["_SIGNEDOUTPUTURIS"]._serialized_end = 1185
59
+ _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 1188
60
+ _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 1365
61
+ _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 1368
62
+ _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 1528
63
+ _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 1531
64
+ _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 1862
65
+ _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 1864
66
+ _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 1907
67
+ _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 1909
68
+ _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 2019
69
+ _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 2021
70
+ _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 2043
71
+ _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 2045
72
+ _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 2158
73
+ _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 2161
74
+ _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 2369
75
+ _globals["_DBSCHEMAINFO"]._serialized_start = 2371
76
+ _globals["_DBSCHEMAINFO"]._serialized_end = 2458
77
+ _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 2461
78
+ _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 2595
79
+ _globals["_GETTABLESREQUEST"]._serialized_start = 2598
80
+ _globals["_GETTABLESREQUEST"]._serialized_end = 2885
81
+ _globals["_TABLEINFO"]._serialized_start = 2888
82
+ _globals["_TABLEINFO"]._serialized_end = 3077
83
+ _globals["_GETTABLESRESPONSE"]._serialized_start = 3079
84
+ _globals["_GETTABLESRESPONSE"]._serialized_end = 3205
85
+ _globals["_SQLQUERYPROGRESSINFO"]._serialized_start = 3207
86
+ _globals["_SQLQUERYPROGRESSINFO"]._serialized_end = 3229
87
+ _globals["_SQLQUERYFAILEDINFO"]._serialized_start = 3231
88
+ _globals["_SQLQUERYFAILEDINFO"]._serialized_end = 3304
89
+ _globals["_POLLSQLQUERYREQUEST"]._serialized_start = 3306
90
+ _globals["_POLLSQLQUERYREQUEST"]._serialized_end = 3362
91
+ _globals["_POLLSQLQUERYRESPONSE"]._serialized_start = 3365
92
+ _globals["_POLLSQLQUERYRESPONSE"]._serialized_end = 3671
93
+ _globals["_SQLSERVICE"]._serialized_start = 3850
94
+ _globals["_SQLSERVICE"]._serialized_end = 4477
95
95
  # @@protoc_insertion_point(module_scope)
@@ -45,10 +45,16 @@ class ExecuteSqlSyncQueryRequestOptions(_message.Message):
45
45
  def __init__(self) -> None: ...
46
46
 
47
47
  class ExecuteSqlAsyncQueryRequestOptions(_message.Message):
48
- __slots__ = ("execution_mode",)
48
+ __slots__ = ("execution_mode", "resource_group")
49
49
  EXECUTION_MODE_FIELD_NUMBER: _ClassVar[int]
50
+ RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
50
51
  execution_mode: ExecuteSqlAsyncExecutionMode
51
- def __init__(self, execution_mode: _Optional[_Union[ExecuteSqlAsyncExecutionMode, str]] = ...) -> None: ...
52
+ resource_group: str
53
+ def __init__(
54
+ self,
55
+ execution_mode: _Optional[_Union[ExecuteSqlAsyncExecutionMode, str]] = ...,
56
+ resource_group: _Optional[str] = ...,
57
+ ) -> None: ...
52
58
 
53
59
  class ExecuteSqlResultPersistenceSettings(_message.Message):
54
60
  __slots__ = ("enabled",)
@@ -57,15 +63,17 @@ class ExecuteSqlResultPersistenceSettings(_message.Message):
57
63
  def __init__(self, enabled: bool = ...) -> None: ...
58
64
 
59
65
  class ExecuteSqlQueryRequest(_message.Message):
60
- __slots__ = ("query", "correlation_id", "persistence_settings", "sync_options", "async_options")
66
+ __slots__ = ("query", "correlation_id", "persistence_settings", "max_memory_bytes", "sync_options", "async_options")
61
67
  QUERY_FIELD_NUMBER: _ClassVar[int]
62
68
  CORRELATION_ID_FIELD_NUMBER: _ClassVar[int]
63
69
  PERSISTENCE_SETTINGS_FIELD_NUMBER: _ClassVar[int]
70
+ MAX_MEMORY_BYTES_FIELD_NUMBER: _ClassVar[int]
64
71
  SYNC_OPTIONS_FIELD_NUMBER: _ClassVar[int]
65
72
  ASYNC_OPTIONS_FIELD_NUMBER: _ClassVar[int]
66
73
  query: str
67
74
  correlation_id: str
68
75
  persistence_settings: ExecuteSqlResultPersistenceSettings
76
+ max_memory_bytes: int
69
77
  sync_options: ExecuteSqlSyncQueryRequestOptions
70
78
  async_options: ExecuteSqlAsyncQueryRequestOptions
71
79
  def __init__(
@@ -73,6 +81,7 @@ class ExecuteSqlQueryRequest(_message.Message):
73
81
  query: _Optional[str] = ...,
74
82
  correlation_id: _Optional[str] = ...,
75
83
  persistence_settings: _Optional[_Union[ExecuteSqlResultPersistenceSettings, _Mapping]] = ...,
84
+ max_memory_bytes: _Optional[int] = ...,
76
85
  sync_options: _Optional[_Union[ExecuteSqlSyncQueryRequestOptions, _Mapping]] = ...,
77
86
  async_options: _Optional[_Union[ExecuteSqlAsyncQueryRequestOptions, _Mapping]] = ...,
78
87
  ) -> None: ...
@@ -15,12 +15,13 @@ _sym_db = _symbol_database.Default()
15
15
 
16
16
  from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
17
17
  from chalk._gen.chalk.common.v1 import online_query_pb2 as chalk_dot_common_dot_v1_dot_online__query__pb2
18
+ from chalk._gen.chalk.server.v1 import builder_pb2 as chalk_dot_server_dot_v1_dot_builder__pb2
18
19
  from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
19
20
  from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
20
21
 
21
22
 
22
23
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
23
- 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'
24
+ 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\x1d\x63halk/server/v1/builder.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"\xa9\x01\n\rContainerSpec\x12\x42\n\x07request\x18\x01 \x01(\x0b\x32#.chalk.server.v1.KubeResourceConfigH\x00R\x07request\x88\x01\x01\x12>\n\x05limit\x18\x02 \x01(\x0b\x32#.chalk.server.v1.KubeResourceConfigH\x01R\x05limit\x88\x01\x01\x42\n\n\x08_requestB\x08\n\x06_limit"\xf8\x04\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\x0fsimpleQueryBulk\x12 \n\x0bpercentiles\x18\x07 \x03(\x03R\x0bpercentiles\x12*\n\x0eimage_override\x18\x08 \x01(\tH\x01R\rimageOverride\x88\x01\x01\x12R\n\x15warmup_container_spec\x18\t \x01(\x0b\x32\x1e.chalk.server.v1.ContainerSpecR\x13warmupContainerSpec\x12\x45\n\x0e\x63ontainer_spec\x18\n \x01(\x0b\x32\x1e.chalk.server.v1.ContainerSpecR\rcontainerSpecB\x0f\n\rquery_requestB\x11\n\x0f_image_override"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'
24
25
  )
25
26
 
26
27
  _globals = globals()
@@ -43,34 +44,36 @@ if _descriptor._USE_C_DESCRIPTORS == False:
43
44
  _globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableResultFiles"]._serialized_options = b"\200}\003"
44
45
  _globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._options = None
45
46
  _globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._serialized_options = b"\200}\003"
46
- _globals["_BENCHMARKSTATUS"]._serialized_start = 1773
47
- _globals["_BENCHMARKSTATUS"]._serialized_end = 2005
48
- _globals["_INPUTFEATURES"]._serialized_start = 186
49
- _globals["_INPUTFEATURES"]._serialized_end = 240
50
- _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_start = 243
51
- _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_end = 482
52
- _globals["_CREATEBENCHMARKREQUEST"]._serialized_start = 485
53
- _globals["_CREATEBENCHMARKREQUEST"]._serialized_end = 865
54
- _globals["_CREATEBENCHMARKRESPONSE"]._serialized_start = 867
55
- _globals["_CREATEBENCHMARKRESPONSE"]._serialized_end = 985
56
- _globals["_BENCHMARKINPUTFILE"]._serialized_start = 988
57
- _globals["_BENCHMARKINPUTFILE"]._serialized_end = 1141
58
- _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_start = 1143
59
- _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_end = 1198
60
- _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_start = 1200
61
- _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_end = 1302
62
- _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_start = 1304
63
- _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_end = 1405
64
- _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_start = 1407
65
- _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_end = 1478
66
- _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_start = 1480
67
- _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_end = 1536
68
- _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_start = 1538
69
- _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_end = 1643
70
- _globals["_GETRESULTFILEURLSREQUEST"]._serialized_start = 1645
71
- _globals["_GETRESULTFILEURLSREQUEST"]._serialized_end = 1702
72
- _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_start = 1704
73
- _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_end = 1770
74
- _globals["_BENCHMARKSERVICE"]._serialized_start = 2008
75
- _globals["_BENCHMARKSERVICE"]._serialized_end = 2634
47
+ _globals["_BENCHMARKSTATUS"]._serialized_start = 2228
48
+ _globals["_BENCHMARKSTATUS"]._serialized_end = 2460
49
+ _globals["_INPUTFEATURES"]._serialized_start = 217
50
+ _globals["_INPUTFEATURES"]._serialized_end = 271
51
+ _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_start = 274
52
+ _globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_end = 513
53
+ _globals["_CONTAINERSPEC"]._serialized_start = 516
54
+ _globals["_CONTAINERSPEC"]._serialized_end = 685
55
+ _globals["_CREATEBENCHMARKREQUEST"]._serialized_start = 688
56
+ _globals["_CREATEBENCHMARKREQUEST"]._serialized_end = 1320
57
+ _globals["_CREATEBENCHMARKRESPONSE"]._serialized_start = 1322
58
+ _globals["_CREATEBENCHMARKRESPONSE"]._serialized_end = 1440
59
+ _globals["_BENCHMARKINPUTFILE"]._serialized_start = 1443
60
+ _globals["_BENCHMARKINPUTFILE"]._serialized_end = 1596
61
+ _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_start = 1598
62
+ _globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_end = 1653
63
+ _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_start = 1655
64
+ _globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_end = 1757
65
+ _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_start = 1759
66
+ _globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_end = 1860
67
+ _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_start = 1862
68
+ _globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_end = 1933
69
+ _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_start = 1935
70
+ _globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_end = 1991
71
+ _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_start = 1993
72
+ _globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_end = 2098
73
+ _globals["_GETRESULTFILEURLSREQUEST"]._serialized_start = 2100
74
+ _globals["_GETRESULTFILEURLSREQUEST"]._serialized_end = 2157
75
+ _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_start = 2159
76
+ _globals["_GETRESULTFILEURLSRESPONSE"]._serialized_end = 2225
77
+ _globals["_BENCHMARKSERVICE"]._serialized_start = 2463
78
+ _globals["_BENCHMARKSERVICE"]._serialized_end = 3089
76
79
  # @@protoc_insertion_point(module_scope)
@@ -1,5 +1,6 @@
1
1
  from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
2
2
  from chalk._gen.chalk.common.v1 import online_query_pb2 as _online_query_pb2
3
+ from chalk._gen.chalk.server.v1 import builder_pb2 as _builder_pb2
3
4
  from google.protobuf import duration_pb2 as _duration_pb2
4
5
  from google.protobuf import timestamp_pb2 as _timestamp_pb2
5
6
  from google.protobuf.internal import containers as _containers
@@ -58,20 +59,51 @@ class SimpleOnlineQueryBulkRequest(_message.Message):
58
59
  output_features: _Optional[_Iterable[str]] = ...,
59
60
  ) -> None: ...
60
61
 
62
+ class ContainerSpec(_message.Message):
63
+ __slots__ = ("request", "limit")
64
+ REQUEST_FIELD_NUMBER: _ClassVar[int]
65
+ LIMIT_FIELD_NUMBER: _ClassVar[int]
66
+ request: _builder_pb2.KubeResourceConfig
67
+ limit: _builder_pb2.KubeResourceConfig
68
+ def __init__(
69
+ self,
70
+ request: _Optional[_Union[_builder_pb2.KubeResourceConfig, _Mapping]] = ...,
71
+ limit: _Optional[_Union[_builder_pb2.KubeResourceConfig, _Mapping]] = ...,
72
+ ) -> None: ...
73
+
61
74
  class CreateBenchmarkRequest(_message.Message):
62
- __slots__ = ("warmup_qps", "warmup_duration", "qps", "duration", "query_bulk", "simple_query_bulk")
75
+ __slots__ = (
76
+ "warmup_qps",
77
+ "warmup_duration",
78
+ "qps",
79
+ "duration",
80
+ "query_bulk",
81
+ "simple_query_bulk",
82
+ "percentiles",
83
+ "image_override",
84
+ "warmup_container_spec",
85
+ "container_spec",
86
+ )
63
87
  WARMUP_QPS_FIELD_NUMBER: _ClassVar[int]
64
88
  WARMUP_DURATION_FIELD_NUMBER: _ClassVar[int]
65
89
  QPS_FIELD_NUMBER: _ClassVar[int]
66
90
  DURATION_FIELD_NUMBER: _ClassVar[int]
67
91
  QUERY_BULK_FIELD_NUMBER: _ClassVar[int]
68
92
  SIMPLE_QUERY_BULK_FIELD_NUMBER: _ClassVar[int]
93
+ PERCENTILES_FIELD_NUMBER: _ClassVar[int]
94
+ IMAGE_OVERRIDE_FIELD_NUMBER: _ClassVar[int]
95
+ WARMUP_CONTAINER_SPEC_FIELD_NUMBER: _ClassVar[int]
96
+ CONTAINER_SPEC_FIELD_NUMBER: _ClassVar[int]
69
97
  warmup_qps: int
70
98
  warmup_duration: _duration_pb2.Duration
71
99
  qps: int
72
100
  duration: _duration_pb2.Duration
73
101
  query_bulk: _online_query_pb2.OnlineQueryBulkRequest
74
102
  simple_query_bulk: SimpleOnlineQueryBulkRequest
103
+ percentiles: _containers.RepeatedScalarFieldContainer[int]
104
+ image_override: str
105
+ warmup_container_spec: ContainerSpec
106
+ container_spec: ContainerSpec
75
107
  def __init__(
76
108
  self,
77
109
  warmup_qps: _Optional[int] = ...,
@@ -80,6 +112,10 @@ class CreateBenchmarkRequest(_message.Message):
80
112
  duration: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
81
113
  query_bulk: _Optional[_Union[_online_query_pb2.OnlineQueryBulkRequest, _Mapping]] = ...,
82
114
  simple_query_bulk: _Optional[_Union[SimpleOnlineQueryBulkRequest, _Mapping]] = ...,
115
+ percentiles: _Optional[_Iterable[int]] = ...,
116
+ image_override: _Optional[str] = ...,
117
+ warmup_container_spec: _Optional[_Union[ContainerSpec, _Mapping]] = ...,
118
+ container_spec: _Optional[_Union[ContainerSpec, _Mapping]] = ...,
83
119
  ) -> None: ...
84
120
 
85
121
  class CreateBenchmarkResponse(_message.Message):
@@ -0,0 +1,78 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+
4
+ import grpc
5
+
6
+ from chalk._gen.chalk.server.v1 import branches_pb2 as chalk_dot_server_dot_v1_dot_branches__pb2
7
+
8
+
9
+ class BranchServiceStub(object):
10
+ """Missing associated documentation comment in .proto file."""
11
+
12
+ def __init__(self, channel):
13
+ """Constructor.
14
+
15
+ Args:
16
+ channel: A grpc.Channel.
17
+ """
18
+ self.ListBranchWithLatestDeployments = channel.unary_unary(
19
+ "/chalk.server.v1.BranchService/ListBranchWithLatestDeployments",
20
+ request_serializer=chalk_dot_server_dot_v1_dot_branches__pb2.ListBranchWithLatestDeploymentsRequest.SerializeToString,
21
+ response_deserializer=chalk_dot_server_dot_v1_dot_branches__pb2.ListBranchWithLatestDeploymentsResponse.FromString,
22
+ )
23
+
24
+
25
+ class BranchServiceServicer(object):
26
+ """Missing associated documentation comment in .proto file."""
27
+
28
+ def ListBranchWithLatestDeployments(self, request, context):
29
+ """Missing associated documentation comment in .proto file."""
30
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
31
+ context.set_details("Method not implemented!")
32
+ raise NotImplementedError("Method not implemented!")
33
+
34
+
35
+ def add_BranchServiceServicer_to_server(servicer, server):
36
+ rpc_method_handlers = {
37
+ "ListBranchWithLatestDeployments": grpc.unary_unary_rpc_method_handler(
38
+ servicer.ListBranchWithLatestDeployments,
39
+ request_deserializer=chalk_dot_server_dot_v1_dot_branches__pb2.ListBranchWithLatestDeploymentsRequest.FromString,
40
+ response_serializer=chalk_dot_server_dot_v1_dot_branches__pb2.ListBranchWithLatestDeploymentsResponse.SerializeToString,
41
+ ),
42
+ }
43
+ generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.BranchService", rpc_method_handlers)
44
+ server.add_generic_rpc_handlers((generic_handler,))
45
+
46
+
47
+ # This class is part of an EXPERIMENTAL API.
48
+ class BranchService(object):
49
+ """Missing associated documentation comment in .proto file."""
50
+
51
+ @staticmethod
52
+ def ListBranchWithLatestDeployments(
53
+ request,
54
+ target,
55
+ options=(),
56
+ channel_credentials=None,
57
+ call_credentials=None,
58
+ insecure=False,
59
+ compression=None,
60
+ wait_for_ready=None,
61
+ timeout=None,
62
+ metadata=None,
63
+ ):
64
+ return grpc.experimental.unary_unary(
65
+ request,
66
+ target,
67
+ "/chalk.server.v1.BranchService/ListBranchWithLatestDeployments",
68
+ chalk_dot_server_dot_v1_dot_branches__pb2.ListBranchWithLatestDeploymentsRequest.SerializeToString,
69
+ chalk_dot_server_dot_v1_dot_branches__pb2.ListBranchWithLatestDeploymentsResponse.FromString,
70
+ options,
71
+ channel_credentials,
72
+ insecure,
73
+ call_credentials,
74
+ compression,
75
+ wait_for_ready,
76
+ timeout,
77
+ metadata,
78
+ )