chalkpy 2.94.5__py3-none-any.whl → 2.94.7__py3-none-any.whl

chalk/_gen/chalk/server/v1/scheduler_pb2.py

@@ -21,7 +21,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1f\x63halk/server/v1/scheduler.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1b\x63halk/server/v1/batch.proto\x1a)chalk/server/v1/scheduled_query_run.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xfe\x05\n\x0f\x43ronResolverRun\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12!\n\x0cresolver_fqn\x18\x03 \x01(\tR\x0bresolverFqn\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12+\n\x11schedule_readable\x18\x05 \x01(\tR\x10scheduleReadable\x12\x36\n\x17schedule_resolver_value\x18\x06 \x01(\tR\x15scheduleResolverValue\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12(\n\rdeployment_id\x18\x08 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x31\n\x03\x65nd\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x03\x65nd\x88\x01\x01\x12:\n\x05\x62\x61tch\x18\n \x01(\x0b\x32\x1f.chalk.server.v1.BatchOperationH\x02R\x05\x62\x61tch\x88\x01\x01\x12\x46\n\x0ctrigger_kind\x18\x0b \x01(\x0e\x32#.chalk.server.v1.CronRunTriggerKindR\x0btriggerKind\x12@\n\x0blower_bound\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\nupperBound\x88\x01\x01\x12$\n\x0bmax_samples\x18\x0e \x01(\x03H\x05R\nmaxSamples\x88\x01\x01\x42\x10\n\x0e_deployment_idB\x06\n\x04_endB\x08\n\x06_batchB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x0e\n\x0c_max_samples"\x9d\x03\n ManualTriggerCronResolverRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12$\n\x0bmax_samples\x18\x02 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12@\n\x0blower_bound\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nupperBound\x88\x01\x01\x12\x30\n\x11timestamping_mode\x18\x06 \x01(\tH\x03R\x10timestampingMode\x88\x01\x01\x12$\n\x0bjob_options\x18\x07 \x01(\tH\x04R\njobOptions\x88\x01\x01\x42\x0e\n\x0c_max_samplesB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x14\n\x12_timestamping_modeB\x0e\n\x0c_job_options"q\n!ManualTriggerCronResolverResponse\x12L\n\x11\x63ron_resolver_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x0f\x63ronResolverRun"\xf0\x04\n"ManualTriggerScheduledQueryRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x03R\x0b\x63ronQueryId\x12p\n\x0fplanner_options\x18\x02 \x03(\x0b\x32G.chalk.server.v1.ManualTriggerScheduledQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\x33\n\x15incremental_resolvers\x18\x03 \x03(\tR\x14incrementalResolvers\x12$\n\x0bmax_samples\x18\x04 \x01(\x03H\x01R\nmaxSamples\x88\x01\x01\x12j\n\renv_overrides\x18\x05 \x03(\x0b\x32\x45.chalk.server.v1.ManualTriggerScheduledQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12(\n\x0f\x63ron_query_name\x18\x06 \x01(\tH\x00R\rcronQueryName\x1aY\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x17\n\x15\x63ron_query_identifierB\x0e\n\x0c_max_samples"y\n#ManualTriggerScheduledQueryResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun*\x9c\x01\n\x12\x43ronRunTriggerKind\x12%\n!CRON_RUN_TRIGGER_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x43RON_RUN_TRIGGER_KIND_MANUAL\x10\x01\x12\x1d\n\x19\x43RON_RUN_TRIGGER_KIND_API\x10\x02\x12\x1e\n\x1a\x43RON_RUN_TRIGGER_KIND_CRON\x10\x03\x32\xac\x02\n\x10SchedulerService\x12\x87\x01\n\x19ManualTriggerCronResolver\x12\x31.chalk.server.v1.ManualTriggerCronResolverRequest\x1a\x32.chalk.server.v1.ManualTriggerCronResolverResponse"\x03\x80}\x11\x12\x8d\x01\n\x1bManualTriggerScheduledQuery\x12\x33.chalk.server.v1.ManualTriggerScheduledQueryRequest\x1a\x34.chalk.server.v1.ManualTriggerScheduledQueryResponse"\x03\x80}\x11\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0eSchedulerProtoP\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'
+    b'\n\x1f\x63halk/server/v1/scheduler.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1b\x63halk/server/v1/batch.proto\x1a)chalk/server/v1/scheduled_query_run.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xfe\x05\n\x0f\x43ronResolverRun\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12!\n\x0cresolver_fqn\x18\x03 \x01(\tR\x0bresolverFqn\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12+\n\x11schedule_readable\x18\x05 \x01(\tR\x10scheduleReadable\x12\x36\n\x17schedule_resolver_value\x18\x06 \x01(\tR\x15scheduleResolverValue\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12(\n\rdeployment_id\x18\x08 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x31\n\x03\x65nd\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x03\x65nd\x88\x01\x01\x12:\n\x05\x62\x61tch\x18\n \x01(\x0b\x32\x1f.chalk.server.v1.BatchOperationH\x02R\x05\x62\x61tch\x88\x01\x01\x12\x46\n\x0ctrigger_kind\x18\x0b \x01(\x0e\x32#.chalk.server.v1.CronRunTriggerKindR\x0btriggerKind\x12@\n\x0blower_bound\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\nupperBound\x88\x01\x01\x12$\n\x0bmax_samples\x18\x0e \x01(\x03H\x05R\nmaxSamples\x88\x01\x01\x42\x10\n\x0e_deployment_idB\x06\n\x04_endB\x08\n\x06_batchB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x0e\n\x0c_max_samples"\x9d\x03\n ManualTriggerCronResolverRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12$\n\x0bmax_samples\x18\x02 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12@\n\x0blower_bound\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nupperBound\x88\x01\x01\x12\x30\n\x11timestamping_mode\x18\x06 \x01(\tH\x03R\x10timestampingMode\x88\x01\x01\x12$\n\x0bjob_options\x18\x07 \x01(\tH\x04R\njobOptions\x88\x01\x01\x42\x0e\n\x0c_max_samplesB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x14\n\x12_timestamping_modeB\x0e\n\x0c_job_options"q\n!ManualTriggerCronResolverResponse\x12L\n\x11\x63ron_resolver_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x0f\x63ronResolverRun"\xf0\x04\n"ManualTriggerScheduledQueryRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x03R\x0b\x63ronQueryId\x12p\n\x0fplanner_options\x18\x02 \x03(\x0b\x32G.chalk.server.v1.ManualTriggerScheduledQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\x33\n\x15incremental_resolvers\x18\x03 \x03(\tR\x14incrementalResolvers\x12$\n\x0bmax_samples\x18\x04 \x01(\x03H\x01R\nmaxSamples\x88\x01\x01\x12j\n\renv_overrides\x18\x05 \x03(\x0b\x32\x45.chalk.server.v1.ManualTriggerScheduledQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12(\n\x0f\x63ron_query_name\x18\x06 \x01(\tH\x00R\rcronQueryName\x1aY\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x17\n\x15\x63ron_query_identifierB\x0e\n\x0c_max_samples"y\n#ManualTriggerScheduledQueryResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun"0\n\x1eGetScheduledResolverRunRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"U\n\x1fGetScheduledResolverRunResponse\x12\x32\n\x03run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x03run"\xa7\x03\n ListScheduledResolverRunsRequest\x12\x37\n\x06\x63ursor\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x06\x63ursor\x88\x01\x01\x12,\n\x0fresolver_filter\x18\x02 \x01(\tH\x01R\x0eresolverFilter\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x12\x35\n\x05start\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\x05start\x88\x01\x01\x12\x31\n\x03\x65nd\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\x03\x65nd\x88\x01\x01\x12J\n\rstatus_filter\x18\x06 \x01(\x0e\x32 .chalk.server.v1.OperationStatusH\x05R\x0cstatusFilter\x88\x01\x01\x42\t\n\x07_cursorB\x12\n\x10_resolver_filterB\x08\n\x06_limitB\x08\n\x06_startB\x06\n\x04_endB\x10\n\x0e_status_filter"Y\n!ListScheduledResolverRunsResponse\x12\x34\n\x04runs\x18\x01 \x03(\x0b\x32 .chalk.server.v1.CronResolverRunR\x04runs*\x9c\x01\n\x12\x43ronRunTriggerKind\x12%\n!CRON_RUN_TRIGGER_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x43RON_RUN_TRIGGER_KIND_MANUAL\x10\x01\x12\x1d\n\x19\x43RON_RUN_TRIGGER_KIND_API\x10\x02\x12\x1e\n\x1a\x43RON_RUN_TRIGGER_KIND_CRON\x10\x03\x32\xba\x04\n\x10SchedulerService\x12\x87\x01\n\x19ManualTriggerCronResolver\x12\x31.chalk.server.v1.ManualTriggerCronResolverRequest\x1a\x32.chalk.server.v1.ManualTriggerCronResolverResponse"\x03\x80}\x11\x12\x8d\x01\n\x1bManualTriggerScheduledQuery\x12\x33.chalk.server.v1.ManualTriggerScheduledQueryRequest\x1a\x34.chalk.server.v1.ManualTriggerScheduledQueryResponse"\x03\x80}\x11\x12\x81\x01\n\x17GetScheduledResolverRun\x12/.chalk.server.v1.GetScheduledResolverRunRequest\x1a\x30.chalk.server.v1.GetScheduledResolverRunResponse"\x03\x80}\x10\x12\x87\x01\n\x19ListScheduledResolverRuns\x12\x31.chalk.server.v1.ListScheduledResolverRunsRequest\x1a\x32.chalk.server.v1.ListScheduledResolverRunsResponse"\x03\x80}\x10\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0eSchedulerProtoP\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'
 )
 
 _globals = globals()
@@ -40,8 +40,12 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerCronResolver"]._serialized_options = b"\200}\021"
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerScheduledQuery"]._options = None
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerScheduledQuery"]._serialized_options = b"\200}\021"
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 2271
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 2427
+    _globals["_SCHEDULERSERVICE"].methods_by_name["GetScheduledResolverRun"]._options = None
+    _globals["_SCHEDULERSERVICE"].methods_by_name["GetScheduledResolverRun"]._serialized_options = b"\200}\020"
+    _globals["_SCHEDULERSERVICE"].methods_by_name["ListScheduledResolverRuns"]._options = None
+    _globals["_SCHEDULERSERVICE"].methods_by_name["ListScheduledResolverRuns"]._serialized_options = b"\200}\020"
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 2925
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 3081
     _globals["_CRONRESOLVERRUN"]._serialized_start = 221
     _globals["_CRONRESOLVERRUN"]._serialized_end = 987
     _globals["_MANUALTRIGGERCRONRESOLVERREQUEST"]._serialized_start = 990
@@ -56,6 +60,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 2104
     _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_start = 2147
     _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_end = 2268
-    _globals["_SCHEDULERSERVICE"]._serialized_start = 2430
-    _globals["_SCHEDULERSERVICE"]._serialized_end = 2730
+    _globals["_GETSCHEDULEDRESOLVERRUNREQUEST"]._serialized_start = 2270
+    _globals["_GETSCHEDULEDRESOLVERRUNREQUEST"]._serialized_end = 2318
+    _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_start = 2320
+    _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_end = 2405
+    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_start = 2408
+    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_end = 2831
+    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_start = 2833
+    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_end = 2922
+    _globals["_SCHEDULERSERVICE"]._serialized_start = 3084
+    _globals["_SCHEDULERSERVICE"]._serialized_end = 3654
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/scheduler_pb2.pyi

@@ -178,3 +178,45 @@ class ManualTriggerScheduledQueryResponse(_message.Message):
     def __init__(
         self, scheduled_query_run: _Optional[_Union[_scheduled_query_run_pb2.ScheduledQueryRun, _Mapping]] = ...
     ) -> None: ...
+
+class GetScheduledResolverRunRequest(_message.Message):
+    __slots__ = ("id",)
+    ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    def __init__(self, id: _Optional[str] = ...) -> None: ...
+
+class GetScheduledResolverRunResponse(_message.Message):
+    __slots__ = ("run",)
+    RUN_FIELD_NUMBER: _ClassVar[int]
+    run: CronResolverRun
+    def __init__(self, run: _Optional[_Union[CronResolverRun, _Mapping]] = ...) -> None: ...
+
+class ListScheduledResolverRunsRequest(_message.Message):
+    __slots__ = ("cursor", "resolver_filter", "limit", "start", "end", "status_filter")
+    CURSOR_FIELD_NUMBER: _ClassVar[int]
+    RESOLVER_FILTER_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    START_FIELD_NUMBER: _ClassVar[int]
+    END_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FILTER_FIELD_NUMBER: _ClassVar[int]
+    cursor: _timestamp_pb2.Timestamp
+    resolver_filter: str
+    limit: int
+    start: _timestamp_pb2.Timestamp
+    end: _timestamp_pb2.Timestamp
+    status_filter: _batch_pb2.OperationStatus
+    def __init__(
+        self,
+        cursor: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        resolver_filter: _Optional[str] = ...,
+        limit: _Optional[int] = ...,
+        start: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        end: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        status_filter: _Optional[_Union[_batch_pb2.OperationStatus, str]] = ...,
+    ) -> None: ...
+
+class ListScheduledResolverRunsResponse(_message.Message):
+    __slots__ = ("runs",)
+    RUNS_FIELD_NUMBER: _ClassVar[int]
+    runs: _containers.RepeatedCompositeFieldContainer[CronResolverRun]
+    def __init__(self, runs: _Optional[_Iterable[_Union[CronResolverRun, _Mapping]]] = ...) -> None: ...

chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py

@@ -25,6 +25,16 @@ class SchedulerServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ManualTriggerScheduledQueryRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ManualTriggerScheduledQueryResponse.FromString,
         )
+        self.GetScheduledResolverRun = channel.unary_unary(
+            "/chalk.server.v1.SchedulerService/GetScheduledResolverRun",
+            request_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.GetScheduledResolverRunRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.GetScheduledResolverRunResponse.FromString,
+        )
+        self.ListScheduledResolverRuns = channel.unary_unary(
+            "/chalk.server.v1.SchedulerService/ListScheduledResolverRuns",
+            request_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsResponse.FromString,
+        )
 
 
 class SchedulerServiceServicer(object):
@@ -42,6 +52,18 @@ class SchedulerServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def GetScheduledResolverRun(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def ListScheduledResolverRuns(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
 
 def add_SchedulerServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -55,6 +77,16 @@ def add_SchedulerServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ManualTriggerScheduledQueryRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ManualTriggerScheduledQueryResponse.SerializeToString,
         ),
+        "GetScheduledResolverRun": grpc.unary_unary_rpc_method_handler(
+            servicer.GetScheduledResolverRun,
+            request_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.GetScheduledResolverRunRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.GetScheduledResolverRunResponse.SerializeToString,
+        ),
+        "ListScheduledResolverRuns": grpc.unary_unary_rpc_method_handler(
+            servicer.ListScheduledResolverRuns,
+            request_deserializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.SchedulerService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -121,3 +153,61 @@ class SchedulerService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def GetScheduledResolverRun(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.SchedulerService/GetScheduledResolverRun",
+            chalk_dot_server_dot_v1_dot_scheduler__pb2.GetScheduledResolverRunRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_scheduler__pb2.GetScheduledResolverRunResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def ListScheduledResolverRuns(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.SchedulerService/ListScheduledResolverRuns",
+            chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_scheduler__pb2.ListScheduledResolverRunsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi

@@ -8,6 +8,10 @@ from abc import (
     abstractmethod,
 )
 from chalk._gen.chalk.server.v1.scheduler_pb2 import (
+    GetScheduledResolverRunRequest,
+    GetScheduledResolverRunResponse,
+    ListScheduledResolverRunsRequest,
+    ListScheduledResolverRunsResponse,
     ManualTriggerCronResolverRequest,
     ManualTriggerCronResolverResponse,
     ManualTriggerScheduledQueryRequest,
@@ -30,6 +34,14 @@ class SchedulerServiceStub:
         ManualTriggerScheduledQueryRequest,
         ManualTriggerScheduledQueryResponse,
     ]
+    GetScheduledResolverRun: UnaryUnaryMultiCallable[
+        GetScheduledResolverRunRequest,
+        GetScheduledResolverRunResponse,
+    ]
+    ListScheduledResolverRuns: UnaryUnaryMultiCallable[
+        ListScheduledResolverRunsRequest,
+        ListScheduledResolverRunsResponse,
+    ]
 
 class SchedulerServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -44,5 +56,17 @@ class SchedulerServiceServicer(metaclass=ABCMeta):
         request: ManualTriggerScheduledQueryRequest,
         context: ServicerContext,
     ) -> ManualTriggerScheduledQueryResponse: ...
+    @abstractmethod
+    def GetScheduledResolverRun(
+        self,
+        request: GetScheduledResolverRunRequest,
+        context: ServicerContext,
+    ) -> GetScheduledResolverRunResponse: ...
+    @abstractmethod
+    def ListScheduledResolverRuns(
+        self,
+        request: ListScheduledResolverRunsRequest,
+        context: ServicerContext,
+    ) -> ListScheduledResolverRunsResponse: ...
 
 def add_SchedulerServiceServicer_to_server(servicer: SchedulerServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/script_tasks_pb2.py

@@ -20,7 +20,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n"chalk/server/v1/script_tasks.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/script_task.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x8d\x01\n\x17\x43reateScriptTaskRequest\x12<\n\x07request\x18\x01 \x01(\x0b\x32".chalk.common.v1.ScriptTaskRequestR\x07request\x12$\n\x0bsource_file\x18\x02 \x01(\x0cH\x00R\nsourceFile\x88\x01\x01\x42\x0e\n\x0c_source_file"3\n\x18\x43reateScriptTaskResponse\x12\x17\n\x07task_id\x18\x01 \x01(\tR\x06taskId"\x9d\x05\n\x0eScriptTaskMeta\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x39\n\x06status\x18\x02 \x01(\x0e\x32!.chalk.common.v1.ScriptTaskStatusR\x06status\x12\x39\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x42\n\x0c\x63ompleted_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x0b\x63ompletedAt\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x05 \x01(\tH\x01R\nbranchName\x88\x01\x01\x12/\n\x11raw_body_filename\x18\x06 \x01(\tH\x02R\x0frawBodyFilename\x88\x01\x01\x12\x1e\n\x08\x61gent_id\x18\x07 \x01(\tH\x03R\x07\x61gentId\x88\x01\x01\x12I\n\x08metadata\x18\x08 \x03(\x0b\x32-.chalk.server.v1.ScriptTaskMeta.MetadataEntryR\x08metadata\x12\x38\n\x04kind\x18\t \x01(\x0e\x32\x1f.chalk.common.v1.ScriptTaskKindH\x04R\x04kind\x88\x01\x01\x12#\n\rdeployment_id\x18\n \x01(\tR\x0c\x64\x65ploymentId\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\x0f\n\r_completed_atB\x0e\n\x0c_branch_nameB\x14\n\x12_raw_body_filenameB\x0b\n\t_agent_idB\x07\n\x05_kind"\x82\x02\n\x16ListScriptTasksRequest\x12\x19\n\x05limit\x18\x01 \x01(\x05H\x00R\x05limit\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x01R\x06\x63ursor\x88\x01\x01\x12;\n\x07\x66ilters\x18\x03 \x03(\x0b\x32!.chalk.common.v1.ScriptTaskFilterR\x07\x66ilters\x12"\n\nstart_time\x18\x04 \x01(\tH\x02R\tstartTime\x88\x01\x01\x12\x1e\n\x08\x65nd_time\x18\x05 \x01(\tH\x03R\x07\x65ndTime\x88\x01\x01\x42\x08\n\x06_limitB\t\n\x07_cursorB\r\n\x0b_start_timeB\x0b\n\t_end_time"\x93\x01\n\x17ListScriptTasksResponse\x12\x42\n\x0cscript_tasks\x18\x01 \x03(\x0b\x32\x1f.chalk.server.v1.ScriptTaskMetaR\x0bscriptTasks\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"<\n\x14GetScriptTaskRequest\x12$\n\x0escript_task_id\x18\x01 \x01(\tR\x0cscriptTaskId"Y\n\x15GetScriptTaskResponse\x12@\n\x0bscript_task\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.ScriptTaskMetaR\nscriptTask"B\n\x1aGetScriptTaskSourceRequest\x12$\n\x0escript_task_id\x18\x01 \x01(\tR\x0cscriptTaskId"m\n\x1bGetScriptTaskSourceResponse\x12&\n\x0fsource_file_url\x18\x01 \x01(\tR\rsourceFileUrl\x12&\n\x0finputs_file_url\x18\x02 \x01(\tR\rinputsFileUrl2\xd1\x03\n\x11ScriptTaskService\x12l\n\x10\x43reateScriptTask\x12(.chalk.server.v1.CreateScriptTaskRequest\x1a).chalk.server.v1.CreateScriptTaskResponse"\x03\x80}\x04\x12l\n\x0fListScriptTasks\x12\'.chalk.server.v1.ListScriptTasksRequest\x1a(.chalk.server.v1.ListScriptTasksResponse"\x06\x90\x02\x01\x80}\x0b\x12\x66\n\rGetScriptTask\x12%.chalk.server.v1.GetScriptTaskRequest\x1a&.chalk.server.v1.GetScriptTaskResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13GetScriptTaskSource\x12+.chalk.server.v1.GetScriptTaskSourceRequest\x1a,.chalk.server.v1.GetScriptTaskSourceResponse"\x06\x90\x02\x01\x80}\x0b\x42\x99\x01\n\x13\x63om.chalk.server.v1B\x10ScriptTasksProtoP\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'
+    b'\n"chalk/server/v1/script_tasks.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/script_task.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x8d\x01\n\x17\x43reateScriptTaskRequest\x12<\n\x07request\x18\x01 \x01(\x0b\x32".chalk.common.v1.ScriptTaskRequestR\x07request\x12$\n\x0bsource_file\x18\x02 \x01(\x0cH\x00R\nsourceFile\x88\x01\x01\x42\x0e\n\x0c_source_file"3\n\x18\x43reateScriptTaskResponse\x12\x17\n\x07task_id\x18\x01 \x01(\tR\x06taskId"\x9d\x05\n\x0eScriptTaskMeta\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x39\n\x06status\x18\x02 \x01(\x0e\x32!.chalk.common.v1.ScriptTaskStatusR\x06status\x12\x39\n\ncreated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x42\n\x0c\x63ompleted_at\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x0b\x63ompletedAt\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x05 \x01(\tH\x01R\nbranchName\x88\x01\x01\x12/\n\x11raw_body_filename\x18\x06 \x01(\tH\x02R\x0frawBodyFilename\x88\x01\x01\x12\x1e\n\x08\x61gent_id\x18\x07 \x01(\tH\x03R\x07\x61gentId\x88\x01\x01\x12I\n\x08metadata\x18\x08 \x03(\x0b\x32-.chalk.server.v1.ScriptTaskMeta.MetadataEntryR\x08metadata\x12\x38\n\x04kind\x18\t \x01(\x0e\x32\x1f.chalk.common.v1.ScriptTaskKindH\x04R\x04kind\x88\x01\x01\x12#\n\rdeployment_id\x18\n \x01(\tR\x0c\x64\x65ploymentId\x1aS\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\x0f\n\r_completed_atB\x0e\n\x0c_branch_nameB\x14\n\x12_raw_body_filenameB\x0b\n\t_agent_idB\x07\n\x05_kind"\x82\x02\n\x16ListScriptTasksRequest\x12\x19\n\x05limit\x18\x01 \x01(\x05H\x00R\x05limit\x88\x01\x01\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x01R\x06\x63ursor\x88\x01\x01\x12;\n\x07\x66ilters\x18\x03 \x03(\x0b\x32!.chalk.common.v1.ScriptTaskFilterR\x07\x66ilters\x12"\n\nstart_time\x18\x04 \x01(\tH\x02R\tstartTime\x88\x01\x01\x12\x1e\n\x08\x65nd_time\x18\x05 \x01(\tH\x03R\x07\x65ndTime\x88\x01\x01\x42\x08\n\x06_limitB\t\n\x07_cursorB\r\n\x0b_start_timeB\x0b\n\t_end_time"\x93\x01\n\x17ListScriptTasksResponse\x12\x42\n\x0cscript_tasks\x18\x01 \x03(\x0b\x32\x1f.chalk.server.v1.ScriptTaskMetaR\x0bscriptTasks\x12$\n\x0bnext_cursor\x18\x02 \x01(\tH\x00R\nnextCursor\x88\x01\x01\x42\x0e\n\x0c_next_cursor"<\n\x14GetScriptTaskRequest\x12$\n\x0escript_task_id\x18\x01 \x01(\tR\x0cscriptTaskId"Y\n\x15GetScriptTaskResponse\x12@\n\x0bscript_task\x18\x01 \x01(\x0b\x32\x1f.chalk.server.v1.ScriptTaskMetaR\nscriptTask"B\n\x1aGetScriptTaskSourceRequest\x12$\n\x0escript_task_id\x18\x01 \x01(\tR\x0cscriptTaskId"m\n\x1bGetScriptTaskSourceResponse\x12&\n\x0fsource_file_url\x18\x01 \x01(\tR\rsourceFileUrl\x12&\n\x0finputs_file_url\x18\x02 \x01(\tR\rinputsFileUrl"?\n\x17\x43\x61ncelScriptTaskRequest\x12$\n\x0escript_task_id\x18\x01 \x01(\tR\x0cscriptTaskId"\x1a\n\x18\x43\x61ncelScriptTaskResponse">\n\x16RerunScriptTaskRequest\x12$\n\x0escript_task_id\x18\x01 \x01(\tR\x0cscriptTaskId"F\n\x17RerunScriptTaskResponse\x12+\n\x12new_script_task_id\x18\x01 \x01(\tR\x0fnewScriptTaskId2\xad\x05\n\x11ScriptTaskService\x12l\n\x10\x43reateScriptTask\x12(.chalk.server.v1.CreateScriptTaskRequest\x1a).chalk.server.v1.CreateScriptTaskResponse"\x03\x80}\x04\x12l\n\x0fListScriptTasks\x12\'.chalk.server.v1.ListScriptTasksRequest\x1a(.chalk.server.v1.ListScriptTasksResponse"\x06\x90\x02\x01\x80}\x0b\x12\x66\n\rGetScriptTask\x12%.chalk.server.v1.GetScriptTaskRequest\x1a&.chalk.server.v1.GetScriptTaskResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13GetScriptTaskSource\x12+.chalk.server.v1.GetScriptTaskSourceRequest\x1a,.chalk.server.v1.GetScriptTaskSourceResponse"\x06\x90\x02\x01\x80}\x0b\x12o\n\x10\x43\x61ncelScriptTask\x12(.chalk.server.v1.CancelScriptTaskRequest\x1a).chalk.server.v1.CancelScriptTaskResponse"\x06\x90\x02\x02\x80}\x04\x12i\n\x0fRerunScriptTask\x12\'.chalk.server.v1.RerunScriptTaskRequest\x1a(.chalk.server.v1.RerunScriptTaskResponse"\x03\x80}\x04\x42\x99\x01\n\x13\x63om.chalk.server.v1B\x10ScriptTasksProtoP\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'
 )
 
 _globals = globals()
@@ -41,6 +41,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SCRIPTTASKSERVICE"].methods_by_name["GetScriptTask"]._serialized_options = b"\220\002\001\200}\013"
     _globals["_SCRIPTTASKSERVICE"].methods_by_name["GetScriptTaskSource"]._options = None
     _globals["_SCRIPTTASKSERVICE"].methods_by_name["GetScriptTaskSource"]._serialized_options = b"\220\002\001\200}\013"
+    _globals["_SCRIPTTASKSERVICE"].methods_by_name["CancelScriptTask"]._options = None
+    _globals["_SCRIPTTASKSERVICE"].methods_by_name["CancelScriptTask"]._serialized_options = b"\220\002\002\200}\004"
+    _globals["_SCRIPTTASKSERVICE"].methods_by_name["RerunScriptTask"]._options = None
+    _globals["_SCRIPTTASKSERVICE"].methods_by_name["RerunScriptTask"]._serialized_options = b"\200}\004"
     _globals["_CREATESCRIPTTASKREQUEST"]._serialized_start = 187
     _globals["_CREATESCRIPTTASKREQUEST"]._serialized_end = 328
     _globals["_CREATESCRIPTTASKRESPONSE"]._serialized_start = 330
@@ -61,6 +65,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_GETSCRIPTTASKSOURCEREQUEST"]._serialized_end = 1685
     _globals["_GETSCRIPTTASKSOURCERESPONSE"]._serialized_start = 1687
     _globals["_GETSCRIPTTASKSOURCERESPONSE"]._serialized_end = 1796
-    _globals["_SCRIPTTASKSERVICE"]._serialized_start = 1799
-    _globals["_SCRIPTTASKSERVICE"]._serialized_end = 2264
+    _globals["_CANCELSCRIPTTASKREQUEST"]._serialized_start = 1798
+    _globals["_CANCELSCRIPTTASKREQUEST"]._serialized_end = 1861
+    _globals["_CANCELSCRIPTTASKRESPONSE"]._serialized_start = 1863
+    _globals["_CANCELSCRIPTTASKRESPONSE"]._serialized_end = 1889
+    _globals["_RERUNSCRIPTTASKREQUEST"]._serialized_start = 1891
+    _globals["_RERUNSCRIPTTASKREQUEST"]._serialized_end = 1953
+    _globals["_RERUNSCRIPTTASKRESPONSE"]._serialized_start = 1955
+    _globals["_RERUNSCRIPTTASKRESPONSE"]._serialized_end = 2025
+    _globals["_SCRIPTTASKSERVICE"]._serialized_start = 2028
+    _globals["_SCRIPTTASKSERVICE"]._serialized_end = 2713
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi

@@ -148,3 +148,25 @@ class GetScriptTaskSourceResponse(_message.Message):
     source_file_url: str
     inputs_file_url: str
     def __init__(self, source_file_url: _Optional[str] = ..., inputs_file_url: _Optional[str] = ...) -> None: ...
+
+class CancelScriptTaskRequest(_message.Message):
+    __slots__ = ("script_task_id",)
+    SCRIPT_TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    script_task_id: str
+    def __init__(self, script_task_id: _Optional[str] = ...) -> None: ...
+
+class CancelScriptTaskResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class RerunScriptTaskRequest(_message.Message):
+    __slots__ = ("script_task_id",)
+    SCRIPT_TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    script_task_id: str
+    def __init__(self, script_task_id: _Optional[str] = ...) -> None: ...
+
+class RerunScriptTaskResponse(_message.Message):
+    __slots__ = ("new_script_task_id",)
+    NEW_SCRIPT_TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    new_script_task_id: str
+    def __init__(self, new_script_task_id: _Optional[str] = ...) -> None: ...

chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py

@@ -35,6 +35,16 @@ class ScriptTaskServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.GetScriptTaskSourceRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.GetScriptTaskSourceResponse.FromString,
         )
+        self.CancelScriptTask = channel.unary_unary(
+            "/chalk.server.v1.ScriptTaskService/CancelScriptTask",
+            request_serializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.CancelScriptTaskRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.CancelScriptTaskResponse.FromString,
+        )
+        self.RerunScriptTask = channel.unary_unary(
+            "/chalk.server.v1.ScriptTaskService/RerunScriptTask",
+            request_serializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.RerunScriptTaskRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.RerunScriptTaskResponse.FromString,
+        )
 
 
 class ScriptTaskServiceServicer(object):
@@ -64,6 +74,18 @@ class ScriptTaskServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def CancelScriptTask(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def RerunScriptTask(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
 
 def add_ScriptTaskServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -87,6 +109,16 @@ def add_ScriptTaskServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.GetScriptTaskSourceRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.GetScriptTaskSourceResponse.SerializeToString,
         ),
+        "CancelScriptTask": grpc.unary_unary_rpc_method_handler(
+            servicer.CancelScriptTask,
+            request_deserializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.CancelScriptTaskRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.CancelScriptTaskResponse.SerializeToString,
+        ),
+        "RerunScriptTask": grpc.unary_unary_rpc_method_handler(
+            servicer.RerunScriptTask,
+            request_deserializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.RerunScriptTaskRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_script__tasks__pb2.RerunScriptTaskResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.ScriptTaskService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -211,3 +243,61 @@ class ScriptTaskService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def CancelScriptTask(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.ScriptTaskService/CancelScriptTask",
+            chalk_dot_server_dot_v1_dot_script__tasks__pb2.CancelScriptTaskRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_script__tasks__pb2.CancelScriptTaskResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def RerunScriptTask(
+        request,
+        target,
+        options=(),
+        channel_credentials=None,
+        call_credentials=None,
+        insecure=False,
+        compression=None,
+        wait_for_ready=None,
+        timeout=None,
+        metadata=None,
+    ):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            "/chalk.server.v1.ScriptTaskService/RerunScriptTask",
+            chalk_dot_server_dot_v1_dot_script__tasks__pb2.RerunScriptTaskRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_script__tasks__pb2.RerunScriptTaskResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi

@@ -8,6 +8,8 @@ from abc import (
     abstractmethod,
 )
 from chalk._gen.chalk.server.v1.script_tasks_pb2 import (
+    CancelScriptTaskRequest,
+    CancelScriptTaskResponse,
     CreateScriptTaskRequest,
     CreateScriptTaskResponse,
     GetScriptTaskRequest,
@@ -16,6 +18,8 @@ from chalk._gen.chalk.server.v1.script_tasks_pb2 import (
     GetScriptTaskSourceResponse,
     ListScriptTasksRequest,
     ListScriptTasksResponse,
+    RerunScriptTaskRequest,
+    RerunScriptTaskResponse,
 )
 from grpc import (
     Channel,
@@ -42,6 +46,14 @@ class ScriptTaskServiceStub:
         GetScriptTaskSourceRequest,
         GetScriptTaskSourceResponse,
     ]
+    CancelScriptTask: UnaryUnaryMultiCallable[
+        CancelScriptTaskRequest,
+        CancelScriptTaskResponse,
+    ]
+    RerunScriptTask: UnaryUnaryMultiCallable[
+        RerunScriptTaskRequest,
+        RerunScriptTaskResponse,
+    ]
 
 class ScriptTaskServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -68,5 +80,17 @@ class ScriptTaskServiceServicer(metaclass=ABCMeta):
         request: GetScriptTaskSourceRequest,
         context: ServicerContext,
     ) -> GetScriptTaskSourceResponse: ...
+    @abstractmethod
+    def CancelScriptTask(
+        self,
+        request: CancelScriptTaskRequest,
+        context: ServicerContext,
+    ) -> CancelScriptTaskResponse: ...
+    @abstractmethod
+    def RerunScriptTask(
+        self,
+        request: RerunScriptTaskRequest,
+        context: ServicerContext,
+    ) -> RerunScriptTaskResponse: ...
 
 def add_ScriptTaskServiceServicer_to_server(servicer: ScriptTaskServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/sql_queries_pb2.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/server/v1/sql_queries.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    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"\x91\x03\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\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_id"\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'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.sql_queries_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.server.v1B\017SqlQueriesProtoP\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"
+    _globals["_SQLQUERIESSERVICE"].methods_by_name["ListSqlQueries"]._options = None
+    _globals["_SQLQUERIESSERVICE"].methods_by_name["ListSqlQueries"]._serialized_options = b"\220\002\001\200}\003"
+    _globals["_SQLQUERIESSERVICE"].methods_by_name["GetSqlQuery"]._options = None
+    _globals["_SQLQUERIESSERVICE"].methods_by_name["GetSqlQuery"]._serialized_options = b"\220\002\001\200}\003"
+    _globals["_SQLQUERY"]._serialized_start = 121
+    _globals["_SQLQUERY"]._serialized_end = 994
+    _globals["_LISTSQLQUERIESREQUEST"]._serialized_start = 997
+    _globals["_LISTSQLQUERIESREQUEST"]._serialized_end = 1398
+    _globals["_LISTSQLQUERIESRESPONSE"]._serialized_start = 1401
+    _globals["_LISTSQLQUERIESRESPONSE"]._serialized_end = 1532
+    _globals["_GETSQLQUERYREQUEST"]._serialized_start = 1534
+    _globals["_GETSQLQUERYREQUEST"]._serialized_end = 1570
+    _globals["_GETSQLQUERYRESPONSE"]._serialized_start = 1572
+    _globals["_GETSQLQUERYRESPONSE"]._serialized_end = 1642
+    _globals["_SQLQUERIESSERVICE"]._serialized_start = 1645
+    _globals["_SQLQUERIESSERVICE"]._serialized_end = 1869
+# @@protoc_insertion_point(module_scope)