chalkpy 2.90.1__py3-none-any.whl → 2.95.3__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (202) hide show
  1. chalk/__init__.py +2 -1
  2. chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
  3. chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
  4. chalk/_gen/chalk/artifacts/v1/chart_pb2.py +16 -16
  5. chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +4 -0
  6. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +8 -7
  7. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +5 -0
  8. chalk/_gen/chalk/common/v1/offline_query_pb2.py +17 -15
  9. chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +25 -0
  10. chalk/_gen/chalk/common/v1/script_task_pb2.py +3 -3
  11. chalk/_gen/chalk/common/v1/script_task_pb2.pyi +2 -0
  12. chalk/_gen/chalk/dataframe/__init__.py +0 -0
  13. chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
  14. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
  15. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +123 -0
  16. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
  17. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
  18. chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
  19. chalk/_gen/chalk/graph/v1/graph_pb2.pyi +25 -0
  20. chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
  21. chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -0
  22. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
  23. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
  24. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
  25. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
  26. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
  27. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
  28. chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +54 -27
  29. chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
  30. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
  31. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
  32. chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
  33. chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
  34. chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
  35. chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
  36. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
  37. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
  38. chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
  39. chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
  40. chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
  41. chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
  42. chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
  43. chalk/_gen/chalk/server/v1/builder_pb2.py +358 -288
  44. chalk/_gen/chalk/server/v1/builder_pb2.pyi +360 -10
  45. chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +225 -0
  46. chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +60 -0
  47. chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
  48. chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
  49. chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
  50. chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
  51. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
  52. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
  53. chalk/_gen/chalk/server/v1/cloud_components_pb2.py +141 -119
  54. chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +106 -4
  55. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +45 -0
  56. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +12 -0
  57. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
  58. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
  59. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
  60. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
  61. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +52 -38
  62. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +62 -1
  63. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +90 -0
  64. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +24 -0
  65. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
  66. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -0
  67. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
  68. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
  69. chalk/_gen/chalk/server/v1/deploy_pb2.py +9 -3
  70. chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
  71. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
  72. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
  73. chalk/_gen/chalk/server/v1/deployment_pb2.py +6 -6
  74. chalk/_gen/chalk/server/v1/deployment_pb2.pyi +20 -0
  75. chalk/_gen/chalk/server/v1/environment_pb2.py +14 -12
  76. chalk/_gen/chalk/server/v1/environment_pb2.pyi +19 -0
  77. chalk/_gen/chalk/server/v1/eventbus_pb2.py +4 -2
  78. chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
  79. chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
  80. chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
  81. chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
  82. chalk/_gen/chalk/server/v1/graph_pb2.py +38 -26
  83. chalk/_gen/chalk/server/v1/graph_pb2.pyi +58 -0
  84. chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +47 -0
  85. chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +18 -0
  86. chalk/_gen/chalk/server/v1/incident_pb2.py +23 -21
  87. chalk/_gen/chalk/server/v1/incident_pb2.pyi +15 -1
  88. chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
  89. chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
  90. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
  91. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
  92. chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
  93. chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
  94. chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
  95. chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
  96. chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
  97. chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
  98. chalk/_gen/chalk/server/v1/log_pb2.py +21 -3
  99. chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
  100. chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
  101. chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
  102. chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
  103. chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
  104. chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
  105. chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
  106. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
  107. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
  108. chalk/_gen/chalk/server/v1/queries_pb2.py +66 -66
  109. chalk/_gen/chalk/server/v1/queries_pb2.pyi +32 -2
  110. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -12
  111. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +16 -3
  112. chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
  113. chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
  114. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
  115. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
  116. chalk/_gen/chalk/server/v1/script_tasks_pb2.py +15 -3
  117. chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +22 -0
  118. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
  119. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -0
  120. chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
  121. chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
  122. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
  123. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
  124. chalk/_gen/chalk/server/v1/sql_queries_pb2.py +48 -0
  125. chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
  126. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
  127. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
  128. chalk/_gen/chalk/server/v1/team_pb2.py +154 -141
  129. chalk/_gen/chalk/server/v1/team_pb2.pyi +30 -2
  130. chalk/_gen/chalk/server/v1/team_pb2_grpc.py +45 -0
  131. chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +12 -0
  132. chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
  133. chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
  134. chalk/_gen/chalk/server/v1/trace_pb2.py +44 -40
  135. chalk/_gen/chalk/server/v1/trace_pb2.pyi +20 -0
  136. chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
  137. chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
  138. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
  139. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
  140. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +16 -10
  141. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +52 -1
  142. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
  143. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
  144. chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
  145. chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
  146. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
  147. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
  148. chalk/_lsp/error_builder.py +11 -0
  149. chalk/_version.py +1 -1
  150. chalk/client/client.py +128 -43
  151. chalk/client/client_async.py +149 -0
  152. chalk/client/client_async_impl.py +22 -0
  153. chalk/client/client_grpc.py +539 -104
  154. chalk/client/client_impl.py +449 -122
  155. chalk/client/dataset.py +7 -1
  156. chalk/client/models.py +98 -0
  157. chalk/client/serialization/model_serialization.py +92 -9
  158. chalk/df/LazyFramePlaceholder.py +1154 -0
  159. chalk/features/_class_property.py +7 -0
  160. chalk/features/_embedding/embedding.py +1 -0
  161. chalk/features/_encoding/converter.py +83 -2
  162. chalk/features/feature_field.py +40 -30
  163. chalk/features/feature_set_decorator.py +1 -0
  164. chalk/features/feature_wrapper.py +42 -3
  165. chalk/features/hooks.py +81 -10
  166. chalk/features/inference.py +33 -31
  167. chalk/features/resolver.py +224 -24
  168. chalk/functions/__init__.py +65 -3
  169. chalk/gitignore/gitignore_parser.py +5 -1
  170. chalk/importer.py +142 -68
  171. chalk/ml/__init__.py +2 -0
  172. chalk/ml/model_hooks.py +194 -26
  173. chalk/ml/model_reference.py +56 -8
  174. chalk/ml/model_version.py +24 -15
  175. chalk/ml/utils.py +20 -17
  176. chalk/operators/_utils.py +10 -3
  177. chalk/parsed/_proto/export.py +22 -0
  178. chalk/parsed/duplicate_input_gql.py +3 -0
  179. chalk/parsed/json_conversions.py +20 -14
  180. chalk/parsed/to_proto.py +16 -4
  181. chalk/parsed/user_types_to_json.py +31 -10
  182. chalk/parsed/validation_from_registries.py +182 -0
  183. chalk/queries/named_query.py +16 -6
  184. chalk/queries/scheduled_query.py +9 -1
  185. chalk/serialization/parsed_annotation.py +24 -11
  186. chalk/sql/__init__.py +18 -0
  187. chalk/sql/_internal/integrations/databricks.py +55 -17
  188. chalk/sql/_internal/integrations/mssql.py +127 -62
  189. chalk/sql/_internal/integrations/redshift.py +4 -0
  190. chalk/sql/_internal/sql_file_resolver.py +53 -9
  191. chalk/sql/_internal/sql_source.py +35 -2
  192. chalk/streams/_kafka_source.py +5 -1
  193. chalk/streams/_windows.py +15 -2
  194. chalk/utils/_otel_version.py +13 -0
  195. chalk/utils/async_helpers.py +2 -2
  196. chalk/utils/missing_dependency.py +5 -4
  197. chalk/utils/tracing.py +185 -95
  198. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/METADATA +4 -6
  199. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/RECORD +202 -146
  200. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
  201. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
  202. {chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/top_level.txt +0 -0
@@ -8,6 +8,10 @@ from abc import (
8
8
  abstractmethod,
9
9
  )
10
10
  from chalk._gen.chalk.server.v1.log_pb2 import (
11
+ GetLogFacetValuesRequest,
12
+ GetLogFacetValuesResponse,
13
+ GetLogFacetsRequest,
14
+ GetLogFacetsResponse,
11
15
  SearchLogEntriesAggregatedRequest,
12
16
  SearchLogEntriesAggregatedResponse,
13
17
  SearchLogEntriesRequest,
@@ -30,6 +34,14 @@ class LogSearchServiceStub:
30
34
  SearchLogEntriesAggregatedRequest,
31
35
  SearchLogEntriesAggregatedResponse,
32
36
  ]
37
+ GetLogFacets: UnaryUnaryMultiCallable[
38
+ GetLogFacetsRequest,
39
+ GetLogFacetsResponse,
40
+ ]
41
+ GetLogFacetValues: UnaryUnaryMultiCallable[
42
+ GetLogFacetValuesRequest,
43
+ GetLogFacetValuesResponse,
44
+ ]
33
45
 
34
46
  class LogSearchServiceServicer(metaclass=ABCMeta):
35
47
  @abstractmethod
@@ -44,5 +56,17 @@ class LogSearchServiceServicer(metaclass=ABCMeta):
44
56
  request: SearchLogEntriesAggregatedRequest,
45
57
  context: ServicerContext,
46
58
  ) -> SearchLogEntriesAggregatedResponse: ...
59
+ @abstractmethod
60
+ def GetLogFacets(
61
+ self,
62
+ request: GetLogFacetsRequest,
63
+ context: ServicerContext,
64
+ ) -> GetLogFacetsResponse: ...
65
+ @abstractmethod
66
+ def GetLogFacetValues(
67
+ self,
68
+ request: GetLogFacetValuesRequest,
69
+ context: ServicerContext,
70
+ ) -> GetLogFacetValuesResponse: ...
47
71
 
48
72
  def add_LogSearchServiceServicer_to_server(servicer: LogSearchServiceServicer, server: Server) -> None: ...
@@ -23,7 +23,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
23
23
 
24
24
 
25
25
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
26
- b'\n$chalk/server/v1/model_registry.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1a\x63halk/flags/v1/flags.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a$chalk/models/v1/model_artifact.proto\x1a google/protobuf/field_mask.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xf8\x02\n\rModelArtifact\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04path\x18\x02 \x01(\tR\x04path\x12\x36\n\x04spec\x18\x03 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\x04spec\x12H\n\x08metadata\x18\x06 \x03(\x0b\x32,.chalk.server.v1.ModelArtifact.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x04 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xc4\x03\n\x0cModelVersion\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12\x45\n\x0emodel_artifact\x18\x04 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x05 \x03(\tR\x07\x61liases\x12G\n\x08metadata\x18\x06 \x03(\x0b\x32+.chalk.server.v1.ModelVersion.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x07 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xed\x04\n\x05Model\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12@\n\x08metadata\x18\x04 \x03(\x0b\x32$.chalk.server.v1.Model.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x05 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12>\n\nupdated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tupdatedAt\x88\x01\x01\x12@\n\x0b\x61rchived_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x12T\n\x14latest_model_version\x18\t \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x03R\x12latestModelVersion\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_byB\r\n\x0b_updated_atB\x0e\n\x0c_archived_atB\x17\n\x15_latest_model_version"`\n\x11ListModelsRequest\x12\x1b\n\x06\x63ursor\x18\x01 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x42\t\n\x07_cursorB\x08\n\x06_limit"z\n\x12ListModelsResponse\x12.\n\x06models\x18\x01 \x03(\x0b\x32\x16.chalk.server.v1.ModelR\x06models\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"c\n\x0fGetModelRequest\x12\x1b\n\x08model_id\x18\x01 \x01(\tH\x00R\x07modelId\x12\x1f\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelNameB\x12\n\x10model_identifier"O\n\x10GetModelResponse\x12\x31\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelH\x00R\x05model\x88\x01\x01\x42\x08\n\x06_model"\xf9\x01\n\x12\x43reateModelRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x02 \x01(\tR\x0b\x64\x65scription\x12M\n\x08metadata\x18\x04 \x03(\x0b\x32\x31.chalk.server.v1.CreateModelRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"C\n\x13\x43reateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\xf8\x02\n\x14UpdateModelOperation\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12%\n\x0b\x64\x65scription\x18\x02 \x01(\tH\x01R\x0b\x64\x65scription\x88\x01\x01\x12O\n\x08metadata\x18\x03 \x03(\x0b\x32\x33.chalk.server.v1.UpdateModelOperation.MetadataEntryR\x08metadata\x12@\n\x0b\x61rchived_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_model_nameB\x0e\n\x0c_descriptionB\x0e\n\x0c_archived_at"\xab\x01\n\x12UpdateModelRequest\x12\x19\n\x08model_id\x18\x01 \x01(\tR\x07modelId\x12=\n\x06update\x18\x02 \x01(\x0b\x32%.chalk.server.v1.UpdateModelOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"C\n\x13UpdateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\x9a\x01\n\x18ListModelVersionsRequest\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x01R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\t\n\x07_cursorB\x08\n\x06_limit"\x97\x01\n\x19ListModelVersionsResponse\x12\x44\n\x0emodel_versions\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\rmodelVersions\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"Q\n\x16GetModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"t\n\x17GetModelVersionResponse\x12G\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x00R\x0cmodelVersion\x88\x01\x01\x42\x10\n\x0e_model_version"\xbf\x02\n\x1a\x43reateModelArtifactRequest\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12U\n\x08metadata\x18\x04 \x03(\x0b\x32\x39.chalk.server.v1.CreateModelArtifactRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"d\n\x1b\x43reateModelArtifactResponse\x12\x45\n\x0emodel_artifact\x18\x01 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact"\xf6\x02\n\x19\x43reateModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x03 \x03(\tR\x07\x61liases\x12T\n\x08metadata\x18\x04 \x03(\x0b\x32\x38.chalk.server.v1.CreateModelVersionRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"`\n\x1a\x43reateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"J\n\x0fModelVersionKey\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"\xe4\x01\n\x1bUpdateModelVersionOperation\x12\x18\n\x07\x61liases\x18\x01 \x03(\tR\x07\x61liases\x12V\n\x08metadata\x18\x02 \x03(\x0b\x32:.chalk.server.v1.UpdateModelVersionOperation.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"\xec\x01\n\x19UpdateModelVersionRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey\x12\x44\n\x06update\x18\x02 \x01(\x0b\x32,.chalk.server.v1.UpdateModelVersionOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"`\n\x1aUpdateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"B\n!GetModelArtifactUploadUrlsRequest\x12\x1d\n\nfile_names\x18\x01 \x03(\tR\tfileNames"\xf5\x01\n"GetModelArtifactUploadUrlsResponse\x12\x64\n\x0bupload_urls\x18\x01 \x03(\x0b\x32\x43.chalk.server.v1.GetModelArtifactUploadUrlsResponse.UploadUrlsEntryR\nuploadUrls\x12*\n\x11model_artifact_id\x18\x02 \x01(\tR\x0fmodelArtifactId\x1a=\n\x0fUploadUrlsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"l\n\x1c\x44ownloadModelArtifactRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey"1\n\x1d\x44ownloadModelArtifactResponse\x12\x10\n\x03uri\x18\x01 \x01(\tR\x03uri"\xfc\x01\n\x19GetModelReferencesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12"\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12(\n\rmodel_version\x18\x03 \x01(\rH\x01R\x0cmodelVersion\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x04 \x01(\tH\x02R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x05 \x01(\x05H\x03R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0e_model_versionB\t\n\x07_cursorB\x08\n\x06_limit"]\n\rModelRelation\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures\x12%\n\x0eoutput_feature\x18\x02 \x01(\tR\routputFeature"\x8d\x03\n\x0eModelReference\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\trelations\x18\x05 \x03(\x0b\x32\x1e.chalk.server.v1.ModelRelationR\trelations\x12\x1c\n\tresolvers\x18\x06 \x03(\tR\tresolvers\x12\\\n\x15source_file_reference\x18\x07 \x01(\x0b\x32#.chalk.graph.v1.SourceFileReferenceH\x00R\x13sourceFileReference\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAtB\x18\n\x16_source_file_reference"\x9e\x01\n\x1aGetModelReferencesResponse\x12J\n\x10model_references\x18\x01 \x03(\x0b\x32\x1f.chalk.server.v1.ModelReferenceR\x0fmodelReferences\x12$\n\x0bnext_cursor\x18\x04 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"\xa2\x01\n\x18GetModelReferenceRequest\x12\x1d\n\x08model_id\x18\x01 \x01(\tB\x02\x18\x01R\x07modelId\x12\x1d\n\nmodel_name\x18\x04 \x01(\tR\tmodelName\x12#\n\rmodel_version\x18\x02 \x01(\x04R\x0cmodelVersion\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId"~\n\x19GetModelReferenceResponse\x12M\n\x0fmodel_reference\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.ModelReferenceH\x00R\x0emodelReference\x88\x01\x01\x42\x12\n\x10_model_reference"\xa6\x01\n\x0cRunCriterion\x12\x15\n\x06run_id\x18\x01 \x01(\tR\x05runId\x12\x1b\n\x06metric\x18\x02 \x01(\tH\x00R\x06metric\x88\x01\x01\x12I\n\tdirection\x18\x03 \x01(\x0e\x32&.chalk.server.v1.RunCriterionDirectionH\x01R\tdirection\x88\x01\x01\x42\t\n\x07_metricB\x0c\n\n_direction"\xe9\x01\n%CreateModelVersionFromArtifactRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12,\n\x11model_artifact_id\x18\x02 \x01(\tH\x00R\x0fmodelArtifactId\x12\x42\n\x0ctraining_run\x18\x03 \x01(\x0b\x32\x1d.chalk.server.v1.RunCriterionH\x00R\x0btrainingRun\x12\x18\n\x07\x61liases\x18\x04 \x03(\tR\x07\x61liasesB\x15\n\x13\x61rtifact_identifier"l\n&CreateModelVersionFromArtifactResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion*\x82\x01\n\x15RunCriterionDirection\x12\'\n#RUN_CRITERION_DIRECTION_UNSPECIFIED\x10\x00\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MAX\x10\x01\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MIN\x10\x02\x32\xec\x0f\n\x14ModelRegistryService\x12]\n\nListModels\x12".chalk.server.v1.ListModelsRequest\x1a#.chalk.server.v1.ListModelsResponse"\x06\x90\x02\x01\x80}\x0b\x12W\n\x08GetModel\x12 .chalk.server.v1.GetModelRequest\x1a!.chalk.server.v1.GetModelResponse"\x06\x90\x02\x01\x80}\x0b\x12\xb4\x01\n\x0b\x43reateModel\x12#.chalk.server.v1.CreateModelRequest\x1a$.chalk.server.v1.CreateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xb4\x01\n\x0bUpdateModel\x12#.chalk.server.v1.UpdateModelRequest\x1a$.chalk.server.v1.UpdateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x11ListModelVersions\x12).chalk.server.v1.ListModelVersionsRequest\x1a*.chalk.server.v1.ListModelVersionsResponse"\x06\x90\x02\x01\x80}\x0b\x12l\n\x0fGetModelVersion\x12\'.chalk.server.v1.GetModelVersionRequest\x1a(.chalk.server.v1.GetModelVersionResponse"\x06\x90\x02\x01\x80}\x0b\x12\xc9\x01\n\x12\x43reateModelVersion\x12*.chalk.server.v1.CreateModelVersionRequest\x1a+.chalk.server.v1.CreateModelVersionResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xcc\x01\n\x13\x43reateModelArtifact\x12+.chalk.server.v1.CreateModelArtifactRequest\x1a,.chalk.server.v1.CreateModelArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xed\x01\n\x1e\x43reateModelVersionFromArtifact\x12\x36.chalk.server.v1.CreateModelVersionFromArtifactRequest\x1a\x37.chalk.server.v1.CreateModelVersionFromArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x12UpdateModelVersion\x12*.chalk.server.v1.UpdateModelVersionRequest\x1a+.chalk.server.v1.UpdateModelVersionResponse"\x03\x80}\x0c\x12u\n\x12GetModelReferences\x12*.chalk.server.v1.GetModelReferencesRequest\x1a+.chalk.server.v1.GetModelReferencesResponse"\x06\x90\x02\x01\x80}\x0b\x12r\n\x11GetModelReference\x12).chalk.server.v1.GetModelReferenceRequest\x1a*.chalk.server.v1.GetModelReferenceResponse"\x06\x90\x02\x01\x80}\x0b\x12\xe1\x01\n\x1aGetModelArtifactUploadUrls\x12\x32.chalk.server.v1.GetModelArtifactUploadUrlsRequest\x1a\x33.chalk.server.v1.GetModelArtifactUploadUrlsResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.B\x9b\x01\n\x13\x63om.chalk.server.v1B\x12ModelRegistryProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
26
+ b'\n$chalk/server/v1/model_registry.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1a\x63halk/flags/v1/flags.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a$chalk/models/v1/model_artifact.proto\x1a google/protobuf/field_mask.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xf8\x02\n\rModelArtifact\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n\x04path\x18\x02 \x01(\tR\x04path\x12\x36\n\x04spec\x18\x03 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\x04spec\x12H\n\x08metadata\x18\x06 \x03(\x0b\x32,.chalk.server.v1.ModelArtifact.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x04 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xc4\x03\n\x0cModelVersion\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12\x45\n\x0emodel_artifact\x18\x04 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x05 \x03(\tR\x07\x61liases\x12G\n\x08metadata\x18\x06 \x03(\x0b\x32+.chalk.server.v1.ModelVersion.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x07 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_by"\xed\x04\n\x05Model\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x03 \x01(\tR\x0b\x64\x65scription\x12@\n\x08metadata\x18\x04 \x03(\x0b\x32$.chalk.server.v1.Model.MetadataEntryR\x08metadata\x12"\n\ncreated_by\x18\x05 \x01(\tH\x00R\tcreatedBy\x88\x01\x01\x12\x39\n\ncreated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12>\n\nupdated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tupdatedAt\x88\x01\x01\x12@\n\x0b\x61rchived_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x12T\n\x14latest_model_version\x18\t \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x03R\x12latestModelVersion\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_created_byB\r\n\x0b_updated_atB\x0e\n\x0c_archived_atB\x17\n\x15_latest_model_version"`\n\x11ListModelsRequest\x12\x1b\n\x06\x63ursor\x18\x01 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x42\t\n\x07_cursorB\x08\n\x06_limit"z\n\x12ListModelsResponse\x12.\n\x06models\x18\x01 \x03(\x0b\x32\x16.chalk.server.v1.ModelR\x06models\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"c\n\x0fGetModelRequest\x12\x1b\n\x08model_id\x18\x01 \x01(\tH\x00R\x07modelId\x12\x1f\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelNameB\x12\n\x10model_identifier"O\n\x10GetModelResponse\x12\x31\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelH\x00R\x05model\x88\x01\x01\x42\x08\n\x06_model"\xf9\x01\n\x12\x43reateModelRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12 \n\x0b\x64\x65scription\x18\x02 \x01(\tR\x0b\x64\x65scription\x12M\n\x08metadata\x18\x04 \x03(\x0b\x32\x31.chalk.server.v1.CreateModelRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"C\n\x13\x43reateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\xf8\x02\n\x14UpdateModelOperation\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12%\n\x0b\x64\x65scription\x18\x02 \x01(\tH\x01R\x0b\x64\x65scription\x88\x01\x01\x12O\n\x08metadata\x18\x03 \x03(\x0b\x32\x33.chalk.server.v1.UpdateModelOperation.MetadataEntryR\x08metadata\x12@\n\x0b\x61rchived_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\narchivedAt\x88\x01\x01\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\r\n\x0b_model_nameB\x0e\n\x0c_descriptionB\x0e\n\x0c_archived_at"\xab\x01\n\x12UpdateModelRequest\x12\x19\n\x08model_id\x18\x01 \x01(\tR\x07modelId\x12=\n\x06update\x18\x02 \x01(\x0b\x32%.chalk.server.v1.UpdateModelOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"C\n\x13UpdateModelResponse\x12,\n\x05model\x18\x01 \x01(\x0b\x32\x16.chalk.server.v1.ModelR\x05model"\x9a\x01\n\x18ListModelVersionsRequest\x12"\n\nmodel_name\x18\x01 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x01R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\t\n\x07_cursorB\x08\n\x06_limit"\x97\x01\n\x19ListModelVersionsResponse\x12\x44\n\x0emodel_versions\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\rmodelVersions\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"Q\n\x16GetModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"t\n\x17GetModelVersionResponse\x12G\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionH\x00R\x0cmodelVersion\x88\x01\x01\x42\x10\n\x0e_model_version"\xbf\x02\n\x1a\x43reateModelArtifactRequest\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12U\n\x08metadata\x18\x04 \x03(\x0b\x32\x39.chalk.server.v1.CreateModelArtifactRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"d\n\x1b\x43reateModelArtifactResponse\x12\x45\n\x0emodel_artifact\x18\x01 \x01(\x0b\x32\x1e.chalk.server.v1.ModelArtifactR\rmodelArtifact"\xf6\x02\n\x19\x43reateModelVersionRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12*\n\x11model_artifact_id\x18\x05 \x01(\tR\x0fmodelArtifactId\x12I\n\x0emodel_artifact\x18\x02 \x01(\x0b\x32".chalk.models.v1.ModelArtifactSpecR\rmodelArtifact\x12\x18\n\x07\x61liases\x18\x03 \x03(\tR\x07\x61liases\x12T\n\x08metadata\x18\x04 \x03(\x0b\x32\x38.chalk.server.v1.CreateModelVersionRequest.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"`\n\x1a\x43reateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"J\n\x0fModelVersionKey\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x02 \x01(\x04R\x07version"\xe4\x01\n\x1bUpdateModelVersionOperation\x12\x18\n\x07\x61liases\x18\x01 \x03(\tR\x07\x61liases\x12V\n\x08metadata\x18\x02 \x03(\x0b\x32:.chalk.server.v1.UpdateModelVersionOperation.MetadataEntryR\x08metadata\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"\xec\x01\n\x19UpdateModelVersionRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey\x12\x44\n\x06update\x18\x02 \x01(\x0b\x32,.chalk.server.v1.UpdateModelVersionOperationR\x06update\x12;\n\x0bupdate_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\nupdateMask"`\n\x1aUpdateModelVersionResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion"B\n!GetModelArtifactUploadUrlsRequest\x12\x1d\n\nfile_names\x18\x01 \x03(\tR\tfileNames"\xf5\x01\n"GetModelArtifactUploadUrlsResponse\x12\x64\n\x0bupload_urls\x18\x01 \x03(\x0b\x32\x43.chalk.server.v1.GetModelArtifactUploadUrlsResponse.UploadUrlsEntryR\nuploadUrls\x12*\n\x11model_artifact_id\x18\x02 \x01(\tR\x0fmodelArtifactId\x1a=\n\x0fUploadUrlsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"l\n\x1c\x44ownloadModelArtifactRequest\x12L\n\x11model_version_key\x18\x01 \x01(\x0b\x32 .chalk.server.v1.ModelVersionKeyR\x0fmodelVersionKey"1\n\x1d\x44ownloadModelArtifactResponse\x12\x10\n\x03uri\x18\x01 \x01(\tR\x03uri"\xfc\x01\n\x19GetModelReferencesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12"\n\nmodel_name\x18\x02 \x01(\tH\x00R\tmodelName\x88\x01\x01\x12(\n\rmodel_version\x18\x03 \x01(\rH\x01R\x0cmodelVersion\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x04 \x01(\tH\x02R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x05 \x01(\x05H\x03R\x05limit\x88\x01\x01\x42\r\n\x0b_model_nameB\x10\n\x0e_model_versionB\t\n\x07_cursorB\x08\n\x06_limit"]\n\rModelRelation\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures\x12%\n\x0eoutput_feature\x18\x02 \x01(\tR\routputFeature"\x8d\x03\n\x0eModelReference\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1d\n\nmodel_name\x18\x02 \x01(\tR\tmodelName\x12\x18\n\x07version\x18\x03 \x01(\x04R\x07version\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\trelations\x18\x05 \x03(\x0b\x32\x1e.chalk.server.v1.ModelRelationR\trelations\x12\x1c\n\tresolvers\x18\x06 \x03(\tR\tresolvers\x12\\\n\x15source_file_reference\x18\x07 \x01(\x0b\x32#.chalk.graph.v1.SourceFileReferenceH\x00R\x13sourceFileReference\x88\x01\x01\x12\x39\n\ncreated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAtB\x18\n\x16_source_file_reference"\x9e\x01\n\x1aGetModelReferencesResponse\x12J\n\x10model_references\x18\x01 \x03(\x0b\x32\x1f.chalk.server.v1.ModelReferenceR\x0fmodelReferences\x12$\n\x0bnext_cursor\x18\x04 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"\xa2\x01\n\x18GetModelReferenceRequest\x12\x1d\n\x08model_id\x18\x01 \x01(\tB\x02\x18\x01R\x07modelId\x12\x1d\n\nmodel_name\x18\x04 \x01(\tR\tmodelName\x12#\n\rmodel_version\x18\x02 \x01(\x04R\x0cmodelVersion\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId"~\n\x19GetModelReferenceResponse\x12M\n\x0fmodel_reference\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.ModelReferenceH\x00R\x0emodelReference\x88\x01\x01\x42\x12\n\x10_model_reference"\xd7\x01\n\x0cRunCriterion\x12\x17\n\x06run_id\x18\x01 \x01(\tH\x00R\x05runId\x12\x1b\n\x08run_name\x18\x04 \x01(\tH\x00R\x07runName\x12\x1b\n\x06metric\x18\x02 \x01(\tH\x01R\x06metric\x88\x01\x01\x12I\n\tdirection\x18\x03 \x01(\x0e\x32&.chalk.server.v1.RunCriterionDirectionH\x02R\tdirection\x88\x01\x01\x42\x10\n\x0erun_identifierB\t\n\x07_metricB\x0c\n\n_direction"\xe9\x01\n%CreateModelVersionFromArtifactRequest\x12\x1d\n\nmodel_name\x18\x01 \x01(\tR\tmodelName\x12,\n\x11model_artifact_id\x18\x02 \x01(\tH\x00R\x0fmodelArtifactId\x12\x42\n\x0ctraining_run\x18\x03 \x01(\x0b\x32\x1d.chalk.server.v1.RunCriterionH\x00R\x0btrainingRun\x12\x18\n\x07\x61liases\x18\x04 \x03(\tR\x07\x61liasesB\x15\n\x13\x61rtifact_identifier"l\n&CreateModelVersionFromArtifactResponse\x12\x42\n\rmodel_version\x18\x01 \x01(\x0b\x32\x1d.chalk.server.v1.ModelVersionR\x0cmodelVersion*\x82\x01\n\x15RunCriterionDirection\x12\'\n#RUN_CRITERION_DIRECTION_UNSPECIFIED\x10\x00\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MAX\x10\x01\x12\x1f\n\x1bRUN_CRITERION_DIRECTION_MIN\x10\x02\x32\xec\x0f\n\x14ModelRegistryService\x12]\n\nListModels\x12".chalk.server.v1.ListModelsRequest\x1a#.chalk.server.v1.ListModelsResponse"\x06\x90\x02\x01\x80}\x0b\x12W\n\x08GetModel\x12 .chalk.server.v1.GetModelRequest\x1a!.chalk.server.v1.GetModelResponse"\x06\x90\x02\x01\x80}\x0b\x12\xb4\x01\n\x0b\x43reateModel\x12#.chalk.server.v1.CreateModelRequest\x1a$.chalk.server.v1.CreateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xb4\x01\n\x0bUpdateModel\x12#.chalk.server.v1.UpdateModelRequest\x1a$.chalk.server.v1.UpdateModelResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x11ListModelVersions\x12).chalk.server.v1.ListModelVersionsRequest\x1a*.chalk.server.v1.ListModelVersionsResponse"\x06\x90\x02\x01\x80}\x0b\x12l\n\x0fGetModelVersion\x12\'.chalk.server.v1.GetModelVersionRequest\x1a(.chalk.server.v1.GetModelVersionResponse"\x06\x90\x02\x01\x80}\x0b\x12\xc9\x01\n\x12\x43reateModelVersion\x12*.chalk.server.v1.CreateModelVersionRequest\x1a+.chalk.server.v1.CreateModelVersionResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xcc\x01\n\x13\x43reateModelArtifact\x12+.chalk.server.v1.CreateModelArtifactRequest\x1a,.chalk.server.v1.CreateModelArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12\xed\x01\n\x1e\x43reateModelVersionFromArtifact\x12\x36.chalk.server.v1.CreateModelVersionFromArtifactRequest\x1a\x37.chalk.server.v1.CreateModelVersionFromArtifactResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.\x12r\n\x12UpdateModelVersion\x12*.chalk.server.v1.UpdateModelVersionRequest\x1a+.chalk.server.v1.UpdateModelVersionResponse"\x03\x80}\x0c\x12u\n\x12GetModelReferences\x12*.chalk.server.v1.GetModelReferencesRequest\x1a+.chalk.server.v1.GetModelReferencesResponse"\x06\x90\x02\x01\x80}\x0b\x12r\n\x11GetModelReference\x12).chalk.server.v1.GetModelReferenceRequest\x1a*.chalk.server.v1.GetModelReferenceResponse"\x06\x90\x02\x01\x80}\x0b\x12\xe1\x01\n\x1aGetModelArtifactUploadUrls\x12\x32.chalk.server.v1.GetModelArtifactUploadUrlsRequest\x1a\x33.chalk.server.v1.GetModelArtifactUploadUrlsResponse"Z\x80}\x0c\x92\xd3\x0eS\n\x16model_registry_enabled\x12\x39This action is not enabled. Please contact Chalk Support.B\x9b\x01\n\x13\x63om.chalk.server.v1B\x12ModelRegistryProtoP\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'
27
27
  )
28
28
 
29
29
  _globals = globals()
@@ -98,8 +98,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
98
98
  _globals["_MODELREGISTRYSERVICE"].methods_by_name[
99
99
  "GetModelArtifactUploadUrls"
100
100
  ]._serialized_options = b"\200}\014\222\323\016S\n\026model_registry_enabled\0229This action is not enabled. Please contact Chalk Support."
101
- _globals["_RUNCRITERIONDIRECTION"]._serialized_start = 7338
102
- _globals["_RUNCRITERIONDIRECTION"]._serialized_end = 7468
101
+ _globals["_RUNCRITERIONDIRECTION"]._serialized_start = 7387
102
+ _globals["_RUNCRITERIONDIRECTION"]._serialized_end = 7517
103
103
  _globals["_MODELARTIFACT"]._serialized_start = 282
104
104
  _globals["_MODELARTIFACT"]._serialized_end = 658
105
105
  _globals["_MODELARTIFACT_METADATAENTRY"]._serialized_start = 560
@@ -187,11 +187,11 @@ if _descriptor._USE_C_DESCRIPTORS == False:
187
187
  _globals["_GETMODELREFERENCERESPONSE"]._serialized_start = 6694
188
188
  _globals["_GETMODELREFERENCERESPONSE"]._serialized_end = 6820
189
189
  _globals["_RUNCRITERION"]._serialized_start = 6823
190
- _globals["_RUNCRITERION"]._serialized_end = 6989
191
- _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_start = 6992
192
- _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_end = 7225
193
- _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_start = 7227
194
- _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_end = 7335
195
- _globals["_MODELREGISTRYSERVICE"]._serialized_start = 7471
196
- _globals["_MODELREGISTRYSERVICE"]._serialized_end = 9499
190
+ _globals["_RUNCRITERION"]._serialized_end = 7038
191
+ _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_start = 7041
192
+ _globals["_CREATEMODELVERSIONFROMARTIFACTREQUEST"]._serialized_end = 7274
193
+ _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_start = 7276
194
+ _globals["_CREATEMODELVERSIONFROMARTIFACTRESPONSE"]._serialized_end = 7384
195
+ _globals["_MODELREGISTRYSERVICE"]._serialized_start = 7520
196
+ _globals["_MODELREGISTRYSERVICE"]._serialized_end = 9548
197
197
  # @@protoc_insertion_point(module_scope)
@@ -566,16 +566,19 @@ class GetModelReferenceResponse(_message.Message):
566
566
  def __init__(self, model_reference: _Optional[_Union[ModelReference, _Mapping]] = ...) -> None: ...
567
567
 
568
568
  class RunCriterion(_message.Message):
569
- __slots__ = ("run_id", "metric", "direction")
569
+ __slots__ = ("run_id", "run_name", "metric", "direction")
570
570
  RUN_ID_FIELD_NUMBER: _ClassVar[int]
571
+ RUN_NAME_FIELD_NUMBER: _ClassVar[int]
571
572
  METRIC_FIELD_NUMBER: _ClassVar[int]
572
573
  DIRECTION_FIELD_NUMBER: _ClassVar[int]
573
574
  run_id: str
575
+ run_name: str
574
576
  metric: str
575
577
  direction: RunCriterionDirection
576
578
  def __init__(
577
579
  self,
578
580
  run_id: _Optional[str] = ...,
581
+ run_name: _Optional[str] = ...,
579
582
  metric: _Optional[str] = ...,
580
583
  direction: _Optional[_Union[RunCriterionDirection, str]] = ...,
581
584
  ) -> None: ...
@@ -0,0 +1,53 @@
1
+ # -*- coding: utf-8 -*-
2
+ # Generated by the protocol buffer compiler. DO NOT EDIT!
3
+ # source: chalk/server/v1/plandebug.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 google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
18
+ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
19
+
20
+
21
+ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
22
+ b'\n\x1f\x63halk/server/v1/plandebug.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"^\n\x18GetQueryPlanStageRequest\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId"\xd8\x02\n\x19GetQueryPlanStageResponse\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId\x12?\n\x0c\x64\x61ta_preview\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructH\x00R\x0b\x64\x61taPreview\x88\x01\x01\x12?\n\x0c\x64\x61ta_summary\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x0b\x64\x61taSummary\x88\x01\x01\x12\x41\n\rgroup_preview\x18\x05 \x01(\x0b\x32\x17.google.protobuf.StructH\x02R\x0cgroupPreview\x88\x01\x01\x42\x0f\n\r_data_previewB\x0f\n\r_data_summaryB\x10\n\x0e_group_preview"l\n&GetQueryPlanStageResolverInputsRequest\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId"\xf8\x01\n\'GetQueryPlanStageResolverInputsResponse\x12:\n\tresolvers\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructH\x00R\tresolvers\x88\x01\x01\x12\x36\n\x07scalars\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x07scalars\x88\x01\x01\x12\x34\n\x06tables\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructH\x02R\x06tables\x88\x01\x01\x42\x0c\n\n_resolversB\n\n\x08_scalarsB\t\n\x07_tables"j\n$GetQueryPlanStageDownloadLinkRequest\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId"\x9b\x02\n%GetQueryPlanStageDownloadLinkResponse\x12"\n\nsigned_url\x18\x01 \x01(\tH\x00R\tsignedUrl\x88\x01\x01\x12;\n\ngroup_urls\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\tgroupUrls\x88\x01\x01\x12\x19\n\x05\x65rror\x18\x03 \x01(\tH\x02R\x05\x65rror\x88\x01\x01\x12?\n\nexpiration\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nexpiration\x88\x01\x01\x42\r\n\x0b_signed_urlB\r\n\x0b_group_urlsB\x08\n\x06_errorB\r\n\x0b_expiration2\xb5\x03\n\x10PlanDebugService\x12o\n\x11GetQueryPlanStage\x12).chalk.server.v1.GetQueryPlanStageRequest\x1a*.chalk.server.v1.GetQueryPlanStageResponse"\x03\x80}\x04\x12\x99\x01\n\x1fGetQueryPlanStageResolverInputs\x12\x37.chalk.server.v1.GetQueryPlanStageResolverInputsRequest\x1a\x38.chalk.server.v1.GetQueryPlanStageResolverInputsResponse"\x03\x80}\x04\x12\x93\x01\n\x1dGetQueryPlanStageDownloadLink\x12\x35.chalk.server.v1.GetQueryPlanStageDownloadLinkRequest\x1a\x36.chalk.server.v1.GetQueryPlanStageDownloadLinkResponse"\x03\x80}\x04\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0ePlandebugProtoP\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
+ )
24
+
25
+ _globals = globals()
26
+ _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
27
+ _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.plandebug_pb2", _globals)
28
+ if _descriptor._USE_C_DESCRIPTORS == False:
29
+ _globals["DESCRIPTOR"]._options = None
30
+ _globals[
31
+ "DESCRIPTOR"
32
+ ]._serialized_options = b"\n\023com.chalk.server.v1B\016PlandebugProtoP\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"
33
+ _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStage"]._options = None
34
+ _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStage"]._serialized_options = b"\200}\004"
35
+ _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageResolverInputs"]._options = None
36
+ _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageResolverInputs"]._serialized_options = b"\200}\004"
37
+ _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageDownloadLink"]._options = None
38
+ _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageDownloadLink"]._serialized_options = b"\200}\004"
39
+ _globals["_GETQUERYPLANSTAGEREQUEST"]._serialized_start = 148
40
+ _globals["_GETQUERYPLANSTAGEREQUEST"]._serialized_end = 242
41
+ _globals["_GETQUERYPLANSTAGERESPONSE"]._serialized_start = 245
42
+ _globals["_GETQUERYPLANSTAGERESPONSE"]._serialized_end = 589
43
+ _globals["_GETQUERYPLANSTAGERESOLVERINPUTSREQUEST"]._serialized_start = 591
44
+ _globals["_GETQUERYPLANSTAGERESOLVERINPUTSREQUEST"]._serialized_end = 699
45
+ _globals["_GETQUERYPLANSTAGERESOLVERINPUTSRESPONSE"]._serialized_start = 702
46
+ _globals["_GETQUERYPLANSTAGERESOLVERINPUTSRESPONSE"]._serialized_end = 950
47
+ _globals["_GETQUERYPLANSTAGEDOWNLOADLINKREQUEST"]._serialized_start = 952
48
+ _globals["_GETQUERYPLANSTAGEDOWNLOADLINKREQUEST"]._serialized_end = 1058
49
+ _globals["_GETQUERYPLANSTAGEDOWNLOADLINKRESPONSE"]._serialized_start = 1061
50
+ _globals["_GETQUERYPLANSTAGEDOWNLOADLINKRESPONSE"]._serialized_end = 1344
51
+ _globals["_PLANDEBUGSERVICE"]._serialized_start = 1347
52
+ _globals["_PLANDEBUGSERVICE"]._serialized_end = 1784
53
+ # @@protoc_insertion_point(module_scope)
@@ -0,0 +1,86 @@
1
+ from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
2
+ from google.protobuf import struct_pb2 as _struct_pb2
3
+ from google.protobuf import timestamp_pb2 as _timestamp_pb2
4
+ from google.protobuf import descriptor as _descriptor
5
+ from google.protobuf import message as _message
6
+ from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
7
+
8
+ DESCRIPTOR: _descriptor.FileDescriptor
9
+
10
+ class GetQueryPlanStageRequest(_message.Message):
11
+ __slots__ = ("operator_id", "operation_id")
12
+ OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
13
+ OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
14
+ operator_id: str
15
+ operation_id: str
16
+ def __init__(self, operator_id: _Optional[str] = ..., operation_id: _Optional[str] = ...) -> None: ...
17
+
18
+ class GetQueryPlanStageResponse(_message.Message):
19
+ __slots__ = ("operator_id", "operation_id", "data_preview", "data_summary", "group_preview")
20
+ OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
21
+ OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
22
+ DATA_PREVIEW_FIELD_NUMBER: _ClassVar[int]
23
+ DATA_SUMMARY_FIELD_NUMBER: _ClassVar[int]
24
+ GROUP_PREVIEW_FIELD_NUMBER: _ClassVar[int]
25
+ operator_id: str
26
+ operation_id: str
27
+ data_preview: _struct_pb2.Struct
28
+ data_summary: _struct_pb2.Struct
29
+ group_preview: _struct_pb2.Struct
30
+ def __init__(
31
+ self,
32
+ operator_id: _Optional[str] = ...,
33
+ operation_id: _Optional[str] = ...,
34
+ data_preview: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
35
+ data_summary: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
36
+ group_preview: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
37
+ ) -> None: ...
38
+
39
+ class GetQueryPlanStageResolverInputsRequest(_message.Message):
40
+ __slots__ = ("operator_id", "operation_id")
41
+ OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
42
+ OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
43
+ operator_id: str
44
+ operation_id: str
45
+ def __init__(self, operator_id: _Optional[str] = ..., operation_id: _Optional[str] = ...) -> None: ...
46
+
47
+ class GetQueryPlanStageResolverInputsResponse(_message.Message):
48
+ __slots__ = ("resolvers", "scalars", "tables")
49
+ RESOLVERS_FIELD_NUMBER: _ClassVar[int]
50
+ SCALARS_FIELD_NUMBER: _ClassVar[int]
51
+ TABLES_FIELD_NUMBER: _ClassVar[int]
52
+ resolvers: _struct_pb2.Struct
53
+ scalars: _struct_pb2.Struct
54
+ tables: _struct_pb2.Struct
55
+ def __init__(
56
+ self,
57
+ resolvers: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
58
+ scalars: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
59
+ tables: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
60
+ ) -> None: ...
61
+
62
+ class GetQueryPlanStageDownloadLinkRequest(_message.Message):
63
+ __slots__ = ("operator_id", "operation_id")
64
+ OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
65
+ OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
66
+ operator_id: str
67
+ operation_id: str
68
+ def __init__(self, operator_id: _Optional[str] = ..., operation_id: _Optional[str] = ...) -> None: ...
69
+
70
+ class GetQueryPlanStageDownloadLinkResponse(_message.Message):
71
+ __slots__ = ("signed_url", "group_urls", "error", "expiration")
72
+ SIGNED_URL_FIELD_NUMBER: _ClassVar[int]
73
+ GROUP_URLS_FIELD_NUMBER: _ClassVar[int]
74
+ ERROR_FIELD_NUMBER: _ClassVar[int]
75
+ EXPIRATION_FIELD_NUMBER: _ClassVar[int]
76
+ signed_url: str
77
+ group_urls: _struct_pb2.Struct
78
+ error: str
79
+ expiration: _timestamp_pb2.Timestamp
80
+ def __init__(
81
+ self,
82
+ signed_url: _Optional[str] = ...,
83
+ group_urls: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
84
+ error: _Optional[str] = ...,
85
+ expiration: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
86
+ ) -> None: ...
@@ -0,0 +1,168 @@
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 plandebug_pb2 as chalk_dot_server_dot_v1_dot_plandebug__pb2
7
+
8
+
9
+ class PlanDebugServiceStub(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.GetQueryPlanStage = channel.unary_unary(
19
+ "/chalk.server.v1.PlanDebugService/GetQueryPlanStage",
20
+ request_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageRequest.SerializeToString,
21
+ response_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResponse.FromString,
22
+ )
23
+ self.GetQueryPlanStageResolverInputs = channel.unary_unary(
24
+ "/chalk.server.v1.PlanDebugService/GetQueryPlanStageResolverInputs",
25
+ request_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsRequest.SerializeToString,
26
+ response_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsResponse.FromString,
27
+ )
28
+ self.GetQueryPlanStageDownloadLink = channel.unary_unary(
29
+ "/chalk.server.v1.PlanDebugService/GetQueryPlanStageDownloadLink",
30
+ request_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkRequest.SerializeToString,
31
+ response_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkResponse.FromString,
32
+ )
33
+
34
+
35
+ class PlanDebugServiceServicer(object):
36
+ """Missing associated documentation comment in .proto file."""
37
+
38
+ def GetQueryPlanStage(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 GetQueryPlanStageResolverInputs(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 GetQueryPlanStageDownloadLink(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_PlanDebugServiceServicer_to_server(servicer, server):
58
+ rpc_method_handlers = {
59
+ "GetQueryPlanStage": grpc.unary_unary_rpc_method_handler(
60
+ servicer.GetQueryPlanStage,
61
+ request_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageRequest.FromString,
62
+ response_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResponse.SerializeToString,
63
+ ),
64
+ "GetQueryPlanStageResolverInputs": grpc.unary_unary_rpc_method_handler(
65
+ servicer.GetQueryPlanStageResolverInputs,
66
+ request_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsRequest.FromString,
67
+ response_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsResponse.SerializeToString,
68
+ ),
69
+ "GetQueryPlanStageDownloadLink": grpc.unary_unary_rpc_method_handler(
70
+ servicer.GetQueryPlanStageDownloadLink,
71
+ request_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkRequest.FromString,
72
+ response_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkResponse.SerializeToString,
73
+ ),
74
+ }
75
+ generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.PlanDebugService", rpc_method_handlers)
76
+ server.add_generic_rpc_handlers((generic_handler,))
77
+
78
+
79
+ # This class is part of an EXPERIMENTAL API.
80
+ class PlanDebugService(object):
81
+ """Missing associated documentation comment in .proto file."""
82
+
83
+ @staticmethod
84
+ def GetQueryPlanStage(
85
+ request,
86
+ target,
87
+ options=(),
88
+ channel_credentials=None,
89
+ call_credentials=None,
90
+ insecure=False,
91
+ compression=None,
92
+ wait_for_ready=None,
93
+ timeout=None,
94
+ metadata=None,
95
+ ):
96
+ return grpc.experimental.unary_unary(
97
+ request,
98
+ target,
99
+ "/chalk.server.v1.PlanDebugService/GetQueryPlanStage",
100
+ chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageRequest.SerializeToString,
101
+ chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResponse.FromString,
102
+ options,
103
+ channel_credentials,
104
+ insecure,
105
+ call_credentials,
106
+ compression,
107
+ wait_for_ready,
108
+ timeout,
109
+ metadata,
110
+ )
111
+
112
+ @staticmethod
113
+ def GetQueryPlanStageResolverInputs(
114
+ request,
115
+ target,
116
+ options=(),
117
+ channel_credentials=None,
118
+ call_credentials=None,
119
+ insecure=False,
120
+ compression=None,
121
+ wait_for_ready=None,
122
+ timeout=None,
123
+ metadata=None,
124
+ ):
125
+ return grpc.experimental.unary_unary(
126
+ request,
127
+ target,
128
+ "/chalk.server.v1.PlanDebugService/GetQueryPlanStageResolverInputs",
129
+ chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsRequest.SerializeToString,
130
+ chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsResponse.FromString,
131
+ options,
132
+ channel_credentials,
133
+ insecure,
134
+ call_credentials,
135
+ compression,
136
+ wait_for_ready,
137
+ timeout,
138
+ metadata,
139
+ )
140
+
141
+ @staticmethod
142
+ def GetQueryPlanStageDownloadLink(
143
+ request,
144
+ target,
145
+ options=(),
146
+ channel_credentials=None,
147
+ call_credentials=None,
148
+ insecure=False,
149
+ compression=None,
150
+ wait_for_ready=None,
151
+ timeout=None,
152
+ metadata=None,
153
+ ):
154
+ return grpc.experimental.unary_unary(
155
+ request,
156
+ target,
157
+ "/chalk.server.v1.PlanDebugService/GetQueryPlanStageDownloadLink",
158
+ chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkRequest.SerializeToString,
159
+ chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkResponse.FromString,
160
+ options,
161
+ channel_credentials,
162
+ insecure,
163
+ call_credentials,
164
+ compression,
165
+ wait_for_ready,
166
+ timeout,
167
+ metadata,
168
+ )
@@ -0,0 +1,60 @@
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.plandebug_pb2 import (
11
+ GetQueryPlanStageDownloadLinkRequest,
12
+ GetQueryPlanStageDownloadLinkResponse,
13
+ GetQueryPlanStageRequest,
14
+ GetQueryPlanStageResolverInputsRequest,
15
+ GetQueryPlanStageResolverInputsResponse,
16
+ GetQueryPlanStageResponse,
17
+ )
18
+ from grpc import (
19
+ Channel,
20
+ Server,
21
+ ServicerContext,
22
+ UnaryUnaryMultiCallable,
23
+ )
24
+
25
+ class PlanDebugServiceStub:
26
+ def __init__(self, channel: Channel) -> None: ...
27
+ GetQueryPlanStage: UnaryUnaryMultiCallable[
28
+ GetQueryPlanStageRequest,
29
+ GetQueryPlanStageResponse,
30
+ ]
31
+ GetQueryPlanStageResolverInputs: UnaryUnaryMultiCallable[
32
+ GetQueryPlanStageResolverInputsRequest,
33
+ GetQueryPlanStageResolverInputsResponse,
34
+ ]
35
+ GetQueryPlanStageDownloadLink: UnaryUnaryMultiCallable[
36
+ GetQueryPlanStageDownloadLinkRequest,
37
+ GetQueryPlanStageDownloadLinkResponse,
38
+ ]
39
+
40
+ class PlanDebugServiceServicer(metaclass=ABCMeta):
41
+ @abstractmethod
42
+ def GetQueryPlanStage(
43
+ self,
44
+ request: GetQueryPlanStageRequest,
45
+ context: ServicerContext,
46
+ ) -> GetQueryPlanStageResponse: ...
47
+ @abstractmethod
48
+ def GetQueryPlanStageResolverInputs(
49
+ self,
50
+ request: GetQueryPlanStageResolverInputsRequest,
51
+ context: ServicerContext,
52
+ ) -> GetQueryPlanStageResolverInputsResponse: ...
53
+ @abstractmethod
54
+ def GetQueryPlanStageDownloadLink(
55
+ self,
56
+ request: GetQueryPlanStageDownloadLinkRequest,
57
+ context: ServicerContext,
58
+ ) -> GetQueryPlanStageDownloadLinkResponse: ...
59
+
60
+ def add_PlanDebugServiceServicer_to_server(servicer: PlanDebugServiceServicer, server: Server) -> None: ...