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

chalk/_gen/chalk/server/v1/offline_queries_pb2.pyi

@@ -326,3 +326,76 @@ class CreateModelTrainingJobRequest(_message.Message):
 class CreateModelTrainingJobResponse(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
+
+class IngestDatasetRequest(_message.Message):
+    __slots__ = (
+        "outputs",
+        "revision_id",
+        "branch",
+        "planner_options",
+        "store_online",
+        "store_offline",
+        "enable_profiling",
+        "online_timestamping_mode",
+        "explain",
+    )
+    class PlannerOptionsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: _struct_pb2.Value
+        def __init__(
+            self, key: _Optional[str] = ..., value: _Optional[_Union[_struct_pb2.Value, _Mapping]] = ...
+        ) -> None: ...
+
+    OUTPUTS_FIELD_NUMBER: _ClassVar[int]
+    REVISION_ID_FIELD_NUMBER: _ClassVar[int]
+    BRANCH_FIELD_NUMBER: _ClassVar[int]
+    PLANNER_OPTIONS_FIELD_NUMBER: _ClassVar[int]
+    STORE_ONLINE_FIELD_NUMBER: _ClassVar[int]
+    STORE_OFFLINE_FIELD_NUMBER: _ClassVar[int]
+    ENABLE_PROFILING_FIELD_NUMBER: _ClassVar[int]
+    ONLINE_TIMESTAMPING_MODE_FIELD_NUMBER: _ClassVar[int]
+    EXPLAIN_FIELD_NUMBER: _ClassVar[int]
+    outputs: _containers.RepeatedScalarFieldContainer[str]
+    revision_id: str
+    branch: str
+    planner_options: _containers.MessageMap[str, _struct_pb2.Value]
+    store_online: bool
+    store_offline: bool
+    enable_profiling: bool
+    online_timestamping_mode: str
+    explain: bool
+    def __init__(
+        self,
+        outputs: _Optional[_Iterable[str]] = ...,
+        revision_id: _Optional[str] = ...,
+        branch: _Optional[str] = ...,
+        planner_options: _Optional[_Mapping[str, _struct_pb2.Value]] = ...,
+        store_online: bool = ...,
+        store_offline: bool = ...,
+        enable_profiling: bool = ...,
+        online_timestamping_mode: _Optional[str] = ...,
+        explain: bool = ...,
+    ) -> None: ...
+
+class IngestDatasetResponse(_message.Message):
+    __slots__ = ("dataset_response",)
+    DATASET_RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    dataset_response: _dataset_response_pb2.DatasetResponse
+    def __init__(
+        self, dataset_response: _Optional[_Union[_dataset_response_pb2.DatasetResponse, _Mapping]] = ...
+    ) -> None: ...
+
+class RetryOfflineQueryShardRequest(_message.Message):
+    __slots__ = ("offline_query_id", "shard_index")
+    OFFLINE_QUERY_ID_FIELD_NUMBER: _ClassVar[int]
+    SHARD_INDEX_FIELD_NUMBER: _ClassVar[int]
+    offline_query_id: str
+    shard_index: int
+    def __init__(self, offline_query_id: _Optional[str] = ..., shard_index: _Optional[int] = ...) -> None: ...
+
+class RetryOfflineQueryShardResponse(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...

chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.py

@@ -41,6 +41,16 @@ class OfflineQueryMetadataServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.CreateModelTrainingJobRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.CreateModelTrainingJobResponse.FromString,
         )
+        self.IngestDataset = channel.unary_unary(
+            "/chalk.server.v1.OfflineQueryMetadataService/IngestDataset",
+            request_serializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.IngestDatasetRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.IngestDatasetResponse.FromString,
+        )
+        self.RetryOfflineQueryShard = channel.unary_unary(
+            "/chalk.server.v1.OfflineQueryMetadataService/RetryOfflineQueryShard",
+            request_serializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.RetryOfflineQueryShardRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.RetryOfflineQueryShardResponse.FromString,
+        )
 
 
 class OfflineQueryMetadataServiceServicer(object):
@@ -81,6 +91,18 @@ class OfflineQueryMetadataServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def IngestDataset(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 RetryOfflineQueryShard(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_OfflineQueryMetadataServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -109,6 +131,16 @@ def add_OfflineQueryMetadataServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.CreateModelTrainingJobRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.CreateModelTrainingJobResponse.SerializeToString,
         ),
+        "IngestDataset": grpc.unary_unary_rpc_method_handler(
+            servicer.IngestDataset,
+            request_deserializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.IngestDatasetRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.IngestDatasetResponse.SerializeToString,
+        ),
+        "RetryOfflineQueryShard": grpc.unary_unary_rpc_method_handler(
+            servicer.RetryOfflineQueryShard,
+            request_deserializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.RetryOfflineQueryShardRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_offline__queries__pb2.RetryOfflineQueryShardResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
         "chalk.server.v1.OfflineQueryMetadataService", rpc_method_handlers
@@ -264,3 +296,61 @@ class OfflineQueryMetadataService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def IngestDataset(
+        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.OfflineQueryMetadataService/IngestDataset",
+            chalk_dot_server_dot_v1_dot_offline__queries__pb2.IngestDatasetRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_offline__queries__pb2.IngestDatasetResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def RetryOfflineQueryShard(
+        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.OfflineQueryMetadataService/RetryOfflineQueryShard",
+            chalk_dot_server_dot_v1_dot_offline__queries__pb2.RetryOfflineQueryShardRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_offline__queries__pb2.RetryOfflineQueryShardResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/offline_queries_pb2_grpc.pyi

@@ -14,8 +14,12 @@ from chalk._gen.chalk.server.v1.offline_queries_pb2 import (
     CreateOfflineQueryJobResponse,
     GetOfflineQueryRequest,
     GetOfflineQueryResponse,
+    IngestDatasetRequest,
+    IngestDatasetResponse,
     ListOfflineQueriesRequest,
     ListOfflineQueriesResponse,
+    RetryOfflineQueryShardRequest,
+    RetryOfflineQueryShardResponse,
 )
 from chalk._gen.chalk.server.v1.performance_summary_pb2 import (
     ListOfflineQueryShardPerformanceSummariesRequest,
@@ -55,6 +59,14 @@ class OfflineQueryMetadataServiceStub:
         CreateModelTrainingJobRequest,
         CreateModelTrainingJobResponse,
     ]
+    IngestDataset: UnaryUnaryMultiCallable[
+        IngestDatasetRequest,
+        IngestDatasetResponse,
+    ]
+    RetryOfflineQueryShard: UnaryUnaryMultiCallable[
+        RetryOfflineQueryShardRequest,
+        RetryOfflineQueryShardResponse,
+    ]
 
 class OfflineQueryMetadataServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -92,6 +104,18 @@ class OfflineQueryMetadataServiceServicer(metaclass=ABCMeta):
         request: CreateModelTrainingJobRequest,
         context: ServicerContext,
     ) -> CreateModelTrainingJobResponse: ...
+    @abstractmethod
+    def IngestDataset(
+        self,
+        request: IngestDatasetRequest,
+        context: ServicerContext,
+    ) -> IngestDatasetResponse: ...
+    @abstractmethod
+    def RetryOfflineQueryShard(
+        self,
+        request: RetryOfflineQueryShardRequest,
+        context: ServicerContext,
+    ) -> RetryOfflineQueryShardResponse: ...
 
 def add_OfflineQueryMetadataServiceServicer_to_server(
     servicer: OfflineQueryMetadataServiceServicer, server: Server

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: ...