chalkpy 2.95.5__py3-none-any.whl → 2.95.7__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/_gen/chalk/protosql/v1/sql_service_pb2.py +47 -47
- chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +12 -3
- chalk/_gen/chalk/server/v1/benchmark_pb2.py +34 -31
- chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +37 -1
- chalk/_gen/chalk/server/v1/branches_pb2_grpc.py +78 -0
- chalk/_gen/chalk/server/v1/builder_pb2.py +310 -310
- chalk/_gen/chalk/server/v1/builder_pb2.pyi +4 -0
- chalk/_gen/chalk/server/v1/monitoring_pb2.py +83 -65
- chalk/_gen/chalk/server/v1/monitoring_pb2.pyi +97 -0
- chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py +90 -0
- chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi +24 -0
- chalk/_gen/chalk/server/v1/scheduler_pb2.py +32 -26
- chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +12 -0
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +45 -0
- chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +12 -0
- chalk/_gen/chalk/server/v1/sql_queries_pb2.py +10 -10
- chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +8 -0
- chalk/_version.py +1 -1
- chalk/client/client_grpc.py +10 -2
- chalk/client/client_impl.py +5 -3
- {chalkpy-2.95.5.dist-info → chalkpy-2.95.7.dist-info}/METADATA +1 -1
- {chalkpy-2.95.5.dist-info → chalkpy-2.95.7.dist-info}/RECORD +25 -24
- {chalkpy-2.95.5.dist-info → chalkpy-2.95.7.dist-info}/WHEEL +0 -0
- {chalkpy-2.95.5.dist-info → chalkpy-2.95.7.dist-info}/entry_points.txt +0 -0
- {chalkpy-2.95.5.dist-info → chalkpy-2.95.7.dist-info}/top_level.txt +0 -0
|
@@ -228,3 +228,15 @@ class ListScheduledResolverRunsResponse(_message.Message):
|
|
|
228
228
|
RUNS_FIELD_NUMBER: _ClassVar[int]
|
|
229
229
|
runs: _containers.RepeatedCompositeFieldContainer[CronResolverRun]
|
|
230
230
|
def __init__(self, runs: _Optional[_Iterable[_Union[CronResolverRun, _Mapping]]] = ...) -> None: ...
|
|
231
|
+
|
|
232
|
+
class CancelScheduledResolverRunRequest(_message.Message):
|
|
233
|
+
__slots__ = ("run_id",)
|
|
234
|
+
RUN_ID_FIELD_NUMBER: _ClassVar[int]
|
|
235
|
+
run_id: str
|
|
236
|
+
def __init__(self, run_id: _Optional[str] = ...) -> None: ...
|
|
237
|
+
|
|
238
|
+
class CancelScheduledResolverRunResponse(_message.Message):
|
|
239
|
+
__slots__ = ("cron_run",)
|
|
240
|
+
CRON_RUN_FIELD_NUMBER: _ClassVar[int]
|
|
241
|
+
cron_run: CronResolverRun
|
|
242
|
+
def __init__(self, cron_run: _Optional[_Union[CronResolverRun, _Mapping]] = ...) -> None: ...
|
|
@@ -35,6 +35,11 @@ class SchedulerServiceStub(object):
|
|
|
35
35
|
request_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsRequest.SerializeToString,
|
|
36
36
|
response_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsResponse.FromString,
|
|
37
37
|
)
|
|
38
|
+
self.CancelScheduledResolverRun = channel.unary_unary(
|
|
39
|
+
"/chalk.server.v1.SchedulerService/CancelScheduledResolverRun",
|
|
40
|
+
request_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.CancelScheduledResolverRunRequest.SerializeToString,
|
|
41
|
+
response_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.CancelScheduledResolverRunResponse.FromString,
|
|
42
|
+
)
|
|
38
43
|
|
|
39
44
|
|
|
40
45
|
class SchedulerServiceServicer(object):
|
|
@@ -64,6 +69,12 @@ class SchedulerServiceServicer(object):
|
|
|
64
69
|
context.set_details("Method not implemented!")
|
|
65
70
|
raise NotImplementedError("Method not implemented!")
|
|
66
71
|
|
|
72
|
+
def CancelScheduledResolverRun(self, request, context):
|
|
73
|
+
"""Missing associated documentation comment in .proto file."""
|
|
74
|
+
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
|
|
75
|
+
context.set_details("Method not implemented!")
|
|
76
|
+
raise NotImplementedError("Method not implemented!")
|
|
77
|
+
|
|
67
78
|
|
|
68
79
|
def add_SchedulerServiceServicer_to_server(servicer, server):
|
|
69
80
|
rpc_method_handlers = {
|
|
@@ -87,6 +98,11 @@ def add_SchedulerServiceServicer_to_server(servicer, server):
|
|
|
87
98
|
request_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsRequest.FromString,
|
|
88
99
|
response_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsResponse.SerializeToString,
|
|
89
100
|
),
|
|
101
|
+
"CancelScheduledResolverRun": grpc.unary_unary_rpc_method_handler(
|
|
102
|
+
servicer.CancelScheduledResolverRun,
|
|
103
|
+
request_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.CancelScheduledResolverRunRequest.FromString,
|
|
104
|
+
response_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.CancelScheduledResolverRunResponse.SerializeToString,
|
|
105
|
+
),
|
|
90
106
|
}
|
|
91
107
|
generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.SchedulerService", rpc_method_handlers)
|
|
92
108
|
server.add_generic_rpc_handlers((generic_handler,))
|
|
@@ -211,3 +227,32 @@ class SchedulerService(object):
|
|
|
211
227
|
timeout,
|
|
212
228
|
metadata,
|
|
213
229
|
)
|
|
230
|
+
|
|
231
|
+
@staticmethod
|
|
232
|
+
def CancelScheduledResolverRun(
|
|
233
|
+
request,
|
|
234
|
+
target,
|
|
235
|
+
options=(),
|
|
236
|
+
channel_credentials=None,
|
|
237
|
+
call_credentials=None,
|
|
238
|
+
insecure=False,
|
|
239
|
+
compression=None,
|
|
240
|
+
wait_for_ready=None,
|
|
241
|
+
timeout=None,
|
|
242
|
+
metadata=None,
|
|
243
|
+
):
|
|
244
|
+
return grpc.experimental.unary_unary(
|
|
245
|
+
request,
|
|
246
|
+
target,
|
|
247
|
+
"/chalk.server.v1.SchedulerService/CancelScheduledResolverRun",
|
|
248
|
+
chalk_dot_server_dot_v1_dot_scheduler__pb2.CancelScheduledResolverRunRequest.SerializeToString,
|
|
249
|
+
chalk_dot_server_dot_v1_dot_scheduler__pb2.CancelScheduledResolverRunResponse.FromString,
|
|
250
|
+
options,
|
|
251
|
+
channel_credentials,
|
|
252
|
+
insecure,
|
|
253
|
+
call_credentials,
|
|
254
|
+
compression,
|
|
255
|
+
wait_for_ready,
|
|
256
|
+
timeout,
|
|
257
|
+
metadata,
|
|
258
|
+
)
|
|
@@ -8,6 +8,8 @@ from abc import (
|
|
|
8
8
|
abstractmethod,
|
|
9
9
|
)
|
|
10
10
|
from chalk._gen.chalk.server.v1.scheduler_pb2 import (
|
|
11
|
+
CancelScheduledResolverRunRequest,
|
|
12
|
+
CancelScheduledResolverRunResponse,
|
|
11
13
|
GetScheduledResolverRunRequest,
|
|
12
14
|
GetScheduledResolverRunResponse,
|
|
13
15
|
ListScheduledResolverRunsRequest,
|
|
@@ -42,6 +44,10 @@ class SchedulerServiceStub:
|
|
|
42
44
|
ListScheduledResolverRunsRequest,
|
|
43
45
|
ListScheduledResolverRunsResponse,
|
|
44
46
|
]
|
|
47
|
+
CancelScheduledResolverRun: UnaryUnaryMultiCallable[
|
|
48
|
+
CancelScheduledResolverRunRequest,
|
|
49
|
+
CancelScheduledResolverRunResponse,
|
|
50
|
+
]
|
|
45
51
|
|
|
46
52
|
class SchedulerServiceServicer(metaclass=ABCMeta):
|
|
47
53
|
@abstractmethod
|
|
@@ -68,5 +74,11 @@ class SchedulerServiceServicer(metaclass=ABCMeta):
|
|
|
68
74
|
request: ListScheduledResolverRunsRequest,
|
|
69
75
|
context: ServicerContext,
|
|
70
76
|
) -> ListScheduledResolverRunsResponse: ...
|
|
77
|
+
@abstractmethod
|
|
78
|
+
def CancelScheduledResolverRun(
|
|
79
|
+
self,
|
|
80
|
+
request: CancelScheduledResolverRunRequest,
|
|
81
|
+
context: ServicerContext,
|
|
82
|
+
) -> CancelScheduledResolverRunResponse: ...
|
|
71
83
|
|
|
72
84
|
def add_SchedulerServiceServicer_to_server(servicer: SchedulerServiceServicer, server: Server) -> None: ...
|
|
@@ -18,7 +18,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
|
|
|
18
18
|
|
|
19
19
|
|
|
20
20
|
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
|
|
21
|
-
b'\n!chalk/server/v1/sql_queries.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xe9\x06\n\x08SqlQuery\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1e\n\x08\x61gent_id\x18\x02 \x01(\tH\x00R\x07\x61gentId\x88\x01\x01\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x01R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x1d\n\nquery_text\x18\x06 \x01(\tR\tqueryText\x12\x1f\n\x08\x64uration\x18\x07 \x01(\x01H\x02R\x08\x64uration\x88\x01\x01\x12\x16\n\x06status\x18\x08 \x01(\x05R\x06status\x12*\n\x0equery_metadata\x18\t \x01(\tH\x03R\rqueryMetadata\x88\x01\x01\x12\x1e\n\x08trace_id\x18\n \x01(\tH\x04R\x07traceId\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x0b \x01(\tH\x05R\nbranchName\x88\x01\x01\x12+\n\x0fhas_plan_stages\x18\x0c \x01(\x08H\x06R\rhasPlanStages\x88\x01\x01\x12$\n\x0b\x65xternal_id\x18\r \x01(\tH\x07R\nexternalId\x88\x01\x01\x12*\n\x0e\x63orrelation_id\x18\x0e \x01(\tH\x08R\rcorrelationId\x88\x01\x01\x12*\n\x0eresource_group\x18\x0f \x01(\tH\tR\rresourceGroup\x88\x01\x01\x12+\n\x0fquery_plan_json\x18\x10 \x01(\tH\nR\rqueryPlanJson\x88\x01\x01\x12/\n\x11output_uri_prefix\x18\x11 \x01(\tH\x0bR\x0foutputUriPrefix\x88\x01\x01\x42\x0b\n\t_agent_idB\x10\n\x0e_deployment_idB\x0b\n\t_durationB\x11\n\x0f_query_metadataB\x0b\n\t_trace_idB\x0e\n\x0c_branch_nameB\x12\n\x10_has_plan_stagesB\x0e\n\x0c_external_idB\x11\n\x0f_correlation_idB\x11\n\x0f_resource_groupB\x12\n\x10_query_plan_jsonB\x14\n\x12_output_uri_prefix"\
|
|
21
|
+
b'\n!chalk/server/v1/sql_queries.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xe9\x06\n\x08SqlQuery\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1e\n\x08\x61gent_id\x18\x02 \x01(\tH\x00R\x07\x61gentId\x88\x01\x01\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x01R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x1d\n\nquery_text\x18\x06 \x01(\tR\tqueryText\x12\x1f\n\x08\x64uration\x18\x07 \x01(\x01H\x02R\x08\x64uration\x88\x01\x01\x12\x16\n\x06status\x18\x08 \x01(\x05R\x06status\x12*\n\x0equery_metadata\x18\t \x01(\tH\x03R\rqueryMetadata\x88\x01\x01\x12\x1e\n\x08trace_id\x18\n \x01(\tH\x04R\x07traceId\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x0b \x01(\tH\x05R\nbranchName\x88\x01\x01\x12+\n\x0fhas_plan_stages\x18\x0c \x01(\x08H\x06R\rhasPlanStages\x88\x01\x01\x12$\n\x0b\x65xternal_id\x18\r \x01(\tH\x07R\nexternalId\x88\x01\x01\x12*\n\x0e\x63orrelation_id\x18\x0e \x01(\tH\x08R\rcorrelationId\x88\x01\x01\x12*\n\x0eresource_group\x18\x0f \x01(\tH\tR\rresourceGroup\x88\x01\x01\x12+\n\x0fquery_plan_json\x18\x10 \x01(\tH\nR\rqueryPlanJson\x88\x01\x01\x12/\n\x11output_uri_prefix\x18\x11 \x01(\tH\x0bR\x0foutputUriPrefix\x88\x01\x01\x42\x0b\n\t_agent_idB\x10\n\x0e_deployment_idB\x0b\n\t_durationB\x11\n\x0f_query_metadataB\x0b\n\t_trace_idB\x0e\n\x0c_branch_nameB\x12\n\x10_has_plan_stagesB\x0e\n\x0c_external_idB\x11\n\x0f_correlation_idB\x11\n\x0f_resource_groupB\x12\n\x10_query_plan_jsonB\x14\n\x12_output_uri_prefix"\x8d\x04\n\x15ListSqlQueriesRequest\x12\x14\n\x05limit\x18\x01 \x01(\x05R\x05limit\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x1e\n\x08\x61gent_id\x18\x03 \x01(\tH\x01R\x07\x61gentId\x88\x01\x01\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x02R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1b\n\x06status\x18\x05 \x01(\x05H\x03R\x06status\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x06 \x01(\tH\x04R\nbranchName\x88\x01\x01\x12$\n\x0b\x65xternal_id\x18\x07 \x01(\tH\x05R\nexternalId\x88\x01\x01\x12*\n\x0e\x63orrelation_id\x18\x08 \x01(\tH\x06R\rcorrelationId\x88\x01\x01\x12\x35\n\x05start\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x07R\x05start\x88\x01\x01\x12\x31\n\x03\x65nd\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x08R\x03\x65nd\x88\x01\x01\x42\t\n\x07_cursorB\x0b\n\t_agent_idB\x10\n\x0e_deployment_idB\t\n\x07_statusB\x0e\n\x0c_branch_nameB\x0e\n\x0c_external_idB\x11\n\x0f_correlation_idB\x08\n\x06_startB\x06\n\x04_end"\x83\x01\n\x16ListSqlQueriesResponse\x12\x33\n\x07queries\x18\x01 \x03(\x0b\x32\x19.chalk.server.v1.SqlQueryR\x07queries\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"$\n\x12GetSqlQueryRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"F\n\x13GetSqlQueryResponse\x12/\n\x05query\x18\x01 \x01(\x0b\x32\x19.chalk.server.v1.SqlQueryR\x05query2\xe0\x01\n\x11SqlQueriesService\x12i\n\x0eListSqlQueries\x12&.chalk.server.v1.ListSqlQueriesRequest\x1a\'.chalk.server.v1.ListSqlQueriesResponse"\x06\x90\x02\x01\x80}\x03\x12`\n\x0bGetSqlQuery\x12#.chalk.server.v1.GetSqlQueryRequest\x1a$.chalk.server.v1.GetSqlQueryResponse"\x06\x90\x02\x01\x80}\x03\x42\x98\x01\n\x13\x63om.chalk.server.v1B\x0fSqlQueriesProtoP\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
22
|
)
|
|
23
23
|
|
|
24
24
|
_globals = globals()
|
|
@@ -36,13 +36,13 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
36
36
|
_globals["_SQLQUERY"]._serialized_start = 121
|
|
37
37
|
_globals["_SQLQUERY"]._serialized_end = 994
|
|
38
38
|
_globals["_LISTSQLQUERIESREQUEST"]._serialized_start = 997
|
|
39
|
-
_globals["_LISTSQLQUERIESREQUEST"]._serialized_end =
|
|
40
|
-
_globals["_LISTSQLQUERIESRESPONSE"]._serialized_start =
|
|
41
|
-
_globals["_LISTSQLQUERIESRESPONSE"]._serialized_end =
|
|
42
|
-
_globals["_GETSQLQUERYREQUEST"]._serialized_start =
|
|
43
|
-
_globals["_GETSQLQUERYREQUEST"]._serialized_end =
|
|
44
|
-
_globals["_GETSQLQUERYRESPONSE"]._serialized_start =
|
|
45
|
-
_globals["_GETSQLQUERYRESPONSE"]._serialized_end =
|
|
46
|
-
_globals["_SQLQUERIESSERVICE"]._serialized_start =
|
|
47
|
-
_globals["_SQLQUERIESSERVICE"]._serialized_end =
|
|
39
|
+
_globals["_LISTSQLQUERIESREQUEST"]._serialized_end = 1522
|
|
40
|
+
_globals["_LISTSQLQUERIESRESPONSE"]._serialized_start = 1525
|
|
41
|
+
_globals["_LISTSQLQUERIESRESPONSE"]._serialized_end = 1656
|
|
42
|
+
_globals["_GETSQLQUERYREQUEST"]._serialized_start = 1658
|
|
43
|
+
_globals["_GETSQLQUERYREQUEST"]._serialized_end = 1694
|
|
44
|
+
_globals["_GETSQLQUERYRESPONSE"]._serialized_start = 1696
|
|
45
|
+
_globals["_GETSQLQUERYRESPONSE"]._serialized_end = 1766
|
|
46
|
+
_globals["_SQLQUERIESSERVICE"]._serialized_start = 1769
|
|
47
|
+
_globals["_SQLQUERIESSERVICE"]._serialized_end = 1993
|
|
48
48
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -98,6 +98,8 @@ class ListSqlQueriesRequest(_message.Message):
|
|
|
98
98
|
"branch_name",
|
|
99
99
|
"external_id",
|
|
100
100
|
"correlation_id",
|
|
101
|
+
"start",
|
|
102
|
+
"end",
|
|
101
103
|
)
|
|
102
104
|
LIMIT_FIELD_NUMBER: _ClassVar[int]
|
|
103
105
|
CURSOR_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -107,6 +109,8 @@ class ListSqlQueriesRequest(_message.Message):
|
|
|
107
109
|
BRANCH_NAME_FIELD_NUMBER: _ClassVar[int]
|
|
108
110
|
EXTERNAL_ID_FIELD_NUMBER: _ClassVar[int]
|
|
109
111
|
CORRELATION_ID_FIELD_NUMBER: _ClassVar[int]
|
|
112
|
+
START_FIELD_NUMBER: _ClassVar[int]
|
|
113
|
+
END_FIELD_NUMBER: _ClassVar[int]
|
|
110
114
|
limit: int
|
|
111
115
|
cursor: str
|
|
112
116
|
agent_id: str
|
|
@@ -115,6 +119,8 @@ class ListSqlQueriesRequest(_message.Message):
|
|
|
115
119
|
branch_name: str
|
|
116
120
|
external_id: str
|
|
117
121
|
correlation_id: str
|
|
122
|
+
start: _timestamp_pb2.Timestamp
|
|
123
|
+
end: _timestamp_pb2.Timestamp
|
|
118
124
|
def __init__(
|
|
119
125
|
self,
|
|
120
126
|
limit: _Optional[int] = ...,
|
|
@@ -125,6 +131,8 @@ class ListSqlQueriesRequest(_message.Message):
|
|
|
125
131
|
branch_name: _Optional[str] = ...,
|
|
126
132
|
external_id: _Optional[str] = ...,
|
|
127
133
|
correlation_id: _Optional[str] = ...,
|
|
134
|
+
start: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
|
|
135
|
+
end: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
|
|
128
136
|
) -> None: ...
|
|
129
137
|
|
|
130
138
|
class ListSqlQueriesResponse(_message.Message):
|
chalk/_version.py
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
__version__ = "2.95.
|
|
1
|
+
__version__ = "2.95.7"
|
chalk/client/client_grpc.py
CHANGED
|
@@ -30,6 +30,7 @@ from chalk._gen.chalk.graph.v1.graph_pb2 import Graph
|
|
|
30
30
|
from chalk._gen.chalk.models.v1 import model_artifact_pb2 as _model_artifact_pb2
|
|
31
31
|
from chalk._gen.chalk.protosql.v1.sql_service_pb2 import (
|
|
32
32
|
ExecuteSqlQueryRequest,
|
|
33
|
+
ExecuteSqlResultPersistenceSettings,
|
|
33
34
|
GetDbCatalogsRequest,
|
|
34
35
|
GetDbSchemasRequest,
|
|
35
36
|
GetTablesRequest,
|
|
@@ -1453,8 +1454,15 @@ class ChalkGRPCClient:
|
|
|
1453
1454
|
"""
|
|
1454
1455
|
return self._stub_refresher.call_team_stub(lambda x: x.ListServiceTokens(ListServiceTokensRequest()))
|
|
1455
1456
|
|
|
1456
|
-
def run_sql(self, sql: str):
|
|
1457
|
-
|
|
1457
|
+
def run_sql(self, sql: str, persistence_settings: Optional[Dict[str, Any]] = None):
|
|
1458
|
+
request = ExecuteSqlQueryRequest(query=sql)
|
|
1459
|
+
|
|
1460
|
+
if persistence_settings is not None:
|
|
1461
|
+
# Convert dict to proto message
|
|
1462
|
+
enabled = persistence_settings.get("enabled", False)
|
|
1463
|
+
request.persistence_settings.CopyFrom(ExecuteSqlResultPersistenceSettings(enabled=enabled))
|
|
1464
|
+
|
|
1465
|
+
return self._stub_refresher.call_sql_stub(lambda x: x.ExecuteSqlQuery(request))
|
|
1458
1466
|
|
|
1459
1467
|
def explain_sql(self, sql: str):
|
|
1460
1468
|
return self._stub_refresher.call_sql_stub(lambda x: x.PlanSqlQuery(PlanSqlQueryRequest(query=sql)))
|
chalk/client/client_impl.py
CHANGED
|
@@ -242,13 +242,15 @@ def _do_query_errors_match(
|
|
|
242
242
|
def _do_resultsets_match(
|
|
243
243
|
actual: Sequence[Any],
|
|
244
244
|
expected: Sequence[Any],
|
|
245
|
+
float_rel_tolerance: float = 1e-6,
|
|
246
|
+
float_abs_tolerance: float = 1e-12,
|
|
245
247
|
) -> bool:
|
|
246
248
|
if len(actual) != len(expected):
|
|
247
249
|
return False
|
|
248
250
|
for a in actual:
|
|
249
251
|
matched = False
|
|
250
252
|
for i, e in enumerate(expected):
|
|
251
|
-
if _does_result_match(a, e):
|
|
253
|
+
if _does_result_match(a, e, float_rel_tolerance, float_abs_tolerance):
|
|
252
254
|
matched = True
|
|
253
255
|
expected = [x for j, x in enumerate(expected) if j != i]
|
|
254
256
|
break
|
|
@@ -5011,7 +5013,7 @@ https://docs.chalk.ai/cli/apply
|
|
|
5011
5013
|
if x.field in expected_features # Filter to only asserted features
|
|
5012
5014
|
]
|
|
5013
5015
|
|
|
5014
|
-
feature_mismatch = not _do_resultsets_match(actuals, expected)
|
|
5016
|
+
feature_mismatch = not _do_resultsets_match(actuals, expected, float_rel_tolerance, float_abs_tolerance)
|
|
5015
5017
|
if feature_mismatch or show_table:
|
|
5016
5018
|
no_expectations = len(expected) == 0
|
|
5017
5019
|
|
|
@@ -5022,7 +5024,7 @@ https://docs.chalk.ai/cli/apply
|
|
|
5022
5024
|
safe = []
|
|
5023
5025
|
|
|
5024
5026
|
for x in actuals:
|
|
5025
|
-
if any(_does_result_match(x, e) for e in expected):
|
|
5027
|
+
if any(_does_result_match(x, e, float_rel_tolerance, float_abs_tolerance) for e in expected):
|
|
5026
5028
|
safe.append(x.fqn) # use fqns, since __eq__ might behave weirdly for feature values
|
|
5027
5029
|
|
|
5028
5030
|
include_errors = any(k.error for k in expected) or any(k.error for k in actuals)
|
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
|
|
2
|
-
chalk/_version.py,sha256=
|
|
2
|
+
chalk/_version.py,sha256=ts6BmSqe_eJx96sHwhbYb1Lg9DwpOIgdQB6kAgplmXE,23
|
|
3
3
|
chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
|
|
4
4
|
chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
|
|
5
5
|
chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
|
|
@@ -292,8 +292,8 @@ chalk/_gen/chalk/nodepools/v1/karpenter_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rK
|
|
|
292
292
|
chalk/_gen/chalk/nodepools/v1/karpenter_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
|
|
293
293
|
chalk/_gen/chalk/protosql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
294
294
|
chalk/_gen/chalk/protosql/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
295
|
-
chalk/_gen/chalk/protosql/v1/sql_service_pb2.py,sha256=
|
|
296
|
-
chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi,sha256=
|
|
295
|
+
chalk/_gen/chalk/protosql/v1/sql_service_pb2.py,sha256=QmLxn2MTHInL-38LtSMp9xEsOxgUiyF-iDH-Vw0n0i0,12403
|
|
296
|
+
chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi,sha256=JGujyZ3zs-_Yddcny7AkNSe-hjvzKl369dzfzNj2W_k,11996
|
|
297
297
|
chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py,sha256=GnW9QXWQSzv96xWRo9lFoSRhyvfNgwlR8tjVbdQRNbw,11923
|
|
298
298
|
chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi,sha256=qoW1zPv6LdxqATJb4jU_4JvIsSG-svO6G1UdqoaGvhg,2589
|
|
299
299
|
chalk/_gen/chalk/python/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
@@ -320,8 +320,8 @@ chalk/_gen/chalk/server/v1/batch_pb2.py,sha256=1MYeWKT95ITKIv6ixORefTcOJ1Ij6Ar8S
|
|
|
320
320
|
chalk/_gen/chalk/server/v1/batch_pb2.pyi,sha256=yvywnYnjW5WaOVL395NNTIuBoYQkcIHD9mVGKt2cwDA,4975
|
|
321
321
|
chalk/_gen/chalk/server/v1/batch_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
|
|
322
322
|
chalk/_gen/chalk/server/v1/batch_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
|
|
323
|
-
chalk/_gen/chalk/server/v1/benchmark_pb2.py,sha256=
|
|
324
|
-
chalk/_gen/chalk/server/v1/benchmark_pb2.pyi,sha256=
|
|
323
|
+
chalk/_gen/chalk/server/v1/benchmark_pb2.py,sha256=CEHh2oJ8xSZhga-cdjxQwnK-e46dmyLbYj2tnycFtJc,9581
|
|
324
|
+
chalk/_gen/chalk/server/v1/benchmark_pb2.pyi,sha256=wTLgWMEHc_fLTHRfmzy0_E7x2q5qX9aIZUkealrZR-o,8134
|
|
325
325
|
chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py,sha256=aLCoZyDVbQ_bxmGiFbCsgUsATE5sAeV9x_qPSXEB0UU,10503
|
|
326
326
|
chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi,sha256=avlT5EFmgySdsTML_QmMh841_eRroMa5s9756KBrcco,2449
|
|
327
327
|
chalk/_gen/chalk/server/v1/billing_pb2.py,sha256=tQWnCYpgPXpEIU8oUD-PimSuaft_PFx3xPT0D7cInvc,12221
|
|
@@ -334,9 +334,10 @@ chalk/_gen/chalk/server/v1/bootstrap_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPY
|
|
|
334
334
|
chalk/_gen/chalk/server/v1/bootstrap_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
|
|
335
335
|
chalk/_gen/chalk/server/v1/branches_pb2.py,sha256=IXvRzrb8kxu1tEKid_Q8vvru-7lBZYcHjqKej4vxxnY,4344
|
|
336
336
|
chalk/_gen/chalk/server/v1/branches_pb2.pyi,sha256=UUimesKEKh7VP7iemsSM2AYPmKCb-eCPXSvFDhaQ-cU,3217
|
|
337
|
+
chalk/_gen/chalk/server/v1/branches_pb2_grpc.py,sha256=kLHMoEIOFFHZMUvVqqRFJbsi1fM7AB7e-kPrMSKsBMU,3026
|
|
337
338
|
chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi,sha256=q_wOuj-HAtT59IaxGcbTxP1eglkjoQCewqXCuzic7_o,988
|
|
338
|
-
chalk/_gen/chalk/server/v1/builder_pb2.py,sha256=
|
|
339
|
-
chalk/_gen/chalk/server/v1/builder_pb2.pyi,sha256=
|
|
339
|
+
chalk/_gen/chalk/server/v1/builder_pb2.py,sha256=G0E-4uJI5JrldYa_wOsRdqA8vRnV2irJ4GnhKZ-F1Ho,97628
|
|
340
|
+
chalk/_gen/chalk/server/v1/builder_pb2.pyi,sha256=YhueGpkT8Sct08NRk17ifVYnZ2cpSf4Qq9vepOTn3eI,90276
|
|
340
341
|
chalk/_gen/chalk/server/v1/builder_pb2_grpc.py,sha256=0SrCdnH91t9969dcvihrM6I2-U7C7dy4yHScFDUmB6U,106516
|
|
341
342
|
chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi,sha256=N3eGK8l_FwcY_O4etT0Sorfp6tI416I0F2Y2JaAU9Po,24277
|
|
342
343
|
chalk/_gen/chalk/server/v1/chart_pb2.py,sha256=YJD-XW0lR_u-nGWTrjKKE5ZRDiXubsk7Va7DMFeNHDg,11764
|
|
@@ -455,10 +456,10 @@ chalk/_gen/chalk/server/v1/model_registry_pb2.py,sha256=c9kDe2xfiJEHXRT0ddm1oCl5
|
|
|
455
456
|
chalk/_gen/chalk/server/v1/model_registry_pb2.pyi,sha256=PdyuvcjZsM_x2kXkodPHkpJwMVROWq1SnUNOs0r5LuM,23841
|
|
456
457
|
chalk/_gen/chalk/server/v1/model_registry_pb2_grpc.py,sha256=JXGw1LWWDQnrpWDWcBMxwqHscxACfiEMKPe0DeEERgA,25781
|
|
457
458
|
chalk/_gen/chalk/server/v1/model_registry_pb2_grpc.pyi,sha256=_diKNW3Atycbmr0nWo33wVCGRrEEbc3RFCySlHswVdU,5230
|
|
458
|
-
chalk/_gen/chalk/server/v1/monitoring_pb2.py,sha256=
|
|
459
|
-
chalk/_gen/chalk/server/v1/monitoring_pb2.pyi,sha256
|
|
460
|
-
chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py,sha256=
|
|
461
|
-
chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi,sha256=
|
|
459
|
+
chalk/_gen/chalk/server/v1/monitoring_pb2.py,sha256=6N-4AAGC7dZdyNVk2MPdXrK-OdNTItGsx7zwSQT_yek,26721
|
|
460
|
+
chalk/_gen/chalk/server/v1/monitoring_pb2.pyi,sha256=vE8FBZYxSVYTB0TmJrAA5vnjF4WYK8ga7DRYY4HzN44,18625
|
|
461
|
+
chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.py,sha256=DX6HbDO1dy8eEuxVRKlYL_ymsTQIXoW1vWPeMee0UsQ,36169
|
|
462
|
+
chalk/_gen/chalk/server/v1/monitoring_pb2_grpc.pyi,sha256=LQj-vgn0FRUMMTzQ7Nk1U5t21wwiOCLzFeG2vkAaG5I,7984
|
|
462
463
|
chalk/_gen/chalk/server/v1/named_query_pb2.py,sha256=zI_GWCj0C9re3N4-rJwiqlEeRkJJ0AbwDRKolym0c1A,4563
|
|
463
464
|
chalk/_gen/chalk/server/v1/named_query_pb2.pyi,sha256=InM5LShJ1PENmIqn63eC8Oa5JUsnVv6jtrnnBgbyLx8,2078
|
|
464
465
|
chalk/_gen/chalk/server/v1/named_query_pb2_grpc.py,sha256=YAm4u0fgfG94MvophF22rNh3tdrSE0z441vbfhmmBCo,6927
|
|
@@ -499,10 +500,10 @@ chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=WJ-nWBvaVoAUy57jZ1k
|
|
|
499
500
|
chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=ePknNWz-VY73vA60QqzyX6ZgimWyJQ6rO_f_DUlNziY,6590
|
|
500
501
|
chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
|
|
501
502
|
chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
|
|
502
|
-
chalk/_gen/chalk/server/v1/scheduler_pb2.py,sha256=
|
|
503
|
-
chalk/_gen/chalk/server/v1/scheduler_pb2.pyi,sha256=
|
|
504
|
-
chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py,sha256=
|
|
505
|
-
chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi,sha256=
|
|
503
|
+
chalk/_gen/chalk/server/v1/scheduler_pb2.py,sha256=fgpRAivQ1ut2cziY-RB9_3fWGOTPxnR8dsdUMwswbTs,11469
|
|
504
|
+
chalk/_gen/chalk/server/v1/scheduler_pb2.pyi,sha256=z1xI9hRF10ZtXmKVewMN-0AnhrUn4OEDQb0YAGTqQ7w,9670
|
|
505
|
+
chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py,sha256=IbohnpM4-Zh4gMb91cxaYcJYH3wsHm8zmtcoJLOt6Jw,10854
|
|
506
|
+
chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi,sha256=kOuptMT4kNP-wZMwo_pUwt6sR3wKqieFeIYRav-1dCU,2665
|
|
506
507
|
chalk/_gen/chalk/server/v1/script_tasks_pb2.py,sha256=pptzSNfGIkr_BOK6EOV_aFJKkCyqd1twzIMGX09SXwg,9050
|
|
507
508
|
chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi,sha256=jaDE1Li64g_tkH2GKIH7n9itfrEiAzSt_RMa0JYsGr8,6486
|
|
508
509
|
chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py,sha256=7P9KY6p2hZL98tKDgFw6-VGYVMSRUSU0Cce0-2cZeOw,12247
|
|
@@ -511,8 +512,8 @@ chalk/_gen/chalk/server/v1/sql_interface_pb2.py,sha256=w2OSrwqhGIK9PAIx9VbXpO812
|
|
|
511
512
|
chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi,sha256=pT4cDJ0BD23FGcIVCvug1DUkOgGfxotzHFRVqSgPLHw,5342
|
|
512
513
|
chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py,sha256=MajKTaBgNlPsDar-Lb--vjDMOwbS-v9SX0QFIc4nyaA,14300
|
|
513
514
|
chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi,sha256=mjySGPdaPhrntQ7FUmrE6JnfG2G0NtBEGYbGbtlwZSs,3390
|
|
514
|
-
chalk/_gen/chalk/server/v1/sql_queries_pb2.py,sha256=
|
|
515
|
-
chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi,sha256=
|
|
515
|
+
chalk/_gen/chalk/server/v1/sql_queries_pb2.py,sha256=DuMtIZikOpIRipasG4xRMBZ0WHb1vtKnpZ_TUXeycd8,5995
|
|
516
|
+
chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi,sha256=HPH921z7g3lavuKY5MC-vdD8B89xEfU7qHNeb4jUTiw,5175
|
|
516
517
|
chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py,sha256=EO7DRoB6i-B7Q501MMKse4USOO2PennuCHYCD7VeOVQ,4632
|
|
517
518
|
chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi,sha256=cpPeZ2umyexiAVfBUZubsRBzkgtNQLZP2PZ4Bd0Lwfs,1381
|
|
518
519
|
chalk/_gen/chalk/server/v1/status_pb2.py,sha256=Bw7uR28OnLRNFn6lATt99VQG4ZXDTHPd_KLS8iIA9FI,6053
|
|
@@ -606,8 +607,8 @@ chalk/client/__init__.py,sha256=wu3WQVzE5gRj6noQQDOdYJUgDaz_9QtbjXH4KuoIlXQ,1782
|
|
|
606
607
|
chalk/client/client.py,sha256=fqw75x4yiAk3CXryGbuF_qg09WDIlluhYeFdQ-BfTG8,103337
|
|
607
608
|
chalk/client/client_async.py,sha256=nFFTWJbdBlb7zksyjOMBY566tZTAyNXQhCnq06LHWl0,50803
|
|
608
609
|
chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
|
|
609
|
-
chalk/client/client_grpc.py,sha256=
|
|
610
|
-
chalk/client/client_impl.py,sha256=
|
|
610
|
+
chalk/client/client_grpc.py,sha256=rCjKVe6uMNfKjat_Fng3SRUoMaut8Vz5QRw02v_d9xc,106699
|
|
611
|
+
chalk/client/client_impl.py,sha256=rvT02yriRo7_WFqbUsHXRjm6DbRyXzw2QjcLwKaW10E,211844
|
|
611
612
|
chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
|
|
612
613
|
chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
|
|
613
614
|
chalk/client/models.py,sha256=FDgEqIeBJq25T9AH38jdWkeyRLiAXfLXfXHpomFJM08,63923
|
|
@@ -818,8 +819,8 @@ chalk/utils/tracing.py,sha256=qAyMRdFBkL4Q2_Bn-C31atDfJirUtBvcSjkYC2R51sM,13005
|
|
|
818
819
|
chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
|
|
819
820
|
chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
820
821
|
chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
|
|
821
|
-
chalkpy-2.95.
|
|
822
|
-
chalkpy-2.95.
|
|
823
|
-
chalkpy-2.95.
|
|
824
|
-
chalkpy-2.95.
|
|
825
|
-
chalkpy-2.95.
|
|
822
|
+
chalkpy-2.95.7.dist-info/METADATA,sha256=ukzrm_YJQA8yEk9UpMNyPC0LzQQDjtjXVTKWQ3hjTzw,27494
|
|
823
|
+
chalkpy-2.95.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
824
|
+
chalkpy-2.95.7.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
|
|
825
|
+
chalkpy-2.95.7.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
|
|
826
|
+
chalkpy-2.95.7.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|