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.
- chalk/__init__.py +2 -1
- chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
- chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
- chalk/_gen/chalk/artifacts/v1/chart_pb2.py +36 -33
- chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +41 -1
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +8 -7
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +5 -0
- chalk/_gen/chalk/common/v1/offline_query_pb2.py +19 -13
- chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +37 -0
- chalk/_gen/chalk/common/v1/online_query_pb2.py +54 -54
- chalk/_gen/chalk/common/v1/online_query_pb2.pyi +13 -1
- chalk/_gen/chalk/common/v1/script_task_pb2.py +13 -11
- chalk/_gen/chalk/common/v1/script_task_pb2.pyi +19 -1
- chalk/_gen/chalk/dataframe/__init__.py +0 -0
- chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +123 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
- chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
- chalk/_gen/chalk/graph/v1/graph_pb2.pyi +25 -0
- chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
- chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
- chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
- chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
- chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +54 -27
- chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
- chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
- chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
- chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
- chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
- chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
- chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
- chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
- chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
- chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
- chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
- chalk/_gen/chalk/server/v1/builder_pb2.py +372 -272
- chalk/_gen/chalk/server/v1/builder_pb2.pyi +479 -12
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +360 -0
- chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +96 -0
- chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
- chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
- chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/cloud_components_pb2.py +153 -107
- chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +146 -4
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +180 -0
- chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +48 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +59 -35
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +127 -1
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +135 -0
- chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +36 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
- chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
- chalk/_gen/chalk/server/v1/datasets_pb2.py +36 -24
- chalk/_gen/chalk/server/v1/datasets_pb2.pyi +71 -2
- chalk/_gen/chalk/server/v1/datasets_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/datasets_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/deploy_pb2.py +9 -3
- chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
- chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/deployment_pb2.py +20 -15
- chalk/_gen/chalk/server/v1/deployment_pb2.pyi +25 -0
- chalk/_gen/chalk/server/v1/environment_pb2.py +25 -15
- chalk/_gen/chalk/server/v1/environment_pb2.pyi +93 -1
- chalk/_gen/chalk/server/v1/eventbus_pb2.py +44 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2.pyi +64 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.py +4 -0
- chalk/_gen/chalk/server/v1/eventbus_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
- chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
- chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
- chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/server/v1/graph_pb2.py +41 -3
- chalk/_gen/chalk/server/v1/graph_pb2.pyi +191 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +92 -0
- chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +32 -0
- chalk/_gen/chalk/server/v1/incident_pb2.py +57 -0
- chalk/_gen/chalk/server/v1/incident_pb2.pyi +165 -0
- chalk/_gen/chalk/server/v1/incident_pb2_grpc.py +4 -0
- chalk/_gen/chalk/server/v1/incident_pb2_grpc.pyi +4 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
- chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
- chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
- chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
- chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/log_pb2.py +21 -3
- chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
- chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.py +73 -0
- chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2.pyi +212 -0
- chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.py +217 -0
- chalk/_gen/chalk/server/v1/metadataplanejobqueue_pb2_grpc.pyi +74 -0
- chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
- chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
- chalk/_gen/chalk/server/v1/monitoring_pb2.py +84 -75
- chalk/_gen/chalk/server/v1/monitoring_pb2.pyi +1 -0
- chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py +136 -0
- chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi +38 -0
- chalk/_gen/chalk/server/v1/offline_queries_pb2.py +32 -10
- chalk/_gen/chalk/server/v1/offline_queries_pb2.pyi +73 -0
- chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
- chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
- chalk/_gen/chalk/server/v1/queries_pb2.py +76 -48
- chalk/_gen/chalk/server/v1/queries_pb2.pyi +155 -2
- chalk/_gen/chalk/server/v1/queries_pb2_grpc.py +180 -0
- chalk/_gen/chalk/server/v1/queries_pb2_grpc.pyi +48 -0
- chalk/_gen/chalk/server/v1/scheduled_query_pb2.py +4 -2
- chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -6
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +75 -2
- chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
- chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2.py +26 -14
- chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +33 -3
- chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
- chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2.py +48 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
- chalk/_gen/chalk/server/v1/team_pb2.py +156 -137
- chalk/_gen/chalk/server/v1/team_pb2.pyi +56 -10
- chalk/_gen/chalk/server/v1/team_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
- chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
- chalk/_gen/chalk/server/v1/trace_pb2.py +50 -28
- chalk/_gen/chalk/server/v1/trace_pb2.pyi +121 -0
- chalk/_gen/chalk/server/v1/trace_pb2_grpc.py +135 -0
- chalk/_gen/chalk/server/v1/trace_pb2_grpc.pyi +42 -0
- chalk/_gen/chalk/server/v1/webhook_pb2.py +9 -3
- chalk/_gen/chalk/server/v1/webhook_pb2.pyi +18 -0
- chalk/_gen/chalk/server/v1/webhook_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/webhook_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
- chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +19 -7
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +96 -3
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
- chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
- chalk/_lsp/error_builder.py +11 -0
- chalk/_monitoring/Chart.py +1 -3
- chalk/_version.py +1 -1
- chalk/cli.py +5 -10
- chalk/client/client.py +178 -64
- chalk/client/client_async.py +154 -0
- chalk/client/client_async_impl.py +22 -0
- chalk/client/client_grpc.py +738 -112
- chalk/client/client_impl.py +541 -136
- chalk/client/dataset.py +27 -6
- chalk/client/models.py +99 -2
- chalk/client/serialization/model_serialization.py +126 -10
- chalk/config/project_config.py +1 -1
- chalk/df/LazyFramePlaceholder.py +1154 -0
- chalk/df/ast_parser.py +2 -10
- chalk/features/_class_property.py +7 -0
- chalk/features/_embedding/embedding.py +1 -0
- chalk/features/_embedding/sentence_transformer.py +1 -1
- chalk/features/_encoding/converter.py +83 -2
- chalk/features/_encoding/pyarrow.py +20 -4
- chalk/features/_encoding/rich.py +1 -3
- chalk/features/_tensor.py +1 -2
- chalk/features/dataframe/_filters.py +14 -5
- chalk/features/dataframe/_impl.py +91 -36
- chalk/features/dataframe/_validation.py +11 -7
- chalk/features/feature_field.py +40 -30
- chalk/features/feature_set.py +1 -2
- chalk/features/feature_set_decorator.py +1 -0
- chalk/features/feature_wrapper.py +42 -3
- chalk/features/hooks.py +81 -12
- chalk/features/inference.py +65 -10
- chalk/features/resolver.py +338 -56
- chalk/features/tag.py +1 -3
- chalk/features/underscore_features.py +2 -1
- chalk/functions/__init__.py +456 -21
- chalk/functions/holidays.py +1 -3
- chalk/gitignore/gitignore_parser.py +5 -1
- chalk/importer.py +186 -74
- chalk/ml/__init__.py +6 -2
- chalk/ml/model_hooks.py +368 -51
- chalk/ml/model_reference.py +68 -10
- chalk/ml/model_version.py +34 -21
- chalk/ml/utils.py +143 -40
- chalk/operators/_utils.py +14 -3
- chalk/parsed/_proto/export.py +22 -0
- chalk/parsed/duplicate_input_gql.py +4 -0
- chalk/parsed/expressions.py +1 -3
- chalk/parsed/json_conversions.py +21 -14
- chalk/parsed/to_proto.py +16 -4
- chalk/parsed/user_types_to_json.py +31 -10
- chalk/parsed/validation_from_registries.py +182 -0
- chalk/queries/named_query.py +16 -6
- chalk/queries/scheduled_query.py +13 -1
- chalk/serialization/parsed_annotation.py +25 -12
- chalk/sql/__init__.py +221 -0
- chalk/sql/_internal/integrations/athena.py +6 -1
- chalk/sql/_internal/integrations/bigquery.py +22 -2
- chalk/sql/_internal/integrations/databricks.py +61 -18
- chalk/sql/_internal/integrations/mssql.py +281 -0
- chalk/sql/_internal/integrations/postgres.py +11 -3
- chalk/sql/_internal/integrations/redshift.py +4 -0
- chalk/sql/_internal/integrations/snowflake.py +11 -2
- chalk/sql/_internal/integrations/util.py +2 -1
- chalk/sql/_internal/sql_file_resolver.py +55 -10
- chalk/sql/_internal/sql_source.py +36 -2
- chalk/streams/__init__.py +1 -3
- chalk/streams/_kafka_source.py +5 -1
- chalk/streams/_windows.py +16 -4
- chalk/streams/types.py +1 -2
- chalk/utils/__init__.py +1 -3
- chalk/utils/_otel_version.py +13 -0
- chalk/utils/async_helpers.py +14 -5
- chalk/utils/df_utils.py +2 -2
- chalk/utils/duration.py +1 -3
- chalk/utils/job_log_display.py +538 -0
- chalk/utils/missing_dependency.py +5 -4
- chalk/utils/notebook.py +255 -2
- chalk/utils/pl_helpers.py +190 -37
- chalk/utils/pydanticutil/pydantic_compat.py +1 -2
- chalk/utils/storage_client.py +246 -0
- chalk/utils/threading.py +1 -3
- chalk/utils/tracing.py +194 -86
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/METADATA +53 -21
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/RECORD +268 -198
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
- {chalkpy-2.89.22.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
- {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"\
|
|
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["
|
|
33
|
-
_globals["
|
|
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"\
|
|
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["
|
|
44
|
-
_globals["
|
|
45
|
-
_globals["
|
|
46
|
-
_globals["
|
|
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 =
|
|
49
|
-
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_start =
|
|
50
|
-
_globals["_CHALKSPAN_ATTRIBUTESENTRY"]._serialized_end =
|
|
51
|
-
_globals["
|
|
52
|
-
_globals["
|
|
53
|
-
_globals["
|
|
54
|
-
_globals["
|
|
55
|
-
_globals["
|
|
56
|
-
_globals["
|
|
57
|
-
_globals["
|
|
58
|
-
_globals["
|
|
59
|
-
_globals["
|
|
60
|
-
_globals["
|
|
61
|
-
_globals["
|
|
62
|
-
_globals["
|
|
63
|
-
_globals["
|
|
64
|
-
_globals["
|
|
65
|
-
_globals["
|
|
66
|
-
_globals["
|
|
67
|
-
_globals["
|
|
68
|
-
_globals["
|
|
69
|
-
_globals["
|
|
70
|
-
_globals["
|
|
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: ...
|