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
@@ -50,6 +50,8 @@ from chalk._gen.chalk.server.v1.cloud_components_pb2 import (
50
50
  ListCloudComponentStorageResponse,
51
51
  ListCloudComponentVpcRequest,
52
52
  ListCloudComponentVpcResponse,
53
+ TestClusterConnectionRequest,
54
+ TestClusterConnectionResponse,
53
55
  UpdateCloudComponentClusterRequest,
54
56
  UpdateCloudComponentClusterResponse,
55
57
  )
@@ -98,6 +100,10 @@ class CloudComponentsServiceStub:
98
100
  DeleteCloudComponentClusterRequest,
99
101
  DeleteCloudComponentClusterResponse,
100
102
  ]
103
+ TestClusterConnection: UnaryUnaryMultiCallable[
104
+ TestClusterConnectionRequest,
105
+ TestClusterConnectionResponse,
106
+ ]
101
107
  CreateCloudComponentStorage: UnaryUnaryMultiCallable[
102
108
  CreateCloudComponentStorageRequest,
103
109
  CreateCloudComponentStorageResponse,
@@ -207,6 +213,12 @@ class CloudComponentsServiceServicer(metaclass=ABCMeta):
207
213
  context: ServicerContext,
208
214
  ) -> DeleteCloudComponentClusterResponse: ...
209
215
  @abstractmethod
216
+ def TestClusterConnection(
217
+ self,
218
+ request: TestClusterConnectionRequest,
219
+ context: ServicerContext,
220
+ ) -> TestClusterConnectionResponse: ...
221
+ @abstractmethod
210
222
  def CreateCloudComponentStorage(
211
223
  self,
212
224
  request: CreateCloudComponentStorageRequest,
@@ -20,7 +20,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
20
20
 
21
21
 
22
22
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
23
- b'\n\'chalk/server/v1/cloud_credentials.proto\x12\x0f\x63halk.server.v1\x1a\x19\x63halk/auth/v1/audit.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/server/v1/environment.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x1d\n\x1bListCloudCredentialsRequest"k\n\x1cListCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x03(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials"\x93\x02\n\x18\x43loudCredentialsResponse\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x17\n\x07team_id\x18\x02 \x01(\tR\x06teamId\x12\x12\n\x04name\x18\x03 \x01(\tR\x04name\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12\x30\n\x04spec\x18\x05 \x01(\x0b\x32\x1c.chalk.server.v1.CloudConfigR\x04spec\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt"w\n\x17\x43loudCredentialsRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04kind\x18\x02 \x01(\tR\x04kind\x12\x34\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x1c.chalk.server.v1.CloudConfigR\x06\x63onfig"k\n\x1d\x43reateCloudCredentialsRequest\x12J\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32(.chalk.server.v1.CloudCredentialsRequestR\x0b\x63redentials"m\n\x1e\x43reateCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials",\n\x1aGetCloudCredentialsRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"j\n\x1bGetCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials"{\n\x1dUpdateCloudCredentialsRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12J\n\x0b\x63redentials\x18\x02 \x01(\x0b\x32(.chalk.server.v1.CloudCredentialsRequestR\x0b\x63redentials"m\n\x1eUpdateCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials"/\n\x1d\x44\x65leteCloudCredentialsRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id" \n\x1e\x44\x65leteCloudCredentialsResponse2\xfd\x05\n\x1e\x43loudAccountCredentialsService\x12{\n\x14ListCloudCredentials\x12,.chalk.server.v1.ListCloudCredentialsRequest\x1a-.chalk.server.v1.ListCloudCredentialsResponse"\x06\x90\x02\x01\x80}\x02\x12x\n\x13GetCloudCredentials\x12+.chalk.server.v1.GetCloudCredentialsRequest\x1a,.chalk.server.v1.GetCloudCredentialsResponse"\x06\x90\x02\x01\x80}\x02\x12\x9f\x01\n\x16\x43reateCloudCredentials\x12..chalk.server.v1.CreateCloudCredentialsRequest\x1a/.chalk.server.v1.CreateCloudCredentialsResponse"$\x80}\x1a\x8a\xd3\x0e\x1d\x08\x02\x12\x19\x43reated cloud credentials\x12\x9f\x01\n\x16UpdateCloudCredentials\x12..chalk.server.v1.UpdateCloudCredentialsRequest\x1a/.chalk.server.v1.UpdateCloudCredentialsResponse"$\x80}\x1a\x8a\xd3\x0e\x1d\x08\x02\x12\x19Updated cloud credentials\x12\x9f\x01\n\x16\x44\x65leteCloudCredentials\x12..chalk.server.v1.DeleteCloudCredentialsRequest\x1a/.chalk.server.v1.DeleteCloudCredentialsResponse"$\x80}\x1a\x8a\xd3\x0e\x1d\x08\x02\x12\x19\x44\x65leted cloud credentialsB\x9e\x01\n\x13\x63om.chalk.server.v1B\x15\x43loudCredentialsProtoP\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'
23
+ b'\n\'chalk/server/v1/cloud_credentials.proto\x12\x0f\x63halk.server.v1\x1a\x19\x63halk/auth/v1/audit.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/server/v1/environment.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x1d\n\x1bListCloudCredentialsRequest"k\n\x1cListCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x03(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials"\x93\x02\n\x18\x43loudCredentialsResponse\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x17\n\x07team_id\x18\x02 \x01(\tR\x06teamId\x12\x12\n\x04name\x18\x03 \x01(\tR\x04name\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12\x30\n\x04spec\x18\x05 \x01(\x0b\x32\x1c.chalk.server.v1.CloudConfigR\x04spec\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt"w\n\x17\x43loudCredentialsRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04kind\x18\x02 \x01(\tR\x04kind\x12\x34\n\x06\x63onfig\x18\x03 \x01(\x0b\x32\x1c.chalk.server.v1.CloudConfigR\x06\x63onfig"k\n\x1d\x43reateCloudCredentialsRequest\x12J\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32(.chalk.server.v1.CloudCredentialsRequestR\x0b\x63redentials"m\n\x1e\x43reateCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials",\n\x1aGetCloudCredentialsRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"j\n\x1bGetCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials"{\n\x1dUpdateCloudCredentialsRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12J\n\x0b\x63redentials\x18\x02 \x01(\x0b\x32(.chalk.server.v1.CloudCredentialsRequestR\x0b\x63redentials"m\n\x1eUpdateCloudCredentialsResponse\x12K\n\x0b\x63redentials\x18\x01 \x01(\x0b\x32).chalk.server.v1.CloudCredentialsResponseR\x0b\x63redentials"/\n\x1d\x44\x65leteCloudCredentialsRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id" \n\x1e\x44\x65leteCloudCredentialsResponse"\x82\x01\n\x1bTestCloudCredentialsRequest\x12\x10\n\x02id\x18\x01 \x01(\tH\x00R\x02id\x12\x42\n\x06\x63onfig\x18\x02 \x01(\x0b\x32(.chalk.server.v1.CloudCredentialsRequestH\x00R\x06\x63onfigB\r\n\x0b\x63redentials"h\n\x1cTestCloudCredentialsResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\x12\x18\n\x07message\x18\x02 \x01(\tR\x07message\x12\x14\n\x05\x65rror\x18\x03 \x01(\tR\x05\x65rror2\xfa\x06\n\x1e\x43loudAccountCredentialsService\x12{\n\x14ListCloudCredentials\x12,.chalk.server.v1.ListCloudCredentialsRequest\x1a-.chalk.server.v1.ListCloudCredentialsResponse"\x06\x90\x02\x01\x80}\x02\x12x\n\x13GetCloudCredentials\x12+.chalk.server.v1.GetCloudCredentialsRequest\x1a,.chalk.server.v1.GetCloudCredentialsResponse"\x06\x90\x02\x01\x80}\x02\x12\x9f\x01\n\x16\x43reateCloudCredentials\x12..chalk.server.v1.CreateCloudCredentialsRequest\x1a/.chalk.server.v1.CreateCloudCredentialsResponse"$\x80}\x1a\x8a\xd3\x0e\x1d\x08\x02\x12\x19\x43reated cloud credentials\x12\x9f\x01\n\x16UpdateCloudCredentials\x12..chalk.server.v1.UpdateCloudCredentialsRequest\x1a/.chalk.server.v1.UpdateCloudCredentialsResponse"$\x80}\x1a\x8a\xd3\x0e\x1d\x08\x02\x12\x19Updated cloud credentials\x12\x9f\x01\n\x16\x44\x65leteCloudCredentials\x12..chalk.server.v1.DeleteCloudCredentialsRequest\x1a/.chalk.server.v1.DeleteCloudCredentialsResponse"$\x80}\x1a\x8a\xd3\x0e\x1d\x08\x02\x12\x19\x44\x65leted cloud credentials\x12{\n\x14TestCloudCredentials\x12,.chalk.server.v1.TestCloudCredentialsRequest\x1a-.chalk.server.v1.TestCloudCredentialsResponse"\x06\x90\x02\x01\x80}\x02\x42\x9e\x01\n\x13\x63om.chalk.server.v1B\x15\x43loudCredentialsProtoP\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
24
  )
25
25
 
26
26
  _globals = globals()
@@ -51,6 +51,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
51
51
  _globals["_CLOUDACCOUNTCREDENTIALSSERVICE"].methods_by_name[
52
52
  "DeleteCloudCredentials"
53
53
  ]._serialized_options = b"\200}\032\212\323\016\035\010\002\022\031Deleted cloud credentials"
54
+ _globals["_CLOUDACCOUNTCREDENTIALSSERVICE"].methods_by_name["TestCloudCredentials"]._options = None
55
+ _globals["_CLOUDACCOUNTCREDENTIALSSERVICE"].methods_by_name[
56
+ "TestCloudCredentials"
57
+ ]._serialized_options = b"\220\002\001\200}\002"
54
58
  _globals["_LISTCLOUDCREDENTIALSREQUEST"]._serialized_start = 188
55
59
  _globals["_LISTCLOUDCREDENTIALSREQUEST"]._serialized_end = 217
56
60
  _globals["_LISTCLOUDCREDENTIALSRESPONSE"]._serialized_start = 219
@@ -75,6 +79,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
75
79
  _globals["_DELETECLOUDCREDENTIALSREQUEST"]._serialized_end = 1384
76
80
  _globals["_DELETECLOUDCREDENTIALSRESPONSE"]._serialized_start = 1386
77
81
  _globals["_DELETECLOUDCREDENTIALSRESPONSE"]._serialized_end = 1418
78
- _globals["_CLOUDACCOUNTCREDENTIALSSERVICE"]._serialized_start = 1421
79
- _globals["_CLOUDACCOUNTCREDENTIALSSERVICE"]._serialized_end = 2186
82
+ _globals["_TESTCLOUDCREDENTIALSREQUEST"]._serialized_start = 1421
83
+ _globals["_TESTCLOUDCREDENTIALSREQUEST"]._serialized_end = 1551
84
+ _globals["_TESTCLOUDCREDENTIALSRESPONSE"]._serialized_start = 1553
85
+ _globals["_TESTCLOUDCREDENTIALSRESPONSE"]._serialized_end = 1657
86
+ _globals["_CLOUDACCOUNTCREDENTIALSSERVICE"]._serialized_start = 1660
87
+ _globals["_CLOUDACCOUNTCREDENTIALSSERVICE"]._serialized_end = 2550
80
88
  # @@protoc_insertion_point(module_scope)
@@ -116,3 +116,23 @@ class DeleteCloudCredentialsRequest(_message.Message):
116
116
  class DeleteCloudCredentialsResponse(_message.Message):
117
117
  __slots__ = ()
118
118
  def __init__(self) -> None: ...
119
+
120
+ class TestCloudCredentialsRequest(_message.Message):
121
+ __slots__ = ("id", "config")
122
+ ID_FIELD_NUMBER: _ClassVar[int]
123
+ CONFIG_FIELD_NUMBER: _ClassVar[int]
124
+ id: str
125
+ config: CloudCredentialsRequest
126
+ def __init__(
127
+ self, id: _Optional[str] = ..., config: _Optional[_Union[CloudCredentialsRequest, _Mapping]] = ...
128
+ ) -> None: ...
129
+
130
+ class TestCloudCredentialsResponse(_message.Message):
131
+ __slots__ = ("success", "message", "error")
132
+ SUCCESS_FIELD_NUMBER: _ClassVar[int]
133
+ MESSAGE_FIELD_NUMBER: _ClassVar[int]
134
+ ERROR_FIELD_NUMBER: _ClassVar[int]
135
+ success: bool
136
+ message: str
137
+ error: str
138
+ def __init__(self, success: bool = ..., message: _Optional[str] = ..., error: _Optional[str] = ...) -> None: ...
@@ -40,6 +40,11 @@ class CloudAccountCredentialsServiceStub(object):
40
40
  request_serializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.DeleteCloudCredentialsRequest.SerializeToString,
41
41
  response_deserializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.DeleteCloudCredentialsResponse.FromString,
42
42
  )
43
+ self.TestCloudCredentials = channel.unary_unary(
44
+ "/chalk.server.v1.CloudAccountCredentialsService/TestCloudCredentials",
45
+ request_serializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.TestCloudCredentialsRequest.SerializeToString,
46
+ response_deserializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.TestCloudCredentialsResponse.FromString,
47
+ )
43
48
 
44
49
 
45
50
  class CloudAccountCredentialsServiceServicer(object):
@@ -75,6 +80,12 @@ class CloudAccountCredentialsServiceServicer(object):
75
80
  context.set_details("Method not implemented!")
76
81
  raise NotImplementedError("Method not implemented!")
77
82
 
83
+ def TestCloudCredentials(self, request, context):
84
+ """Missing associated documentation comment in .proto file."""
85
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
86
+ context.set_details("Method not implemented!")
87
+ raise NotImplementedError("Method not implemented!")
88
+
78
89
 
79
90
  def add_CloudAccountCredentialsServiceServicer_to_server(servicer, server):
80
91
  rpc_method_handlers = {
@@ -103,6 +114,11 @@ def add_CloudAccountCredentialsServiceServicer_to_server(servicer, server):
103
114
  request_deserializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.DeleteCloudCredentialsRequest.FromString,
104
115
  response_serializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.DeleteCloudCredentialsResponse.SerializeToString,
105
116
  ),
117
+ "TestCloudCredentials": grpc.unary_unary_rpc_method_handler(
118
+ servicer.TestCloudCredentials,
119
+ request_deserializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.TestCloudCredentialsRequest.FromString,
120
+ response_serializer=chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.TestCloudCredentialsResponse.SerializeToString,
121
+ ),
106
122
  }
107
123
  generic_handler = grpc.method_handlers_generic_handler(
108
124
  "chalk.server.v1.CloudAccountCredentialsService", rpc_method_handlers
@@ -258,3 +274,32 @@ class CloudAccountCredentialsService(object):
258
274
  timeout,
259
275
  metadata,
260
276
  )
277
+
278
+ @staticmethod
279
+ def TestCloudCredentials(
280
+ request,
281
+ target,
282
+ options=(),
283
+ channel_credentials=None,
284
+ call_credentials=None,
285
+ insecure=False,
286
+ compression=None,
287
+ wait_for_ready=None,
288
+ timeout=None,
289
+ metadata=None,
290
+ ):
291
+ return grpc.experimental.unary_unary(
292
+ request,
293
+ target,
294
+ "/chalk.server.v1.CloudAccountCredentialsService/TestCloudCredentials",
295
+ chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.TestCloudCredentialsRequest.SerializeToString,
296
+ chalk_dot_server_dot_v1_dot_cloud__credentials__pb2.TestCloudCredentialsResponse.FromString,
297
+ options,
298
+ channel_credentials,
299
+ insecure,
300
+ call_credentials,
301
+ compression,
302
+ wait_for_ready,
303
+ timeout,
304
+ metadata,
305
+ )
@@ -16,6 +16,8 @@ from chalk._gen.chalk.server.v1.cloud_credentials_pb2 import (
16
16
  GetCloudCredentialsResponse,
17
17
  ListCloudCredentialsRequest,
18
18
  ListCloudCredentialsResponse,
19
+ TestCloudCredentialsRequest,
20
+ TestCloudCredentialsResponse,
19
21
  UpdateCloudCredentialsRequest,
20
22
  UpdateCloudCredentialsResponse,
21
23
  )
@@ -48,6 +50,10 @@ class CloudAccountCredentialsServiceStub:
48
50
  DeleteCloudCredentialsRequest,
49
51
  DeleteCloudCredentialsResponse,
50
52
  ]
53
+ TestCloudCredentials: UnaryUnaryMultiCallable[
54
+ TestCloudCredentialsRequest,
55
+ TestCloudCredentialsResponse,
56
+ ]
51
57
 
52
58
  class CloudAccountCredentialsServiceServicer(metaclass=ABCMeta):
53
59
  @abstractmethod
@@ -80,6 +86,12 @@ class CloudAccountCredentialsServiceServicer(metaclass=ABCMeta):
80
86
  request: DeleteCloudCredentialsRequest,
81
87
  context: ServicerContext,
82
88
  ) -> DeleteCloudCredentialsResponse: ...
89
+ @abstractmethod
90
+ def TestCloudCredentials(
91
+ self,
92
+ request: TestCloudCredentialsRequest,
93
+ context: ServicerContext,
94
+ ) -> TestCloudCredentialsResponse: ...
83
95
 
84
96
  def add_CloudAccountCredentialsServiceServicer_to_server(
85
97
  servicer: CloudAccountCredentialsServiceServicer, server: Server
@@ -20,7 +20,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
20
20
 
21
21
 
22
22
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
23
- b'\n\'chalk/server/v1/dataplanejobqueue.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\'chalk/kubernetes/v1/resourcequota.proto\x1a&chalk/kubernetes/v1/scaledobject.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x9c\t\n\x0cJobQueueItem\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12\x39\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12\x1e\n\x08job_name\x18\x05 \x01(\tH\x00R\x07jobName\x88\x01\x01\x12\x1f\n\x0b\x61ttempt_idx\x18\x06 \x01(\x03R\nattemptIdx\x12\x34\n\x05state\x18\x07 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x42\n\x0cscheduled_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x0bscheduledAt\x88\x01\x01\x12\x31\n\x04kind\x18\t \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12\x19\n\x08job_args\x18\n \x01(\x0cR\x07jobArgs\x12%\n\x0eresource_group\x18\x0b \x01(\tR\rresourceGroup\x12\x42\n\x0c\x66inalized_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\x0b\x66inalizedAt\x88\x01\x01\x12K\n\x11last_attempted_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\x0flastAttemptedAt\x88\x01\x01\x12!\n\x0c\x61ttempted_by\x18\x0e \x03(\tR\x0b\x61ttemptedBy\x12K\n\x11last_heartbeat_at\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\x0flastHeartbeatAt\x88\x01\x01\x12&\n\x0coperation_id\x18\x10 \x01(\tH\x05R\x0boperationId\x88\x01\x01\x12Y\n\x18\x63\x61ncelation_requested_at\x18\x11 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x06R\x16\x63\x61ncelationRequestedAt\x88\x01\x01\x12!\n\x0cmax_attempts\x18\x12 \x01(\x03R\x0bmaxAttempts\x12\x34\n\x16mainline_deployment_id\x18\x13 \x01(\tR\x14mainlineDeploymentId\x12"\n\x08shard_id\x18\x14 \x01(\x03\x42\x02\x18\x01H\x07R\x07shardId\x88\x01\x01\x12 \n\tjob_index\x18\x15 \x01(\x03H\x08R\x08jobIndex\x88\x01\x01\x42\x0b\n\t_job_nameB\x0f\n\r_scheduled_atB\x0f\n\r_finalized_atB\x14\n\x12_last_attempted_atB\x14\n\x12_last_heartbeat_atB\x0f\n\r_operation_idB\x1b\n\x19_cancelation_requested_atB\x0b\n\t_shard_idB\x0c\n\n_job_index"-\n\x1bGetDataPlaneJobQueueRequest\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id"O\n\x1cGetDataPlaneJobQueueResponse\x12/\n\x03job\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.JobQueueItemR\x03job"\xa4\x02\n\x1cListDataPlaneJobQueueRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId\x12\x34\n\x05state\x18\x03 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x31\n\x04kind\x18\x04 \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12\x14\n\x05limit\x18\x05 \x01(\x05R\x05limit\x12\x16\n\x06offset\x18\x06 \x01(\x05R\x06offset\x12!\n\x0coperation_id\x18\x07 \x01(\tR\x0boperationId"h\n\x1dListDataPlaneJobQueueResponse\x12\x31\n\x04jobs\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.JobQueueItemR\x04jobs\x12\x14\n\x05total\x18\x02 \x01(\x05R\x05total"t\n$GetJobQueueAuxiliaryResourcesRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12%\n\x0eresource_group\x18\x02 \x01(\tR\rresourceGroup"\x8f\x03\n%GetJobQueueAuxiliaryResourcesResponse\x12\x8f\x01\n\x19\x64\x65ployment_scaled_objects\x18\x01 \x03(\x0b\x32S.chalk.server.v1.GetJobQueueAuxiliaryResourcesResponse.DeploymentScaledObjectsEntryR\x17\x64\x65ploymentScaledObjects\x12W\n\x0eresource_quota\x18\x02 \x01(\x0b\x32\x30.chalk.kubernetes.v1.KubernetesResourceQuotaDataR\rresourceQuota\x1a{\n\x1c\x44\x65ploymentScaledObjectsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x45\n\x05value\x18\x02 \x01(\x0b\x32/.chalk.kubernetes.v1.KubernetesScaledObjectDataR\x05value:\x02\x38\x01"\xc5\x04\n\x12JobQueueRowSummary\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12\x39\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x1f\n\x0b\x61ttempt_idx\x18\x06 \x01(\x05R\nattemptIdx\x12\x34\n\x05state\x18\x07 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x42\n\x0c\x66inalized_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x0b\x66inalizedAt\x88\x01\x01\x12K\n\x11last_attempted_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x0flastAttemptedAt\x88\x01\x01\x12%\n\x0e\x61ttempter_idxs\x18\x0e \x03(\x05R\rattempterIdxs\x12K\n\x11last_heartbeat_at\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\x0flastHeartbeatAt\x88\x01\x01\x12!\n\x0cmax_attempts\x18\x12 \x01(\x05R\x0bmaxAttempts\x12\x1c\n\x07job_idx\x18\x15 \x01(\x05H\x03R\x06jobIdx\x88\x01\x01\x42\x0f\n\r_finalized_atB\x14\n\x12_last_attempted_atB\x14\n\x12_last_heartbeat_atB\n\n\x08_job_idx"\xb8\x03\n\x18JobQueueOperationSummary\x12v\n\x15indexed_row_summaries\x18\x01 \x03(\x0b\x32\x42.chalk.server.v1.JobQueueOperationSummary.IndexedRowSummariesEntryR\x13indexedRowSummaries\x12.\n\x13resource_group_name\x18\x03 \x01(\tR\x11resourceGroupName\x12\x34\n\x16mainline_deployment_id\x18\x04 \x01(\tR\x14mainlineDeploymentId\x12\x31\n\x04kind\x18\x05 \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12\x1e\n\nattempters\x18\x06 \x03(\tR\nattempters\x1ak\n\x18IndexedRowSummariesEntry\x12\x10\n\x03key\x18\x01 \x01(\x05R\x03key\x12\x39\n\x05value\x18\x02 \x01(\x0b\x32#.chalk.server.v1.JobQueueRowSummaryR\x05value:\x02\x38\x01"\xbb\x01\n"GetJobQueueOperationSummaryRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x00R\x05limit\x88\x01\x01\x12\x1b\n\x06offset\x18\x04 \x01(\x05H\x01R\x06offset\x88\x01\x01\x42\x08\n\x06_limitB\t\n\x07_offset"{\n#GetJobQueueOperationSummaryResponse\x12H\n\x07summary\x18\x01 \x01(\x0b\x32).chalk.server.v1.JobQueueOperationSummaryH\x00R\x07summary\x88\x01\x01\x42\n\n\x08_summary"\x94\x04\n\x0fJobQueueAttempt\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12\x39\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12 \n\x0cjob_queue_id\x18\x05 \x01(\x05R\njobQueueId\x12\x1f\n\x0b\x61ttempt_idx\x18\x06 \x01(\x03R\nattemptIdx\x12\x37\n\tqueued_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x08queuedAt\x12>\n\nstarted_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tstartedAt\x88\x01\x01\x12&\n\x0c\x61ttempted_by\x18\t \x01(\tH\x01R\x0b\x61ttemptedBy\x88\x01\x01\x12\x14\n\x05state\x18\n \x01(\x05R\x05state\x12@\n\x0b\x66inished_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nfinishedAt\x88\x01\x01\x42\r\n\x0b_started_atB\x0f\n\r_attempted_byB\x0e\n\x0c_finished_at"\x8e\x01\n\x1bListJobQueueAttemptsRequest\x12%\n\x0cjob_queue_id\x18\x01 \x01(\x05H\x00R\njobQueueId\x88\x01\x01\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x01R\x0boperationId\x88\x01\x01\x42\x0f\n\r_job_queue_idB\x0f\n\r_operation_id"\\\n\x1cListJobQueueAttemptsResponse\x12<\n\x08\x61ttempts\x18\x01 \x03(\x0b\x32 .chalk.server.v1.JobQueueAttemptR\x08\x61ttempts*\xe4\x01\n\rJobQueueState\x12\x1f\n\x1bJOB_QUEUE_STATE_UNSPECIFIED\x10\x00\x12\x1d\n\x19JOB_QUEUE_STATE_SCHEDULED\x10\x01\x12\x1b\n\x17JOB_QUEUE_STATE_RUNNING\x10\x02\x12\x1d\n\x19JOB_QUEUE_STATE_COMPLETED\x10\x03\x12\x1a\n\x16JOB_QUEUE_STATE_FAILED\x10\x04\x12\x1c\n\x18JOB_QUEUE_STATE_CANCELED\x10\x05\x12\x1d\n\x19JOB_QUEUE_STATE_NOT_READY\x10\x06*\x9a\x01\n\x0cJobQueueKind\x12\x1e\n\x1aJOB_QUEUE_KIND_UNSPECIFIED\x10\x00\x12&\n"JOB_QUEUE_KIND_ASYNC_OFFLINE_QUERY\x10\x01\x12"\n\x1eJOB_QUEUE_KIND_SCHEDULED_QUERY\x10\x02\x12\x1e\n\x1aJOB_QUEUE_KIND_SCRIPT_TASK\x10\x03\x32\xc0\x05\n\x18\x44\x61taPlaneJobQueueService\x12{\n\x14GetDataPlaneJobQueue\x12,.chalk.server.v1.GetDataPlaneJobQueueRequest\x1a-.chalk.server.v1.GetDataPlaneJobQueueResponse"\x06\x90\x02\x01\x80}\x10\x12~\n\x15ListDataPlaneJobQueue\x12-.chalk.server.v1.ListDataPlaneJobQueueRequest\x1a..chalk.server.v1.ListDataPlaneJobQueueResponse"\x06\x90\x02\x01\x80}\x10\x12\x96\x01\n\x1dGetJobQueueAuxiliaryResources\x12\x35.chalk.server.v1.GetJobQueueAuxiliaryResourcesRequest\x1a\x36.chalk.server.v1.GetJobQueueAuxiliaryResourcesResponse"\x06\x90\x02\x01\x80}\x0b\x12\x90\x01\n\x1bGetJobQueueOperationSummary\x12\x33.chalk.server.v1.GetJobQueueOperationSummaryRequest\x1a\x34.chalk.server.v1.GetJobQueueOperationSummaryResponse"\x06\x90\x02\x01\x80}\x10\x12{\n\x14ListJobQueueAttempts\x12,.chalk.server.v1.ListJobQueueAttemptsRequest\x1a-.chalk.server.v1.ListJobQueueAttemptsResponse"\x06\x90\x02\x01\x80}\x10\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16\x44\x61taplanejobqueueProtoP\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'
23
+ b'\n\'chalk/server/v1/dataplanejobqueue.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\'chalk/kubernetes/v1/resourcequota.proto\x1a&chalk/kubernetes/v1/scaledobject.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xae\n\n\x0cJobQueueItem\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12\x39\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12\x1e\n\x08job_name\x18\x05 \x01(\tH\x00R\x07jobName\x88\x01\x01\x12\x1f\n\x0b\x61ttempt_idx\x18\x06 \x01(\x03R\nattemptIdx\x12\x34\n\x05state\x18\x07 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x42\n\x0cscheduled_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x0bscheduledAt\x88\x01\x01\x12\x31\n\x04kind\x18\t \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12\x19\n\x08job_args\x18\n \x01(\x0cR\x07jobArgs\x12%\n\x0eresource_group\x18\x0b \x01(\tR\rresourceGroup\x12\x42\n\x0c\x66inalized_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\x0b\x66inalizedAt\x88\x01\x01\x12K\n\x11last_attempted_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\x0flastAttemptedAt\x88\x01\x01\x12!\n\x0c\x61ttempted_by\x18\x0e \x03(\tR\x0b\x61ttemptedBy\x12K\n\x11last_heartbeat_at\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\x0flastHeartbeatAt\x88\x01\x01\x12&\n\x0coperation_id\x18\x10 \x01(\tH\x05R\x0boperationId\x88\x01\x01\x12Y\n\x18\x63\x61ncelation_requested_at\x18\x11 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x06R\x16\x63\x61ncelationRequestedAt\x88\x01\x01\x12!\n\x0cmax_attempts\x18\x12 \x01(\x03R\x0bmaxAttempts\x12\x34\n\x16mainline_deployment_id\x18\x13 \x01(\tR\x14mainlineDeploymentId\x12"\n\x08shard_id\x18\x14 \x01(\x03\x42\x02\x18\x01H\x07R\x07shardId\x88\x01\x01\x12 \n\tjob_index\x18\x15 \x01(\x03H\x08R\x08jobIndex\x88\x01\x01\x12*\n\x0e\x66\x65tch_priority\x18\x16 \x01(\x03H\tR\rfetchPriority\x88\x01\x01\x12\x37\n\x15workflow_execution_id\x18\x17 \x01(\tH\nR\x13workflowExecutionId\x88\x01\x01\x42\x0b\n\t_job_nameB\x0f\n\r_scheduled_atB\x0f\n\r_finalized_atB\x14\n\x12_last_attempted_atB\x14\n\x12_last_heartbeat_atB\x0f\n\r_operation_idB\x1b\n\x19_cancelation_requested_atB\x0b\n\t_shard_idB\x0c\n\n_job_indexB\x11\n\x0f_fetch_priorityB\x18\n\x16_workflow_execution_id"-\n\x1bGetDataPlaneJobQueueRequest\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id"O\n\x1cGetDataPlaneJobQueueResponse\x12/\n\x03job\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.JobQueueItemR\x03job"\xa4\x02\n\x1cListDataPlaneJobQueueRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId\x12\x34\n\x05state\x18\x03 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x31\n\x04kind\x18\x04 \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12\x14\n\x05limit\x18\x05 \x01(\x05R\x05limit\x12\x16\n\x06offset\x18\x06 \x01(\x05R\x06offset\x12!\n\x0coperation_id\x18\x07 \x01(\tR\x0boperationId"h\n\x1dListDataPlaneJobQueueResponse\x12\x31\n\x04jobs\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.JobQueueItemR\x04jobs\x12\x14\n\x05total\x18\x02 \x01(\x05R\x05total"t\n$GetJobQueueAuxiliaryResourcesRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12%\n\x0eresource_group\x18\x02 \x01(\tR\rresourceGroup"\x8f\x03\n%GetJobQueueAuxiliaryResourcesResponse\x12\x8f\x01\n\x19\x64\x65ployment_scaled_objects\x18\x01 \x03(\x0b\x32S.chalk.server.v1.GetJobQueueAuxiliaryResourcesResponse.DeploymentScaledObjectsEntryR\x17\x64\x65ploymentScaledObjects\x12W\n\x0eresource_quota\x18\x02 \x01(\x0b\x32\x30.chalk.kubernetes.v1.KubernetesResourceQuotaDataR\rresourceQuota\x1a{\n\x1c\x44\x65ploymentScaledObjectsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x45\n\x05value\x18\x02 \x01(\x0b\x32/.chalk.kubernetes.v1.KubernetesScaledObjectDataR\x05value:\x02\x38\x01"\xc5\x04\n\x12JobQueueRowSummary\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12\x39\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x1f\n\x0b\x61ttempt_idx\x18\x06 \x01(\x05R\nattemptIdx\x12\x34\n\x05state\x18\x07 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x42\n\x0c\x66inalized_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x0b\x66inalizedAt\x88\x01\x01\x12K\n\x11last_attempted_at\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x0flastAttemptedAt\x88\x01\x01\x12%\n\x0e\x61ttempter_idxs\x18\x0e \x03(\x05R\rattempterIdxs\x12K\n\x11last_heartbeat_at\x18\x0f \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\x0flastHeartbeatAt\x88\x01\x01\x12!\n\x0cmax_attempts\x18\x12 \x01(\x05R\x0bmaxAttempts\x12\x1c\n\x07job_idx\x18\x15 \x01(\x05H\x03R\x06jobIdx\x88\x01\x01\x42\x0f\n\r_finalized_atB\x14\n\x12_last_attempted_atB\x14\n\x12_last_heartbeat_atB\n\n\x08_job_idx"\xdc\x04\n\x18JobQueueOperationSummary\x12v\n\x15indexed_row_summaries\x18\x01 \x03(\x0b\x32\x42.chalk.server.v1.JobQueueOperationSummary.IndexedRowSummariesEntryR\x13indexedRowSummaries\x12.\n\x13resource_group_name\x18\x03 \x01(\tR\x11resourceGroupName\x12\x34\n\x16mainline_deployment_id\x18\x04 \x01(\tR\x14mainlineDeploymentId\x12\x31\n\x04kind\x18\x05 \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12\x1e\n\nattempters\x18\x06 \x03(\tR\nattempters\x12\x37\n\x15workflow_execution_id\x18\x07 \x01(\tH\x00R\x13workflowExecutionId\x88\x01\x01\x12O\n$unsatisfied_dependency_operation_ids\x18\x08 \x03(\tR!unsatisfiedDependencyOperationIds\x1ak\n\x18IndexedRowSummariesEntry\x12\x10\n\x03key\x18\x01 \x01(\x05R\x03key\x12\x39\n\x05value\x18\x02 \x01(\x0b\x32#.chalk.server.v1.JobQueueRowSummaryR\x05value:\x02\x38\x01\x42\x18\n\x16_workflow_execution_id"\xbb\x01\n"GetJobQueueOperationSummaryRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x00R\x05limit\x88\x01\x01\x12\x1b\n\x06offset\x18\x04 \x01(\x05H\x01R\x06offset\x88\x01\x01\x42\x08\n\x06_limitB\t\n\x07_offset"{\n#GetJobQueueOperationSummaryResponse\x12H\n\x07summary\x18\x01 \x01(\x0b\x32).chalk.server.v1.JobQueueOperationSummaryH\x00R\x07summary\x88\x01\x01\x42\n\n\x08_summary"\x94\x04\n\x0fJobQueueAttempt\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12\x39\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12 \n\x0cjob_queue_id\x18\x05 \x01(\x05R\njobQueueId\x12\x1f\n\x0b\x61ttempt_idx\x18\x06 \x01(\x03R\nattemptIdx\x12\x37\n\tqueued_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x08queuedAt\x12>\n\nstarted_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tstartedAt\x88\x01\x01\x12&\n\x0c\x61ttempted_by\x18\t \x01(\tH\x01R\x0b\x61ttemptedBy\x88\x01\x01\x12\x14\n\x05state\x18\n \x01(\x05R\x05state\x12@\n\x0b\x66inished_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nfinishedAt\x88\x01\x01\x42\r\n\x0b_started_atB\x0f\n\r_attempted_byB\x0e\n\x0c_finished_at"\x8e\x01\n\x1bListJobQueueAttemptsRequest\x12%\n\x0cjob_queue_id\x18\x01 \x01(\x05H\x00R\njobQueueId\x88\x01\x01\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x01R\x0boperationId\x88\x01\x01\x42\x0f\n\r_job_queue_idB\x0f\n\r_operation_id"\\\n\x1cListJobQueueAttemptsResponse\x12<\n\x08\x61ttempts\x18\x01 \x03(\x0b\x32 .chalk.server.v1.JobQueueAttemptR\x08\x61ttempts"Q\n\x11OperationAndShard\x12!\n\x0coperation_id\x18\x01 \x01(\tR\x0boperationId\x12\x19\n\x08shard_id\x18\x02 \x01(\x03R\x07shardId"\xff\x01\n\x1d\x46orceCancelJobQueueJobRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12*\n\x0f\x62y_job_queue_id\x18\x02 \x01(\x05H\x00R\x0c\x62yJobQueueId\x88\x01\x01\x12\\\n\x16\x62y_operation_and_shard\x18\x03 \x01(\x0b\x32".chalk.server.v1.OperationAndShardH\x01R\x13\x62yOperationAndShard\x88\x01\x01\x42\x12\n\x10_by_job_queue_idB\x19\n\x17_by_operation_and_shard" \n\x1e\x46orceCancelJobQueueJobResponse"`\n\x1e\x43\x61ncelWorkflowExecutionRequest\x12\x34\n\x15workflow_execution_id\x18\x01 \x01(\tH\x00R\x13workflowExecutionIdB\x08\n\x06method"!\n\x1f\x43\x61ncelWorkflowExecutionResponse*\x81\x02\n\rJobQueueState\x12\x1f\n\x1bJOB_QUEUE_STATE_UNSPECIFIED\x10\x00\x12\x1d\n\x19JOB_QUEUE_STATE_SCHEDULED\x10\x01\x12\x1b\n\x17JOB_QUEUE_STATE_RUNNING\x10\x02\x12\x1d\n\x19JOB_QUEUE_STATE_COMPLETED\x10\x03\x12\x1a\n\x16JOB_QUEUE_STATE_FAILED\x10\x04\x12\x1c\n\x18JOB_QUEUE_STATE_CANCELED\x10\x05\x12\x1d\n\x19JOB_QUEUE_STATE_NOT_READY\x10\x06\x12\x1b\n\x17JOB_QUEUE_STATE_WAITING\x10\x07*\x9a\x01\n\x0cJobQueueKind\x12\x1e\n\x1aJOB_QUEUE_KIND_UNSPECIFIED\x10\x00\x12&\n"JOB_QUEUE_KIND_ASYNC_OFFLINE_QUERY\x10\x01\x12"\n\x1eJOB_QUEUE_KIND_SCHEDULED_QUERY\x10\x02\x12\x1e\n\x1aJOB_QUEUE_KIND_SCRIPT_TASK\x10\x03\x32\xc4\x07\n\x18\x44\x61taPlaneJobQueueService\x12{\n\x14GetDataPlaneJobQueue\x12,.chalk.server.v1.GetDataPlaneJobQueueRequest\x1a-.chalk.server.v1.GetDataPlaneJobQueueResponse"\x06\x90\x02\x01\x80}\x10\x12~\n\x15ListDataPlaneJobQueue\x12-.chalk.server.v1.ListDataPlaneJobQueueRequest\x1a..chalk.server.v1.ListDataPlaneJobQueueResponse"\x06\x90\x02\x01\x80}\x10\x12\x96\x01\n\x1dGetJobQueueAuxiliaryResources\x12\x35.chalk.server.v1.GetJobQueueAuxiliaryResourcesRequest\x1a\x36.chalk.server.v1.GetJobQueueAuxiliaryResourcesResponse"\x06\x90\x02\x01\x80}\x0b\x12\x90\x01\n\x1bGetJobQueueOperationSummary\x12\x33.chalk.server.v1.GetJobQueueOperationSummaryRequest\x1a\x34.chalk.server.v1.GetJobQueueOperationSummaryResponse"\x06\x90\x02\x01\x80}\x10\x12{\n\x14ListJobQueueAttempts\x12,.chalk.server.v1.ListJobQueueAttemptsRequest\x1a-.chalk.server.v1.ListJobQueueAttemptsResponse"\x06\x90\x02\x01\x80}\x10\x12~\n\x16\x46orceCancelJobQueueJob\x12..chalk.server.v1.ForceCancelJobQueueJobRequest\x1a/.chalk.server.v1.ForceCancelJobQueueJobResponse"\x03\x80}\x04\x12\x81\x01\n\x17\x43\x61ncelWorkflowExecution\x12/.chalk.server.v1.CancelWorkflowExecutionRequest\x1a\x30.chalk.server.v1.CancelWorkflowExecutionResponse"\x03\x80}\x04\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16\x44\x61taplanejobqueueProtoP\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
24
  )
25
25
 
26
26
  _globals = globals()
@@ -57,42 +57,56 @@ if _descriptor._USE_C_DESCRIPTORS == False:
57
57
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name[
58
58
  "ListJobQueueAttempts"
59
59
  ]._serialized_options = b"\220\002\001\200}\020"
60
- _globals["_JOBQUEUESTATE"]._serialized_start = 4556
61
- _globals["_JOBQUEUESTATE"]._serialized_end = 4784
62
- _globals["_JOBQUEUEKIND"]._serialized_start = 4787
63
- _globals["_JOBQUEUEKIND"]._serialized_end = 4941
60
+ _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["ForceCancelJobQueueJob"]._options = None
61
+ _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["ForceCancelJobQueueJob"]._serialized_options = b"\200}\004"
62
+ _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["CancelWorkflowExecution"]._options = None
63
+ _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["CancelWorkflowExecution"]._serialized_options = b"\200}\004"
64
+ _globals["_JOBQUEUESTATE"]._serialized_start = 5374
65
+ _globals["_JOBQUEUESTATE"]._serialized_end = 5631
66
+ _globals["_JOBQUEUEKIND"]._serialized_start = 5634
67
+ _globals["_JOBQUEUEKIND"]._serialized_end = 5788
64
68
  _globals["_JOBQUEUEITEM"]._serialized_start = 208
65
- _globals["_JOBQUEUEITEM"]._serialized_end = 1388
66
- _globals["_GETDATAPLANEJOBQUEUEREQUEST"]._serialized_start = 1390
67
- _globals["_GETDATAPLANEJOBQUEUEREQUEST"]._serialized_end = 1435
68
- _globals["_GETDATAPLANEJOBQUEUERESPONSE"]._serialized_start = 1437
69
- _globals["_GETDATAPLANEJOBQUEUERESPONSE"]._serialized_end = 1516
70
- _globals["_LISTDATAPLANEJOBQUEUEREQUEST"]._serialized_start = 1519
71
- _globals["_LISTDATAPLANEJOBQUEUEREQUEST"]._serialized_end = 1811
72
- _globals["_LISTDATAPLANEJOBQUEUERESPONSE"]._serialized_start = 1813
73
- _globals["_LISTDATAPLANEJOBQUEUERESPONSE"]._serialized_end = 1917
74
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESREQUEST"]._serialized_start = 1919
75
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESREQUEST"]._serialized_end = 2035
76
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE"]._serialized_start = 2038
77
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE"]._serialized_end = 2437
78
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE_DEPLOYMENTSCALEDOBJECTSENTRY"]._serialized_start = 2314
79
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE_DEPLOYMENTSCALEDOBJECTSENTRY"]._serialized_end = 2437
80
- _globals["_JOBQUEUEROWSUMMARY"]._serialized_start = 2440
81
- _globals["_JOBQUEUEROWSUMMARY"]._serialized_end = 3021
82
- _globals["_JOBQUEUEOPERATIONSUMMARY"]._serialized_start = 3024
83
- _globals["_JOBQUEUEOPERATIONSUMMARY"]._serialized_end = 3464
84
- _globals["_JOBQUEUEOPERATIONSUMMARY_INDEXEDROWSUMMARIESENTRY"]._serialized_start = 3357
85
- _globals["_JOBQUEUEOPERATIONSUMMARY_INDEXEDROWSUMMARIESENTRY"]._serialized_end = 3464
86
- _globals["_GETJOBQUEUEOPERATIONSUMMARYREQUEST"]._serialized_start = 3467
87
- _globals["_GETJOBQUEUEOPERATIONSUMMARYREQUEST"]._serialized_end = 3654
88
- _globals["_GETJOBQUEUEOPERATIONSUMMARYRESPONSE"]._serialized_start = 3656
89
- _globals["_GETJOBQUEUEOPERATIONSUMMARYRESPONSE"]._serialized_end = 3779
90
- _globals["_JOBQUEUEATTEMPT"]._serialized_start = 3782
91
- _globals["_JOBQUEUEATTEMPT"]._serialized_end = 4314
92
- _globals["_LISTJOBQUEUEATTEMPTSREQUEST"]._serialized_start = 4317
93
- _globals["_LISTJOBQUEUEATTEMPTSREQUEST"]._serialized_end = 4459
94
- _globals["_LISTJOBQUEUEATTEMPTSRESPONSE"]._serialized_start = 4461
95
- _globals["_LISTJOBQUEUEATTEMPTSRESPONSE"]._serialized_end = 4553
96
- _globals["_DATAPLANEJOBQUEUESERVICE"]._serialized_start = 4944
97
- _globals["_DATAPLANEJOBQUEUESERVICE"]._serialized_end = 5648
69
+ _globals["_JOBQUEUEITEM"]._serialized_end = 1534
70
+ _globals["_GETDATAPLANEJOBQUEUEREQUEST"]._serialized_start = 1536
71
+ _globals["_GETDATAPLANEJOBQUEUEREQUEST"]._serialized_end = 1581
72
+ _globals["_GETDATAPLANEJOBQUEUERESPONSE"]._serialized_start = 1583
73
+ _globals["_GETDATAPLANEJOBQUEUERESPONSE"]._serialized_end = 1662
74
+ _globals["_LISTDATAPLANEJOBQUEUEREQUEST"]._serialized_start = 1665
75
+ _globals["_LISTDATAPLANEJOBQUEUEREQUEST"]._serialized_end = 1957
76
+ _globals["_LISTDATAPLANEJOBQUEUERESPONSE"]._serialized_start = 1959
77
+ _globals["_LISTDATAPLANEJOBQUEUERESPONSE"]._serialized_end = 2063
78
+ _globals["_GETJOBQUEUEAUXILIARYRESOURCESREQUEST"]._serialized_start = 2065
79
+ _globals["_GETJOBQUEUEAUXILIARYRESOURCESREQUEST"]._serialized_end = 2181
80
+ _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE"]._serialized_start = 2184
81
+ _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE"]._serialized_end = 2583
82
+ _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE_DEPLOYMENTSCALEDOBJECTSENTRY"]._serialized_start = 2460
83
+ _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE_DEPLOYMENTSCALEDOBJECTSENTRY"]._serialized_end = 2583
84
+ _globals["_JOBQUEUEROWSUMMARY"]._serialized_start = 2586
85
+ _globals["_JOBQUEUEROWSUMMARY"]._serialized_end = 3167
86
+ _globals["_JOBQUEUEOPERATIONSUMMARY"]._serialized_start = 3170
87
+ _globals["_JOBQUEUEOPERATIONSUMMARY"]._serialized_end = 3774
88
+ _globals["_JOBQUEUEOPERATIONSUMMARY_INDEXEDROWSUMMARIESENTRY"]._serialized_start = 3641
89
+ _globals["_JOBQUEUEOPERATIONSUMMARY_INDEXEDROWSUMMARIESENTRY"]._serialized_end = 3748
90
+ _globals["_GETJOBQUEUEOPERATIONSUMMARYREQUEST"]._serialized_start = 3777
91
+ _globals["_GETJOBQUEUEOPERATIONSUMMARYREQUEST"]._serialized_end = 3964
92
+ _globals["_GETJOBQUEUEOPERATIONSUMMARYRESPONSE"]._serialized_start = 3966
93
+ _globals["_GETJOBQUEUEOPERATIONSUMMARYRESPONSE"]._serialized_end = 4089
94
+ _globals["_JOBQUEUEATTEMPT"]._serialized_start = 4092
95
+ _globals["_JOBQUEUEATTEMPT"]._serialized_end = 4624
96
+ _globals["_LISTJOBQUEUEATTEMPTSREQUEST"]._serialized_start = 4627
97
+ _globals["_LISTJOBQUEUEATTEMPTSREQUEST"]._serialized_end = 4769
98
+ _globals["_LISTJOBQUEUEATTEMPTSRESPONSE"]._serialized_start = 4771
99
+ _globals["_LISTJOBQUEUEATTEMPTSRESPONSE"]._serialized_end = 4863
100
+ _globals["_OPERATIONANDSHARD"]._serialized_start = 4865
101
+ _globals["_OPERATIONANDSHARD"]._serialized_end = 4946
102
+ _globals["_FORCECANCELJOBQUEUEJOBREQUEST"]._serialized_start = 4949
103
+ _globals["_FORCECANCELJOBQUEUEJOBREQUEST"]._serialized_end = 5204
104
+ _globals["_FORCECANCELJOBQUEUEJOBRESPONSE"]._serialized_start = 5206
105
+ _globals["_FORCECANCELJOBQUEUEJOBRESPONSE"]._serialized_end = 5238
106
+ _globals["_CANCELWORKFLOWEXECUTIONREQUEST"]._serialized_start = 5240
107
+ _globals["_CANCELWORKFLOWEXECUTIONREQUEST"]._serialized_end = 5336
108
+ _globals["_CANCELWORKFLOWEXECUTIONRESPONSE"]._serialized_start = 5338
109
+ _globals["_CANCELWORKFLOWEXECUTIONRESPONSE"]._serialized_end = 5371
110
+ _globals["_DATAPLANEJOBQUEUESERVICE"]._serialized_start = 5791
111
+ _globals["_DATAPLANEJOBQUEUESERVICE"]._serialized_end = 6755
98
112
  # @@protoc_insertion_point(module_scope)
@@ -25,6 +25,7 @@ class JobQueueState(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
25
25
  JOB_QUEUE_STATE_FAILED: _ClassVar[JobQueueState]
26
26
  JOB_QUEUE_STATE_CANCELED: _ClassVar[JobQueueState]
27
27
  JOB_QUEUE_STATE_NOT_READY: _ClassVar[JobQueueState]
28
+ JOB_QUEUE_STATE_WAITING: _ClassVar[JobQueueState]
28
29
 
29
30
  class JobQueueKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
30
31
  __slots__ = ()
@@ -40,6 +41,7 @@ JOB_QUEUE_STATE_COMPLETED: JobQueueState
40
41
  JOB_QUEUE_STATE_FAILED: JobQueueState
41
42
  JOB_QUEUE_STATE_CANCELED: JobQueueState
42
43
  JOB_QUEUE_STATE_NOT_READY: JobQueueState
44
+ JOB_QUEUE_STATE_WAITING: JobQueueState
43
45
  JOB_QUEUE_KIND_UNSPECIFIED: JobQueueKind
44
46
  JOB_QUEUE_KIND_ASYNC_OFFLINE_QUERY: JobQueueKind
45
47
  JOB_QUEUE_KIND_SCHEDULED_QUERY: JobQueueKind
@@ -68,6 +70,8 @@ class JobQueueItem(_message.Message):
68
70
  "mainline_deployment_id",
69
71
  "shard_id",
70
72
  "job_index",
73
+ "fetch_priority",
74
+ "workflow_execution_id",
71
75
  )
72
76
  ID_FIELD_NUMBER: _ClassVar[int]
73
77
  CREATED_AT_FIELD_NUMBER: _ClassVar[int]
@@ -90,6 +94,8 @@ class JobQueueItem(_message.Message):
90
94
  MAINLINE_DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
91
95
  SHARD_ID_FIELD_NUMBER: _ClassVar[int]
92
96
  JOB_INDEX_FIELD_NUMBER: _ClassVar[int]
97
+ FETCH_PRIORITY_FIELD_NUMBER: _ClassVar[int]
98
+ WORKFLOW_EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
93
99
  id: int
94
100
  created_at: _timestamp_pb2.Timestamp
95
101
  environment_id: str
@@ -111,6 +117,8 @@ class JobQueueItem(_message.Message):
111
117
  mainline_deployment_id: str
112
118
  shard_id: int
113
119
  job_index: int
120
+ fetch_priority: int
121
+ workflow_execution_id: str
114
122
  def __init__(
115
123
  self,
116
124
  id: _Optional[int] = ...,
@@ -134,6 +142,8 @@ class JobQueueItem(_message.Message):
134
142
  mainline_deployment_id: _Optional[str] = ...,
135
143
  shard_id: _Optional[int] = ...,
136
144
  job_index: _Optional[int] = ...,
145
+ fetch_priority: _Optional[int] = ...,
146
+ workflow_execution_id: _Optional[str] = ...,
137
147
  ) -> None: ...
138
148
 
139
149
  class GetDataPlaneJobQueueRequest(_message.Message):
@@ -265,7 +275,15 @@ class JobQueueRowSummary(_message.Message):
265
275
  ) -> None: ...
266
276
 
267
277
  class JobQueueOperationSummary(_message.Message):
268
- __slots__ = ("indexed_row_summaries", "resource_group_name", "mainline_deployment_id", "kind", "attempters")
278
+ __slots__ = (
279
+ "indexed_row_summaries",
280
+ "resource_group_name",
281
+ "mainline_deployment_id",
282
+ "kind",
283
+ "attempters",
284
+ "workflow_execution_id",
285
+ "unsatisfied_dependency_operation_ids",
286
+ )
269
287
  class IndexedRowSummariesEntry(_message.Message):
270
288
  __slots__ = ("key", "value")
271
289
  KEY_FIELD_NUMBER: _ClassVar[int]
@@ -281,11 +299,15 @@ class JobQueueOperationSummary(_message.Message):
281
299
  MAINLINE_DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
282
300
  KIND_FIELD_NUMBER: _ClassVar[int]
283
301
  ATTEMPTERS_FIELD_NUMBER: _ClassVar[int]
302
+ WORKFLOW_EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
303
+ UNSATISFIED_DEPENDENCY_OPERATION_IDS_FIELD_NUMBER: _ClassVar[int]
284
304
  indexed_row_summaries: _containers.MessageMap[int, JobQueueRowSummary]
285
305
  resource_group_name: str
286
306
  mainline_deployment_id: str
287
307
  kind: JobQueueKind
288
308
  attempters: _containers.RepeatedScalarFieldContainer[str]
309
+ workflow_execution_id: str
310
+ unsatisfied_dependency_operation_ids: _containers.RepeatedScalarFieldContainer[str]
289
311
  def __init__(
290
312
  self,
291
313
  indexed_row_summaries: _Optional[_Mapping[int, JobQueueRowSummary]] = ...,
@@ -293,6 +315,8 @@ class JobQueueOperationSummary(_message.Message):
293
315
  mainline_deployment_id: _Optional[str] = ...,
294
316
  kind: _Optional[_Union[JobQueueKind, str]] = ...,
295
317
  attempters: _Optional[_Iterable[str]] = ...,
318
+ workflow_execution_id: _Optional[str] = ...,
319
+ unsatisfied_dependency_operation_ids: _Optional[_Iterable[str]] = ...,
296
320
  ) -> None: ...
297
321
 
298
322
  class GetJobQueueOperationSummaryRequest(_message.Message):
@@ -383,3 +407,40 @@ class ListJobQueueAttemptsResponse(_message.Message):
383
407
  ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
384
408
  attempts: _containers.RepeatedCompositeFieldContainer[JobQueueAttempt]
385
409
  def __init__(self, attempts: _Optional[_Iterable[_Union[JobQueueAttempt, _Mapping]]] = ...) -> None: ...
410
+
411
+ class OperationAndShard(_message.Message):
412
+ __slots__ = ("operation_id", "shard_id")
413
+ OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
414
+ SHARD_ID_FIELD_NUMBER: _ClassVar[int]
415
+ operation_id: str
416
+ shard_id: int
417
+ def __init__(self, operation_id: _Optional[str] = ..., shard_id: _Optional[int] = ...) -> None: ...
418
+
419
+ class ForceCancelJobQueueJobRequest(_message.Message):
420
+ __slots__ = ("environment_id", "by_job_queue_id", "by_operation_and_shard")
421
+ ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
422
+ BY_JOB_QUEUE_ID_FIELD_NUMBER: _ClassVar[int]
423
+ BY_OPERATION_AND_SHARD_FIELD_NUMBER: _ClassVar[int]
424
+ environment_id: str
425
+ by_job_queue_id: int
426
+ by_operation_and_shard: OperationAndShard
427
+ def __init__(
428
+ self,
429
+ environment_id: _Optional[str] = ...,
430
+ by_job_queue_id: _Optional[int] = ...,
431
+ by_operation_and_shard: _Optional[_Union[OperationAndShard, _Mapping]] = ...,
432
+ ) -> None: ...
433
+
434
+ class ForceCancelJobQueueJobResponse(_message.Message):
435
+ __slots__ = ()
436
+ def __init__(self) -> None: ...
437
+
438
+ class CancelWorkflowExecutionRequest(_message.Message):
439
+ __slots__ = ("workflow_execution_id",)
440
+ WORKFLOW_EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
441
+ workflow_execution_id: str
442
+ def __init__(self, workflow_execution_id: _Optional[str] = ...) -> None: ...
443
+
444
+ class CancelWorkflowExecutionResponse(_message.Message):
445
+ __slots__ = ()
446
+ def __init__(self) -> None: ...
@@ -40,6 +40,16 @@ class DataPlaneJobQueueServiceStub(object):
40
40
  request_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsRequest.SerializeToString,
41
41
  response_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsResponse.FromString,
42
42
  )
43
+ self.ForceCancelJobQueueJob = channel.unary_unary(
44
+ "/chalk.server.v1.DataPlaneJobQueueService/ForceCancelJobQueueJob",
45
+ request_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ForceCancelJobQueueJobRequest.SerializeToString,
46
+ response_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ForceCancelJobQueueJobResponse.FromString,
47
+ )
48
+ self.CancelWorkflowExecution = channel.unary_unary(
49
+ "/chalk.server.v1.DataPlaneJobQueueService/CancelWorkflowExecution",
50
+ request_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.CancelWorkflowExecutionRequest.SerializeToString,
51
+ response_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.CancelWorkflowExecutionResponse.FromString,
52
+ )
43
53
 
44
54
 
45
55
  class DataPlaneJobQueueServiceServicer(object):
@@ -75,6 +85,18 @@ class DataPlaneJobQueueServiceServicer(object):
75
85
  context.set_details("Method not implemented!")
76
86
  raise NotImplementedError("Method not implemented!")
77
87
 
88
+ def ForceCancelJobQueueJob(self, request, context):
89
+ """Missing associated documentation comment in .proto file."""
90
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
91
+ context.set_details("Method not implemented!")
92
+ raise NotImplementedError("Method not implemented!")
93
+
94
+ def CancelWorkflowExecution(self, request, context):
95
+ """Missing associated documentation comment in .proto file."""
96
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
97
+ context.set_details("Method not implemented!")
98
+ raise NotImplementedError("Method not implemented!")
99
+
78
100
 
79
101
  def add_DataPlaneJobQueueServiceServicer_to_server(servicer, server):
80
102
  rpc_method_handlers = {
@@ -103,6 +125,16 @@ def add_DataPlaneJobQueueServiceServicer_to_server(servicer, server):
103
125
  request_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsRequest.FromString,
104
126
  response_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsResponse.SerializeToString,
105
127
  ),
128
+ "ForceCancelJobQueueJob": grpc.unary_unary_rpc_method_handler(
129
+ servicer.ForceCancelJobQueueJob,
130
+ request_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ForceCancelJobQueueJobRequest.FromString,
131
+ response_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ForceCancelJobQueueJobResponse.SerializeToString,
132
+ ),
133
+ "CancelWorkflowExecution": grpc.unary_unary_rpc_method_handler(
134
+ servicer.CancelWorkflowExecution,
135
+ request_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.CancelWorkflowExecutionRequest.FromString,
136
+ response_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.CancelWorkflowExecutionResponse.SerializeToString,
137
+ ),
106
138
  }
107
139
  generic_handler = grpc.method_handlers_generic_handler(
108
140
  "chalk.server.v1.DataPlaneJobQueueService", rpc_method_handlers
@@ -258,3 +290,61 @@ class DataPlaneJobQueueService(object):
258
290
  timeout,
259
291
  metadata,
260
292
  )
293
+
294
+ @staticmethod
295
+ def ForceCancelJobQueueJob(
296
+ request,
297
+ target,
298
+ options=(),
299
+ channel_credentials=None,
300
+ call_credentials=None,
301
+ insecure=False,
302
+ compression=None,
303
+ wait_for_ready=None,
304
+ timeout=None,
305
+ metadata=None,
306
+ ):
307
+ return grpc.experimental.unary_unary(
308
+ request,
309
+ target,
310
+ "/chalk.server.v1.DataPlaneJobQueueService/ForceCancelJobQueueJob",
311
+ chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ForceCancelJobQueueJobRequest.SerializeToString,
312
+ chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ForceCancelJobQueueJobResponse.FromString,
313
+ options,
314
+ channel_credentials,
315
+ insecure,
316
+ call_credentials,
317
+ compression,
318
+ wait_for_ready,
319
+ timeout,
320
+ metadata,
321
+ )
322
+
323
+ @staticmethod
324
+ def CancelWorkflowExecution(
325
+ request,
326
+ target,
327
+ options=(),
328
+ channel_credentials=None,
329
+ call_credentials=None,
330
+ insecure=False,
331
+ compression=None,
332
+ wait_for_ready=None,
333
+ timeout=None,
334
+ metadata=None,
335
+ ):
336
+ return grpc.experimental.unary_unary(
337
+ request,
338
+ target,
339
+ "/chalk.server.v1.DataPlaneJobQueueService/CancelWorkflowExecution",
340
+ chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.CancelWorkflowExecutionRequest.SerializeToString,
341
+ chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.CancelWorkflowExecutionResponse.FromString,
342
+ options,
343
+ channel_credentials,
344
+ insecure,
345
+ call_credentials,
346
+ compression,
347
+ wait_for_ready,
348
+ timeout,
349
+ metadata,
350
+ )
@@ -8,6 +8,10 @@ from abc import (
8
8
  abstractmethod,
9
9
  )
10
10
  from chalk._gen.chalk.server.v1.dataplanejobqueue_pb2 import (
11
+ CancelWorkflowExecutionRequest,
12
+ CancelWorkflowExecutionResponse,
13
+ ForceCancelJobQueueJobRequest,
14
+ ForceCancelJobQueueJobResponse,
11
15
  GetDataPlaneJobQueueRequest,
12
16
  GetDataPlaneJobQueueResponse,
13
17
  GetJobQueueAuxiliaryResourcesRequest,
@@ -48,6 +52,14 @@ class DataPlaneJobQueueServiceStub:
48
52
  ListJobQueueAttemptsRequest,
49
53
  ListJobQueueAttemptsResponse,
50
54
  ]
55
+ ForceCancelJobQueueJob: UnaryUnaryMultiCallable[
56
+ ForceCancelJobQueueJobRequest,
57
+ ForceCancelJobQueueJobResponse,
58
+ ]
59
+ CancelWorkflowExecution: UnaryUnaryMultiCallable[
60
+ CancelWorkflowExecutionRequest,
61
+ CancelWorkflowExecutionResponse,
62
+ ]
51
63
 
52
64
  class DataPlaneJobQueueServiceServicer(metaclass=ABCMeta):
53
65
  @abstractmethod
@@ -80,6 +92,18 @@ class DataPlaneJobQueueServiceServicer(metaclass=ABCMeta):
80
92
  request: ListJobQueueAttemptsRequest,
81
93
  context: ServicerContext,
82
94
  ) -> ListJobQueueAttemptsResponse: ...
95
+ @abstractmethod
96
+ def ForceCancelJobQueueJob(
97
+ self,
98
+ request: ForceCancelJobQueueJobRequest,
99
+ context: ServicerContext,
100
+ ) -> ForceCancelJobQueueJobResponse: ...
101
+ @abstractmethod
102
+ def CancelWorkflowExecution(
103
+ self,
104
+ request: CancelWorkflowExecutionRequest,
105
+ context: ServicerContext,
106
+ ) -> CancelWorkflowExecutionResponse: ...
83
107
 
84
108
  def add_DataPlaneJobQueueServiceServicer_to_server(
85
109
  servicer: DataPlaneJobQueueServiceServicer, server: Server