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
@@ -0,0 +1,90 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: chalk/server/v1/dataplaneworkflows.proto
4
+ # Protobuf Python Version: 4.25.3
5
+ """Generated protocol buffer code."""
6
+
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import descriptor_pool as _descriptor_pool
9
+ from google.protobuf import symbol_database as _symbol_database
10
+ from google.protobuf.internal import builder as _builder
11
+ # @@protoc_insertion_point(imports)
12
+
13
+ _sym_db = _symbol_database.Default()
14
+
15
+
16
+ from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
17
+ from chalk._gen.chalk.server.v1 import dataplanejobqueue_pb2 as chalk_dot_server_dot_v1_dot_dataplanejobqueue__pb2
18
+ from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
19
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
20
+
21
+
22
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
23
+ b'\n(chalk/server/v1/dataplaneworkflows.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\'chalk/server/v1/dataplanejobqueue.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xc5\x05\n\x11WorkflowExecution\x12\x0e\n\x02id\x18\x01 \x01(\tR\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\x31\n\x04kind\x18\x05 \x01(\x0e\x32\x1d.chalk.server.v1.WorkflowKindR\x04kind\x12@\n\x06status\x18\x06 \x01(\x0e\x32(.chalk.server.v1.WorkflowExecutionStatusR\x06status\x12$\n\x0b\x62ranch_name\x18\x07 \x01(\tH\x00R\nbranchName\x88\x01\x01\x12\x39\n\x16mainline_deployment_id\x18\x08 \x01(\tH\x01R\x14mainlineDeploymentId\x88\x01\x01\x12\x1e\n\x08\x61gent_id\x18\t \x01(\tH\x02R\x07\x61gentId\x88\x01\x01\x12\x39\n\tmeta_data\x18\n \x01(\x0b\x32\x17.google.protobuf.StructH\x03R\x08metaData\x88\x01\x01\x12>\n\nupdated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\tupdatedAt\x88\x01\x01\x12\x42\n\x0c\x66inalized_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x05R\x0b\x66inalizedAt\x88\x01\x01\x42\x0e\n\x0c_branch_nameB\x19\n\x17_mainline_deployment_idB\x0b\n\t_agent_idB\x0c\n\n_meta_dataB\r\n\x0b_updated_atB\x0f\n\r_finalized_at"-\n\x1bGetDataPlaneWorkflowRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"^\n\x1cGetDataPlaneWorkflowResponse\x12>\n\x08workflow\x18\x01 \x01(\x0b\x32".chalk.server.v1.WorkflowExecutionR\x08workflow"\xc2\x02\n\x1dListDataPlaneWorkflowsRequest\x12)\n\x0e\x65nvironment_id\x18\x01 \x01(\tB\x02\x18\x01R\renvironmentId\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId\x12\x31\n\x04kind\x18\x03 \x01(\x0e\x32\x1d.chalk.server.v1.WorkflowKindR\x04kind\x12@\n\x06status\x18\x04 \x01(\x0e\x32(.chalk.server.v1.WorkflowExecutionStatusR\x06status\x12\x14\n\x05limit\x18\x05 \x01(\x05R\x05limit\x12\x16\n\x06offset\x18\x06 \x01(\x05R\x06offset\x12 \n\tid_filter\x18\x07 \x01(\tH\x00R\x08idFilter\x88\x01\x01\x42\x0c\n\n_id_filter"x\n\x1eListDataPlaneWorkflowsResponse\x12@\n\tworkflows\x18\x01 \x03(\x0b\x32".chalk.server.v1.WorkflowExecutionR\tworkflows\x12\x14\n\x05total\x18\x02 \x01(\x05R\x05total"M\n\x17GetWorkflowGraphRequest\x12\x32\n\x15workflow_execution_id\x18\x01 \x01(\tR\x13workflowExecutionId"\xae\x02\n\x1eJobQueueRowSummaryForWorkflows\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12\x19\n\x08job_name\x18\x02 \x01(\tR\x07jobName\x12\x34\n\x05state\x18\x03 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x1b\n\tjob_index\x18\x04 \x01(\x03R\x08jobIndex\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x42\n\x0c\x66inalized_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x0b\x66inalizedAt\x88\x01\x01\x42\x0f\n\r_finalized_at"\xc3\x04\n\x11WorkflowGraphNode\x12$\n\x0cjob_queue_id\x18\x01 \x01(\x05\x42\x02\x18\x01R\njobQueueId\x12\x1d\n\x08job_name\x18\x03 \x01(\tB\x02\x18\x01R\x07jobName\x12$\n\tjob_index\x18\x04 \x01(\x03\x42\x02\x18\x01H\x00R\x08jobIndex\x88\x01\x01\x12\x38\n\x05state\x18\x05 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateB\x02\x18\x01R\x05state\x12=\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x02\x18\x01R\tcreatedAt\x12\x46\n\x0c\x66inalized_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x02\x18\x01H\x01R\x0b\x66inalizedAt\x88\x01\x01\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId\x12\x31\n\x04kind\x18\x06 \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12%\n\x0eresource_group\x18\t \x01(\tR\rresourceGroup\x12\x66\n\x17job_queue_row_summaries\x18\n \x03(\x0b\x32/.chalk.server.v1.JobQueueRowSummaryForWorkflowsR\x14jobQueueRowSummariesB\x0c\n\n_job_indexB\x0f\n\r_finalized_at"\xbe\x02\n\x11WorkflowGraphEdge\x12\x32\n\x13source_operation_id\x18\x01 \x01(\tB\x02\x18\x01R\x11sourceOperationId\x12\x32\n\x13target_operation_id\x18\x02 \x01(\tB\x02\x18\x01R\x11targetOperationId\x12\x42\n\x0csatisfied_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x0bsatisfiedAt\x88\x01\x01\x12\x34\n\x16\x64\x65pendent_operation_id\x18\x04 \x01(\tR\x14\x64\x65pendentOperationId\x12\x36\n\x17\x64\x65pendency_operation_id\x18\x05 \x01(\tR\x15\x64\x65pendencyOperationIdB\x0f\n\r_satisfied_at"\x8e\x01\n\x18GetWorkflowGraphResponse\x12\x38\n\x05nodes\x18\x01 \x03(\x0b\x32".chalk.server.v1.WorkflowGraphNodeR\x05nodes\x12\x38\n\x05\x65\x64ges\x18\x02 \x03(\x0b\x32".chalk.server.v1.WorkflowGraphEdgeR\x05\x65\x64ges*\x82\x01\n\x0cWorkflowKind\x12\x1d\n\x19WORKFLOW_KIND_UNSPECIFIED\x10\x00\x12\x18\n\x14WORKFLOW_KIND_MANUAL\x10\x01\x12\x16\n\x12WORKFLOW_KIND_CRON\x10\x02\x12!\n\x1dWORKFLOW_KIND_SCHEDULED_QUERY\x10\x03*\x88\x02\n\x17WorkflowExecutionStatus\x12)\n%WORKFLOW_EXECUTION_STATUS_UNSPECIFIED\x10\x00\x12$\n WORKFLOW_EXECUTION_STATUS_QUEUED\x10\x01\x12%\n!WORKFLOW_EXECUTION_STATUS_WORKING\x10\x02\x12\'\n#WORKFLOW_EXECUTION_STATUS_COMPLETED\x10\x03\x12$\n WORKFLOW_EXECUTION_STATUS_FAILED\x10\x04\x12&\n"WORKFLOW_EXECUTION_STATUS_CANCELED\x10\x05\x32\x8d\x03\n\x19\x44\x61taPlaneWorkflowsService\x12{\n\x14GetDataPlaneWorkflow\x12,.chalk.server.v1.GetDataPlaneWorkflowRequest\x1a-.chalk.server.v1.GetDataPlaneWorkflowResponse"\x06\x90\x02\x01\x80}\x10\x12\x81\x01\n\x16ListDataPlaneWorkflows\x12..chalk.server.v1.ListDataPlaneWorkflowsRequest\x1a/.chalk.server.v1.ListDataPlaneWorkflowsResponse"\x06\x90\x02\x01\x80}\x10\x12o\n\x10GetWorkflowGraph\x12(.chalk.server.v1.GetWorkflowGraphRequest\x1a).chalk.server.v1.GetWorkflowGraphResponse"\x06\x90\x02\x01\x80}\x10\x42\xa0\x01\n\x13\x63om.chalk.server.v1B\x17\x44\x61taplaneworkflowsProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
24
+ )
25
+
26
+ _globals = globals()
27
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
28
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.dataplaneworkflows_pb2", _globals)
29
+ if _descriptor._USE_C_DESCRIPTORS == False:
30
+ _globals["DESCRIPTOR"]._options = None
31
+ _globals[
32
+ "DESCRIPTOR"
33
+ ]._serialized_options = b"\n\023com.chalk.server.v1B\027DataplaneworkflowsProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
34
+ _globals["_LISTDATAPLANEWORKFLOWSREQUEST"].fields_by_name["environment_id"]._options = None
35
+ _globals["_LISTDATAPLANEWORKFLOWSREQUEST"].fields_by_name["environment_id"]._serialized_options = b"\030\001"
36
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["job_queue_id"]._options = None
37
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["job_queue_id"]._serialized_options = b"\030\001"
38
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["job_name"]._options = None
39
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["job_name"]._serialized_options = b"\030\001"
40
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["job_index"]._options = None
41
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["job_index"]._serialized_options = b"\030\001"
42
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["state"]._options = None
43
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["state"]._serialized_options = b"\030\001"
44
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["created_at"]._options = None
45
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["created_at"]._serialized_options = b"\030\001"
46
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["finalized_at"]._options = None
47
+ _globals["_WORKFLOWGRAPHNODE"].fields_by_name["finalized_at"]._serialized_options = b"\030\001"
48
+ _globals["_WORKFLOWGRAPHEDGE"].fields_by_name["source_operation_id"]._options = None
49
+ _globals["_WORKFLOWGRAPHEDGE"].fields_by_name["source_operation_id"]._serialized_options = b"\030\001"
50
+ _globals["_WORKFLOWGRAPHEDGE"].fields_by_name["target_operation_id"]._options = None
51
+ _globals["_WORKFLOWGRAPHEDGE"].fields_by_name["target_operation_id"]._serialized_options = b"\030\001"
52
+ _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name["GetDataPlaneWorkflow"]._options = None
53
+ _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name[
54
+ "GetDataPlaneWorkflow"
55
+ ]._serialized_options = b"\220\002\001\200}\020"
56
+ _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name["ListDataPlaneWorkflows"]._options = None
57
+ _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name[
58
+ "ListDataPlaneWorkflows"
59
+ ]._serialized_options = b"\220\002\001\200}\020"
60
+ _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name["GetWorkflowGraph"]._options = None
61
+ _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name[
62
+ "GetWorkflowGraph"
63
+ ]._serialized_options = b"\220\002\001\200}\020"
64
+ _globals["_WORKFLOWKIND"]._serialized_start = 2933
65
+ _globals["_WORKFLOWKIND"]._serialized_end = 3063
66
+ _globals["_WORKFLOWEXECUTIONSTATUS"]._serialized_start = 3066
67
+ _globals["_WORKFLOWEXECUTIONSTATUS"]._serialized_end = 3330
68
+ _globals["_WORKFLOWEXECUTION"]._serialized_start = 199
69
+ _globals["_WORKFLOWEXECUTION"]._serialized_end = 908
70
+ _globals["_GETDATAPLANEWORKFLOWREQUEST"]._serialized_start = 910
71
+ _globals["_GETDATAPLANEWORKFLOWREQUEST"]._serialized_end = 955
72
+ _globals["_GETDATAPLANEWORKFLOWRESPONSE"]._serialized_start = 957
73
+ _globals["_GETDATAPLANEWORKFLOWRESPONSE"]._serialized_end = 1051
74
+ _globals["_LISTDATAPLANEWORKFLOWSREQUEST"]._serialized_start = 1054
75
+ _globals["_LISTDATAPLANEWORKFLOWSREQUEST"]._serialized_end = 1376
76
+ _globals["_LISTDATAPLANEWORKFLOWSRESPONSE"]._serialized_start = 1378
77
+ _globals["_LISTDATAPLANEWORKFLOWSRESPONSE"]._serialized_end = 1498
78
+ _globals["_GETWORKFLOWGRAPHREQUEST"]._serialized_start = 1500
79
+ _globals["_GETWORKFLOWGRAPHREQUEST"]._serialized_end = 1577
80
+ _globals["_JOBQUEUEROWSUMMARYFORWORKFLOWS"]._serialized_start = 1580
81
+ _globals["_JOBQUEUEROWSUMMARYFORWORKFLOWS"]._serialized_end = 1882
82
+ _globals["_WORKFLOWGRAPHNODE"]._serialized_start = 1885
83
+ _globals["_WORKFLOWGRAPHNODE"]._serialized_end = 2464
84
+ _globals["_WORKFLOWGRAPHEDGE"]._serialized_start = 2467
85
+ _globals["_WORKFLOWGRAPHEDGE"]._serialized_end = 2785
86
+ _globals["_GETWORKFLOWGRAPHRESPONSE"]._serialized_start = 2788
87
+ _globals["_GETWORKFLOWGRAPHRESPONSE"]._serialized_end = 2930
88
+ _globals["_DATAPLANEWORKFLOWSSERVICE"]._serialized_start = 3333
89
+ _globals["_DATAPLANEWORKFLOWSSERVICE"]._serialized_end = 3730
90
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,264 @@
1
+ from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
2
+ from chalk._gen.chalk.server.v1 import dataplanejobqueue_pb2 as _dataplanejobqueue_pb2
3
+ from google.protobuf import struct_pb2 as _struct_pb2
4
+ from google.protobuf import timestamp_pb2 as _timestamp_pb2
5
+ from google.protobuf.internal import containers as _containers
6
+ from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
7
+ from google.protobuf import descriptor as _descriptor
8
+ from google.protobuf import message as _message
9
+ from typing import (
10
+ ClassVar as _ClassVar,
11
+ Iterable as _Iterable,
12
+ Mapping as _Mapping,
13
+ Optional as _Optional,
14
+ Union as _Union,
15
+ )
16
+
17
+ DESCRIPTOR: _descriptor.FileDescriptor
18
+
19
+ class WorkflowKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
20
+ __slots__ = ()
21
+ WORKFLOW_KIND_UNSPECIFIED: _ClassVar[WorkflowKind]
22
+ WORKFLOW_KIND_MANUAL: _ClassVar[WorkflowKind]
23
+ WORKFLOW_KIND_CRON: _ClassVar[WorkflowKind]
24
+ WORKFLOW_KIND_SCHEDULED_QUERY: _ClassVar[WorkflowKind]
25
+
26
+ class WorkflowExecutionStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
27
+ __slots__ = ()
28
+ WORKFLOW_EXECUTION_STATUS_UNSPECIFIED: _ClassVar[WorkflowExecutionStatus]
29
+ WORKFLOW_EXECUTION_STATUS_QUEUED: _ClassVar[WorkflowExecutionStatus]
30
+ WORKFLOW_EXECUTION_STATUS_WORKING: _ClassVar[WorkflowExecutionStatus]
31
+ WORKFLOW_EXECUTION_STATUS_COMPLETED: _ClassVar[WorkflowExecutionStatus]
32
+ WORKFLOW_EXECUTION_STATUS_FAILED: _ClassVar[WorkflowExecutionStatus]
33
+ WORKFLOW_EXECUTION_STATUS_CANCELED: _ClassVar[WorkflowExecutionStatus]
34
+
35
+ WORKFLOW_KIND_UNSPECIFIED: WorkflowKind
36
+ WORKFLOW_KIND_MANUAL: WorkflowKind
37
+ WORKFLOW_KIND_CRON: WorkflowKind
38
+ WORKFLOW_KIND_SCHEDULED_QUERY: WorkflowKind
39
+ WORKFLOW_EXECUTION_STATUS_UNSPECIFIED: WorkflowExecutionStatus
40
+ WORKFLOW_EXECUTION_STATUS_QUEUED: WorkflowExecutionStatus
41
+ WORKFLOW_EXECUTION_STATUS_WORKING: WorkflowExecutionStatus
42
+ WORKFLOW_EXECUTION_STATUS_COMPLETED: WorkflowExecutionStatus
43
+ WORKFLOW_EXECUTION_STATUS_FAILED: WorkflowExecutionStatus
44
+ WORKFLOW_EXECUTION_STATUS_CANCELED: WorkflowExecutionStatus
45
+
46
+ class WorkflowExecution(_message.Message):
47
+ __slots__ = (
48
+ "id",
49
+ "created_at",
50
+ "environment_id",
51
+ "deployment_id",
52
+ "kind",
53
+ "status",
54
+ "branch_name",
55
+ "mainline_deployment_id",
56
+ "agent_id",
57
+ "meta_data",
58
+ "updated_at",
59
+ "finalized_at",
60
+ )
61
+ ID_FIELD_NUMBER: _ClassVar[int]
62
+ CREATED_AT_FIELD_NUMBER: _ClassVar[int]
63
+ ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
64
+ DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
65
+ KIND_FIELD_NUMBER: _ClassVar[int]
66
+ STATUS_FIELD_NUMBER: _ClassVar[int]
67
+ BRANCH_NAME_FIELD_NUMBER: _ClassVar[int]
68
+ MAINLINE_DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
69
+ AGENT_ID_FIELD_NUMBER: _ClassVar[int]
70
+ META_DATA_FIELD_NUMBER: _ClassVar[int]
71
+ UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
72
+ FINALIZED_AT_FIELD_NUMBER: _ClassVar[int]
73
+ id: str
74
+ created_at: _timestamp_pb2.Timestamp
75
+ environment_id: str
76
+ deployment_id: str
77
+ kind: WorkflowKind
78
+ status: WorkflowExecutionStatus
79
+ branch_name: str
80
+ mainline_deployment_id: str
81
+ agent_id: str
82
+ meta_data: _struct_pb2.Struct
83
+ updated_at: _timestamp_pb2.Timestamp
84
+ finalized_at: _timestamp_pb2.Timestamp
85
+ def __init__(
86
+ self,
87
+ id: _Optional[str] = ...,
88
+ created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
89
+ environment_id: _Optional[str] = ...,
90
+ deployment_id: _Optional[str] = ...,
91
+ kind: _Optional[_Union[WorkflowKind, str]] = ...,
92
+ status: _Optional[_Union[WorkflowExecutionStatus, str]] = ...,
93
+ branch_name: _Optional[str] = ...,
94
+ mainline_deployment_id: _Optional[str] = ...,
95
+ agent_id: _Optional[str] = ...,
96
+ meta_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
97
+ updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
98
+ finalized_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
99
+ ) -> None: ...
100
+
101
+ class GetDataPlaneWorkflowRequest(_message.Message):
102
+ __slots__ = ("id",)
103
+ ID_FIELD_NUMBER: _ClassVar[int]
104
+ id: str
105
+ def __init__(self, id: _Optional[str] = ...) -> None: ...
106
+
107
+ class GetDataPlaneWorkflowResponse(_message.Message):
108
+ __slots__ = ("workflow",)
109
+ WORKFLOW_FIELD_NUMBER: _ClassVar[int]
110
+ workflow: WorkflowExecution
111
+ def __init__(self, workflow: _Optional[_Union[WorkflowExecution, _Mapping]] = ...) -> None: ...
112
+
113
+ class ListDataPlaneWorkflowsRequest(_message.Message):
114
+ __slots__ = ("environment_id", "deployment_id", "kind", "status", "limit", "offset", "id_filter")
115
+ ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
116
+ DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
117
+ KIND_FIELD_NUMBER: _ClassVar[int]
118
+ STATUS_FIELD_NUMBER: _ClassVar[int]
119
+ LIMIT_FIELD_NUMBER: _ClassVar[int]
120
+ OFFSET_FIELD_NUMBER: _ClassVar[int]
121
+ ID_FILTER_FIELD_NUMBER: _ClassVar[int]
122
+ environment_id: str
123
+ deployment_id: str
124
+ kind: WorkflowKind
125
+ status: WorkflowExecutionStatus
126
+ limit: int
127
+ offset: int
128
+ id_filter: str
129
+ def __init__(
130
+ self,
131
+ environment_id: _Optional[str] = ...,
132
+ deployment_id: _Optional[str] = ...,
133
+ kind: _Optional[_Union[WorkflowKind, str]] = ...,
134
+ status: _Optional[_Union[WorkflowExecutionStatus, str]] = ...,
135
+ limit: _Optional[int] = ...,
136
+ offset: _Optional[int] = ...,
137
+ id_filter: _Optional[str] = ...,
138
+ ) -> None: ...
139
+
140
+ class ListDataPlaneWorkflowsResponse(_message.Message):
141
+ __slots__ = ("workflows", "total")
142
+ WORKFLOWS_FIELD_NUMBER: _ClassVar[int]
143
+ TOTAL_FIELD_NUMBER: _ClassVar[int]
144
+ workflows: _containers.RepeatedCompositeFieldContainer[WorkflowExecution]
145
+ total: int
146
+ def __init__(
147
+ self, workflows: _Optional[_Iterable[_Union[WorkflowExecution, _Mapping]]] = ..., total: _Optional[int] = ...
148
+ ) -> None: ...
149
+
150
+ class GetWorkflowGraphRequest(_message.Message):
151
+ __slots__ = ("workflow_execution_id",)
152
+ WORKFLOW_EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
153
+ workflow_execution_id: str
154
+ def __init__(self, workflow_execution_id: _Optional[str] = ...) -> None: ...
155
+
156
+ class JobQueueRowSummaryForWorkflows(_message.Message):
157
+ __slots__ = ("id", "job_name", "state", "job_index", "created_at", "finalized_at")
158
+ ID_FIELD_NUMBER: _ClassVar[int]
159
+ JOB_NAME_FIELD_NUMBER: _ClassVar[int]
160
+ STATE_FIELD_NUMBER: _ClassVar[int]
161
+ JOB_INDEX_FIELD_NUMBER: _ClassVar[int]
162
+ CREATED_AT_FIELD_NUMBER: _ClassVar[int]
163
+ FINALIZED_AT_FIELD_NUMBER: _ClassVar[int]
164
+ id: int
165
+ job_name: str
166
+ state: _dataplanejobqueue_pb2.JobQueueState
167
+ job_index: int
168
+ created_at: _timestamp_pb2.Timestamp
169
+ finalized_at: _timestamp_pb2.Timestamp
170
+ def __init__(
171
+ self,
172
+ id: _Optional[int] = ...,
173
+ job_name: _Optional[str] = ...,
174
+ state: _Optional[_Union[_dataplanejobqueue_pb2.JobQueueState, str]] = ...,
175
+ job_index: _Optional[int] = ...,
176
+ created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
177
+ finalized_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
178
+ ) -> None: ...
179
+
180
+ class WorkflowGraphNode(_message.Message):
181
+ __slots__ = (
182
+ "job_queue_id",
183
+ "job_name",
184
+ "job_index",
185
+ "state",
186
+ "created_at",
187
+ "finalized_at",
188
+ "operation_id",
189
+ "kind",
190
+ "resource_group",
191
+ "job_queue_row_summaries",
192
+ )
193
+ JOB_QUEUE_ID_FIELD_NUMBER: _ClassVar[int]
194
+ JOB_NAME_FIELD_NUMBER: _ClassVar[int]
195
+ JOB_INDEX_FIELD_NUMBER: _ClassVar[int]
196
+ STATE_FIELD_NUMBER: _ClassVar[int]
197
+ CREATED_AT_FIELD_NUMBER: _ClassVar[int]
198
+ FINALIZED_AT_FIELD_NUMBER: _ClassVar[int]
199
+ OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
200
+ KIND_FIELD_NUMBER: _ClassVar[int]
201
+ RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
202
+ JOB_QUEUE_ROW_SUMMARIES_FIELD_NUMBER: _ClassVar[int]
203
+ job_queue_id: int
204
+ job_name: str
205
+ job_index: int
206
+ state: _dataplanejobqueue_pb2.JobQueueState
207
+ created_at: _timestamp_pb2.Timestamp
208
+ finalized_at: _timestamp_pb2.Timestamp
209
+ operation_id: str
210
+ kind: _dataplanejobqueue_pb2.JobQueueKind
211
+ resource_group: str
212
+ job_queue_row_summaries: _containers.RepeatedCompositeFieldContainer[JobQueueRowSummaryForWorkflows]
213
+ def __init__(
214
+ self,
215
+ job_queue_id: _Optional[int] = ...,
216
+ job_name: _Optional[str] = ...,
217
+ job_index: _Optional[int] = ...,
218
+ state: _Optional[_Union[_dataplanejobqueue_pb2.JobQueueState, str]] = ...,
219
+ created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
220
+ finalized_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
221
+ operation_id: _Optional[str] = ...,
222
+ kind: _Optional[_Union[_dataplanejobqueue_pb2.JobQueueKind, str]] = ...,
223
+ resource_group: _Optional[str] = ...,
224
+ job_queue_row_summaries: _Optional[_Iterable[_Union[JobQueueRowSummaryForWorkflows, _Mapping]]] = ...,
225
+ ) -> None: ...
226
+
227
+ class WorkflowGraphEdge(_message.Message):
228
+ __slots__ = (
229
+ "source_operation_id",
230
+ "target_operation_id",
231
+ "satisfied_at",
232
+ "dependent_operation_id",
233
+ "dependency_operation_id",
234
+ )
235
+ SOURCE_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
236
+ TARGET_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
237
+ SATISFIED_AT_FIELD_NUMBER: _ClassVar[int]
238
+ DEPENDENT_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
239
+ DEPENDENCY_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
240
+ source_operation_id: str
241
+ target_operation_id: str
242
+ satisfied_at: _timestamp_pb2.Timestamp
243
+ dependent_operation_id: str
244
+ dependency_operation_id: str
245
+ def __init__(
246
+ self,
247
+ source_operation_id: _Optional[str] = ...,
248
+ target_operation_id: _Optional[str] = ...,
249
+ satisfied_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
250
+ dependent_operation_id: _Optional[str] = ...,
251
+ dependency_operation_id: _Optional[str] = ...,
252
+ ) -> None: ...
253
+
254
+ class GetWorkflowGraphResponse(_message.Message):
255
+ __slots__ = ("nodes", "edges")
256
+ NODES_FIELD_NUMBER: _ClassVar[int]
257
+ EDGES_FIELD_NUMBER: _ClassVar[int]
258
+ nodes: _containers.RepeatedCompositeFieldContainer[WorkflowGraphNode]
259
+ edges: _containers.RepeatedCompositeFieldContainer[WorkflowGraphEdge]
260
+ def __init__(
261
+ self,
262
+ nodes: _Optional[_Iterable[_Union[WorkflowGraphNode, _Mapping]]] = ...,
263
+ edges: _Optional[_Iterable[_Union[WorkflowGraphEdge, _Mapping]]] = ...,
264
+ ) -> None: ...
@@ -0,0 +1,170 @@
1
+ # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
+ """Client and server classes corresponding to protobuf-defined services."""
3
+
4
+ import grpc
5
+
6
+ from chalk._gen.chalk.server.v1 import dataplaneworkflows_pb2 as chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2
7
+
8
+
9
+ class DataPlaneWorkflowsServiceStub(object):
10
+ """Missing associated documentation comment in .proto file."""
11
+
12
+ def __init__(self, channel):
13
+ """Constructor.
14
+
15
+ Args:
16
+ channel: A grpc.Channel.
17
+ """
18
+ self.GetDataPlaneWorkflow = channel.unary_unary(
19
+ "/chalk.server.v1.DataPlaneWorkflowsService/GetDataPlaneWorkflow",
20
+ request_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowRequest.SerializeToString,
21
+ response_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowResponse.FromString,
22
+ )
23
+ self.ListDataPlaneWorkflows = channel.unary_unary(
24
+ "/chalk.server.v1.DataPlaneWorkflowsService/ListDataPlaneWorkflows",
25
+ request_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsRequest.SerializeToString,
26
+ response_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsResponse.FromString,
27
+ )
28
+ self.GetWorkflowGraph = channel.unary_unary(
29
+ "/chalk.server.v1.DataPlaneWorkflowsService/GetWorkflowGraph",
30
+ request_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphRequest.SerializeToString,
31
+ response_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphResponse.FromString,
32
+ )
33
+
34
+
35
+ class DataPlaneWorkflowsServiceServicer(object):
36
+ """Missing associated documentation comment in .proto file."""
37
+
38
+ def GetDataPlaneWorkflow(self, request, context):
39
+ """Missing associated documentation comment in .proto file."""
40
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
41
+ context.set_details("Method not implemented!")
42
+ raise NotImplementedError("Method not implemented!")
43
+
44
+ def ListDataPlaneWorkflows(self, request, context):
45
+ """Missing associated documentation comment in .proto file."""
46
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
47
+ context.set_details("Method not implemented!")
48
+ raise NotImplementedError("Method not implemented!")
49
+
50
+ def GetWorkflowGraph(self, request, context):
51
+ """Missing associated documentation comment in .proto file."""
52
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
53
+ context.set_details("Method not implemented!")
54
+ raise NotImplementedError("Method not implemented!")
55
+
56
+
57
+ def add_DataPlaneWorkflowsServiceServicer_to_server(servicer, server):
58
+ rpc_method_handlers = {
59
+ "GetDataPlaneWorkflow": grpc.unary_unary_rpc_method_handler(
60
+ servicer.GetDataPlaneWorkflow,
61
+ request_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowRequest.FromString,
62
+ response_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowResponse.SerializeToString,
63
+ ),
64
+ "ListDataPlaneWorkflows": grpc.unary_unary_rpc_method_handler(
65
+ servicer.ListDataPlaneWorkflows,
66
+ request_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsRequest.FromString,
67
+ response_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsResponse.SerializeToString,
68
+ ),
69
+ "GetWorkflowGraph": grpc.unary_unary_rpc_method_handler(
70
+ servicer.GetWorkflowGraph,
71
+ request_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphRequest.FromString,
72
+ response_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphResponse.SerializeToString,
73
+ ),
74
+ }
75
+ generic_handler = grpc.method_handlers_generic_handler(
76
+ "chalk.server.v1.DataPlaneWorkflowsService", rpc_method_handlers
77
+ )
78
+ server.add_generic_rpc_handlers((generic_handler,))
79
+
80
+
81
+ # This class is part of an EXPERIMENTAL API.
82
+ class DataPlaneWorkflowsService(object):
83
+ """Missing associated documentation comment in .proto file."""
84
+
85
+ @staticmethod
86
+ def GetDataPlaneWorkflow(
87
+ request,
88
+ target,
89
+ options=(),
90
+ channel_credentials=None,
91
+ call_credentials=None,
92
+ insecure=False,
93
+ compression=None,
94
+ wait_for_ready=None,
95
+ timeout=None,
96
+ metadata=None,
97
+ ):
98
+ return grpc.experimental.unary_unary(
99
+ request,
100
+ target,
101
+ "/chalk.server.v1.DataPlaneWorkflowsService/GetDataPlaneWorkflow",
102
+ chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowRequest.SerializeToString,
103
+ chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowResponse.FromString,
104
+ options,
105
+ channel_credentials,
106
+ insecure,
107
+ call_credentials,
108
+ compression,
109
+ wait_for_ready,
110
+ timeout,
111
+ metadata,
112
+ )
113
+
114
+ @staticmethod
115
+ def ListDataPlaneWorkflows(
116
+ request,
117
+ target,
118
+ options=(),
119
+ channel_credentials=None,
120
+ call_credentials=None,
121
+ insecure=False,
122
+ compression=None,
123
+ wait_for_ready=None,
124
+ timeout=None,
125
+ metadata=None,
126
+ ):
127
+ return grpc.experimental.unary_unary(
128
+ request,
129
+ target,
130
+ "/chalk.server.v1.DataPlaneWorkflowsService/ListDataPlaneWorkflows",
131
+ chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsRequest.SerializeToString,
132
+ chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsResponse.FromString,
133
+ options,
134
+ channel_credentials,
135
+ insecure,
136
+ call_credentials,
137
+ compression,
138
+ wait_for_ready,
139
+ timeout,
140
+ metadata,
141
+ )
142
+
143
+ @staticmethod
144
+ def GetWorkflowGraph(
145
+ request,
146
+ target,
147
+ options=(),
148
+ channel_credentials=None,
149
+ call_credentials=None,
150
+ insecure=False,
151
+ compression=None,
152
+ wait_for_ready=None,
153
+ timeout=None,
154
+ metadata=None,
155
+ ):
156
+ return grpc.experimental.unary_unary(
157
+ request,
158
+ target,
159
+ "/chalk.server.v1.DataPlaneWorkflowsService/GetWorkflowGraph",
160
+ chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphRequest.SerializeToString,
161
+ chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphResponse.FromString,
162
+ options,
163
+ channel_credentials,
164
+ insecure,
165
+ call_credentials,
166
+ compression,
167
+ wait_for_ready,
168
+ timeout,
169
+ metadata,
170
+ )
@@ -0,0 +1,62 @@
1
+ """
2
+ @generated by mypy-protobuf. Do not edit manually!
3
+ isort:skip_file
4
+ """
5
+
6
+ from abc import (
7
+ ABCMeta,
8
+ abstractmethod,
9
+ )
10
+ from chalk._gen.chalk.server.v1.dataplaneworkflows_pb2 import (
11
+ GetDataPlaneWorkflowRequest,
12
+ GetDataPlaneWorkflowResponse,
13
+ GetWorkflowGraphRequest,
14
+ GetWorkflowGraphResponse,
15
+ ListDataPlaneWorkflowsRequest,
16
+ ListDataPlaneWorkflowsResponse,
17
+ )
18
+ from grpc import (
19
+ Channel,
20
+ Server,
21
+ ServicerContext,
22
+ UnaryUnaryMultiCallable,
23
+ )
24
+
25
+ class DataPlaneWorkflowsServiceStub:
26
+ def __init__(self, channel: Channel) -> None: ...
27
+ GetDataPlaneWorkflow: UnaryUnaryMultiCallable[
28
+ GetDataPlaneWorkflowRequest,
29
+ GetDataPlaneWorkflowResponse,
30
+ ]
31
+ ListDataPlaneWorkflows: UnaryUnaryMultiCallable[
32
+ ListDataPlaneWorkflowsRequest,
33
+ ListDataPlaneWorkflowsResponse,
34
+ ]
35
+ GetWorkflowGraph: UnaryUnaryMultiCallable[
36
+ GetWorkflowGraphRequest,
37
+ GetWorkflowGraphResponse,
38
+ ]
39
+
40
+ class DataPlaneWorkflowsServiceServicer(metaclass=ABCMeta):
41
+ @abstractmethod
42
+ def GetDataPlaneWorkflow(
43
+ self,
44
+ request: GetDataPlaneWorkflowRequest,
45
+ context: ServicerContext,
46
+ ) -> GetDataPlaneWorkflowResponse: ...
47
+ @abstractmethod
48
+ def ListDataPlaneWorkflows(
49
+ self,
50
+ request: ListDataPlaneWorkflowsRequest,
51
+ context: ServicerContext,
52
+ ) -> ListDataPlaneWorkflowsResponse: ...
53
+ @abstractmethod
54
+ def GetWorkflowGraph(
55
+ self,
56
+ request: GetWorkflowGraphRequest,
57
+ context: ServicerContext,
58
+ ) -> GetWorkflowGraphResponse: ...
59
+
60
+ def add_DataPlaneWorkflowsServiceServicer_to_server(
61
+ servicer: DataPlaneWorkflowsServiceServicer, server: Server
62
+ ) -> None: ...