chalkpy 2.94.4__py3-none-any.whl → 2.94.6__py3-none-any.whl

chalk/_gen/chalk/server/v1/log_pb2_grpc.py

@@ -25,6 +25,16 @@ class LogSearchServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_log__pb2.SearchLogEntriesAggregatedRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_log__pb2.SearchLogEntriesAggregatedResponse.FromString,
         )
+        self.GetLogFacets = channel.unary_unary(
+            "/chalk.server.v1.LogSearchService/GetLogFacets",
+            request_serializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetsResponse.FromString,
+        )
+        self.GetLogFacetValues = channel.unary_unary(
+            "/chalk.server.v1.LogSearchService/GetLogFacetValues",
+            request_serializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetValuesRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetValuesResponse.FromString,
+        )
 
 
 class LogSearchServiceServicer(object):
@@ -42,6 +52,18 @@ class LogSearchServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def GetLogFacets(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 GetLogFacetValues(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_LogSearchServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -55,6 +77,16 @@ def add_LogSearchServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_log__pb2.SearchLogEntriesAggregatedRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_log__pb2.SearchLogEntriesAggregatedResponse.SerializeToString,
         ),
+        "GetLogFacets": grpc.unary_unary_rpc_method_handler(
+            servicer.GetLogFacets,
+            request_deserializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetsResponse.SerializeToString,
+        ),
+        "GetLogFacetValues": grpc.unary_unary_rpc_method_handler(
+            servicer.GetLogFacetValues,
+            request_deserializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetValuesRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetValuesResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.LogSearchService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -121,3 +153,61 @@ class LogSearchService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def GetLogFacets(
+        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.LogSearchService/GetLogFacets",
+            chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetLogFacetValues(
+        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.LogSearchService/GetLogFacetValues",
+            chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetValuesRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_log__pb2.GetLogFacetValuesResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi

@@ -8,6 +8,10 @@ from abc import (
     abstractmethod,
 )
 from chalk._gen.chalk.server.v1.log_pb2 import (
+    GetLogFacetValuesRequest,
+    GetLogFacetValuesResponse,
+    GetLogFacetsRequest,
+    GetLogFacetsResponse,
     SearchLogEntriesAggregatedRequest,
     SearchLogEntriesAggregatedResponse,
     SearchLogEntriesRequest,
@@ -30,6 +34,14 @@ class LogSearchServiceStub:
         SearchLogEntriesAggregatedRequest,
         SearchLogEntriesAggregatedResponse,
     ]
+    GetLogFacets: UnaryUnaryMultiCallable[
+        GetLogFacetsRequest,
+        GetLogFacetsResponse,
+    ]
+    GetLogFacetValues: UnaryUnaryMultiCallable[
+        GetLogFacetValuesRequest,
+        GetLogFacetValuesResponse,
+    ]
 
 class LogSearchServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -44,5 +56,17 @@ class LogSearchServiceServicer(metaclass=ABCMeta):
         request: SearchLogEntriesAggregatedRequest,
         context: ServicerContext,
     ) -> SearchLogEntriesAggregatedResponse: ...
+    @abstractmethod
+    def GetLogFacets(
+        self,
+        request: GetLogFacetsRequest,
+        context: ServicerContext,
+    ) -> GetLogFacetsResponse: ...
+    @abstractmethod
+    def GetLogFacetValues(
+        self,
+        request: GetLogFacetValuesRequest,
+        context: ServicerContext,
+    ) -> GetLogFacetValuesResponse: ...
 
 def add_LogSearchServiceServicer_to_server(servicer: LogSearchServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/plandebug_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/server/v1/plandebug.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 struct_pb2 as google_dot_protobuf_dot_struct__pb2
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n\x1f\x63halk/server/v1/plandebug.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"^\n\x18GetQueryPlanStageRequest\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId"\xd8\x02\n\x19GetQueryPlanStageResponse\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId\x12?\n\x0c\x64\x61ta_preview\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructH\x00R\x0b\x64\x61taPreview\x88\x01\x01\x12?\n\x0c\x64\x61ta_summary\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x0b\x64\x61taSummary\x88\x01\x01\x12\x41\n\rgroup_preview\x18\x05 \x01(\x0b\x32\x17.google.protobuf.StructH\x02R\x0cgroupPreview\x88\x01\x01\x42\x0f\n\r_data_previewB\x0f\n\r_data_summaryB\x10\n\x0e_group_preview"l\n&GetQueryPlanStageResolverInputsRequest\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId"\xf8\x01\n\'GetQueryPlanStageResolverInputsResponse\x12:\n\tresolvers\x18\x01 \x01(\x0b\x32\x17.google.protobuf.StructH\x00R\tresolvers\x88\x01\x01\x12\x36\n\x07scalars\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x07scalars\x88\x01\x01\x12\x34\n\x06tables\x18\x03 \x01(\x0b\x32\x17.google.protobuf.StructH\x02R\x06tables\x88\x01\x01\x42\x0c\n\n_resolversB\n\n\x08_scalarsB\t\n\x07_tables"j\n$GetQueryPlanStageDownloadLinkRequest\x12\x1f\n\x0boperator_id\x18\x01 \x01(\tR\noperatorId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId"\x9b\x02\n%GetQueryPlanStageDownloadLinkResponse\x12"\n\nsigned_url\x18\x01 \x01(\tH\x00R\tsignedUrl\x88\x01\x01\x12;\n\ngroup_urls\x18\x02 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\tgroupUrls\x88\x01\x01\x12\x19\n\x05\x65rror\x18\x03 \x01(\tH\x02R\x05\x65rror\x88\x01\x01\x12?\n\nexpiration\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nexpiration\x88\x01\x01\x42\r\n\x0b_signed_urlB\r\n\x0b_group_urlsB\x08\n\x06_errorB\r\n\x0b_expiration2\xb5\x03\n\x10PlanDebugService\x12o\n\x11GetQueryPlanStage\x12).chalk.server.v1.GetQueryPlanStageRequest\x1a*.chalk.server.v1.GetQueryPlanStageResponse"\x03\x80}\x04\x12\x99\x01\n\x1fGetQueryPlanStageResolverInputs\x12\x37.chalk.server.v1.GetQueryPlanStageResolverInputsRequest\x1a\x38.chalk.server.v1.GetQueryPlanStageResolverInputsResponse"\x03\x80}\x04\x12\x93\x01\n\x1dGetQueryPlanStageDownloadLink\x12\x35.chalk.server.v1.GetQueryPlanStageDownloadLinkRequest\x1a\x36.chalk.server.v1.GetQueryPlanStageDownloadLinkResponse"\x03\x80}\x04\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0ePlandebugProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.server.v1.plandebug_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.server.v1B\016PlandebugProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
+    _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStage"]._options = None
+    _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStage"]._serialized_options = b"\200}\004"
+    _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageResolverInputs"]._options = None
+    _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageResolverInputs"]._serialized_options = b"\200}\004"
+    _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageDownloadLink"]._options = None
+    _globals["_PLANDEBUGSERVICE"].methods_by_name["GetQueryPlanStageDownloadLink"]._serialized_options = b"\200}\004"
+    _globals["_GETQUERYPLANSTAGEREQUEST"]._serialized_start = 148
+    _globals["_GETQUERYPLANSTAGEREQUEST"]._serialized_end = 242
+    _globals["_GETQUERYPLANSTAGERESPONSE"]._serialized_start = 245
+    _globals["_GETQUERYPLANSTAGERESPONSE"]._serialized_end = 589
+    _globals["_GETQUERYPLANSTAGERESOLVERINPUTSREQUEST"]._serialized_start = 591
+    _globals["_GETQUERYPLANSTAGERESOLVERINPUTSREQUEST"]._serialized_end = 699
+    _globals["_GETQUERYPLANSTAGERESOLVERINPUTSRESPONSE"]._serialized_start = 702
+    _globals["_GETQUERYPLANSTAGERESOLVERINPUTSRESPONSE"]._serialized_end = 950
+    _globals["_GETQUERYPLANSTAGEDOWNLOADLINKREQUEST"]._serialized_start = 952
+    _globals["_GETQUERYPLANSTAGEDOWNLOADLINKREQUEST"]._serialized_end = 1058
+    _globals["_GETQUERYPLANSTAGEDOWNLOADLINKRESPONSE"]._serialized_start = 1061
+    _globals["_GETQUERYPLANSTAGEDOWNLOADLINKRESPONSE"]._serialized_end = 1344
+    _globals["_PLANDEBUGSERVICE"]._serialized_start = 1347
+    _globals["_PLANDEBUGSERVICE"]._serialized_end = 1784
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/plandebug_pb2.pyi

@@ -0,0 +1,86 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class GetQueryPlanStageRequest(_message.Message):
+    __slots__ = ("operator_id", "operation_id")
+    OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    operator_id: str
+    operation_id: str
+    def __init__(self, operator_id: _Optional[str] = ..., operation_id: _Optional[str] = ...) -> None: ...
+
+class GetQueryPlanStageResponse(_message.Message):
+    __slots__ = ("operator_id", "operation_id", "data_preview", "data_summary", "group_preview")
+    OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    DATA_PREVIEW_FIELD_NUMBER: _ClassVar[int]
+    DATA_SUMMARY_FIELD_NUMBER: _ClassVar[int]
+    GROUP_PREVIEW_FIELD_NUMBER: _ClassVar[int]
+    operator_id: str
+    operation_id: str
+    data_preview: _struct_pb2.Struct
+    data_summary: _struct_pb2.Struct
+    group_preview: _struct_pb2.Struct
+    def __init__(
+        self,
+        operator_id: _Optional[str] = ...,
+        operation_id: _Optional[str] = ...,
+        data_preview: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        data_summary: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        group_preview: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+    ) -> None: ...
+
+class GetQueryPlanStageResolverInputsRequest(_message.Message):
+    __slots__ = ("operator_id", "operation_id")
+    OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    operator_id: str
+    operation_id: str
+    def __init__(self, operator_id: _Optional[str] = ..., operation_id: _Optional[str] = ...) -> None: ...
+
+class GetQueryPlanStageResolverInputsResponse(_message.Message):
+    __slots__ = ("resolvers", "scalars", "tables")
+    RESOLVERS_FIELD_NUMBER: _ClassVar[int]
+    SCALARS_FIELD_NUMBER: _ClassVar[int]
+    TABLES_FIELD_NUMBER: _ClassVar[int]
+    resolvers: _struct_pb2.Struct
+    scalars: _struct_pb2.Struct
+    tables: _struct_pb2.Struct
+    def __init__(
+        self,
+        resolvers: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        scalars: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        tables: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+    ) -> None: ...
+
+class GetQueryPlanStageDownloadLinkRequest(_message.Message):
+    __slots__ = ("operator_id", "operation_id")
+    OPERATOR_ID_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    operator_id: str
+    operation_id: str
+    def __init__(self, operator_id: _Optional[str] = ..., operation_id: _Optional[str] = ...) -> None: ...
+
+class GetQueryPlanStageDownloadLinkResponse(_message.Message):
+    __slots__ = ("signed_url", "group_urls", "error", "expiration")
+    SIGNED_URL_FIELD_NUMBER: _ClassVar[int]
+    GROUP_URLS_FIELD_NUMBER: _ClassVar[int]
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    EXPIRATION_FIELD_NUMBER: _ClassVar[int]
+    signed_url: str
+    group_urls: _struct_pb2.Struct
+    error: str
+    expiration: _timestamp_pb2.Timestamp
+    def __init__(
+        self,
+        signed_url: _Optional[str] = ...,
+        group_urls: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        error: _Optional[str] = ...,
+        expiration: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+    ) -> None: ...

chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py

@@ -0,0 +1,168 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+
+import grpc
+
+from chalk._gen.chalk.server.v1 import plandebug_pb2 as chalk_dot_server_dot_v1_dot_plandebug__pb2
+
+
+class PlanDebugServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetQueryPlanStage = channel.unary_unary(
+            "/chalk.server.v1.PlanDebugService/GetQueryPlanStage",
+            request_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResponse.FromString,
+        )
+        self.GetQueryPlanStageResolverInputs = channel.unary_unary(
+            "/chalk.server.v1.PlanDebugService/GetQueryPlanStageResolverInputs",
+            request_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsResponse.FromString,
+        )
+        self.GetQueryPlanStageDownloadLink = channel.unary_unary(
+            "/chalk.server.v1.PlanDebugService/GetQueryPlanStageDownloadLink",
+            request_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkResponse.FromString,
+        )
+
+
+class PlanDebugServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetQueryPlanStage(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 GetQueryPlanStageResolverInputs(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 GetQueryPlanStageDownloadLink(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_PlanDebugServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "GetQueryPlanStage": grpc.unary_unary_rpc_method_handler(
+            servicer.GetQueryPlanStage,
+            request_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResponse.SerializeToString,
+        ),
+        "GetQueryPlanStageResolverInputs": grpc.unary_unary_rpc_method_handler(
+            servicer.GetQueryPlanStageResolverInputs,
+            request_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsResponse.SerializeToString,
+        ),
+        "GetQueryPlanStageDownloadLink": grpc.unary_unary_rpc_method_handler(
+            servicer.GetQueryPlanStageDownloadLink,
+            request_deserializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.PlanDebugService", rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class PlanDebugService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetQueryPlanStage(
+        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.PlanDebugService/GetQueryPlanStage",
+            chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetQueryPlanStageResolverInputs(
+        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.PlanDebugService/GetQueryPlanStageResolverInputs",
+            chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageResolverInputsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetQueryPlanStageDownloadLink(
+        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.PlanDebugService/GetQueryPlanStageDownloadLink",
+            chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_plandebug__pb2.GetQueryPlanStageDownloadLinkResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi

@@ -0,0 +1,60 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.server.v1.plandebug_pb2 import (
+    GetQueryPlanStageDownloadLinkRequest,
+    GetQueryPlanStageDownloadLinkResponse,
+    GetQueryPlanStageRequest,
+    GetQueryPlanStageResolverInputsRequest,
+    GetQueryPlanStageResolverInputsResponse,
+    GetQueryPlanStageResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class PlanDebugServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    GetQueryPlanStage: UnaryUnaryMultiCallable[
+        GetQueryPlanStageRequest,
+        GetQueryPlanStageResponse,
+    ]
+    GetQueryPlanStageResolverInputs: UnaryUnaryMultiCallable[
+        GetQueryPlanStageResolverInputsRequest,
+        GetQueryPlanStageResolverInputsResponse,
+    ]
+    GetQueryPlanStageDownloadLink: UnaryUnaryMultiCallable[
+        GetQueryPlanStageDownloadLinkRequest,
+        GetQueryPlanStageDownloadLinkResponse,
+    ]
+
+class PlanDebugServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def GetQueryPlanStage(
+        self,
+        request: GetQueryPlanStageRequest,
+        context: ServicerContext,
+    ) -> GetQueryPlanStageResponse: ...
+    @abstractmethod
+    def GetQueryPlanStageResolverInputs(
+        self,
+        request: GetQueryPlanStageResolverInputsRequest,
+        context: ServicerContext,
+    ) -> GetQueryPlanStageResolverInputsResponse: ...
+    @abstractmethod
+    def GetQueryPlanStageDownloadLink(
+        self,
+        request: GetQueryPlanStageDownloadLinkRequest,
+        context: ServicerContext,
+    ) -> GetQueryPlanStageDownloadLinkResponse: ...
+
+def add_PlanDebugServiceServicer_to_server(servicer: PlanDebugServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py

@@ -19,7 +19,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n)chalk/server/v1/scheduled_query_run.proto\x12\x0f\x63halk.server.v1\x1a%chalk/server/v1/offline_queries.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xa4\x06\n\x11ScheduledQueryRun\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1a\n\x06run_id\x18\x04 \x01(\tH\x01R\x05runId\x88\x01\x01\x12"\n\rcron_query_id\x18\x05 \x01(\x03R\x0b\x63ronQueryId\x12\x33\n\x16\x63ron_query_schedule_id\x18\x06 \x01(\x03R\x13\x63ronQueryScheduleId\x12\x1b\n\tcron_name\x18\x07 \x01(\tR\x08\x63ronName\x12-\n\x10gcr_execution_id\x18\x08 \x01(\tH\x02R\x0egcrExecutionId\x88\x01\x01\x12%\n\x0cgcr_job_name\x18\t \x01(\tH\x03R\ngcrJobName\x88\x01\x01\x12(\n\x10offline_query_id\x18\n \x01(\tR\x0eofflineQueryId\x12\x39\n\ncreated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12@\n\x06status\x18\r \x01(\x0e\x32(.chalk.server.v1.ScheduledQueryRunStatusR\x06status\x12\x35\n\x14\x62locker_operation_id\x18\x0e \x01(\tH\x04R\x12\x62lockerOperationId\x88\x01\x01\x12\x37\n\x15workflow_execution_id\x18\x0f \x01(\tH\x05R\x13workflowExecutionId\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_run_idB\x13\n\x11_gcr_execution_idB\x0f\n\r_gcr_job_nameB\x17\n\x15_blocker_operation_idB\x18\n\x16_workflow_execution_id"\xa7\x01\n\x1bGetScheduledQueryRunRequest\x12\x17\n\x06run_id\x18\x01 \x01(\x03H\x00R\x05runId\x12*\n\x10offline_query_id\x18\x02 \x01(\tH\x00R\x0eofflineQueryId\x12\x35\n\x08get_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x07getMaskB\x0c\n\nidentifier"\xd1\x01\n\x1cGetScheduledQueryRunResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun\x12K\n\roffline_query\x18\x02 \x01(\x0b\x32!.chalk.server.v1.OfflineQueryMetaH\x00R\x0cofflineQuery\x88\x01\x01\x42\x10\n\x0e_offline_query"\x9d\x01\n\x1cGetScheduledQueryRunsRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12\x1d\n\tcron_name\x18\x04 \x01(\tH\x00R\x08\x63ronName\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor\x12\x14\n\x05limit\x18\x03 \x01(\x05R\x05limitB\x0c\n\nidentifier"o\n\x1dGetScheduledQueryRunsResponse\x12\x36\n\x04runs\x18\x01 \x03(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x04runs\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor*\x8f\x03\n\x17ScheduledQueryRunStatus\x12*\n&SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED\x10\x00\x12+\n\'SCHEDULED_QUERY_RUN_STATUS_INITIALIZING\x10\x01\x12*\n&SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED\x10\x02\x12&\n"SCHEDULED_QUERY_RUN_STATUS_SKIPPED\x10\x03\x12%\n!SCHEDULED_QUERY_RUN_STATUS_QUEUED\x10\x04\x12&\n"SCHEDULED_QUERY_RUN_STATUS_WORKING\x10\x05\x12(\n$SCHEDULED_QUERY_RUN_STATUS_COMPLETED\x10\x06\x12%\n!SCHEDULED_QUERY_RUN_STATUS_FAILED\x10\x07\x12\'\n#SCHEDULED_QUERY_RUN_STATUS_CANCELED\x10\x08\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16ScheduledQueryRunProtoP\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/scheduled_query_run.proto\x12\x0f\x63halk.server.v1\x1a%chalk/server/v1/offline_queries.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xa4\x06\n\x11ScheduledQueryRun\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1a\n\x06run_id\x18\x04 \x01(\tH\x01R\x05runId\x88\x01\x01\x12"\n\rcron_query_id\x18\x05 \x01(\x03R\x0b\x63ronQueryId\x12\x33\n\x16\x63ron_query_schedule_id\x18\x06 \x01(\x03R\x13\x63ronQueryScheduleId\x12\x1b\n\tcron_name\x18\x07 \x01(\tR\x08\x63ronName\x12-\n\x10gcr_execution_id\x18\x08 \x01(\tH\x02R\x0egcrExecutionId\x88\x01\x01\x12%\n\x0cgcr_job_name\x18\t \x01(\tH\x03R\ngcrJobName\x88\x01\x01\x12(\n\x10offline_query_id\x18\n \x01(\tR\x0eofflineQueryId\x12\x39\n\ncreated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12@\n\x06status\x18\r \x01(\x0e\x32(.chalk.server.v1.ScheduledQueryRunStatusR\x06status\x12\x35\n\x14\x62locker_operation_id\x18\x0e \x01(\tH\x04R\x12\x62lockerOperationId\x88\x01\x01\x12\x37\n\x15workflow_execution_id\x18\x0f \x01(\tH\x05R\x13workflowExecutionId\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_run_idB\x13\n\x11_gcr_execution_idB\x0f\n\r_gcr_job_nameB\x17\n\x15_blocker_operation_idB\x18\n\x16_workflow_execution_id"\xdd\x01\n\x1bGetScheduledQueryRunRequest\x12\x17\n\x06run_id\x18\x01 \x01(\x03H\x00R\x05runId\x12*\n\x10offline_query_id\x18\x02 \x01(\tH\x00R\x0eofflineQueryId\x12\x34\n\x15workflow_execution_id\x18\x04 \x01(\tH\x00R\x13workflowExecutionId\x12\x35\n\x08get_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x07getMaskB\x0c\n\nidentifier"\xd1\x01\n\x1cGetScheduledQueryRunResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun\x12K\n\roffline_query\x18\x02 \x01(\x0b\x32!.chalk.server.v1.OfflineQueryMetaH\x00R\x0cofflineQuery\x88\x01\x01\x42\x10\n\x0e_offline_query"\x9d\x01\n\x1cGetScheduledQueryRunsRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12\x1d\n\tcron_name\x18\x04 \x01(\tH\x00R\x08\x63ronName\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor\x12\x14\n\x05limit\x18\x03 \x01(\x05R\x05limitB\x0c\n\nidentifier"o\n\x1dGetScheduledQueryRunsResponse\x12\x36\n\x04runs\x18\x01 \x03(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x04runs\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor*\x8f\x03\n\x17ScheduledQueryRunStatus\x12*\n&SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED\x10\x00\x12+\n\'SCHEDULED_QUERY_RUN_STATUS_INITIALIZING\x10\x01\x12*\n&SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED\x10\x02\x12&\n"SCHEDULED_QUERY_RUN_STATUS_SKIPPED\x10\x03\x12%\n!SCHEDULED_QUERY_RUN_STATUS_QUEUED\x10\x04\x12&\n"SCHEDULED_QUERY_RUN_STATUS_WORKING\x10\x05\x12(\n$SCHEDULED_QUERY_RUN_STATUS_COMPLETED\x10\x06\x12%\n!SCHEDULED_QUERY_RUN_STATUS_FAILED\x10\x07\x12\'\n#SCHEDULED_QUERY_RUN_STATUS_CANCELED\x10\x08\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16ScheduledQueryRunProtoP\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()
@@ -30,16 +30,16 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals[
         "DESCRIPTOR"
     ]._serialized_options = b"\n\023com.chalk.server.v1B\026ScheduledQueryRunProtoP\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["_SCHEDULEDQUERYRUNSTATUS"]._serialized_start = 1631
-    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 2030
+    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_start = 1685
+    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 2084
     _globals["_SCHEDULEDQUERYRUN"]._serialized_start = 169
     _globals["_SCHEDULEDQUERYRUN"]._serialized_end = 973
     _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_start = 976
-    _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_end = 1143
-    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_start = 1146
-    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_end = 1355
-    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_start = 1358
-    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_end = 1515
-    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_start = 1517
-    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_end = 1628
+    _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_end = 1197
+    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_start = 1200
+    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_end = 1409
+    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_start = 1412
+    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_end = 1569
+    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_start = 1571
+    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_end = 1682
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi

@@ -105,17 +105,20 @@ class ScheduledQueryRun(_message.Message):
     ) -> None: ...
 
 class GetScheduledQueryRunRequest(_message.Message):
-    __slots__ = ("run_id", "offline_query_id", "get_mask")
+    __slots__ = ("run_id", "offline_query_id", "workflow_execution_id", "get_mask")
     RUN_ID_FIELD_NUMBER: _ClassVar[int]
     OFFLINE_QUERY_ID_FIELD_NUMBER: _ClassVar[int]
+    WORKFLOW_EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
     GET_MASK_FIELD_NUMBER: _ClassVar[int]
     run_id: int
     offline_query_id: str
+    workflow_execution_id: str
     get_mask: _field_mask_pb2.FieldMask
     def __init__(
         self,
         run_id: _Optional[int] = ...,
         offline_query_id: _Optional[str] = ...,
+        workflow_execution_id: _Optional[str] = ...,
         get_mask: _Optional[_Union[_field_mask_pb2.FieldMask, _Mapping]] = ...,
     ) -> None: ...