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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (268) hide show
  1. chalk/__init__.py +2 -1
  2. chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
  3. chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
  4. chalk/_gen/chalk/artifacts/v1/chart_pb2.py +36 -33
  5. chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +41 -1
  6. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +8 -7
  7. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +5 -0
  8. chalk/_gen/chalk/common/v1/offline_query_pb2.py +19 -13
  9. chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +37 -0
  10. chalk/_gen/chalk/common/v1/online_query_pb2.py +54 -54
  11. chalk/_gen/chalk/common/v1/online_query_pb2.pyi +13 -1
  12. chalk/_gen/chalk/common/v1/script_task_pb2.py +13 -11
  13. chalk/_gen/chalk/common/v1/script_task_pb2.pyi +19 -1
  14. chalk/_gen/chalk/dataframe/__init__.py +0 -0
  15. chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
  16. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
  17. chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +123 -0
  18. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
  19. chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
  20. chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
  21. chalk/_gen/chalk/graph/v1/graph_pb2.pyi +25 -0
  22. chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
  23. chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -0
  24. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
  25. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
  26. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
  27. chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
  28. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
  29. chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
  30. chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +54 -27
  31. chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
  32. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
  33. chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
  34. chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
  35. chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
  36. chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
  37. chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
  38. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
  39. chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
  40. chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
  41. chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
  42. chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
  43. chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
  44. chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
  45. chalk/_gen/chalk/server/v1/builder_pb2.py +372 -272
  46. chalk/_gen/chalk/server/v1/builder_pb2.pyi +479 -12
  47. chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +360 -0
  48. chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +96 -0
  49. chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
  50. chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
  51. chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
  52. chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
  53. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
  54. chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
  55. chalk/_gen/chalk/server/v1/cloud_components_pb2.py +153 -107
  56. chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +146 -4
  57. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +180 -0
  58. chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +48 -0
  59. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
  60. chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
  61. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
  62. chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
  63. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +59 -35
  64. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +127 -1
  65. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +135 -0
  66. chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +36 -0
  67. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
  68. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -0
  69. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
  70. chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
  71. chalk/_gen/chalk/server/v1/datasets_pb2.py +36 -24
  72. chalk/_gen/chalk/server/v1/datasets_pb2.pyi +71 -2
  73. chalk/_gen/chalk/server/v1/datasets_pb2_grpc.py +45 -0
  74. chalk/_gen/chalk/server/v1/datasets_pb2_grpc.pyi +12 -0
  75. chalk/_gen/chalk/server/v1/deploy_pb2.py +9 -3
  76. chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
  77. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
  78. chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
  79. chalk/_gen/chalk/server/v1/deployment_pb2.py +20 -15
  80. chalk/_gen/chalk/server/v1/deployment_pb2.pyi +25 -0
  81. chalk/_gen/chalk/server/v1/environment_pb2.py +25 -15
  82. chalk/_gen/chalk/server/v1/environment_pb2.pyi +93 -1
  83. chalk/_gen/chalk/server/v1/eventbus_pb2.py +44 -0
  84. chalk/_gen/chalk/server/v1/eventbus_pb2.pyi +64 -0
  85. chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.py +4 -0
  86. chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.pyi +4 -0
  87. chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
  88. chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
  89. chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
  90. chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
  91. chalk/_gen/chalk/server/v1/graph_pb2.py +41 -3
  92. chalk/_gen/chalk/server/v1/graph_pb2.pyi +191 -0
  93. chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +92 -0
  94. chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +32 -0
  95. chalk/_gen/chalk/server/v1/incident_pb2.py +57 -0
  96. chalk/_gen/chalk/server/v1/incident_pb2.pyi +165 -0
  97. chalk/_gen/chalk/server/v1/incident_pb2_grpc.py +4 -0
  98. chalk/_gen/chalk/server/v1/incident_pb2_grpc.pyi +4 -0
  99. chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
  100. chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
  101. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
  102. chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
  103. chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
  104. chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
  105. chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
  106. chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
  107. chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
  108. chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
  109. chalk/_gen/chalk/server/v1/log_pb2.py +21 -3
  110. chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
  111. chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
  112. chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
  113. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.py +73 -0
  114. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.pyi +212 -0
  115. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.py +217 -0
  116. chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.pyi +74 -0
  117. chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
  118. chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
  119. chalk/_gen/chalk/server/v1/monitoring_pb2.py +84 -75
  120. chalk/_gen/chalk/server/v1/monitoring_pb2.pyi +1 -0
  121. chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py +136 -0
  122. chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi +38 -0
  123. chalk/_gen/chalk/server/v1/offline_queries_pb2.py +32 -10
  124. chalk/_gen/chalk/server/v1/offline_queries_pb2.pyi +73 -0
  125. chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.py +90 -0
  126. chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.pyi +24 -0
  127. chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
  128. chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
  129. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
  130. chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
  131. chalk/_gen/chalk/server/v1/queries_pb2.py +76 -48
  132. chalk/_gen/chalk/server/v1/queries_pb2.pyi +155 -2
  133. chalk/_gen/chalk/server/v1/queries_pb2_grpc.py +180 -0
  134. chalk/_gen/chalk/server/v1/queries_pb2_grpc.pyi +48 -0
  135. chalk/_gen/chalk/server/v1/scheduled_query_pb2.py +4 -2
  136. chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.py +45 -0
  137. chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.pyi +12 -0
  138. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -6
  139. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +75 -2
  140. chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
  141. chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
  142. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
  143. chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
  144. chalk/_gen/chalk/server/v1/script_tasks_pb2.py +26 -14
  145. chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +33 -3
  146. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
  147. chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -0
  148. chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
  149. chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
  150. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
  151. chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
  152. chalk/_gen/chalk/server/v1/sql_queries_pb2.py +48 -0
  153. chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
  154. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
  155. chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
  156. chalk/_gen/chalk/server/v1/team_pb2.py +156 -137
  157. chalk/_gen/chalk/server/v1/team_pb2.pyi +56 -10
  158. chalk/_gen/chalk/server/v1/team_pb2_grpc.py +90 -0
  159. chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +24 -0
  160. chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
  161. chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
  162. chalk/_gen/chalk/server/v1/trace_pb2.py +50 -28
  163. chalk/_gen/chalk/server/v1/trace_pb2.pyi +121 -0
  164. chalk/_gen/chalk/server/v1/trace_pb2_grpc.py +135 -0
  165. chalk/_gen/chalk/server/v1/trace_pb2_grpc.pyi +42 -0
  166. chalk/_gen/chalk/server/v1/webhook_pb2.py +9 -3
  167. chalk/_gen/chalk/server/v1/webhook_pb2.pyi +18 -0
  168. chalk/_gen/chalk/server/v1/webhook_pb2_grpc.py +45 -0
  169. chalk/_gen/chalk/server/v1/webhook_pb2_grpc.pyi +12 -0
  170. chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
  171. chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
  172. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
  173. chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
  174. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +19 -7
  175. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +96 -3
  176. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
  177. chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
  178. chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
  179. chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
  180. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
  181. chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
  182. chalk/_lsp/error_builder.py +11 -0
  183. chalk/_monitoring/Chart.py +1 -3
  184. chalk/_version.py +1 -1
  185. chalk/cli.py +5 -10
  186. chalk/client/client.py +178 -64
  187. chalk/client/client_async.py +154 -0
  188. chalk/client/client_async_impl.py +22 -0
  189. chalk/client/client_grpc.py +738 -112
  190. chalk/client/client_impl.py +541 -136
  191. chalk/client/dataset.py +27 -6
  192. chalk/client/models.py +99 -2
  193. chalk/client/serialization/model_serialization.py +126 -10
  194. chalk/config/project_config.py +1 -1
  195. chalk/df/LazyFramePlaceholder.py +1154 -0
  196. chalk/df/ast_parser.py +2 -10
  197. chalk/features/_class_property.py +7 -0
  198. chalk/features/_embedding/embedding.py +1 -0
  199. chalk/features/_embedding/sentence_transformer.py +1 -1
  200. chalk/features/_encoding/converter.py +83 -2
  201. chalk/features/_encoding/pyarrow.py +20 -4
  202. chalk/features/_encoding/rich.py +1 -3
  203. chalk/features/_tensor.py +1 -2
  204. chalk/features/dataframe/_filters.py +14 -5
  205. chalk/features/dataframe/_impl.py +91 -36
  206. chalk/features/dataframe/_validation.py +11 -7
  207. chalk/features/feature_field.py +40 -30
  208. chalk/features/feature_set.py +1 -2
  209. chalk/features/feature_set_decorator.py +1 -0
  210. chalk/features/feature_wrapper.py +42 -3
  211. chalk/features/hooks.py +81 -12
  212. chalk/features/inference.py +65 -10
  213. chalk/features/resolver.py +338 -56
  214. chalk/features/tag.py +1 -3
  215. chalk/features/underscore_features.py +2 -1
  216. chalk/functions/__init__.py +456 -21
  217. chalk/functions/holidays.py +1 -3
  218. chalk/gitignore/gitignore_parser.py +5 -1
  219. chalk/importer.py +186 -74
  220. chalk/ml/__init__.py +6 -2
  221. chalk/ml/model_hooks.py +368 -51
  222. chalk/ml/model_reference.py +68 -10
  223. chalk/ml/model_version.py +34 -21
  224. chalk/ml/utils.py +143 -40
  225. chalk/operators/_utils.py +14 -3
  226. chalk/parsed/_proto/export.py +22 -0
  227. chalk/parsed/duplicate_input_gql.py +4 -0
  228. chalk/parsed/expressions.py +1 -3
  229. chalk/parsed/json_conversions.py +21 -14
  230. chalk/parsed/to_proto.py +16 -4
  231. chalk/parsed/user_types_to_json.py +31 -10
  232. chalk/parsed/validation_from_registries.py +182 -0
  233. chalk/queries/named_query.py +16 -6
  234. chalk/queries/scheduled_query.py +13 -1
  235. chalk/serialization/parsed_annotation.py +25 -12
  236. chalk/sql/__init__.py +221 -0
  237. chalk/sql/_internal/integrations/athena.py +6 -1
  238. chalk/sql/_internal/integrations/bigquery.py +22 -2
  239. chalk/sql/_internal/integrations/databricks.py +61 -18
  240. chalk/sql/_internal/integrations/mssql.py +281 -0
  241. chalk/sql/_internal/integrations/postgres.py +11 -3
  242. chalk/sql/_internal/integrations/redshift.py +4 -0
  243. chalk/sql/_internal/integrations/snowflake.py +11 -2
  244. chalk/sql/_internal/integrations/util.py +2 -1
  245. chalk/sql/_internal/sql_file_resolver.py +55 -10
  246. chalk/sql/_internal/sql_source.py +36 -2
  247. chalk/streams/__init__.py +1 -3
  248. chalk/streams/_kafka_source.py +5 -1
  249. chalk/streams/_windows.py +16 -4
  250. chalk/streams/types.py +1 -2
  251. chalk/utils/__init__.py +1 -3
  252. chalk/utils/_otel_version.py +13 -0
  253. chalk/utils/async_helpers.py +14 -5
  254. chalk/utils/df_utils.py +2 -2
  255. chalk/utils/duration.py +1 -3
  256. chalk/utils/job_log_display.py +538 -0
  257. chalk/utils/missing_dependency.py +5 -4
  258. chalk/utils/notebook.py +255 -2
  259. chalk/utils/pl_helpers.py +190 -37
  260. chalk/utils/pydanticutil/pydantic_compat.py +1 -2
  261. chalk/utils/storage_client.py +246 -0
  262. chalk/utils/threading.py +1 -3
  263. chalk/utils/tracing.py +194 -86
  264. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/METADATA +53 -21
  265. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/RECORD +268 -198
  266. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
  267. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
  268. {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/top_level.txt +0 -0
@@ -20,6 +20,11 @@ class TeamServiceStub(object):
20
20
  request_serializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvRequest.SerializeToString,
21
21
  response_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvResponse.FromString,
22
22
  )
23
+ self.GetEnvIncludingArchived = channel.unary_unary(
24
+ "/chalk.server.v1.TeamService/GetEnvIncludingArchived",
25
+ request_serializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvIncludingArchivedRequest.SerializeToString,
26
+ response_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvIncludingArchivedResponse.FromString,
27
+ )
23
28
  self.GetEnvironments = channel.unary_unary(
24
29
  "/chalk.server.v1.TeamService/GetEnvironments",
25
30
  request_serializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvironmentsRequest.SerializeToString,
@@ -70,6 +75,11 @@ class TeamServiceStub(object):
70
75
  request_serializer=chalk_dot_server_dot_v1_dot_team__pb2.UpdateEnvironmentRequest.SerializeToString,
71
76
  response_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.UpdateEnvironmentResponse.FromString,
72
77
  )
78
+ self.CreateVectorDBConfiguration = channel.unary_unary(
79
+ "/chalk.server.v1.TeamService/CreateVectorDBConfiguration",
80
+ request_serializer=chalk_dot_server_dot_v1_dot_team__pb2.CreateVectorDBConfigurationRequest.SerializeToString,
81
+ response_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.CreateVectorDBConfigurationResponse.FromString,
82
+ )
73
83
  self.GetAvailablePermissions = channel.unary_unary(
74
84
  "/chalk.server.v1.TeamService/GetAvailablePermissions",
75
85
  request_serializer=chalk_dot_server_dot_v1_dot_team__pb2.GetAvailablePermissionsRequest.SerializeToString,
@@ -151,6 +161,12 @@ class TeamServiceServicer(object):
151
161
  context.set_details("Method not implemented!")
152
162
  raise NotImplementedError("Method not implemented!")
153
163
 
164
+ def GetEnvIncludingArchived(self, request, context):
165
+ """Missing associated documentation comment in .proto file."""
166
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
167
+ context.set_details("Method not implemented!")
168
+ raise NotImplementedError("Method not implemented!")
169
+
154
170
  def GetEnvironments(self, request, context):
155
171
  """Missing associated documentation comment in .proto file."""
156
172
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -211,6 +227,12 @@ class TeamServiceServicer(object):
211
227
  context.set_details("Method not implemented!")
212
228
  raise NotImplementedError("Method not implemented!")
213
229
 
230
+ def CreateVectorDBConfiguration(self, request, context):
231
+ """Missing associated documentation comment in .proto file."""
232
+ context.set_code(grpc.StatusCode.UNIMPLEMENTED)
233
+ context.set_details("Method not implemented!")
234
+ raise NotImplementedError("Method not implemented!")
235
+
214
236
  def GetAvailablePermissions(self, request, context):
215
237
  """Missing associated documentation comment in .proto file."""
216
238
  context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -308,6 +330,11 @@ def add_TeamServiceServicer_to_server(servicer, server):
308
330
  request_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvRequest.FromString,
309
331
  response_serializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvResponse.SerializeToString,
310
332
  ),
333
+ "GetEnvIncludingArchived": grpc.unary_unary_rpc_method_handler(
334
+ servicer.GetEnvIncludingArchived,
335
+ request_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvIncludingArchivedRequest.FromString,
336
+ response_serializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvIncludingArchivedResponse.SerializeToString,
337
+ ),
311
338
  "GetEnvironments": grpc.unary_unary_rpc_method_handler(
312
339
  servicer.GetEnvironments,
313
340
  request_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.GetEnvironmentsRequest.FromString,
@@ -358,6 +385,11 @@ def add_TeamServiceServicer_to_server(servicer, server):
358
385
  request_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.UpdateEnvironmentRequest.FromString,
359
386
  response_serializer=chalk_dot_server_dot_v1_dot_team__pb2.UpdateEnvironmentResponse.SerializeToString,
360
387
  ),
388
+ "CreateVectorDBConfiguration": grpc.unary_unary_rpc_method_handler(
389
+ servicer.CreateVectorDBConfiguration,
390
+ request_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.CreateVectorDBConfigurationRequest.FromString,
391
+ response_serializer=chalk_dot_server_dot_v1_dot_team__pb2.CreateVectorDBConfigurationResponse.SerializeToString,
392
+ ),
361
393
  "GetAvailablePermissions": grpc.unary_unary_rpc_method_handler(
362
394
  servicer.GetAvailablePermissions,
363
395
  request_deserializer=chalk_dot_server_dot_v1_dot_team__pb2.GetAvailablePermissionsRequest.FromString,
@@ -466,6 +498,35 @@ class TeamService(object):
466
498
  metadata,
467
499
  )
468
500
 
501
+ @staticmethod
502
+ def GetEnvIncludingArchived(
503
+ request,
504
+ target,
505
+ options=(),
506
+ channel_credentials=None,
507
+ call_credentials=None,
508
+ insecure=False,
509
+ compression=None,
510
+ wait_for_ready=None,
511
+ timeout=None,
512
+ metadata=None,
513
+ ):
514
+ return grpc.experimental.unary_unary(
515
+ request,
516
+ target,
517
+ "/chalk.server.v1.TeamService/GetEnvIncludingArchived",
518
+ chalk_dot_server_dot_v1_dot_team__pb2.GetEnvIncludingArchivedRequest.SerializeToString,
519
+ chalk_dot_server_dot_v1_dot_team__pb2.GetEnvIncludingArchivedResponse.FromString,
520
+ options,
521
+ channel_credentials,
522
+ insecure,
523
+ call_credentials,
524
+ compression,
525
+ wait_for_ready,
526
+ timeout,
527
+ metadata,
528
+ )
529
+
469
530
  @staticmethod
470
531
  def GetEnvironments(
471
532
  request,
@@ -756,6 +817,35 @@ class TeamService(object):
756
817
  metadata,
757
818
  )
758
819
 
820
+ @staticmethod
821
+ def CreateVectorDBConfiguration(
822
+ request,
823
+ target,
824
+ options=(),
825
+ channel_credentials=None,
826
+ call_credentials=None,
827
+ insecure=False,
828
+ compression=None,
829
+ wait_for_ready=None,
830
+ timeout=None,
831
+ metadata=None,
832
+ ):
833
+ return grpc.experimental.unary_unary(
834
+ request,
835
+ target,
836
+ "/chalk.server.v1.TeamService/CreateVectorDBConfiguration",
837
+ chalk_dot_server_dot_v1_dot_team__pb2.CreateVectorDBConfigurationRequest.SerializeToString,
838
+ chalk_dot_server_dot_v1_dot_team__pb2.CreateVectorDBConfigurationResponse.FromString,
839
+ options,
840
+ channel_credentials,
841
+ insecure,
842
+ call_credentials,
843
+ compression,
844
+ wait_for_ready,
845
+ timeout,
846
+ metadata,
847
+ )
848
+
759
849
  @staticmethod
760
850
  def GetAvailablePermissions(
761
851
  request,
@@ -20,6 +20,8 @@ from chalk._gen.chalk.server.v1.team_pb2 import (
20
20
  CreateServiceTokenResponse,
21
21
  CreateTeamRequest,
22
22
  CreateTeamResponse,
23
+ CreateVectorDBConfigurationRequest,
24
+ CreateVectorDBConfigurationResponse,
23
25
  DeactivateUserRequest,
24
26
  DeactivateUserResponse,
25
27
  DeleteServiceTokenRequest,
@@ -32,6 +34,8 @@ from chalk._gen.chalk.server.v1.team_pb2 import (
32
34
  GetAvailablePermissionsResponse,
33
35
  GetDisplayAgentRequest,
34
36
  GetDisplayAgentResponse,
37
+ GetEnvIncludingArchivedRequest,
38
+ GetEnvIncludingArchivedResponse,
35
39
  GetEnvRequest,
36
40
  GetEnvResponse,
37
41
  GetEnvironmentsRequest,
@@ -72,6 +76,10 @@ class TeamServiceStub:
72
76
  GetEnvRequest,
73
77
  GetEnvResponse,
74
78
  ]
79
+ GetEnvIncludingArchived: UnaryUnaryMultiCallable[
80
+ GetEnvIncludingArchivedRequest,
81
+ GetEnvIncludingArchivedResponse,
82
+ ]
75
83
  GetEnvironments: UnaryUnaryMultiCallable[
76
84
  GetEnvironmentsRequest,
77
85
  GetEnvironmentsResponse,
@@ -112,6 +120,10 @@ class TeamServiceStub:
112
120
  UpdateEnvironmentRequest,
113
121
  UpdateEnvironmentResponse,
114
122
  ]
123
+ CreateVectorDBConfiguration: UnaryUnaryMultiCallable[
124
+ CreateVectorDBConfigurationRequest,
125
+ CreateVectorDBConfigurationResponse,
126
+ ]
115
127
  GetAvailablePermissions: UnaryUnaryMultiCallable[
116
128
  GetAvailablePermissionsRequest,
117
129
  GetAvailablePermissionsResponse,
@@ -182,6 +194,12 @@ class TeamServiceServicer(metaclass=ABCMeta):
182
194
  context: ServicerContext,
183
195
  ) -> GetEnvResponse: ...
184
196
  @abstractmethod
197
+ def GetEnvIncludingArchived(
198
+ self,
199
+ request: GetEnvIncludingArchivedRequest,
200
+ context: ServicerContext,
201
+ ) -> GetEnvIncludingArchivedResponse: ...
202
+ @abstractmethod
185
203
  def GetEnvironments(
186
204
  self,
187
205
  request: GetEnvironmentsRequest,
@@ -242,6 +260,12 @@ class TeamServiceServicer(metaclass=ABCMeta):
242
260
  context: ServicerContext,
243
261
  ) -> UpdateEnvironmentResponse: ...
244
262
  @abstractmethod
263
+ def CreateVectorDBConfiguration(
264
+ self,
265
+ request: CreateVectorDBConfigurationRequest,
266
+ context: ServicerContext,
267
+ ) -> CreateVectorDBConfigurationResponse: ...
268
+ @abstractmethod
245
269
  def GetAvailablePermissions(
246
270
  self,
247
271
  request: GetAvailablePermissionsRequest,
@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
14
14
 
15
15
 
16
16
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
17
- b'\n\x1b\x63halk/server/v1/topic.proto\x12\x0f\x63halk.server.v1"\'\n\x08SQSTopic\x12\x1b\n\tqueue_url\x18\x01 \x01(\tR\x08queueUrl"G\n\x0bPubSubTopic\x12\x1d\n\nproject_id\x18\x01 \x01(\tR\tprojectId\x12\x19\n\x08topic_id\x18\x02 \x01(\tR\x07topicId"\x8d\x01\n\x05Topic\x12\x38\n\tsqs_topic\x18\x01 \x01(\x0b\x32\x19.chalk.server.v1.SQSTopicH\x00R\x08sqsTopic\x12\x41\n\x0cpubsub_topic\x18\x02 \x01(\x0b\x32\x1c.chalk.server.v1.PubSubTopicH\x00R\x0bpubsubTopicB\x07\n\x05topicB\x93\x01\n\x13\x63om.chalk.server.v1B\nTopicProtoP\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'
17
+ b'\n\x1b\x63halk/server/v1/topic.proto\x12\x0f\x63halk.server.v1"\'\n\x08SQSTopic\x12\x1b\n\tqueue_url\x18\x01 \x01(\tR\x08queueUrl"G\n\x0bPubSubTopic\x12\x1d\n\nproject_id\x18\x01 \x01(\tR\tprojectId\x12\x19\n\x08topic_id\x18\x02 \x01(\tR\x07topicId"#\n\rInMemoryTopic\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name"\xd7\x01\n\x05Topic\x12\x38\n\tsqs_topic\x18\x01 \x01(\x0b\x32\x19.chalk.server.v1.SQSTopicH\x00R\x08sqsTopic\x12\x41\n\x0cpubsub_topic\x18\x02 \x01(\x0b\x32\x1c.chalk.server.v1.PubSubTopicH\x00R\x0bpubsubTopic\x12H\n\x0fin_memory_topic\x18\x03 \x01(\x0b\x32\x1e.chalk.server.v1.InMemoryTopicH\x00R\rinMemoryTopicB\x07\n\x05topicB\x93\x01\n\x13\x63om.chalk.server.v1B\nTopicProtoP\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'
18
18
  )
19
19
 
20
20
  _globals = globals()
@@ -29,6 +29,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
29
29
  _globals["_SQSTOPIC"]._serialized_end = 87
30
30
  _globals["_PUBSUBTOPIC"]._serialized_start = 89
31
31
  _globals["_PUBSUBTOPIC"]._serialized_end = 160
32
- _globals["_TOPIC"]._serialized_start = 163
33
- _globals["_TOPIC"]._serialized_end = 304
32
+ _globals["_INMEMORYTOPIC"]._serialized_start = 162
33
+ _globals["_INMEMORYTOPIC"]._serialized_end = 197
34
+ _globals["_TOPIC"]._serialized_start = 200
35
+ _globals["_TOPIC"]._serialized_end = 415
34
36
  # @@protoc_insertion_point(module_scope)
@@ -18,14 +18,23 @@ class PubSubTopic(_message.Message):
18
18
  topic_id: str
19
19
  def __init__(self, project_id: _Optional[str] = ..., topic_id: _Optional[str] = ...) -> None: ...
20
20
 
21
+ class InMemoryTopic(_message.Message):
22
+ __slots__ = ("name",)
23
+ NAME_FIELD_NUMBER: _ClassVar[int]
24
+ name: str
25
+ def __init__(self, name: _Optional[str] = ...) -> None: ...
26
+
21
27
  class Topic(_message.Message):
22
- __slots__ = ("sqs_topic", "pubsub_topic")
28
+ __slots__ = ("sqs_topic", "pubsub_topic", "in_memory_topic")
23
29
  SQS_TOPIC_FIELD_NUMBER: _ClassVar[int]
24
30
  PUBSUB_TOPIC_FIELD_NUMBER: _ClassVar[int]
31
+ IN_MEMORY_TOPIC_FIELD_NUMBER: _ClassVar[int]
25
32
  sqs_topic: SQSTopic
26
33
  pubsub_topic: PubSubTopic
34
+ in_memory_topic: InMemoryTopic
27
35
  def __init__(
28
36
  self,
29
37
  sqs_topic: _Optional[_Union[SQSTopic, _Mapping]] = ...,
30
38
  pubsub_topic: _Optional[_Union[PubSubTopic, _Mapping]] = ...,
39
+ in_memory_topic: _Optional[_Union[InMemoryTopic, _Mapping]] = ...,
31
40
  ) -> None: ...
@@ -19,7 +19,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
19
19
 
20
20
 
21
21
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
22
- b'\n\x1b\x63halk/server/v1/trace.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xb5\x05\n\tChalkSpan\x12\x17\n\x07span_id\x18\x01 \x01(\tR\x06spanId\x12\x19\n\x08trace_id\x18\x02 \x01(\tR\x07traceId\x12)\n\x0eparent_span_id\x18\x03 \x01(\tH\x00R\x0cparentSpanId\x88\x01\x01\x12%\n\x0eoperation_name\x18\x04 \x01(\tR\roperationName\x12\x39\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12\x35\n\x08\x65nd_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x07\x65ndTime\x12\x35\n\x08\x64uration\x18\x07 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12\x38\n\x06status\x18\x08 \x01(\x0b\x32 .chalk.server.v1.ChalkSpanStatusR\x06status\x12J\n\nattributes\x18\t \x03(\x0b\x32*.chalk.server.v1.ChalkSpan.AttributesEntryR\nattributes\x12\x37\n\x06\x65vents\x18\n \x03(\x0b\x32\x1f.chalk.server.v1.ChalkSpanEventR\x06\x65vents\x12\x34\n\x05links\x18\x0b \x03(\x0b\x32\x1e.chalk.server.v1.ChalkSpanLinkR\x05links\x12\x32\n\x04kind\x18\x0c \x01(\x0e\x32\x1e.chalk.server.v1.ChalkSpanKindR\x04kind\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_parent_span_id"~\n\x0f\x43halkSpanStatus\x12\x34\n\x04\x63ode\x18\x01 \x01(\x0e\x32 .chalk.server.v1.ChalkStatusCodeR\x04\x63ode\x12%\n\x0b\x64\x65scription\x18\x02 \x01(\tH\x00R\x0b\x64\x65scription\x88\x01\x01\x42\x0e\n\x0c_description"\xee\x01\n\x0e\x43halkSpanEvent\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x38\n\ttimestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\ttimestamp\x12O\n\nattributes\x18\x03 \x03(\x0b\x32/.chalk.server.v1.ChalkSpanEvent.AttributesEntryR\nattributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\xd2\x01\n\rChalkSpanLink\x12\x19\n\x08trace_id\x18\x01 \x01(\tR\x07traceId\x12\x17\n\x07span_id\x18\x02 \x01(\tR\x06spanId\x12N\n\nattributes\x18\x03 \x03(\x0b\x32..chalk.server.v1.ChalkSpanLink.AttributesEntryR\nattributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\xe1\x02\n\nChalkTrace\x12\x19\n\x08trace_id\x18\x01 \x01(\tR\x07traceId\x12\x30\n\x05spans\x18\x02 \x03(\x0b\x32\x1a.chalk.server.v1.ChalkSpanR\x05spans\x12%\n\x0croot_span_id\x18\x03 \x01(\tH\x00R\nrootSpanId\x88\x01\x01\x12!\n\x0cservice_name\x18\x04 \x01(\tR\x0bserviceName\x12\x64\n\x13resource_attributes\x18\x05 \x03(\x0b\x32\x33.chalk.server.v1.ChalkTrace.ResourceAttributesEntryR\x12resourceAttributes\x1a\x45\n\x17ResourceAttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0f\n\r_root_span_id"w\n\x0fGetTraceRequest\x12&\n\x0coperation_id\x18\x01 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x1e\n\x08trace_id\x18\x02 \x01(\tH\x01R\x07traceId\x88\x01\x01\x42\x0f\n\r_operation_idB\x0b\n\t_trace_id"E\n\x10GetTraceResponse\x12\x31\n\x05trace\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.ChalkTraceR\x05trace*k\n\x0f\x43halkStatusCode\x12!\n\x1d\x43HALK_STATUS_CODE_UNSPECIFIED\x10\x00\x12\x18\n\x14\x43HALK_STATUS_CODE_OK\x10\x01\x12\x1b\n\x17\x43HALK_STATUS_CODE_ERROR\x10\x02*\xc2\x01\n\rChalkSpanKind\x12\x1f\n\x1b\x43HALK_SPAN_KIND_UNSPECIFIED\x10\x00\x12\x1a\n\x16\x43HALK_SPAN_KIND_SERVER\x10\x01\x12\x1a\n\x16\x43HALK_SPAN_KIND_CLIENT\x10\x02\x12\x1c\n\x18\x43HALK_SPAN_KIND_PRODUCER\x10\x03\x12\x1c\n\x18\x43HALK_SPAN_KIND_CONSUMER\x10\x04\x12\x1c\n\x18\x43HALK_SPAN_KIND_INTERNAL\x10\x05\x32g\n\x0cTraceService\x12W\n\x08GetTrace\x12 .chalk.server.v1.GetTraceRequest\x1a!.chalk.server.v1.GetTraceResponse"\x06\x90\x02\x01\x80}\x06\x42\x93\x01\n\x13\x63om.chalk.server.v1B\nTraceProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
22
+ b'\n\x1b\x63halk/server/v1/trace.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xe1\x06\n\tChalkSpan\x12\x17\n\x07span_id\x18\x01 \x01(\tR\x06spanId\x12\x19\n\x08trace_id\x18\x02 \x01(\tR\x07traceId\x12)\n\x0eparent_span_id\x18\x03 \x01(\tH\x00R\x0cparentSpanId\x88\x01\x01\x12%\n\x0eoperation_name\x18\x04 \x01(\tR\roperationName\x12\x39\n\nstart_time\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tstartTime\x12\x35\n\x08\x65nd_time\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x07\x65ndTime\x12\x35\n\x08\x64uration\x18\x07 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12\x38\n\x06status\x18\x08 \x01(\x0b\x32 .chalk.server.v1.ChalkSpanStatusR\x06status\x12J\n\nattributes\x18\t \x03(\x0b\x32*.chalk.server.v1.ChalkSpan.AttributesEntryR\nattributes\x12\x37\n\x06\x65vents\x18\n \x03(\x0b\x32\x1f.chalk.server.v1.ChalkSpanEventR\x06\x65vents\x12\x34\n\x05links\x18\x0b \x03(\x0b\x32\x1e.chalk.server.v1.ChalkSpanLinkR\x05links\x12\x32\n\x04kind\x18\x0c \x01(\x0e\x32\x1e.chalk.server.v1.ChalkSpanKindR\x04kind\x12\x63\n\x13resource_attributes\x18\r \x03(\x0b\x32\x32.chalk.server.v1.ChalkSpan.ResourceAttributesEntryR\x12resourceAttributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1a\x45\n\x17ResourceAttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_parent_span_id"~\n\x0f\x43halkSpanStatus\x12\x34\n\x04\x63ode\x18\x01 \x01(\x0e\x32 .chalk.server.v1.ChalkStatusCodeR\x04\x63ode\x12%\n\x0b\x64\x65scription\x18\x02 \x01(\tH\x00R\x0b\x64\x65scription\x88\x01\x01\x42\x0e\n\x0c_description"\xee\x01\n\x0e\x43halkSpanEvent\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x38\n\ttimestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\ttimestamp\x12O\n\nattributes\x18\x03 \x03(\x0b\x32/.chalk.server.v1.ChalkSpanEvent.AttributesEntryR\nattributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\xd2\x01\n\rChalkSpanLink\x12\x19\n\x08trace_id\x18\x01 \x01(\tR\x07traceId\x12\x17\n\x07span_id\x18\x02 \x01(\tR\x06spanId\x12N\n\nattributes\x18\x03 \x03(\x0b\x32..chalk.server.v1.ChalkSpanLink.AttributesEntryR\nattributes\x1a=\n\x0f\x41ttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\xe1\x02\n\nChalkTrace\x12\x19\n\x08trace_id\x18\x01 \x01(\tR\x07traceId\x12\x30\n\x05spans\x18\x02 \x03(\x0b\x32\x1a.chalk.server.v1.ChalkSpanR\x05spans\x12%\n\x0croot_span_id\x18\x03 \x01(\tH\x00R\nrootSpanId\x88\x01\x01\x12!\n\x0cservice_name\x18\x04 \x01(\tR\x0bserviceName\x12\x64\n\x13resource_attributes\x18\x05 \x03(\x0b\x32\x33.chalk.server.v1.ChalkTrace.ResourceAttributesEntryR\x12resourceAttributes\x1a\x45\n\x17ResourceAttributesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0f\n\r_root_span_id"w\n\x0fGetTraceRequest\x12&\n\x0coperation_id\x18\x01 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x1e\n\x08trace_id\x18\x02 \x01(\tH\x01R\x07traceId\x88\x01\x01\x42\x0f\n\r_operation_idB\x0b\n\t_trace_id"E\n\x10GetTraceResponse\x12\x31\n\x05trace\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.ChalkTraceR\x05trace"\xeb\x02\n\x10ListTraceRequest\x12>\n\nstart_time\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tstartTime\x88\x01\x01\x12:\n\x08\x65nd_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x07\x65ndTime\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x12&\n\x0cservice_name\x18\x04 \x01(\tH\x03R\x0bserviceName\x88\x01\x01\x12 \n\tspan_name\x18\x05 \x01(\tH\x04R\x08spanName\x88\x01\x01\x12"\n\npage_token\x18\x06 \x01(\tH\x05R\tpageToken\x88\x01\x01\x42\r\n\x0b_start_timeB\x0b\n\t_end_timeB\x08\n\x06_limitB\x0f\n\r_service_nameB\x0c\n\n_span_nameB\r\n\x0b_page_token"\x89\x01\n\x11ListTraceResponse\x12\x33\n\x06traces\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.ChalkTraceR\x06traces\x12+\n\x0fnext_page_token\x18\x02 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token"D\n\x0eGetSpanRequest\x12\x17\n\x07span_id\x18\x01 \x01(\tR\x06spanId\x12\x19\n\x08trace_id\x18\x02 \x01(\tR\x07traceId"A\n\x0fGetSpanResponse\x12.\n\x04span\x18\x01 \x01(\x0b\x32\x1a.chalk.server.v1.ChalkSpanR\x04span"\xbe\x05\n\x0fListSpanRequest\x12\x1e\n\x08trace_id\x18\x01 \x01(\tH\x00R\x07traceId\x88\x01\x01\x12>\n\nstart_time\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tstartTime\x88\x01\x01\x12:\n\x08\x65nd_time\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\x07\x65ndTime\x88\x01\x01\x12\x19\n\x05limit\x18\x04 \x01(\x05H\x03R\x05limit\x88\x01\x01\x12"\n\npage_token\x18\x05 \x01(\tH\x04R\tpageToken\x88\x01\x01\x12)\n\x0eparent_span_id\x18\x06 \x01(\tH\x05R\x0cparentSpanId\x88\x01\x01\x12*\n\x0eoperation_name\x18\x07 \x01(\tH\x06R\roperationName\x88\x01\x01\x12&\n\x0cservice_name\x18\x08 \x01(\tH\x07R\x0bserviceName\x88\x01\x01\x12\x46\n\x0bstatus_code\x18\t \x01(\x0e\x32 .chalk.server.v1.ChalkStatusCodeH\x08R\nstatusCode\x88\x01\x01\x12+\n\x0fmin_duration_us\x18\n \x01(\x03H\tR\rminDurationUs\x88\x01\x01\x12+\n\x0fmax_duration_us\x18\x0b \x01(\x03H\nR\rmaxDurationUs\x88\x01\x01\x42\x0b\n\t_trace_idB\r\n\x0b_start_timeB\x0b\n\t_end_timeB\x08\n\x06_limitB\r\n\x0b_page_tokenB\x11\n\x0f_parent_span_idB\x11\n\x0f_operation_nameB\x0f\n\r_service_nameB\x0e\n\x0c_status_codeB\x12\n\x10_min_duration_usB\x12\n\x10_max_duration_us"\x85\x01\n\x10ListSpanResponse\x12\x30\n\x05spans\x18\x01 \x03(\x0b\x32\x1a.chalk.server.v1.ChalkSpanR\x05spans\x12+\n\x0fnext_page_token\x18\x02 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token*k\n\x0f\x43halkStatusCode\x12!\n\x1d\x43HALK_STATUS_CODE_UNSPECIFIED\x10\x00\x12\x18\n\x14\x43HALK_STATUS_CODE_OK\x10\x01\x12\x1b\n\x17\x43HALK_STATUS_CODE_ERROR\x10\x02*\xc2\x01\n\rChalkSpanKind\x12\x1f\n\x1b\x43HALK_SPAN_KIND_UNSPECIFIED\x10\x00\x12\x1a\n\x16\x43HALK_SPAN_KIND_SERVER\x10\x01\x12\x1a\n\x16\x43HALK_SPAN_KIND_CLIENT\x10\x02\x12\x1c\n\x18\x43HALK_SPAN_KIND_PRODUCER\x10\x03\x12\x1c\n\x18\x43HALK_SPAN_KIND_CONSUMER\x10\x04\x12\x1c\n\x18\x43HALK_SPAN_KIND_INTERNAL\x10\x05\x32\xf2\x02\n\x0cTraceService\x12W\n\x08GetTrace\x12 .chalk.server.v1.GetTraceRequest\x1a!.chalk.server.v1.GetTraceResponse"\x06\x90\x02\x01\x80}\x06\x12Z\n\tListTrace\x12!.chalk.server.v1.ListTraceRequest\x1a".chalk.server.v1.ListTraceResponse"\x06\x90\x02\x01\x80}\x06\x12T\n\x07GetSpan\x12\x1f.chalk.server.v1.GetSpanRequest\x1a .chalk.server.v1.GetSpanResponse"\x06\x90\x02\x01\x80}\x06\x12W\n\x08ListSpan\x12 .chalk.server.v1.ListSpanRequest\x1a!.chalk.server.v1.ListSpanResponse"\x06\x90\x02\x01\x80}\x06\x42\x93\x01\n\x13\x63om.chalk.server.v1B\nTraceProtoP\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
23
  )
24
24
 
25
25
  _globals = globals()
@@ -32,6 +32,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
32
32
  ]._serialized_options = b"\n\023com.chalk.server.v1B\nTraceProtoP\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
33
  _globals["_CHALKSPAN_ATTRIBUTESENTRY"]._options = None
34
34
  _globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_options = b"8\001"
35
+ _globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._options = None
36
+ _globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._serialized_options = b"8\001"
35
37
  _globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._options = None
36
38
  _globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_options = b"8\001"
37
39
  _globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._options = None
@@ -40,32 +42,52 @@ if _descriptor._USE_C_DESCRIPTORS == False:
40
42
  _globals["_CHALKTRACE_RESOURCEATTRIBUTESENTRY"]._serialized_options = b"8\001"
41
43
  _globals["_TRACESERVICE"].methods_by_name["GetTrace"]._options = None
42
44
  _globals["_TRACESERVICE"].methods_by_name["GetTrace"]._serialized_options = b"\220\002\001\200}\006"
43
- _globals["_CHALKSTATUSCODE"]._serialized_start = 1972
44
- _globals["_CHALKSTATUSCODE"]._serialized_end = 2079
45
- _globals["_CHALKSPANKIND"]._serialized_start = 2082
46
- _globals["_CHALKSPANKIND"]._serialized_end = 2276
45
+ _globals["_TRACESERVICE"].methods_by_name["ListTrace"]._options = None
46
+ _globals["_TRACESERVICE"].methods_by_name["ListTrace"]._serialized_options = b"\220\002\001\200}\006"
47
+ _globals["_TRACESERVICE"].methods_by_name["GetSpan"]._options = None
48
+ _globals["_TRACESERVICE"].methods_by_name["GetSpan"]._serialized_options = b"\220\002\001\200}\006"
49
+ _globals["_TRACESERVICE"].methods_by_name["ListSpan"]._options = None
50
+ _globals["_TRACESERVICE"].methods_by_name["ListSpan"]._serialized_options = b"\220\002\001\200}\006"
51
+ _globals["_CHALKSTATUSCODE"]._serialized_start = 3628
52
+ _globals["_CHALKSTATUSCODE"]._serialized_end = 3735
53
+ _globals["_CHALKSPANKIND"]._serialized_start = 3738
54
+ _globals["_CHALKSPANKIND"]._serialized_end = 3932
47
55
  _globals["_CHALKSPAN"]._serialized_start = 147
48
- _globals["_CHALKSPAN"]._serialized_end = 840
49
- _globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_start = 760
50
- _globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_end = 821
51
- _globals["_CHALKSPANSTATUS"]._serialized_start = 842
52
- _globals["_CHALKSPANSTATUS"]._serialized_end = 968
53
- _globals["_CHALKSPANEVENT"]._serialized_start = 971
54
- _globals["_CHALKSPANEVENT"]._serialized_end = 1209
55
- _globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_start = 760
56
- _globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_end = 821
57
- _globals["_CHALKSPANLINK"]._serialized_start = 1212
58
- _globals["_CHALKSPANLINK"]._serialized_end = 1422
59
- _globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._serialized_start = 760
60
- _globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._serialized_end = 821
61
- _globals["_CHALKTRACE"]._serialized_start = 1425
62
- _globals["_CHALKTRACE"]._serialized_end = 1778
63
- _globals["_CHALKTRACE_RESOURCEATTRIBUTESENTRY"]._serialized_start = 1692
64
- _globals["_CHALKTRACE_RESOURCEATTRIBUTESENTRY"]._serialized_end = 1761
65
- _globals["_GETTRACEREQUEST"]._serialized_start = 1780
66
- _globals["_GETTRACEREQUEST"]._serialized_end = 1899
67
- _globals["_GETTRACERESPONSE"]._serialized_start = 1901
68
- _globals["_GETTRACERESPONSE"]._serialized_end = 1970
69
- _globals["_TRACESERVICE"]._serialized_start = 2278
70
- _globals["_TRACESERVICE"]._serialized_end = 2381
56
+ _globals["_CHALKSPAN"]._serialized_end = 1012
57
+ _globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_start = 861
58
+ _globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_end = 922
59
+ _globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._serialized_start = 924
60
+ _globals["_CHALKSPAN_RESOURCEATTRIBUTESENTRY"]._serialized_end = 993
61
+ _globals["_CHALKSPANSTATUS"]._serialized_start = 1014
62
+ _globals["_CHALKSPANSTATUS"]._serialized_end = 1140
63
+ _globals["_CHALKSPANEVENT"]._serialized_start = 1143
64
+ _globals["_CHALKSPANEVENT"]._serialized_end = 1381
65
+ _globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_start = 861
66
+ _globals["_CHALKSPANEVENT_ATTRIBUTESENTRY"]._serialized_end = 922
67
+ _globals["_CHALKSPANLINK"]._serialized_start = 1384
68
+ _globals["_CHALKSPANLINK"]._serialized_end = 1594
69
+ _globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._serialized_start = 861
70
+ _globals["_CHALKSPANLINK_ATTRIBUTESENTRY"]._serialized_end = 922
71
+ _globals["_CHALKTRACE"]._serialized_start = 1597
72
+ _globals["_CHALKTRACE"]._serialized_end = 1950
73
+ _globals["_CHALKTRACE_RESOURCEATTRIBUTESENTRY"]._serialized_start = 924
74
+ _globals["_CHALKTRACE_RESOURCEATTRIBUTESENTRY"]._serialized_end = 993
75
+ _globals["_GETTRACEREQUEST"]._serialized_start = 1952
76
+ _globals["_GETTRACEREQUEST"]._serialized_end = 2071
77
+ _globals["_GETTRACERESPONSE"]._serialized_start = 2073
78
+ _globals["_GETTRACERESPONSE"]._serialized_end = 2142
79
+ _globals["_LISTTRACEREQUEST"]._serialized_start = 2145
80
+ _globals["_LISTTRACEREQUEST"]._serialized_end = 2508
81
+ _globals["_LISTTRACERESPONSE"]._serialized_start = 2511
82
+ _globals["_LISTTRACERESPONSE"]._serialized_end = 2648
83
+ _globals["_GETSPANREQUEST"]._serialized_start = 2650
84
+ _globals["_GETSPANREQUEST"]._serialized_end = 2718
85
+ _globals["_GETSPANRESPONSE"]._serialized_start = 2720
86
+ _globals["_GETSPANRESPONSE"]._serialized_end = 2785
87
+ _globals["_LISTSPANREQUEST"]._serialized_start = 2788
88
+ _globals["_LISTSPANREQUEST"]._serialized_end = 3490
89
+ _globals["_LISTSPANRESPONSE"]._serialized_start = 3493
90
+ _globals["_LISTSPANRESPONSE"]._serialized_end = 3626
91
+ _globals["_TRACESERVICE"]._serialized_start = 3935
92
+ _globals["_TRACESERVICE"]._serialized_end = 4305
71
93
  # @@protoc_insertion_point(module_scope)
@@ -54,6 +54,7 @@ class ChalkSpan(_message.Message):
54
54
  "events",
55
55
  "links",
56
56
  "kind",
57
+ "resource_attributes",
57
58
  )
58
59
  class AttributesEntry(_message.Message):
59
60
  __slots__ = ("key", "value")
@@ -63,6 +64,14 @@ class ChalkSpan(_message.Message):
63
64
  value: str
64
65
  def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
65
66
 
67
+ class ResourceAttributesEntry(_message.Message):
68
+ __slots__ = ("key", "value")
69
+ KEY_FIELD_NUMBER: _ClassVar[int]
70
+ VALUE_FIELD_NUMBER: _ClassVar[int]
71
+ key: str
72
+ value: str
73
+ def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
74
+
66
75
  SPAN_ID_FIELD_NUMBER: _ClassVar[int]
67
76
  TRACE_ID_FIELD_NUMBER: _ClassVar[int]
68
77
  PARENT_SPAN_ID_FIELD_NUMBER: _ClassVar[int]
@@ -75,6 +84,7 @@ class ChalkSpan(_message.Message):
75
84
  EVENTS_FIELD_NUMBER: _ClassVar[int]
76
85
  LINKS_FIELD_NUMBER: _ClassVar[int]
77
86
  KIND_FIELD_NUMBER: _ClassVar[int]
87
+ RESOURCE_ATTRIBUTES_FIELD_NUMBER: _ClassVar[int]
78
88
  span_id: str
79
89
  trace_id: str
80
90
  parent_span_id: str
@@ -87,6 +97,7 @@ class ChalkSpan(_message.Message):
87
97
  events: _containers.RepeatedCompositeFieldContainer[ChalkSpanEvent]
88
98
  links: _containers.RepeatedCompositeFieldContainer[ChalkSpanLink]
89
99
  kind: ChalkSpanKind
100
+ resource_attributes: _containers.ScalarMap[str, str]
90
101
  def __init__(
91
102
  self,
92
103
  span_id: _Optional[str] = ...,
@@ -101,6 +112,7 @@ class ChalkSpan(_message.Message):
101
112
  events: _Optional[_Iterable[_Union[ChalkSpanEvent, _Mapping]]] = ...,
102
113
  links: _Optional[_Iterable[_Union[ChalkSpanLink, _Mapping]]] = ...,
103
114
  kind: _Optional[_Union[ChalkSpanKind, str]] = ...,
115
+ resource_attributes: _Optional[_Mapping[str, str]] = ...,
104
116
  ) -> None: ...
105
117
 
106
118
  class ChalkSpanStatus(_message.Message):
@@ -201,3 +213,112 @@ class GetTraceResponse(_message.Message):
201
213
  TRACE_FIELD_NUMBER: _ClassVar[int]
202
214
  trace: ChalkTrace
203
215
  def __init__(self, trace: _Optional[_Union[ChalkTrace, _Mapping]] = ...) -> None: ...
216
+
217
+ class ListTraceRequest(_message.Message):
218
+ __slots__ = ("start_time", "end_time", "limit", "service_name", "span_name", "page_token")
219
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
220
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
221
+ LIMIT_FIELD_NUMBER: _ClassVar[int]
222
+ SERVICE_NAME_FIELD_NUMBER: _ClassVar[int]
223
+ SPAN_NAME_FIELD_NUMBER: _ClassVar[int]
224
+ PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
225
+ start_time: _timestamp_pb2.Timestamp
226
+ end_time: _timestamp_pb2.Timestamp
227
+ limit: int
228
+ service_name: str
229
+ span_name: str
230
+ page_token: str
231
+ def __init__(
232
+ self,
233
+ start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
234
+ end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
235
+ limit: _Optional[int] = ...,
236
+ service_name: _Optional[str] = ...,
237
+ span_name: _Optional[str] = ...,
238
+ page_token: _Optional[str] = ...,
239
+ ) -> None: ...
240
+
241
+ class ListTraceResponse(_message.Message):
242
+ __slots__ = ("traces", "next_page_token")
243
+ TRACES_FIELD_NUMBER: _ClassVar[int]
244
+ NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
245
+ traces: _containers.RepeatedCompositeFieldContainer[ChalkTrace]
246
+ next_page_token: str
247
+ def __init__(
248
+ self, traces: _Optional[_Iterable[_Union[ChalkTrace, _Mapping]]] = ..., next_page_token: _Optional[str] = ...
249
+ ) -> None: ...
250
+
251
+ class GetSpanRequest(_message.Message):
252
+ __slots__ = ("span_id", "trace_id")
253
+ SPAN_ID_FIELD_NUMBER: _ClassVar[int]
254
+ TRACE_ID_FIELD_NUMBER: _ClassVar[int]
255
+ span_id: str
256
+ trace_id: str
257
+ def __init__(self, span_id: _Optional[str] = ..., trace_id: _Optional[str] = ...) -> None: ...
258
+
259
+ class GetSpanResponse(_message.Message):
260
+ __slots__ = ("span",)
261
+ SPAN_FIELD_NUMBER: _ClassVar[int]
262
+ span: ChalkSpan
263
+ def __init__(self, span: _Optional[_Union[ChalkSpan, _Mapping]] = ...) -> None: ...
264
+
265
+ class ListSpanRequest(_message.Message):
266
+ __slots__ = (
267
+ "trace_id",
268
+ "start_time",
269
+ "end_time",
270
+ "limit",
271
+ "page_token",
272
+ "parent_span_id",
273
+ "operation_name",
274
+ "service_name",
275
+ "status_code",
276
+ "min_duration_us",
277
+ "max_duration_us",
278
+ )
279
+ TRACE_ID_FIELD_NUMBER: _ClassVar[int]
280
+ START_TIME_FIELD_NUMBER: _ClassVar[int]
281
+ END_TIME_FIELD_NUMBER: _ClassVar[int]
282
+ LIMIT_FIELD_NUMBER: _ClassVar[int]
283
+ PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
284
+ PARENT_SPAN_ID_FIELD_NUMBER: _ClassVar[int]
285
+ OPERATION_NAME_FIELD_NUMBER: _ClassVar[int]
286
+ SERVICE_NAME_FIELD_NUMBER: _ClassVar[int]
287
+ STATUS_CODE_FIELD_NUMBER: _ClassVar[int]
288
+ MIN_DURATION_US_FIELD_NUMBER: _ClassVar[int]
289
+ MAX_DURATION_US_FIELD_NUMBER: _ClassVar[int]
290
+ trace_id: str
291
+ start_time: _timestamp_pb2.Timestamp
292
+ end_time: _timestamp_pb2.Timestamp
293
+ limit: int
294
+ page_token: str
295
+ parent_span_id: str
296
+ operation_name: str
297
+ service_name: str
298
+ status_code: ChalkStatusCode
299
+ min_duration_us: int
300
+ max_duration_us: int
301
+ def __init__(
302
+ self,
303
+ trace_id: _Optional[str] = ...,
304
+ start_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
305
+ end_time: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
306
+ limit: _Optional[int] = ...,
307
+ page_token: _Optional[str] = ...,
308
+ parent_span_id: _Optional[str] = ...,
309
+ operation_name: _Optional[str] = ...,
310
+ service_name: _Optional[str] = ...,
311
+ status_code: _Optional[_Union[ChalkStatusCode, str]] = ...,
312
+ min_duration_us: _Optional[int] = ...,
313
+ max_duration_us: _Optional[int] = ...,
314
+ ) -> None: ...
315
+
316
+ class ListSpanResponse(_message.Message):
317
+ __slots__ = ("spans", "next_page_token")
318
+ SPANS_FIELD_NUMBER: _ClassVar[int]
319
+ NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
320
+ spans: _containers.RepeatedCompositeFieldContainer[ChalkSpan]
321
+ next_page_token: str
322
+ def __init__(
323
+ self, spans: _Optional[_Iterable[_Union[ChalkSpan, _Mapping]]] = ..., next_page_token: _Optional[str] = ...
324
+ ) -> None: ...