chalkpy 2.90.1__py3-none-any.whl → 2.95.3__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 (202) hide show
  1. chalk/__init__.py +2 -1
  2. chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
  3. chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
  4. chalk/_gen/chalk/artifacts/v1/chart_pb2.py +16 -16
  5. chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +4 -0
  6. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +8 -7
  7. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +5 -0
  8. chalk/_gen/chalk/common/v1/offline_query_pb2.py +17 -15
  9. chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +25 -0
  10. chalk/_gen/chalk/common/v1/script_task_pb2.py +3 -3
  11. chalk/_gen/chalk/common/v1/script_task_pb2.pyi +2 -0
  12. chalk/_gen/chalk/dataframe/__init__.py +0 -0
  13. chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
  14. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
  15. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +123 -0
  16. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
  17. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
  18. chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
  19. chalk/_gen/chalk/graph/v1/graph_pb2.pyi +25 -0
  20. chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
  21. chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -0
  22. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
  23. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
  24. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
  25. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
  26. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
  27. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
  28. chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +54 -27
  29. chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
  30. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
  31. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
  32. chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
  33. chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
  34. chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
  35. chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
  36. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
  37. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
  38. chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
  39. chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
  40. chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
  41. chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
  42. chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
  43. chalk/_gen/chalk/server/v1/builder_pb2.py +358 -288
  44. chalk/_gen/chalk/server/v1/builder_pb2.pyi +360 -10
  45. chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +225 -0
  46. chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +60 -0
  47. chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
  48. chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
  49. chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
  50. chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
  51. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
  52. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
  53. chalk/_gen/chalk/server/v1/cloud_components_pb2.py +141 -119
  54. chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +106 -4
  55. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +45 -0
  56. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +12 -0
  57. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
  58. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
  59. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
  60. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
  61. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +52 -38
  62. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +62 -1
  63. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +90 -0
  64. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +24 -0
  65. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
  66. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -0
  67. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
  68. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
  69. chalk/_gen/chalk/server/v1/deploy_pb2.py +9 -3
  70. chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
  71. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
  72. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
  73. chalk/_gen/chalk/server/v1/deployment_pb2.py +6 -6
  74. chalk/_gen/chalk/server/v1/deployment_pb2.pyi +20 -0
  75. chalk/_gen/chalk/server/v1/environment_pb2.py +14 -12
  76. chalk/_gen/chalk/server/v1/environment_pb2.pyi +19 -0
  77. chalk/_gen/chalk/server/v1/eventbus_pb2.py +4 -2
  78. chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
  79. chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
  80. chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
  81. chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
  82. chalk/_gen/chalk/server/v1/graph_pb2.py +38 -26
  83. chalk/_gen/chalk/server/v1/graph_pb2.pyi +58 -0
  84. chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +47 -0
  85. chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +18 -0
  86. chalk/_gen/chalk/server/v1/incident_pb2.py +23 -21
  87. chalk/_gen/chalk/server/v1/incident_pb2.pyi +15 -1
  88. chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
  89. chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
  90. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
  91. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
  92. chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
  93. chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
  94. chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
  95. chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
  96. chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
  97. chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
  98. chalk/_gen/chalk/server/v1/log_pb2.py +21 -3
  99. chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
  100. chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
  101. chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
  102. chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
  103. chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
  104. chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
  105. chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
  106. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
  107. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
  108. chalk/_gen/chalk/server/v1/queries_pb2.py +66 -66
  109. chalk/_gen/chalk/server/v1/queries_pb2.pyi +32 -2
  110. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -12
  111. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +16 -3
  112. chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
  113. chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
  114. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
  115. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
  116. chalk/_gen/chalk/server/v1/script_tasks_pb2.py +15 -3
  117. chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +22 -0
  118. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
  119. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -0
  120. chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
  121. chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
  122. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
  123. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
  124. chalk/_gen/chalk/server/v1/sql_queries_pb2.py +48 -0
  125. chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
  126. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
  127. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
  128. chalk/_gen/chalk/server/v1/team_pb2.py +154 -141
  129. chalk/_gen/chalk/server/v1/team_pb2.pyi +30 -2
  130. chalk/_gen/chalk/server/v1/team_pb2_grpc.py +45 -0
  131. chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +12 -0
  132. chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
  133. chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
  134. chalk/_gen/chalk/server/v1/trace_pb2.py +44 -40
  135. chalk/_gen/chalk/server/v1/trace_pb2.pyi +20 -0
  136. chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
  137. chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
  138. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
  139. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
  140. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +16 -10
  141. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +52 -1
  142. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
  143. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
  144. chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
  145. chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
  146. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
  147. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
  148. chalk/_lsp/error_builder.py +11 -0
  149. chalk/_version.py +1 -1
  150. chalk/client/client.py +128 -43
  151. chalk/client/client_async.py +149 -0
  152. chalk/client/client_async_impl.py +22 -0
  153. chalk/client/client_grpc.py +539 -104
  154. chalk/client/client_impl.py +449 -122
  155. chalk/client/dataset.py +7 -1
  156. chalk/client/models.py +98 -0
  157. chalk/client/serialization/model_serialization.py +92 -9
  158. chalk/df/LazyFramePlaceholder.py +1154 -0
  159. chalk/features/_class_property.py +7 -0
  160. chalk/features/_embedding/embedding.py +1 -0
  161. chalk/features/_encoding/converter.py +83 -2
  162. chalk/features/feature_field.py +40 -30
  163. chalk/features/feature_set_decorator.py +1 -0
  164. chalk/features/feature_wrapper.py +42 -3
  165. chalk/features/hooks.py +81 -10
  166. chalk/features/inference.py +33 -31
  167. chalk/features/resolver.py +224 -24
  168. chalk/functions/__init__.py +65 -3
  169. chalk/gitignore/gitignore_parser.py +5 -1
  170. chalk/importer.py +142 -68
  171. chalk/ml/__init__.py +2 -0
  172. chalk/ml/model_hooks.py +194 -26
  173. chalk/ml/model_reference.py +56 -8
  174. chalk/ml/model_version.py +24 -15
  175. chalk/ml/utils.py +20 -17
  176. chalk/operators/_utils.py +10 -3
  177. chalk/parsed/_proto/export.py +22 -0
  178. chalk/parsed/duplicate_input_gql.py +3 -0
  179. chalk/parsed/json_conversions.py +20 -14
  180. chalk/parsed/to_proto.py +16 -4
  181. chalk/parsed/user_types_to_json.py +31 -10
  182. chalk/parsed/validation_from_registries.py +182 -0
  183. chalk/queries/named_query.py +16 -6
  184. chalk/queries/scheduled_query.py +9 -1
  185. chalk/serialization/parsed_annotation.py +24 -11
  186. chalk/sql/__init__.py +18 -0
  187. chalk/sql/_internal/integrations/databricks.py +55 -17
  188. chalk/sql/_internal/integrations/mssql.py +127 -62
  189. chalk/sql/_internal/integrations/redshift.py +4 -0
  190. chalk/sql/_internal/sql_file_resolver.py +53 -9
  191. chalk/sql/_internal/sql_source.py +35 -2
  192. chalk/streams/_kafka_source.py +5 -1
  193. chalk/streams/_windows.py +15 -2
  194. chalk/utils/_otel_version.py +13 -0
  195. chalk/utils/async_helpers.py +2 -2
  196. chalk/utils/missing_dependency.py +5 -4
  197. chalk/utils/tracing.py +185 -95
  198. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/METADATA +4 -6
  199. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/RECORD +202 -146
  200. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
  201. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
  202. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/top_level.txt +0 -0
@@ -215,6 +215,11 @@ class BuilderServiceStub(object):
215
215
  request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentRequest.SerializeToString,
216
216
  response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentResponse.FromString,
217
217
  )
218
+ self.PrepareDeployment = channel.unary_unary(
219
+ "/chalk.server.v1.BuilderService/PrepareDeployment",
220
+ request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentRequest.SerializeToString,
221
+ response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentResponse.FromString,
222
+ )
218
223
  self.GetTelemetryDeployment = channel.unary_unary(
219
224
  "/chalk.server.v1.BuilderService/GetTelemetryDeployment",
220
225
  request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetTelemetryDeploymentRequest.SerializeToString,
@@ -255,6 +260,26 @@ class BuilderServiceStub(object):
255
260
  request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentRequest.SerializeToString,
256
261
  response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentResponse.FromString,
257
262
  )
263
+ self.SuspendClusterGateway = channel.unary_unary(
264
+ "/chalk.server.v1.BuilderService/SuspendClusterGateway",
265
+ request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayRequest.SerializeToString,
266
+ response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayResponse.FromString,
267
+ )
268
+ self.ResumeClusterGateway = channel.unary_unary(
269
+ "/chalk.server.v1.BuilderService/ResumeClusterGateway",
270
+ request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayRequest.SerializeToString,
271
+ response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayResponse.FromString,
272
+ )
273
+ self.SuspendClusterBackgroundPersistence = channel.unary_unary(
274
+ "/chalk.server.v1.BuilderService/SuspendClusterBackgroundPersistence",
275
+ request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceRequest.SerializeToString,
276
+ response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceResponse.FromString,
277
+ )
278
+ self.ResumeClusterBackgroundPersistence = channel.unary_unary(
279
+ "/chalk.server.v1.BuilderService/ResumeClusterBackgroundPersistence",
280
+ request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceRequest.SerializeToString,
281
+ response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceResponse.FromString,
282
+ )
258
283
 
259
284
 
260
285
  class BuilderServiceServicer(object):
@@ -510,6 +535,12 @@ class BuilderServiceServicer(object):
510
535
  context.set_details("Method not implemented!")
511
536
  raise NotImplementedError("Method not implemented!")
512
537
 
538
+ def PrepareDeployment(self, request, context):
539
+ """Missing associated documentation comment in .proto file."""
540
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
541
+ context.set_details("Method not implemented!")
542
+ raise NotImplementedError("Method not implemented!")
543
+
513
544
  def GetTelemetryDeployment(self, request, context):
514
545
  """Missing associated documentation comment in .proto file."""
515
546
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -558,6 +589,30 @@ class BuilderServiceServicer(object):
558
589
  context.set_details("Method not implemented!")
559
590
  raise NotImplementedError("Method not implemented!")
560
591
 
592
+ def SuspendClusterGateway(self, request, context):
593
+ """Missing associated documentation comment in .proto file."""
594
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
595
+ context.set_details("Method not implemented!")
596
+ raise NotImplementedError("Method not implemented!")
597
+
598
+ def ResumeClusterGateway(self, request, context):
599
+ """Missing associated documentation comment in .proto file."""
600
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
601
+ context.set_details("Method not implemented!")
602
+ raise NotImplementedError("Method not implemented!")
603
+
604
+ def SuspendClusterBackgroundPersistence(self, request, context):
605
+ """Missing associated documentation comment in .proto file."""
606
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
607
+ context.set_details("Method not implemented!")
608
+ raise NotImplementedError("Method not implemented!")
609
+
610
+ def ResumeClusterBackgroundPersistence(self, request, context):
611
+ """Missing associated documentation comment in .proto file."""
612
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
613
+ context.set_details("Method not implemented!")
614
+ raise NotImplementedError("Method not implemented!")
615
+
561
616
 
562
617
  def add_BuilderServiceServicer_to_server(servicer, server):
563
618
  rpc_method_handlers = {
@@ -761,6 +816,11 @@ def add_BuilderServiceServicer_to_server(servicer, server):
761
816
  request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentRequest.FromString,
762
817
  response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentResponse.SerializeToString,
763
818
  ),
819
+ "PrepareDeployment": grpc.unary_unary_rpc_method_handler(
820
+ servicer.PrepareDeployment,
821
+ request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentRequest.FromString,
822
+ response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentResponse.SerializeToString,
823
+ ),
764
824
  "GetTelemetryDeployment": grpc.unary_unary_rpc_method_handler(
765
825
  servicer.GetTelemetryDeployment,
766
826
  request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetTelemetryDeploymentRequest.FromString,
@@ -801,6 +861,26 @@ def add_BuilderServiceServicer_to_server(servicer, server):
801
861
  request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentRequest.FromString,
802
862
  response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentResponse.SerializeToString,
803
863
  ),
864
+ "SuspendClusterGateway": grpc.unary_unary_rpc_method_handler(
865
+ servicer.SuspendClusterGateway,
866
+ request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayRequest.FromString,
867
+ response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayResponse.SerializeToString,
868
+ ),
869
+ "ResumeClusterGateway": grpc.unary_unary_rpc_method_handler(
870
+ servicer.ResumeClusterGateway,
871
+ request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayRequest.FromString,
872
+ response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayResponse.SerializeToString,
873
+ ),
874
+ "SuspendClusterBackgroundPersistence": grpc.unary_unary_rpc_method_handler(
875
+ servicer.SuspendClusterBackgroundPersistence,
876
+ request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceRequest.FromString,
877
+ response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceResponse.SerializeToString,
878
+ ),
879
+ "ResumeClusterBackgroundPersistence": grpc.unary_unary_rpc_method_handler(
880
+ servicer.ResumeClusterBackgroundPersistence,
881
+ request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceRequest.FromString,
882
+ response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceResponse.SerializeToString,
883
+ ),
804
884
  }
805
885
  generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.BuilderService", rpc_method_handlers)
806
886
  server.add_generic_rpc_handlers((generic_handler,))
@@ -1970,6 +2050,35 @@ class BuilderService(object):
1970
2050
  metadata,
1971
2051
  )
1972
2052
 
2053
+ @staticmethod
2054
+ def PrepareDeployment(
2055
+ request,
2056
+ target,
2057
+ options=(),
2058
+ channel_credentials=None,
2059
+ call_credentials=None,
2060
+ insecure=False,
2061
+ compression=None,
2062
+ wait_for_ready=None,
2063
+ timeout=None,
2064
+ metadata=None,
2065
+ ):
2066
+ return grpc.experimental.unary_unary(
2067
+ request,
2068
+ target,
2069
+ "/chalk.server.v1.BuilderService/PrepareDeployment",
2070
+ chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentRequest.SerializeToString,
2071
+ chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentResponse.FromString,
2072
+ options,
2073
+ channel_credentials,
2074
+ insecure,
2075
+ call_credentials,
2076
+ compression,
2077
+ wait_for_ready,
2078
+ timeout,
2079
+ metadata,
2080
+ )
2081
+
1973
2082
  @staticmethod
1974
2083
  def GetTelemetryDeployment(
1975
2084
  request,
@@ -2202,6 +2311,122 @@ class BuilderService(object):
2202
2311
  metadata,
2203
2312
  )
2204
2313
 
2314
+ @staticmethod
2315
+ def SuspendClusterGateway(
2316
+ request,
2317
+ target,
2318
+ options=(),
2319
+ channel_credentials=None,
2320
+ call_credentials=None,
2321
+ insecure=False,
2322
+ compression=None,
2323
+ wait_for_ready=None,
2324
+ timeout=None,
2325
+ metadata=None,
2326
+ ):
2327
+ return grpc.experimental.unary_unary(
2328
+ request,
2329
+ target,
2330
+ "/chalk.server.v1.BuilderService/SuspendClusterGateway",
2331
+ chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayRequest.SerializeToString,
2332
+ chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayResponse.FromString,
2333
+ options,
2334
+ channel_credentials,
2335
+ insecure,
2336
+ call_credentials,
2337
+ compression,
2338
+ wait_for_ready,
2339
+ timeout,
2340
+ metadata,
2341
+ )
2342
+
2343
+ @staticmethod
2344
+ def ResumeClusterGateway(
2345
+ request,
2346
+ target,
2347
+ options=(),
2348
+ channel_credentials=None,
2349
+ call_credentials=None,
2350
+ insecure=False,
2351
+ compression=None,
2352
+ wait_for_ready=None,
2353
+ timeout=None,
2354
+ metadata=None,
2355
+ ):
2356
+ return grpc.experimental.unary_unary(
2357
+ request,
2358
+ target,
2359
+ "/chalk.server.v1.BuilderService/ResumeClusterGateway",
2360
+ chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayRequest.SerializeToString,
2361
+ chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayResponse.FromString,
2362
+ options,
2363
+ channel_credentials,
2364
+ insecure,
2365
+ call_credentials,
2366
+ compression,
2367
+ wait_for_ready,
2368
+ timeout,
2369
+ metadata,
2370
+ )
2371
+
2372
+ @staticmethod
2373
+ def SuspendClusterBackgroundPersistence(
2374
+ request,
2375
+ target,
2376
+ options=(),
2377
+ channel_credentials=None,
2378
+ call_credentials=None,
2379
+ insecure=False,
2380
+ compression=None,
2381
+ wait_for_ready=None,
2382
+ timeout=None,
2383
+ metadata=None,
2384
+ ):
2385
+ return grpc.experimental.unary_unary(
2386
+ request,
2387
+ target,
2388
+ "/chalk.server.v1.BuilderService/SuspendClusterBackgroundPersistence",
2389
+ chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceRequest.SerializeToString,
2390
+ chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceResponse.FromString,
2391
+ options,
2392
+ channel_credentials,
2393
+ insecure,
2394
+ call_credentials,
2395
+ compression,
2396
+ wait_for_ready,
2397
+ timeout,
2398
+ metadata,
2399
+ )
2400
+
2401
+ @staticmethod
2402
+ def ResumeClusterBackgroundPersistence(
2403
+ request,
2404
+ target,
2405
+ options=(),
2406
+ channel_credentials=None,
2407
+ call_credentials=None,
2408
+ insecure=False,
2409
+ compression=None,
2410
+ wait_for_ready=None,
2411
+ timeout=None,
2412
+ metadata=None,
2413
+ ):
2414
+ return grpc.experimental.unary_unary(
2415
+ request,
2416
+ target,
2417
+ "/chalk.server.v1.BuilderService/ResumeClusterBackgroundPersistence",
2418
+ chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceRequest.SerializeToString,
2419
+ chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceResponse.FromString,
2420
+ options,
2421
+ channel_credentials,
2422
+ insecure,
2423
+ call_credentials,
2424
+ compression,
2425
+ wait_for_ready,
2426
+ timeout,
2427
+ metadata,
2428
+ )
2429
+
2205
2430
 
2206
2431
  class ClusterBuilderServiceStub(object):
2207
2432
  """Missing associated documentation comment in .proto file."""
@@ -84,10 +84,16 @@ from chalk._gen.chalk.server.v1.builder_pb2 import (
84
84
  MigrateClusterTimescaleDBResponse,
85
85
  MigrateTelemetryDeploymentRequest,
86
86
  MigrateTelemetryDeploymentResponse,
87
+ PrepareDeploymentRequest,
88
+ PrepareDeploymentResponse,
87
89
  RebuildDeploymentRequest,
88
90
  RebuildDeploymentResponse,
89
91
  RedeployDeploymentRequest,
90
92
  RedeployDeploymentResponse,
93
+ ResumeClusterBackgroundPersistenceRequest,
94
+ ResumeClusterBackgroundPersistenceResponse,
95
+ ResumeClusterGatewayRequest,
96
+ ResumeClusterGatewayResponse,
91
97
  ResumeEnvironmentRequest,
92
98
  ResumeEnvironmentResponse,
93
99
  ScaleBranchRequest,
@@ -96,6 +102,10 @@ from chalk._gen.chalk.server.v1.builder_pb2 import (
96
102
  SetTagWeightsResponse,
97
103
  StartBranchRequest,
98
104
  StartBranchResponse,
105
+ SuspendClusterBackgroundPersistenceRequest,
106
+ SuspendClusterBackgroundPersistenceResponse,
107
+ SuspendClusterGatewayRequest,
108
+ SuspendClusterGatewayResponse,
99
109
  SuspendEnvironmentRequest,
100
110
  SuspendEnvironmentResponse,
101
111
  UpdateEnvironmentVariablesRequest,
@@ -294,6 +304,10 @@ class BuilderServiceStub:
294
304
  CreateDeploymentRequest,
295
305
  CreateDeploymentResponse,
296
306
  ]
307
+ PrepareDeployment: UnaryUnaryMultiCallable[
308
+ PrepareDeploymentRequest,
309
+ PrepareDeploymentResponse,
310
+ ]
297
311
  GetTelemetryDeployment: UnaryUnaryMultiCallable[
298
312
  GetTelemetryDeploymentRequest,
299
313
  GetTelemetryDeploymentResponse,
@@ -326,6 +340,22 @@ class BuilderServiceStub:
326
340
  ResumeEnvironmentRequest,
327
341
  ResumeEnvironmentResponse,
328
342
  ]
343
+ SuspendClusterGateway: UnaryUnaryMultiCallable[
344
+ SuspendClusterGatewayRequest,
345
+ SuspendClusterGatewayResponse,
346
+ ]
347
+ ResumeClusterGateway: UnaryUnaryMultiCallable[
348
+ ResumeClusterGatewayRequest,
349
+ ResumeClusterGatewayResponse,
350
+ ]
351
+ SuspendClusterBackgroundPersistence: UnaryUnaryMultiCallable[
352
+ SuspendClusterBackgroundPersistenceRequest,
353
+ SuspendClusterBackgroundPersistenceResponse,
354
+ ]
355
+ ResumeClusterBackgroundPersistence: UnaryUnaryMultiCallable[
356
+ ResumeClusterBackgroundPersistenceRequest,
357
+ ResumeClusterBackgroundPersistenceResponse,
358
+ ]
329
359
 
330
360
  class BuilderServiceServicer(metaclass=ABCMeta):
331
361
  @abstractmethod
@@ -585,6 +615,12 @@ class BuilderServiceServicer(metaclass=ABCMeta):
585
615
  context: ServicerContext,
586
616
  ) -> CreateDeploymentResponse: ...
587
617
  @abstractmethod
618
+ def PrepareDeployment(
619
+ self,
620
+ request: PrepareDeploymentRequest,
621
+ context: ServicerContext,
622
+ ) -> PrepareDeploymentResponse: ...
623
+ @abstractmethod
588
624
  def GetTelemetryDeployment(
589
625
  self,
590
626
  request: GetTelemetryDeploymentRequest,
@@ -632,6 +668,30 @@ class BuilderServiceServicer(metaclass=ABCMeta):
632
668
  request: ResumeEnvironmentRequest,
633
669
  context: ServicerContext,
634
670
  ) -> ResumeEnvironmentResponse: ...
671
+ @abstractmethod
672
+ def SuspendClusterGateway(
673
+ self,
674
+ request: SuspendClusterGatewayRequest,
675
+ context: ServicerContext,
676
+ ) -> SuspendClusterGatewayResponse: ...
677
+ @abstractmethod
678
+ def ResumeClusterGateway(
679
+ self,
680
+ request: ResumeClusterGatewayRequest,
681
+ context: ServicerContext,
682
+ ) -> ResumeClusterGatewayResponse: ...
683
+ @abstractmethod
684
+ def SuspendClusterBackgroundPersistence(
685
+ self,
686
+ request: SuspendClusterBackgroundPersistenceRequest,
687
+ context: ServicerContext,
688
+ ) -> SuspendClusterBackgroundPersistenceResponse: ...
689
+ @abstractmethod
690
+ def ResumeClusterBackgroundPersistence(
691
+ self,
692
+ request: ResumeClusterBackgroundPersistenceRequest,
693
+ context: ServicerContext,
694
+ ) -> ResumeClusterBackgroundPersistenceResponse: ...
635
695
 
636
696
  def add_BuilderServiceServicer_to_server(servicer: BuilderServiceServicer, server: Server) -> None: ...
637
697
 
@@ -22,7 +22,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
22
22
 
23
23
 
24
24
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
25
- b'\n\x1b\x63halk/server/v1/chart.proto\x12\x0f\x63halk.server.v1\x1a\x1e\x63halk/artifacts/v1/chart.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a)chalk/chart/v1/densetimeserieschart.proto\x1a\x1egoogle/protobuf/duration.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"L\n\x06Series\x12\x16\n\x06points\x18\x01 \x03(\x01R\x06points\x12\x14\n\x05label\x18\x02 \x01(\tR\x05label\x12\x14\n\x05units\x18\x03 \x01(\tR\x05units"t\n\x05\x43hart\x12\x14\n\x05title\x18\x01 \x01(\tR\x05title\x12/\n\x06series\x18\x02 \x03(\x0b\x32\x17.chalk.server.v1.SeriesR\x06series\x12$\n\x0ex_timestamp_ms\x18\x04 \x03(\x03R\x0cxTimestampMs"0\n\x05Point\x12\x19\n\x05value\x18\x01 \x01(\x03H\x00R\x05value\x88\x01\x01:\x02\x18\x01\x42\x08\n\x06_value"l\n\nTimeSeries\x12.\n\x06points\x18\x01 \x03(\x0b\x32\x16.chalk.server.v1.PointR\x06points\x12\x14\n\x05label\x18\x02 \x01(\tR\x05label\x12\x14\n\x05units\x18\x03 \x01(\tR\x05units:\x02\x18\x01"\xd7\x01\n\x0fTimeSeriesChart\x12\x14\n\x05title\x18\x01 \x01(\tR\x05title\x12\x33\n\x06series\x18\x02 \x03(\x0b\x32\x1b.chalk.server.v1.TimeSeriesR\x06series\x12\x35\n\x08x_series\x18\x03 \x03(\x0b\x32\x1a.google.protobuf.TimestampR\x07xSeries\x12>\n\rwindow_period\x18\x04 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0cwindowPeriod:\x02\x18\x01"\xbe\x01\n\x11ListChartsFilters\x12P\n\x10link_entity_kind\x18\x01 \x01(\x0e\x32!.chalk.artifacts.v1.ChartLinkKindH\x00R\x0elinkEntityKind\x88\x01\x01\x12-\n\x10linked_entity_id\x18\x02 \x01(\tH\x01R\x0elinkedEntityId\x88\x01\x01\x42\x13\n\x11_link_entity_kindB\x13\n\x11_linked_entity_id"m\n\x12ListChartPageToken\x12@\n\x0e\x63reated_at_hwm\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0c\x63reatedAtHwm\x12\x15\n\x06id_hwm\x18\x02 \x01(\tR\x05idHwm"\xba\x01\n\x11ListChartsRequest\x12\x41\n\x07\x66ilters\x18\x01 \x01(\x0b\x32".chalk.server.v1.ListChartsFiltersH\x00R\x07\x66ilters\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x12"\n\npage_token\x18\x03 \x01(\tH\x02R\tpageToken\x88\x01\x01\x42\n\n\x08_filtersB\x08\n\x06_limitB\r\n\x0b_page_token"\xda\x01\n\x12ListChartsResponse\x12<\n\x06\x63harts\x18\x01 \x03(\x0b\x32 .chalk.artifacts.v1.MetricConfigB\x02\x18\x01R\x06\x63harts\x12\x45\n\x11\x63harts_with_links\x18\x02 \x03(\x0b\x32\x19.chalk.artifacts.v1.ChartR\x0f\x63hartsWithLinks\x12+\n\x0fnext_page_token\x18\x03 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token"\x89\x03\n\x1bUpdateMetricConfigOperation\x12\x17\n\x04name\x18\x01 \x01(\tH\x00R\x04name\x88\x01\x01\x12(\n\rwindow_period\x18\x02 \x01(\tH\x01R\x0cwindowPeriod\x88\x01\x01\x12>\n\x06series\x18\x03 \x03(\x0b\x32&.chalk.artifacts.v1.MetricConfigSeriesR\x06series\x12=\n\x08\x66ormulas\x18\x04 \x03(\x0b\x32!.chalk.artifacts.v1.MetricFormulaR\x08\x66ormulas\x12?\n\x07trigger\x18\x05 \x01(\x0b\x32 .chalk.artifacts.v1.AlertTriggerH\x02R\x07trigger\x88\x01\x01\x12,\n\x0fgraph_generated\x18\x06 \x01(\x08H\x03R\x0egraphGenerated\x88\x01\x01\x42\x07\n\x05_nameB\x10\n\x0e_window_periodB\n\n\x08_triggerB\x12\n\x10_graph_generated"\xc2\x03\n\x12\x43reateChartRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12#\n\rwindow_period\x18\x02 \x01(\tR\x0cwindowPeriod\x12>\n\x06series\x18\x03 \x03(\x0b\x32&.chalk.artifacts.v1.MetricConfigSeriesR\x06series\x12=\n\x08\x66ormulas\x18\x04 \x03(\x0b\x32!.chalk.artifacts.v1.MetricFormulaR\x08\x66ormulas\x12:\n\x07trigger\x18\x05 \x01(\x0b\x32 .chalk.artifacts.v1.AlertTriggerR\x07trigger\x12K\n\x10link_entity_kind\x18\x06 \x01(\x0e\x32!.chalk.artifacts.v1.ChartLinkKindR\x0elinkEntityKind\x12-\n\x10linked_entity_id\x18\x07 \x01(\tH\x00R\x0elinkedEntityId\x88\x01\x01\x12\'\n\x0fgraph_generated\x18\x08 \x01(\x08R\x0egraphGeneratedB\x13\n\x11_linked_entity_id"U\n\x13\x43reateChartResponse\x12\x34\n\x05\x63hart\x18\x02 \x01(\x0b\x32\x19.chalk.artifacts.v1.ChartH\x00R\x05\x63hart\x88\x01\x01\x42\x08\n\x06_chart"\xc8\x01\n\x19UpdateMetricConfigRequest\x12(\n\x10metric_config_id\x18\x01 \x01(\tR\x0emetricConfigId\x12\x44\n\x06update\x18\x02 \x01(\x0b\x32,.chalk.server.v1.UpdateMetricConfigOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"c\n\x1aUpdateMetricConfigResponse\x12\x45\n\rmetric_config\x18\x01 \x01(\x0b\x32 .chalk.artifacts.v1.MetricConfigR\x0cmetricConfig"\x9e\x02\n\x17GetChartSnapshotRequest\x12\x45\n\rmetric_config\x18\x01 \x01(\x0b\x32 .chalk.artifacts.v1.MetricConfigR\x0cmetricConfig\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12\x35\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x07\x65ndTime\x12\x32\n\x13use_start_as_origin\x18\x04 \x01(\x08H\x00R\x10useStartAsOrigin\x88\x01\x01\x42\x16\n\x14_use_start_as_origin"\xcf\x01\n\x18GetChartSnapshotResponse\x12<\n\x06\x63harts\x18\x01 \x03(\x0b\x32$.chalk.chart.v1.DenseTimeSeriesChartR\x06\x63harts\x12\x35\n\x08x_series\x18\x02 \x03(\x0b\x32\x1a.google.protobuf.TimestampR\x07xSeries\x12>\n\rwindow_period\x18\x03 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0cwindowPeriod">\n\x12\x44\x65leteChartRequest\x12(\n\x10metric_config_id\x18\x01 \x01(\tR\x0emetricConfigId"\x15\n\x13\x44\x65leteChartResponse2\x8b\x04\n\rChartsService\x12Z\n\nListCharts\x12".chalk.server.v1.ListChartsRequest\x1a#.chalk.server.v1.ListChartsResponse"\x03\x80}\x06\x12l\n\x10GetChartSnapshot\x12(.chalk.server.v1.GetChartSnapshotRequest\x1a).chalk.server.v1.GetChartSnapshotResponse"\x03\x80}\x06\x12r\n\x12UpdateMetricConfig\x12*.chalk.server.v1.UpdateMetricConfigRequest\x1a+.chalk.server.v1.UpdateMetricConfigResponse"\x03\x80}\x05\x12]\n\x0b\x43reateChart\x12#.chalk.server.v1.CreateChartRequest\x1a$.chalk.server.v1.CreateChartResponse"\x03\x80}\x05\x12]\n\x0b\x44\x65leteChart\x12#.chalk.server.v1.DeleteChartRequest\x1a$.chalk.server.v1.DeleteChartResponse"\x03\x80}\x05\x42\x93\x01\n\x13\x63om.chalk.server.v1B\nChartProtoP\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'
25
+ b'\n\x1b\x63halk/server/v1/chart.proto\x12\x0f\x63halk.server.v1\x1a\x1e\x63halk/artifacts/v1/chart.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a)chalk/chart/v1/densetimeserieschart.proto\x1a\x1egoogle/protobuf/duration.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"L\n\x06Series\x12\x16\n\x06points\x18\x01 \x03(\x01R\x06points\x12\x14\n\x05label\x18\x02 \x01(\tR\x05label\x12\x14\n\x05units\x18\x03 \x01(\tR\x05units"t\n\x05\x43hart\x12\x14\n\x05title\x18\x01 \x01(\tR\x05title\x12/\n\x06series\x18\x02 \x03(\x0b\x32\x17.chalk.server.v1.SeriesR\x06series\x12$\n\x0ex_timestamp_ms\x18\x04 \x03(\x03R\x0cxTimestampMs"0\n\x05Point\x12\x19\n\x05value\x18\x01 \x01(\x03H\x00R\x05value\x88\x01\x01:\x02\x18\x01\x42\x08\n\x06_value"l\n\nTimeSeries\x12.\n\x06points\x18\x01 \x03(\x0b\x32\x16.chalk.server.v1.PointR\x06points\x12\x14\n\x05label\x18\x02 \x01(\tR\x05label\x12\x14\n\x05units\x18\x03 \x01(\tR\x05units:\x02\x18\x01"\xd7\x01\n\x0fTimeSeriesChart\x12\x14\n\x05title\x18\x01 \x01(\tR\x05title\x12\x33\n\x06series\x18\x02 \x03(\x0b\x32\x1b.chalk.server.v1.TimeSeriesR\x06series\x12\x35\n\x08x_series\x18\x03 \x03(\x0b\x32\x1a.google.protobuf.TimestampR\x07xSeries\x12>\n\rwindow_period\x18\x04 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0cwindowPeriod:\x02\x18\x01"\xbe\x01\n\x11ListChartsFilters\x12P\n\x10link_entity_kind\x18\x01 \x01(\x0e\x32!.chalk.artifacts.v1.ChartLinkKindH\x00R\x0elinkEntityKind\x88\x01\x01\x12-\n\x10linked_entity_id\x18\x02 \x01(\tH\x01R\x0elinkedEntityId\x88\x01\x01\x42\x13\n\x11_link_entity_kindB\x13\n\x11_linked_entity_id"m\n\x12ListChartPageToken\x12@\n\x0e\x63reated_at_hwm\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0c\x63reatedAtHwm\x12\x15\n\x06id_hwm\x18\x02 \x01(\tR\x05idHwm"\xba\x01\n\x11ListChartsRequest\x12\x41\n\x07\x66ilters\x18\x01 \x01(\x0b\x32".chalk.server.v1.ListChartsFiltersH\x00R\x07\x66ilters\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x12"\n\npage_token\x18\x03 \x01(\tH\x02R\tpageToken\x88\x01\x01\x42\n\n\x08_filtersB\x08\n\x06_limitB\r\n\x0b_page_token"\xda\x01\n\x12ListChartsResponse\x12<\n\x06\x63harts\x18\x01 \x03(\x0b\x32 .chalk.artifacts.v1.MetricConfigB\x02\x18\x01R\x06\x63harts\x12\x45\n\x11\x63harts_with_links\x18\x02 \x03(\x0b\x32\x19.chalk.artifacts.v1.ChartR\x0f\x63hartsWithLinks\x12+\n\x0fnext_page_token\x18\x03 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token"\x89\x03\n\x1bUpdateMetricConfigOperation\x12\x17\n\x04name\x18\x01 \x01(\tH\x00R\x04name\x88\x01\x01\x12(\n\rwindow_period\x18\x02 \x01(\tH\x01R\x0cwindowPeriod\x88\x01\x01\x12>\n\x06series\x18\x03 \x03(\x0b\x32&.chalk.artifacts.v1.MetricConfigSeriesR\x06series\x12=\n\x08\x66ormulas\x18\x04 \x03(\x0b\x32!.chalk.artifacts.v1.MetricFormulaR\x08\x66ormulas\x12?\n\x07trigger\x18\x05 \x01(\x0b\x32 .chalk.artifacts.v1.AlertTriggerH\x02R\x07trigger\x88\x01\x01\x12,\n\x0fgraph_generated\x18\x06 \x01(\x08H\x03R\x0egraphGenerated\x88\x01\x01\x42\x07\n\x05_nameB\x10\n\x0e_window_periodB\n\n\x08_triggerB\x12\n\x10_graph_generated"\xc2\x03\n\x12\x43reateChartRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12#\n\rwindow_period\x18\x02 \x01(\tR\x0cwindowPeriod\x12>\n\x06series\x18\x03 \x03(\x0b\x32&.chalk.artifacts.v1.MetricConfigSeriesR\x06series\x12=\n\x08\x66ormulas\x18\x04 \x03(\x0b\x32!.chalk.artifacts.v1.MetricFormulaR\x08\x66ormulas\x12:\n\x07trigger\x18\x05 \x01(\x0b\x32 .chalk.artifacts.v1.AlertTriggerR\x07trigger\x12K\n\x10link_entity_kind\x18\x06 \x01(\x0e\x32!.chalk.artifacts.v1.ChartLinkKindR\x0elinkEntityKind\x12-\n\x10linked_entity_id\x18\x07 \x01(\tH\x00R\x0elinkedEntityId\x88\x01\x01\x12\'\n\x0fgraph_generated\x18\x08 \x01(\x08R\x0egraphGeneratedB\x13\n\x11_linked_entity_id"U\n\x13\x43reateChartResponse\x12\x34\n\x05\x63hart\x18\x02 \x01(\x0b\x32\x19.chalk.artifacts.v1.ChartH\x00R\x05\x63hart\x88\x01\x01\x42\x08\n\x06_chart"\xc8\x01\n\x19UpdateMetricConfigRequest\x12(\n\x10metric_config_id\x18\x01 \x01(\tR\x0emetricConfigId\x12\x44\n\x06update\x18\x02 \x01(\x0b\x32,.chalk.server.v1.UpdateMetricConfigOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"c\n\x1aUpdateMetricConfigResponse\x12\x45\n\rmetric_config\x18\x01 \x01(\x0b\x32 .chalk.artifacts.v1.MetricConfigR\x0cmetricConfig"\xd1\x03\n\x17GetChartSnapshotRequest\x12\x45\n\rmetric_config\x18\x01 \x01(\x0b\x32 .chalk.artifacts.v1.MetricConfigR\x0cmetricConfig\x12\x39\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12\x35\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x07\x65ndTime\x12\x32\n\x13use_start_as_origin\x18\x04 \x01(\x08H\x00R\x10useStartAsOrigin\x88\x01\x01\x12<\n\x18use_sketch_metrics_table\x18\x05 \x01(\x08H\x01R\x15useSketchMetricsTable\x88\x01\x01\x12:\n\x17return_sql_query_string\x18\x06 \x01(\x08H\x02R\x14returnSqlQueryString\x88\x01\x01\x42\x16\n\x14_use_start_as_originB\x1b\n\x19_use_sketch_metrics_tableB\x1a\n\x18_return_sql_query_string"\xfb\x01\n\x18GetChartSnapshotResponse\x12<\n\x06\x63harts\x18\x01 \x03(\x0b\x32$.chalk.chart.v1.DenseTimeSeriesChartR\x06\x63harts\x12\x35\n\x08x_series\x18\x02 \x03(\x0b\x32\x1a.google.protobuf.TimestampR\x07xSeries\x12>\n\rwindow_period\x18\x03 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0cwindowPeriod\x12*\n\x11sql_query_strings\x18\x04 \x03(\tR\x0fsqlQueryStrings">\n\x12\x44\x65leteChartRequest\x12(\n\x10metric_config_id\x18\x01 \x01(\tR\x0emetricConfigId"\x15\n\x13\x44\x65leteChartResponse2\x8b\x04\n\rChartsService\x12Z\n\nListCharts\x12".chalk.server.v1.ListChartsRequest\x1a#.chalk.server.v1.ListChartsResponse"\x03\x80}\x06\x12l\n\x10GetChartSnapshot\x12(.chalk.server.v1.GetChartSnapshotRequest\x1a).chalk.server.v1.GetChartSnapshotResponse"\x03\x80}\x06\x12r\n\x12UpdateMetricConfig\x12*.chalk.server.v1.UpdateMetricConfigRequest\x1a+.chalk.server.v1.UpdateMetricConfigResponse"\x03\x80}\x05\x12]\n\x0b\x43reateChart\x12#.chalk.server.v1.CreateChartRequest\x1a$.chalk.server.v1.CreateChartResponse"\x03\x80}\x05\x12]\n\x0b\x44\x65leteChart\x12#.chalk.server.v1.DeleteChartRequest\x1a$.chalk.server.v1.DeleteChartResponse"\x03\x80}\x05\x42\x93\x01\n\x13\x63om.chalk.server.v1B\nChartProtoP\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'
26
26
  )
27
27
 
28
28
  _globals = globals()
@@ -80,13 +80,13 @@ if _descriptor._USE_C_DESCRIPTORS == False:
80
80
  _globals["_UPDATEMETRICCONFIGRESPONSE"]._serialized_start = 2682
81
81
  _globals["_UPDATEMETRICCONFIGRESPONSE"]._serialized_end = 2781
82
82
  _globals["_GETCHARTSNAPSHOTREQUEST"]._serialized_start = 2784
83
- _globals["_GETCHARTSNAPSHOTREQUEST"]._serialized_end = 3070
84
- _globals["_GETCHARTSNAPSHOTRESPONSE"]._serialized_start = 3073
85
- _globals["_GETCHARTSNAPSHOTRESPONSE"]._serialized_end = 3280
86
- _globals["_DELETECHARTREQUEST"]._serialized_start = 3282
87
- _globals["_DELETECHARTREQUEST"]._serialized_end = 3344
88
- _globals["_DELETECHARTRESPONSE"]._serialized_start = 3346
89
- _globals["_DELETECHARTRESPONSE"]._serialized_end = 3367
90
- _globals["_CHARTSSERVICE"]._serialized_start = 3370
91
- _globals["_CHARTSSERVICE"]._serialized_end = 3893
83
+ _globals["_GETCHARTSNAPSHOTREQUEST"]._serialized_end = 3249
84
+ _globals["_GETCHARTSNAPSHOTRESPONSE"]._serialized_start = 3252
85
+ _globals["_GETCHARTSNAPSHOTRESPONSE"]._serialized_end = 3503
86
+ _globals["_DELETECHARTREQUEST"]._serialized_start = 3505
87
+ _globals["_DELETECHARTREQUEST"]._serialized_end = 3567
88
+ _globals["_DELETECHARTRESPONSE"]._serialized_start = 3569
89
+ _globals["_DELETECHARTRESPONSE"]._serialized_end = 3590
90
+ _globals["_CHARTSSERVICE"]._serialized_start = 3593
91
+ _globals["_CHARTSSERVICE"]._serialized_end = 4116
92
92
  # @@protoc_insertion_point(module_scope)
@@ -226,36 +226,52 @@ class UpdateMetricConfigResponse(_message.Message):
226
226
  def __init__(self, metric_config: _Optional[_Union[_chart_pb2.MetricConfig, _Mapping]] = ...) -> None: ...
227
227
 
228
228
  class GetChartSnapshotRequest(_message.Message):
229
- __slots__ = ("metric_config", "start_time", "end_time", "use_start_as_origin")
229
+ __slots__ = (
230
+ "metric_config",
231
+ "start_time",
232
+ "end_time",
233
+ "use_start_as_origin",
234
+ "use_sketch_metrics_table",
235
+ "return_sql_query_string",
236
+ )
230
237
  METRIC_CONFIG_FIELD_NUMBER: _ClassVar[int]
231
238
  START_TIME_FIELD_NUMBER: _ClassVar[int]
232
239
  END_TIME_FIELD_NUMBER: _ClassVar[int]
233
240
  USE_START_AS_ORIGIN_FIELD_NUMBER: _ClassVar[int]
241
+ USE_SKETCH_METRICS_TABLE_FIELD_NUMBER: _ClassVar[int]
242
+ RETURN_SQL_QUERY_STRING_FIELD_NUMBER: _ClassVar[int]
234
243
  metric_config: _chart_pb2.MetricConfig
235
244
  start_time: _timestamp_pb2.Timestamp
236
245
  end_time: _timestamp_pb2.Timestamp
237
246
  use_start_as_origin: bool
247
+ use_sketch_metrics_table: bool
248
+ return_sql_query_string: bool
238
249
  def __init__(
239
250
  self,
240
251
  metric_config: _Optional[_Union[_chart_pb2.MetricConfig, _Mapping]] = ...,
241
252
  start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
242
253
  end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
243
254
  use_start_as_origin: bool = ...,
255
+ use_sketch_metrics_table: bool = ...,
256
+ return_sql_query_string: bool = ...,
244
257
  ) -> None: ...
245
258
 
246
259
  class GetChartSnapshotResponse(_message.Message):
247
- __slots__ = ("charts", "x_series", "window_period")
260
+ __slots__ = ("charts", "x_series", "window_period", "sql_query_strings")
248
261
  CHARTS_FIELD_NUMBER: _ClassVar[int]
249
262
  X_SERIES_FIELD_NUMBER: _ClassVar[int]
250
263
  WINDOW_PERIOD_FIELD_NUMBER: _ClassVar[int]
264
+ SQL_QUERY_STRINGS_FIELD_NUMBER: _ClassVar[int]
251
265
  charts: _containers.RepeatedCompositeFieldContainer[_densetimeserieschart_pb2.DenseTimeSeriesChart]
252
266
  x_series: _containers.RepeatedCompositeFieldContainer[_timestamp_pb2.Timestamp]
253
267
  window_period: _duration_pb2.Duration
268
+ sql_query_strings: _containers.RepeatedScalarFieldContainer[str]
254
269
  def __init__(
255
270
  self,
256
271
  charts: _Optional[_Iterable[_Union[_densetimeserieschart_pb2.DenseTimeSeriesChart, _Mapping]]] = ...,
257
272
  x_series: _Optional[_Iterable[_Union[_timestamp_pb2.Timestamp, _Mapping]]] = ...,
258
273
  window_period: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
274
+ sql_query_strings: _Optional[_Iterable[str]] = ...,
259
275
  ) -> None: ...
260
276
 
261
277
  class DeleteChartRequest(_message.Message):
@@ -0,0 +1,42 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: chalk/server/v1/clickhouse.proto
4
+ # Protobuf Python Version: 4.25.3
5
+ """Generated protocol buffer code."""
6
+
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import symbol_database as _symbol_database
10
+ from google.protobuf.internal import builder as _builder
11
+ # @@protoc_insertion_point(imports)
12
+
13
+ _sym_db = _symbol_database.Default()
14
+
15
+
16
+ from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
17
+ from chalk._gen.chalk.utils.v1 import sensitive_pb2 as chalk_dot_utils_dot_v1_dot_sensitive__pb2
18
+
19
+
20
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
21
+ b'\n chalk/server/v1/clickhouse.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1e\x63halk/utils/v1/sensitive.proto"\x19\n\x17GetClickhouseUriRequest"2\n\x18GetClickhouseUriResponse\x12\x16\n\x03uri\x18\x01 \x01(\tB\x04\xd8\xa1\'\x01R\x03uri2\x84\x01\n\x11\x43lickhouseService\x12o\n\x10GetClickhouseUri\x12(.chalk.server.v1.GetClickhouseUriRequest\x1a).chalk.server.v1.GetClickhouseUriResponse"\x06\x90\x02\x01\x80}\x0b\x42\x98\x01\n\x13\x63om.chalk.server.v1B\x0f\x43lickhouseProtoP\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'
22
+ )
23
+
24
+ _globals = globals()
25
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
26
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.clickhouse_pb2", _globals)
27
+ if _descriptor._USE_C_DESCRIPTORS == False:
28
+ _globals["DESCRIPTOR"]._options = None
29
+ _globals[
30
+ "DESCRIPTOR"
31
+ ]._serialized_options = b"\n\023com.chalk.server.v1B\017ClickhouseProtoP\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"
32
+ _globals["_GETCLICKHOUSEURIRESPONSE"].fields_by_name["uri"]._options = None
33
+ _globals["_GETCLICKHOUSEURIRESPONSE"].fields_by_name["uri"]._serialized_options = b"\330\241'\001"
34
+ _globals["_CLICKHOUSESERVICE"].methods_by_name["GetClickhouseUri"]._options = None
35
+ _globals["_CLICKHOUSESERVICE"].methods_by_name["GetClickhouseUri"]._serialized_options = b"\220\002\001\200}\013"
36
+ _globals["_GETCLICKHOUSEURIREQUEST"]._serialized_start = 118
37
+ _globals["_GETCLICKHOUSEURIREQUEST"]._serialized_end = 143
38
+ _globals["_GETCLICKHOUSEURIRESPONSE"]._serialized_start = 145
39
+ _globals["_GETCLICKHOUSEURIRESPONSE"]._serialized_end = 195
40
+ _globals["_CLICKHOUSESERVICE"]._serialized_start = 198
41
+ _globals["_CLICKHOUSESERVICE"]._serialized_end = 330
42
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,17 @@
1
+ from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
2
+ from chalk._gen.chalk.utils.v1 import sensitive_pb2 as _sensitive_pb2
3
+ from google.protobuf import descriptor as _descriptor
4
+ from google.protobuf import message as _message
5
+ from typing import ClassVar as _ClassVar, Optional as _Optional
6
+
7
+ DESCRIPTOR: _descriptor.FileDescriptor
8
+
9
+ class GetClickhouseUriRequest(_message.Message):
10
+ __slots__ = ()
11
+ def __init__(self) -> None: ...
12
+
13
+ class GetClickhouseUriResponse(_message.Message):
14
+ __slots__ = ("uri",)
15
+ URI_FIELD_NUMBER: _ClassVar[int]
16
+ uri: str
17
+ def __init__(self, uri: _Optional[str] = ...) -> None: ...
@@ -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 clickhouse_pb2 as chalk_dot_server_dot_v1_dot_clickhouse__pb2
7
+
8
+
9
+ class ClickhouseServiceStub(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.GetClickhouseUri = channel.unary_unary(
19
+ "/chalk.server.v1.ClickhouseService/GetClickhouseUri",
20
+ request_serializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriRequest.SerializeToString,
21
+ response_deserializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriResponse.FromString,
22
+ )
23
+
24
+
25
+ class ClickhouseServiceServicer(object):
26
+ """Missing associated documentation comment in .proto file."""
27
+
28
+ def GetClickhouseUri(self, request, context):
29
+ """Get the Clickhouse connection URI for an environment"""
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_ClickhouseServiceServicer_to_server(servicer, server):
36
+ rpc_method_handlers = {
37
+ "GetClickhouseUri": grpc.unary_unary_rpc_method_handler(
38
+ servicer.GetClickhouseUri,
39
+ request_deserializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriRequest.FromString,
40
+ response_serializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriResponse.SerializeToString,
41
+ ),
42
+ }
43
+ generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.ClickhouseService", rpc_method_handlers)
44
+ server.add_generic_rpc_handlers((generic_handler,))
45
+
46
+
47
+ # This class is part of an EXPERIMENTAL API.
48
+ class ClickhouseService(object):
49
+ """Missing associated documentation comment in .proto file."""
50
+
51
+ @staticmethod
52
+ def GetClickhouseUri(
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.ClickhouseService/GetClickhouseUri",
68
+ chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriRequest.SerializeToString,
69
+ chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriResponse.FromString,
70
+ options,
71
+ channel_credentials,
72
+ insecure,
73
+ call_credentials,
74
+ compression,
75
+ wait_for_ready,
76
+ timeout,
77
+ metadata,
78
+ )