chalkpy 2.89.22__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 (268) 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 +36 -33
  5. chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +41 -1
  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 +19 -13
  9. chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +37 -0
  10. chalk/_gen/chalk/common/v1/online_query_pb2.py +54 -54
  11. chalk/_gen/chalk/common/v1/online_query_pb2.pyi +13 -1
  12. chalk/_gen/chalk/common/v1/script_task_pb2.py +13 -11
  13. chalk/_gen/chalk/common/v1/script_task_pb2.pyi +19 -1
  14. chalk/_gen/chalk/dataframe/__init__.py +0 -0
  15. chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
  16. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
  17. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +123 -0
  18. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
  19. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
  20. chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
  21. chalk/_gen/chalk/graph/v1/graph_pb2.pyi +25 -0
  22. chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
  23. chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -0
  24. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
  25. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
  26. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
  27. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
  28. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
  29. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
  30. chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +54 -27
  31. chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
  32. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
  33. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
  34. chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
  35. chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
  36. chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
  37. chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
  38. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
  39. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
  40. chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
  41. chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
  42. chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
  43. chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
  44. chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
  45. chalk/_gen/chalk/server/v1/builder_pb2.py +372 -272
  46. chalk/_gen/chalk/server/v1/builder_pb2.pyi +479 -12
  47. chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +360 -0
  48. chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +96 -0
  49. chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
  50. chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
  51. chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
  52. chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
  53. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
  54. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
  55. chalk/_gen/chalk/server/v1/cloud_components_pb2.py +153 -107
  56. chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +146 -4
  57. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +180 -0
  58. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +48 -0
  59. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
  60. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
  61. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
  62. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
  63. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +59 -35
  64. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +127 -1
  65. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +135 -0
  66. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +36 -0
  67. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
  68. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -0
  69. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
  70. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
  71. chalk/_gen/chalk/server/v1/datasets_pb2.py +36 -24
  72. chalk/_gen/chalk/server/v1/datasets_pb2.pyi +71 -2
  73. chalk/_gen/chalk/server/v1/datasets_pb2_grpc.py +45 -0
  74. chalk/_gen/chalk/server/v1/datasets_pb2_grpc.pyi +12 -0
  75. chalk/_gen/chalk/server/v1/deploy_pb2.py +9 -3
  76. chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
  77. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
  78. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
  79. chalk/_gen/chalk/server/v1/deployment_pb2.py +20 -15
  80. chalk/_gen/chalk/server/v1/deployment_pb2.pyi +25 -0
  81. chalk/_gen/chalk/server/v1/environment_pb2.py +25 -15
  82. chalk/_gen/chalk/server/v1/environment_pb2.pyi +93 -1
  83. chalk/_gen/chalk/server/v1/eventbus_pb2.py +44 -0
  84. chalk/_gen/chalk/server/v1/eventbus_pb2.pyi +64 -0
  85. chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.py +4 -0
  86. chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.pyi +4 -0
  87. chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
  88. chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
  89. chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
  90. chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
  91. chalk/_gen/chalk/server/v1/graph_pb2.py +41 -3
  92. chalk/_gen/chalk/server/v1/graph_pb2.pyi +191 -0
  93. chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +92 -0
  94. chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +32 -0
  95. chalk/_gen/chalk/server/v1/incident_pb2.py +57 -0
  96. chalk/_gen/chalk/server/v1/incident_pb2.pyi +165 -0
  97. chalk/_gen/chalk/server/v1/incident_pb2_grpc.py +4 -0
  98. chalk/_gen/chalk/server/v1/incident_pb2_grpc.pyi +4 -0
  99. chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
  100. chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
  101. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
  102. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
  103. chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
  104. chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
  105. chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
  106. chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
  107. chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
  108. chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
  109. chalk/_gen/chalk/server/v1/log_pb2.py +21 -3
  110. chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
  111. chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
  112. chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
  113. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.py +73 -0
  114. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.pyi +212 -0
  115. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.py +217 -0
  116. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.pyi +74 -0
  117. chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
  118. chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
  119. chalk/_gen/chalk/server/v1/monitoring_pb2.py +84 -75
  120. chalk/_gen/chalk/server/v1/monitoring_pb2.pyi +1 -0
  121. chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py +136 -0
  122. chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi +38 -0
  123. chalk/_gen/chalk/server/v1/offline_queries_pb2.py +32 -10
  124. chalk/_gen/chalk/server/v1/offline_queries_pb2.pyi +73 -0
  125. chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.py +90 -0
  126. chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.pyi +24 -0
  127. chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
  128. chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
  129. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
  130. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
  131. chalk/_gen/chalk/server/v1/queries_pb2.py +76 -48
  132. chalk/_gen/chalk/server/v1/queries_pb2.pyi +155 -2
  133. chalk/_gen/chalk/server/v1/queries_pb2_grpc.py +180 -0
  134. chalk/_gen/chalk/server/v1/queries_pb2_grpc.pyi +48 -0
  135. chalk/_gen/chalk/server/v1/scheduled_query_pb2.py +4 -2
  136. chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.py +45 -0
  137. chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.pyi +12 -0
  138. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -6
  139. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +75 -2
  140. chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
  141. chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
  142. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
  143. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
  144. chalk/_gen/chalk/server/v1/script_tasks_pb2.py +26 -14
  145. chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +33 -3
  146. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
  147. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -0
  148. chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
  149. chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
  150. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
  151. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
  152. chalk/_gen/chalk/server/v1/sql_queries_pb2.py +48 -0
  153. chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
  154. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
  155. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
  156. chalk/_gen/chalk/server/v1/team_pb2.py +156 -137
  157. chalk/_gen/chalk/server/v1/team_pb2.pyi +56 -10
  158. chalk/_gen/chalk/server/v1/team_pb2_grpc.py +90 -0
  159. chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +24 -0
  160. chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
  161. chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
  162. chalk/_gen/chalk/server/v1/trace_pb2.py +50 -28
  163. chalk/_gen/chalk/server/v1/trace_pb2.pyi +121 -0
  164. chalk/_gen/chalk/server/v1/trace_pb2_grpc.py +135 -0
  165. chalk/_gen/chalk/server/v1/trace_pb2_grpc.pyi +42 -0
  166. chalk/_gen/chalk/server/v1/webhook_pb2.py +9 -3
  167. chalk/_gen/chalk/server/v1/webhook_pb2.pyi +18 -0
  168. chalk/_gen/chalk/server/v1/webhook_pb2_grpc.py +45 -0
  169. chalk/_gen/chalk/server/v1/webhook_pb2_grpc.pyi +12 -0
  170. chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
  171. chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
  172. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
  173. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
  174. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +19 -7
  175. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +96 -3
  176. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
  177. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
  178. chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
  179. chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
  180. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
  181. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
  182. chalk/_lsp/error_builder.py +11 -0
  183. chalk/_monitoring/Chart.py +1 -3
  184. chalk/_version.py +1 -1
  185. chalk/cli.py +5 -10
  186. chalk/client/client.py +178 -64
  187. chalk/client/client_async.py +154 -0
  188. chalk/client/client_async_impl.py +22 -0
  189. chalk/client/client_grpc.py +738 -112
  190. chalk/client/client_impl.py +541 -136
  191. chalk/client/dataset.py +27 -6
  192. chalk/client/models.py +99 -2
  193. chalk/client/serialization/model_serialization.py +126 -10
  194. chalk/config/project_config.py +1 -1
  195. chalk/df/LazyFramePlaceholder.py +1154 -0
  196. chalk/df/ast_parser.py +2 -10
  197. chalk/features/_class_property.py +7 -0
  198. chalk/features/_embedding/embedding.py +1 -0
  199. chalk/features/_embedding/sentence_transformer.py +1 -1
  200. chalk/features/_encoding/converter.py +83 -2
  201. chalk/features/_encoding/pyarrow.py +20 -4
  202. chalk/features/_encoding/rich.py +1 -3
  203. chalk/features/_tensor.py +1 -2
  204. chalk/features/dataframe/_filters.py +14 -5
  205. chalk/features/dataframe/_impl.py +91 -36
  206. chalk/features/dataframe/_validation.py +11 -7
  207. chalk/features/feature_field.py +40 -30
  208. chalk/features/feature_set.py +1 -2
  209. chalk/features/feature_set_decorator.py +1 -0
  210. chalk/features/feature_wrapper.py +42 -3
  211. chalk/features/hooks.py +81 -12
  212. chalk/features/inference.py +65 -10
  213. chalk/features/resolver.py +338 -56
  214. chalk/features/tag.py +1 -3
  215. chalk/features/underscore_features.py +2 -1
  216. chalk/functions/__init__.py +456 -21
  217. chalk/functions/holidays.py +1 -3
  218. chalk/gitignore/gitignore_parser.py +5 -1
  219. chalk/importer.py +186 -74
  220. chalk/ml/__init__.py +6 -2
  221. chalk/ml/model_hooks.py +368 -51
  222. chalk/ml/model_reference.py +68 -10
  223. chalk/ml/model_version.py +34 -21
  224. chalk/ml/utils.py +143 -40
  225. chalk/operators/_utils.py +14 -3
  226. chalk/parsed/_proto/export.py +22 -0
  227. chalk/parsed/duplicate_input_gql.py +4 -0
  228. chalk/parsed/expressions.py +1 -3
  229. chalk/parsed/json_conversions.py +21 -14
  230. chalk/parsed/to_proto.py +16 -4
  231. chalk/parsed/user_types_to_json.py +31 -10
  232. chalk/parsed/validation_from_registries.py +182 -0
  233. chalk/queries/named_query.py +16 -6
  234. chalk/queries/scheduled_query.py +13 -1
  235. chalk/serialization/parsed_annotation.py +25 -12
  236. chalk/sql/__init__.py +221 -0
  237. chalk/sql/_internal/integrations/athena.py +6 -1
  238. chalk/sql/_internal/integrations/bigquery.py +22 -2
  239. chalk/sql/_internal/integrations/databricks.py +61 -18
  240. chalk/sql/_internal/integrations/mssql.py +281 -0
  241. chalk/sql/_internal/integrations/postgres.py +11 -3
  242. chalk/sql/_internal/integrations/redshift.py +4 -0
  243. chalk/sql/_internal/integrations/snowflake.py +11 -2
  244. chalk/sql/_internal/integrations/util.py +2 -1
  245. chalk/sql/_internal/sql_file_resolver.py +55 -10
  246. chalk/sql/_internal/sql_source.py +36 -2
  247. chalk/streams/__init__.py +1 -3
  248. chalk/streams/_kafka_source.py +5 -1
  249. chalk/streams/_windows.py +16 -4
  250. chalk/streams/types.py +1 -2
  251. chalk/utils/__init__.py +1 -3
  252. chalk/utils/_otel_version.py +13 -0
  253. chalk/utils/async_helpers.py +14 -5
  254. chalk/utils/df_utils.py +2 -2
  255. chalk/utils/duration.py +1 -3
  256. chalk/utils/job_log_display.py +538 -0
  257. chalk/utils/missing_dependency.py +5 -4
  258. chalk/utils/notebook.py +255 -2
  259. chalk/utils/pl_helpers.py +190 -37
  260. chalk/utils/pydanticutil/pydantic_compat.py +1 -2
  261. chalk/utils/storage_client.py +246 -0
  262. chalk/utils/threading.py +1 -3
  263. chalk/utils/tracing.py +194 -86
  264. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/METADATA +53 -21
  265. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/RECORD +268 -198
  266. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
  267. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
  268. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/top_level.txt +0 -0
@@ -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*\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\xc3\x04\n\x18\x44\x61taPlaneJobQueueService\x12{\n\x14GetDataPlaneJobQueue\x12,.chalk.server.v1.GetDataPlaneJobQueueRequest\x1a-.chalk.server.v1.GetDataPlaneJobQueueResponse"\x06\x90\x02\x01\x80}\x1b\x12~\n\x15ListDataPlaneJobQueue\x12-.chalk.server.v1.ListDataPlaneJobQueueRequest\x1a..chalk.server.v1.ListDataPlaneJobQueueResponse"\x06\x90\x02\x01\x80}\x1b\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}\x1b\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()
@@ -40,11 +40,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
40
40
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["GetDataPlaneJobQueue"]._options = None
41
41
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name[
42
42
  "GetDataPlaneJobQueue"
43
- ]._serialized_options = b"\220\002\001\200}\033"
43
+ ]._serialized_options = b"\220\002\001\200}\020"
44
44
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["ListDataPlaneJobQueue"]._options = None
45
45
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name[
46
46
  "ListDataPlaneJobQueue"
47
- ]._serialized_options = b"\220\002\001\200}\033"
47
+ ]._serialized_options = b"\220\002\001\200}\020"
48
48
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["GetJobQueueAuxiliaryResources"]._options = None
49
49
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name[
50
50
  "GetJobQueueAuxiliaryResources"
@@ -52,37 +52,61 @@ if _descriptor._USE_C_DESCRIPTORS == False:
52
52
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["GetJobQueueOperationSummary"]._options = None
53
53
  _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name[
54
54
  "GetJobQueueOperationSummary"
55
- ]._serialized_options = b"\220\002\001\200}\033"
56
- _globals["_JOBQUEUESTATE"]._serialized_start = 3782
57
- _globals["_JOBQUEUESTATE"]._serialized_end = 4010
58
- _globals["_JOBQUEUEKIND"]._serialized_start = 4013
59
- _globals["_JOBQUEUEKIND"]._serialized_end = 4167
55
+ ]._serialized_options = b"\220\002\001\200}\020"
56
+ _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name["ListJobQueueAttempts"]._options = None
57
+ _globals["_DATAPLANEJOBQUEUESERVICE"].methods_by_name[
58
+ "ListJobQueueAttempts"
59
+ ]._serialized_options = b"\220\002\001\200}\020"
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
60
68
  _globals["_JOBQUEUEITEM"]._serialized_start = 208
61
- _globals["_JOBQUEUEITEM"]._serialized_end = 1388
62
- _globals["_GETDATAPLANEJOBQUEUEREQUEST"]._serialized_start = 1390
63
- _globals["_GETDATAPLANEJOBQUEUEREQUEST"]._serialized_end = 1435
64
- _globals["_GETDATAPLANEJOBQUEUERESPONSE"]._serialized_start = 1437
65
- _globals["_GETDATAPLANEJOBQUEUERESPONSE"]._serialized_end = 1516
66
- _globals["_LISTDATAPLANEJOBQUEUEREQUEST"]._serialized_start = 1519
67
- _globals["_LISTDATAPLANEJOBQUEUEREQUEST"]._serialized_end = 1811
68
- _globals["_LISTDATAPLANEJOBQUEUERESPONSE"]._serialized_start = 1813
69
- _globals["_LISTDATAPLANEJOBQUEUERESPONSE"]._serialized_end = 1917
70
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESREQUEST"]._serialized_start = 1919
71
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESREQUEST"]._serialized_end = 2035
72
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE"]._serialized_start = 2038
73
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE"]._serialized_end = 2437
74
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE_DEPLOYMENTSCALEDOBJECTSENTRY"]._serialized_start = 2314
75
- _globals["_GETJOBQUEUEAUXILIARYRESOURCESRESPONSE_DEPLOYMENTSCALEDOBJECTSENTRY"]._serialized_end = 2437
76
- _globals["_JOBQUEUEROWSUMMARY"]._serialized_start = 2440
77
- _globals["_JOBQUEUEROWSUMMARY"]._serialized_end = 3021
78
- _globals["_JOBQUEUEOPERATIONSUMMARY"]._serialized_start = 3024
79
- _globals["_JOBQUEUEOPERATIONSUMMARY"]._serialized_end = 3464
80
- _globals["_JOBQUEUEOPERATIONSUMMARY_INDEXEDROWSUMMARIESENTRY"]._serialized_start = 3357
81
- _globals["_JOBQUEUEOPERATIONSUMMARY_INDEXEDROWSUMMARIESENTRY"]._serialized_end = 3464
82
- _globals["_GETJOBQUEUEOPERATIONSUMMARYREQUEST"]._serialized_start = 3467
83
- _globals["_GETJOBQUEUEOPERATIONSUMMARYREQUEST"]._serialized_end = 3654
84
- _globals["_GETJOBQUEUEOPERATIONSUMMARYRESPONSE"]._serialized_start = 3656
85
- _globals["_GETJOBQUEUEOPERATIONSUMMARYRESPONSE"]._serialized_end = 3779
86
- _globals["_DATAPLANEJOBQUEUESERVICE"]._serialized_start = 4170
87
- _globals["_DATAPLANEJOBQUEUESERVICE"]._serialized_end = 4749
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
88
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):
@@ -318,3 +342,105 @@ class GetJobQueueOperationSummaryResponse(_message.Message):
318
342
  SUMMARY_FIELD_NUMBER: _ClassVar[int]
319
343
  summary: JobQueueOperationSummary
320
344
  def __init__(self, summary: _Optional[_Union[JobQueueOperationSummary, _Mapping]] = ...) -> None: ...
345
+
346
+ class JobQueueAttempt(_message.Message):
347
+ __slots__ = (
348
+ "id",
349
+ "created_at",
350
+ "environment_id",
351
+ "deployment_id",
352
+ "job_queue_id",
353
+ "attempt_idx",
354
+ "queued_at",
355
+ "started_at",
356
+ "attempted_by",
357
+ "state",
358
+ "finished_at",
359
+ )
360
+ ID_FIELD_NUMBER: _ClassVar[int]
361
+ CREATED_AT_FIELD_NUMBER: _ClassVar[int]
362
+ ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
363
+ DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
364
+ JOB_QUEUE_ID_FIELD_NUMBER: _ClassVar[int]
365
+ ATTEMPT_IDX_FIELD_NUMBER: _ClassVar[int]
366
+ QUEUED_AT_FIELD_NUMBER: _ClassVar[int]
367
+ STARTED_AT_FIELD_NUMBER: _ClassVar[int]
368
+ ATTEMPTED_BY_FIELD_NUMBER: _ClassVar[int]
369
+ STATE_FIELD_NUMBER: _ClassVar[int]
370
+ FINISHED_AT_FIELD_NUMBER: _ClassVar[int]
371
+ id: int
372
+ created_at: _timestamp_pb2.Timestamp
373
+ environment_id: str
374
+ deployment_id: str
375
+ job_queue_id: int
376
+ attempt_idx: int
377
+ queued_at: _timestamp_pb2.Timestamp
378
+ started_at: _timestamp_pb2.Timestamp
379
+ attempted_by: str
380
+ state: int
381
+ finished_at: _timestamp_pb2.Timestamp
382
+ def __init__(
383
+ self,
384
+ id: _Optional[int] = ...,
385
+ created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
386
+ environment_id: _Optional[str] = ...,
387
+ deployment_id: _Optional[str] = ...,
388
+ job_queue_id: _Optional[int] = ...,
389
+ attempt_idx: _Optional[int] = ...,
390
+ queued_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
391
+ started_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
392
+ attempted_by: _Optional[str] = ...,
393
+ state: _Optional[int] = ...,
394
+ finished_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
395
+ ) -> None: ...
396
+
397
+ class ListJobQueueAttemptsRequest(_message.Message):
398
+ __slots__ = ("job_queue_id", "operation_id")
399
+ JOB_QUEUE_ID_FIELD_NUMBER: _ClassVar[int]
400
+ OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
401
+ job_queue_id: int
402
+ operation_id: str
403
+ def __init__(self, job_queue_id: _Optional[int] = ..., operation_id: _Optional[str] = ...) -> None: ...
404
+
405
+ class ListJobQueueAttemptsResponse(_message.Message):
406
+ __slots__ = ("attempts",)
407
+ ATTEMPTS_FIELD_NUMBER: _ClassVar[int]
408
+ attempts: _containers.RepeatedCompositeFieldContainer[JobQueueAttempt]
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: ...
@@ -35,6 +35,21 @@ class DataPlaneJobQueueServiceStub(object):
35
35
  request_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.GetJobQueueOperationSummaryRequest.SerializeToString,
36
36
  response_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.GetJobQueueOperationSummaryResponse.FromString,
37
37
  )
38
+ self.ListJobQueueAttempts = channel.unary_unary(
39
+ "/chalk.server.v1.DataPlaneJobQueueService/ListJobQueueAttempts",
40
+ request_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsRequest.SerializeToString,
41
+ response_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsResponse.FromString,
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
+ )
38
53
 
39
54
 
40
55
  class DataPlaneJobQueueServiceServicer(object):
@@ -64,6 +79,24 @@ class DataPlaneJobQueueServiceServicer(object):
64
79
  context.set_details("Method not implemented!")
65
80
  raise NotImplementedError("Method not implemented!")
66
81
 
82
+ def ListJobQueueAttempts(self, request, context):
83
+ """Missing associated documentation comment in .proto file."""
84
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
85
+ context.set_details("Method not implemented!")
86
+ raise NotImplementedError("Method not implemented!")
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
+
67
100
 
68
101
  def add_DataPlaneJobQueueServiceServicer_to_server(servicer, server):
69
102
  rpc_method_handlers = {
@@ -87,6 +120,21 @@ def add_DataPlaneJobQueueServiceServicer_to_server(servicer, server):
87
120
  request_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.GetJobQueueOperationSummaryRequest.FromString,
88
121
  response_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.GetJobQueueOperationSummaryResponse.SerializeToString,
89
122
  ),
123
+ "ListJobQueueAttempts": grpc.unary_unary_rpc_method_handler(
124
+ servicer.ListJobQueueAttempts,
125
+ request_deserializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsRequest.FromString,
126
+ response_serializer=chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsResponse.SerializeToString,
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
+ ),
90
138
  }
91
139
  generic_handler = grpc.method_handlers_generic_handler(
92
140
  "chalk.server.v1.DataPlaneJobQueueService", rpc_method_handlers
@@ -213,3 +261,90 @@ class DataPlaneJobQueueService(object):
213
261
  timeout,
214
262
  metadata,
215
263
  )
264
+
265
+ @staticmethod
266
+ def ListJobQueueAttempts(
267
+ request,
268
+ target,
269
+ options=(),
270
+ channel_credentials=None,
271
+ call_credentials=None,
272
+ insecure=False,
273
+ compression=None,
274
+ wait_for_ready=None,
275
+ timeout=None,
276
+ metadata=None,
277
+ ):
278
+ return grpc.experimental.unary_unary(
279
+ request,
280
+ target,
281
+ "/chalk.server.v1.DataPlaneJobQueueService/ListJobQueueAttempts",
282
+ chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsRequest.SerializeToString,
283
+ chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2.ListJobQueueAttemptsResponse.FromString,
284
+ options,
285
+ channel_credentials,
286
+ insecure,
287
+ call_credentials,
288
+ compression,
289
+ wait_for_ready,
290
+ timeout,
291
+ metadata,
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,
@@ -16,6 +20,8 @@ from chalk._gen.chalk.server.v1.dataplanejobqueue_pb2 import (
16
20
  GetJobQueueOperationSummaryResponse,
17
21
  ListDataPlaneJobQueueRequest,
18
22
  ListDataPlaneJobQueueResponse,
23
+ ListJobQueueAttemptsRequest,
24
+ ListJobQueueAttemptsResponse,
19
25
  )
20
26
  from grpc import (
21
27
  Channel,
@@ -42,6 +48,18 @@ class DataPlaneJobQueueServiceStub:
42
48
  GetJobQueueOperationSummaryRequest,
43
49
  GetJobQueueOperationSummaryResponse,
44
50
  ]
51
+ ListJobQueueAttempts: UnaryUnaryMultiCallable[
52
+ ListJobQueueAttemptsRequest,
53
+ ListJobQueueAttemptsResponse,
54
+ ]
55
+ ForceCancelJobQueueJob: UnaryUnaryMultiCallable[
56
+ ForceCancelJobQueueJobRequest,
57
+ ForceCancelJobQueueJobResponse,
58
+ ]
59
+ CancelWorkflowExecution: UnaryUnaryMultiCallable[
60
+ CancelWorkflowExecutionRequest,
61
+ CancelWorkflowExecutionResponse,
62
+ ]
45
63
 
46
64
  class DataPlaneJobQueueServiceServicer(metaclass=ABCMeta):
47
65
  @abstractmethod
@@ -68,6 +86,24 @@ class DataPlaneJobQueueServiceServicer(metaclass=ABCMeta):
68
86
  request: GetJobQueueOperationSummaryRequest,
69
87
  context: ServicerContext,
70
88
  ) -> GetJobQueueOperationSummaryResponse: ...
89
+ @abstractmethod
90
+ def ListJobQueueAttempts(
91
+ self,
92
+ request: ListJobQueueAttemptsRequest,
93
+ context: ServicerContext,
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: ...
71
107
 
72
108
  def add_DataPlaneJobQueueServiceServicer_to_server(
73
109
  servicer: DataPlaneJobQueueServiceServicer, server: Server