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

chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi

@@ -0,0 +1,88 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.streaming.v1.debug_service_pb2 import (
+    DisableDebugModeRequest,
+    DisableDebugModeResponse,
+    EnableDebugModeRequest,
+    EnableDebugModeResponse,
+    GetDebugMessagesRequest,
+    GetDebugMessagesResponse,
+    GetDebugModeStatusRequest,
+    GetDebugModeStatusResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class StreamingDebugServiceStub:
+    """Streaming Debug Service
+    Provides APIs for managing and querying streaming resolver debug mode
+    """
+
+    def __init__(self, channel: Channel) -> None: ...
+    EnableDebugMode: UnaryUnaryMultiCallable[
+        EnableDebugModeRequest,
+        EnableDebugModeResponse,
+    ]
+    """Enable debug mode for a streaming resolver"""
+    DisableDebugMode: UnaryUnaryMultiCallable[
+        DisableDebugModeRequest,
+        DisableDebugModeResponse,
+    ]
+    """Disable debug mode for a streaming resolver"""
+    GetDebugModeStatus: UnaryUnaryMultiCallable[
+        GetDebugModeStatusRequest,
+        GetDebugModeStatusResponse,
+    ]
+    """Get the current debug mode status for a resolver"""
+    GetDebugMessages: UnaryUnaryMultiCallable[
+        GetDebugMessagesRequest,
+        GetDebugMessagesResponse,
+    ]
+    """Get recent debug messages as parquet"""
+
+class StreamingDebugServiceServicer(metaclass=ABCMeta):
+    """Streaming Debug Service
+    Provides APIs for managing and querying streaming resolver debug mode
+    """
+
+    @abstractmethod
+    def EnableDebugMode(
+        self,
+        request: EnableDebugModeRequest,
+        context: ServicerContext,
+    ) -> EnableDebugModeResponse:
+        """Enable debug mode for a streaming resolver"""
+    @abstractmethod
+    def DisableDebugMode(
+        self,
+        request: DisableDebugModeRequest,
+        context: ServicerContext,
+    ) -> DisableDebugModeResponse:
+        """Disable debug mode for a streaming resolver"""
+    @abstractmethod
+    def GetDebugModeStatus(
+        self,
+        request: GetDebugModeStatusRequest,
+        context: ServicerContext,
+    ) -> GetDebugModeStatusResponse:
+        """Get the current debug mode status for a resolver"""
+    @abstractmethod
+    def GetDebugMessages(
+        self,
+        request: GetDebugMessagesRequest,
+        context: ServicerContext,
+    ) -> GetDebugMessagesResponse:
+        """Get recent debug messages as parquet"""
+
+def add_StreamingDebugServiceServicer_to_server(servicer: StreamingDebugServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py

@@ -18,7 +18,7 @@ from chalk._gen.chalk.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\xb1\x01\n!SimpleStreamingUnaryInvokeRequest\x12\x34\n\x16streaming_resolver_fqn\x18\x01 \x01(\tR\x14streamingResolverFqn\x12\x1d\n\ninput_data\x18\x02 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x42\x0f\n\r_operation_id"\xe7\x02\n"SimpleStreamingUnaryInvokeResponse\x12(\n\x10num_rows_succeed\x18\x01 \x01(\x03R\x0enumRowsSucceed\x12&\n\x0fnum_rows_failed\x18\x02 \x01(\x03R\rnumRowsFailed\x12(\n\x10num_rows_skipped\x18\x03 \x01(\x03R\x0enumRowsSkipped\x12\x36\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12$\n\x0boutput_data\x18\x05 \x01(\x0cH\x01R\noutputData\x88\x01\x01\x12M\n\x10\x65xecution_errors\x18\x06 \x03(\x0b\x32".chalk.streaming.v1.StreamingErrorR\x0f\x65xecutionErrorsB\x08\n\x06_errorB\x0e\n\x0c_output_data*\xc5\x01\n\x0e\x45xecutionPhase\x12\x1f\n\x1b\x45XECUTION_PHASE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_PHASE_PARSE\x10\x01\x12\x1b\n\x17\x45XECUTION_PHASE_MAPPING\x10\x02\x12\x1f\n\x1b\x45XECUTION_PHASE_PERSISTENCE\x10\x03\x12\x1f\n\x1b\x45XECUTION_PHASE_AGGREGATION\x10\x04\x12\x18\n\x14\x45XECUTION_PHASE_SINK\x10\x05\x32\xab\x01\n\x16SimpleStreamingService\x12\x90\x01\n\x1aSimpleStreamingUnaryInvoke\x12\x35.chalk.streaming.v1.SimpleStreamingUnaryInvokeRequest\x1a\x36.chalk.streaming.v1.SimpleStreamingUnaryInvokeResponse"\x03\x80}\x02\x42\x9f\x01\n\x16\x63om.chalk.streaming.v1B\x1bSimpleStreamingServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
+    b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\xd6\x01\n!SimpleStreamingUnaryInvokeRequest\x12\x34\n\x16streaming_resolver_fqn\x18\x01 \x01(\tR\x14streamingResolverFqn\x12\x1d\n\ninput_data\x18\x02 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x04 \x01(\x08H\x01R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_operation_idB\x08\n\x06_debug"\xe7\x02\n"SimpleStreamingUnaryInvokeResponse\x12(\n\x10num_rows_succeed\x18\x01 \x01(\x03R\x0enumRowsSucceed\x12&\n\x0fnum_rows_failed\x18\x02 \x01(\x03R\rnumRowsFailed\x12(\n\x10num_rows_skipped\x18\x03 \x01(\x03R\x0enumRowsSkipped\x12\x36\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12$\n\x0boutput_data\x18\x05 \x01(\x0cH\x01R\noutputData\x88\x01\x01\x12M\n\x10\x65xecution_errors\x18\x06 \x03(\x0b\x32".chalk.streaming.v1.StreamingErrorR\x0f\x65xecutionErrorsB\x08\n\x06_errorB\x0e\n\x0c_output_data"\xb5\x02\n\x1cTestStreamingResolverRequest\x12&\n\x0cresolver_fqn\x18\x01 \x01(\tH\x00R\x0bresolverFqn\x88\x01\x01\x12@\n\x1astatic_stream_resolver_b64\x18\x02 \x01(\tH\x01R\x17staticStreamResolverB64\x88\x01\x01\x12\x1d\n\ninput_data\x18\x03 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x04 \x01(\tH\x02R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x03R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_resolver_fqnB\x1d\n\x1b_static_stream_resolver_b64B\x0f\n\r_operation_idB\x08\n\x06_debug"\xe1\x01\n\x1dTestStreamingResolverResponse\x12\x44\n\x06status\x18\x01 \x01(\x0e\x32,.chalk.streaming.v1.TestStreamResolverStatusR\x06status\x12\x1e\n\x08\x64\x61ta_uri\x18\x02 \x01(\tH\x00R\x07\x64\x61taUri\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12\x18\n\x07message\x18\x04 \x01(\tR\x07messageB\x0b\n\t_data_uri*\xc5\x01\n\x0e\x45xecutionPhase\x12\x1f\n\x1b\x45XECUTION_PHASE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_PHASE_PARSE\x10\x01\x12\x1b\n\x17\x45XECUTION_PHASE_MAPPING\x10\x02\x12\x1f\n\x1b\x45XECUTION_PHASE_PERSISTENCE\x10\x03\x12\x1f\n\x1b\x45XECUTION_PHASE_AGGREGATION\x10\x04\x12\x18\n\x14\x45XECUTION_PHASE_SINK\x10\x05*\x99\x01\n\x18TestStreamResolverStatus\x12+\n\'TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED\x10\x00\x12\'\n#TEST_STREAM_RESOLVER_STATUS_SUCCESS\x10\x01\x12\'\n#TEST_STREAM_RESOLVER_STATUS_FAILURE\x10\x02\x32\xaf\x02\n\x16SimpleStreamingService\x12\x90\x01\n\x1aSimpleStreamingUnaryInvoke\x12\x35.chalk.streaming.v1.SimpleStreamingUnaryInvokeRequest\x1a\x36.chalk.streaming.v1.SimpleStreamingUnaryInvokeResponse"\x03\x80}\x02\x12\x81\x01\n\x15TestStreamingResolver\x12\x30.chalk.streaming.v1.TestStreamingResolverRequest\x1a\x31.chalk.streaming.v1.TestStreamingResolverResponse"\x03\x80}\x02\x42\x9f\x01\n\x16\x63om.chalk.streaming.v1B\x1bSimpleStreamingServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -29,18 +29,24 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals[
         "DESCRIPTOR"
     ]._serialized_options = b"\n\026com.chalk.streaming.v1B\033SimpleStreamingServiceProtoP\001\242\002\003CSX\252\002\022Chalk.Streaming.V1\312\002\022Chalk\\Streaming\\V1\342\002\036Chalk\\Streaming\\V1\\GPBMetadata\352\002\024Chalk::Streaming::V1"
-    _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"].fields_by_name["error"]._options = None
-    _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"].fields_by_name["error"]._serialized_options = b"\030\001"
     _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["SimpleStreamingUnaryInvoke"]._options = None
     _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["SimpleStreamingUnaryInvoke"]._serialized_options = b"\200}\002"
-    _globals["_EXECUTIONPHASE"]._serialized_start = 847
-    _globals["_EXECUTIONPHASE"]._serialized_end = 1044
+    _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["TestStreamingResolver"]._options = None
+    _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["TestStreamingResolver"]._serialized_options = b"\200}\002"
+    _globals["_EXECUTIONPHASE"]._serialized_start = 1424
+    _globals["_EXECUTIONPHASE"]._serialized_end = 1621
+    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_start = 1624
+    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_end = 1777
     _globals["_STREAMINGERROR"]._serialized_start = 142
     _globals["_STREAMINGERROR"]._serialized_end = 302
     _globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_start = 305
-    _globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_end = 482
-    _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_start = 485
-    _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_end = 844
-    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 1047
-    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 1218
+    _globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_end = 519
+    _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_start = 522
+    _globals["_SIMPLESTREAMINGUNARYINVOKERESPONSE"]._serialized_end = 881
+    _globals["_TESTSTREAMINGRESOLVERREQUEST"]._serialized_start = 884
+    _globals["_TESTSTREAMINGRESOLVERREQUEST"]._serialized_end = 1193
+    _globals["_TESTSTREAMINGRESOLVERRESPONSE"]._serialized_start = 1196
+    _globals["_TESTSTREAMINGRESOLVERRESPONSE"]._serialized_end = 1421
+    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 1780
+    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 2083
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi

@@ -23,12 +23,21 @@ class ExecutionPhase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     EXECUTION_PHASE_AGGREGATION: _ClassVar[ExecutionPhase]
     EXECUTION_PHASE_SINK: _ClassVar[ExecutionPhase]
 
+class TestStreamResolverStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED: _ClassVar[TestStreamResolverStatus]
+    TEST_STREAM_RESOLVER_STATUS_SUCCESS: _ClassVar[TestStreamResolverStatus]
+    TEST_STREAM_RESOLVER_STATUS_FAILURE: _ClassVar[TestStreamResolverStatus]
+
 EXECUTION_PHASE_UNSPECIFIED: ExecutionPhase
 EXECUTION_PHASE_PARSE: ExecutionPhase
 EXECUTION_PHASE_MAPPING: ExecutionPhase
 EXECUTION_PHASE_PERSISTENCE: ExecutionPhase
 EXECUTION_PHASE_AGGREGATION: ExecutionPhase
 EXECUTION_PHASE_SINK: ExecutionPhase
+TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED: TestStreamResolverStatus
+TEST_STREAM_RESOLVER_STATUS_SUCCESS: TestStreamResolverStatus
+TEST_STREAM_RESOLVER_STATUS_FAILURE: TestStreamResolverStatus
 
 class StreamingError(_message.Message):
     __slots__ = ("error", "phase", "operation_id")
@@ -46,18 +55,21 @@ class StreamingError(_message.Message):
     ) -> None: ...
 
 class SimpleStreamingUnaryInvokeRequest(_message.Message):
-    __slots__ = ("streaming_resolver_fqn", "input_data", "operation_id")
+    __slots__ = ("streaming_resolver_fqn", "input_data", "operation_id", "debug")
     STREAMING_RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
     INPUT_DATA_FIELD_NUMBER: _ClassVar[int]
     OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    DEBUG_FIELD_NUMBER: _ClassVar[int]
     streaming_resolver_fqn: str
     input_data: bytes
     operation_id: str
+    debug: bool
     def __init__(
         self,
         streaming_resolver_fqn: _Optional[str] = ...,
         input_data: _Optional[bytes] = ...,
         operation_id: _Optional[str] = ...,
+        debug: bool = ...,
     ) -> None: ...
 
 class SimpleStreamingUnaryInvokeResponse(_message.Message):
@@ -83,3 +95,42 @@ class SimpleStreamingUnaryInvokeResponse(_message.Message):
         output_data: _Optional[bytes] = ...,
         execution_errors: _Optional[_Iterable[_Union[StreamingError, _Mapping]]] = ...,
     ) -> None: ...
+
+class TestStreamingResolverRequest(_message.Message):
+    __slots__ = ("resolver_fqn", "static_stream_resolver_b64", "input_data", "operation_id", "debug")
+    RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
+    STATIC_STREAM_RESOLVER_B64_FIELD_NUMBER: _ClassVar[int]
+    INPUT_DATA_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    DEBUG_FIELD_NUMBER: _ClassVar[int]
+    resolver_fqn: str
+    static_stream_resolver_b64: str
+    input_data: bytes
+    operation_id: str
+    debug: bool
+    def __init__(
+        self,
+        resolver_fqn: _Optional[str] = ...,
+        static_stream_resolver_b64: _Optional[str] = ...,
+        input_data: _Optional[bytes] = ...,
+        operation_id: _Optional[str] = ...,
+        debug: bool = ...,
+    ) -> None: ...
+
+class TestStreamingResolverResponse(_message.Message):
+    __slots__ = ("status", "data_uri", "errors", "message")
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    DATA_URI_FIELD_NUMBER: _ClassVar[int]
+    ERRORS_FIELD_NUMBER: _ClassVar[int]
+    MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    status: TestStreamResolverStatus
+    data_uri: str
+    errors: _containers.RepeatedCompositeFieldContainer[_chalk_error_pb2.ChalkError]
+    message: str
+    def __init__(
+        self,
+        status: _Optional[_Union[TestStreamResolverStatus, str]] = ...,
+        data_uri: _Optional[str] = ...,
+        errors: _Optional[_Iterable[_Union[_chalk_error_pb2.ChalkError, _Mapping]]] = ...,
+        message: _Optional[str] = ...,
+    ) -> None: ...

chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py

@@ -22,6 +22,11 @@ class SimpleStreamingServiceStub(object):
             request_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeRequest.SerializeToString,
             response_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeResponse.FromString,
         )
+        self.TestStreamingResolver = channel.unary_unary(
+            "/chalk.streaming.v1.SimpleStreamingService/TestStreamingResolver",
+            request_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverRequest.SerializeToString,
+            response_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverResponse.FromString,
+        )
 
 
 class SimpleStreamingServiceServicer(object):
@@ -35,6 +40,15 @@ class SimpleStreamingServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def TestStreamingResolver(self, request, context):
+        """Test a streaming resolver with provided messages.
+        Supports testing both deployed resolvers and static/undeployed resolvers.
+        Returns a signed URL to download the test results.
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
 
 def add_SimpleStreamingServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -43,6 +57,11 @@ def add_SimpleStreamingServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeRequest.FromString,
             response_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.SimpleStreamingUnaryInvokeResponse.SerializeToString,
         ),
+        "TestStreamingResolver": grpc.unary_unary_rpc_method_handler(
+            servicer.TestStreamingResolver,
+            request_deserializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverRequest.FromString,
+            response_serializer=chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
         "chalk.streaming.v1.SimpleStreamingService", rpc_method_handlers
@@ -82,3 +101,32 @@ class SimpleStreamingService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def TestStreamingResolver(
+        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.streaming.v1.SimpleStreamingService/TestStreamingResolver",
+            chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverRequest.SerializeToString,
+            chalk_dot_streaming_dot_v1_dot_simple__streaming__service__pb2.TestStreamingResolverResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi

@@ -10,6 +10,8 @@ from abc import (
 from chalk._gen.chalk.streaming.v1.simple_streaming_service_pb2 import (
     SimpleStreamingUnaryInvokeRequest,
     SimpleStreamingUnaryInvokeResponse,
+    TestStreamingResolverRequest,
+    TestStreamingResolverResponse,
 )
 from grpc import (
     Channel,
@@ -27,6 +29,14 @@ class SimpleStreamingServiceStub:
     """Runs a simple streaming plan with the given request.
     This is a simplified version of the streaming invoker service.
     """
+    TestStreamingResolver: UnaryUnaryMultiCallable[
+        TestStreamingResolverRequest,
+        TestStreamingResolverResponse,
+    ]
+    """Test a streaming resolver with provided messages.
+    Supports testing both deployed resolvers and static/undeployed resolvers.
+    Returns a signed URL to download the test results.
+    """
 
 class SimpleStreamingServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -38,5 +48,15 @@ class SimpleStreamingServiceServicer(metaclass=ABCMeta):
         """Runs a simple streaming plan with the given request.
         This is a simplified version of the streaming invoker service.
         """
+    @abstractmethod
+    def TestStreamingResolver(
+        self,
+        request: TestStreamingResolverRequest,
+        context: ServicerContext,
+    ) -> TestStreamingResolverResponse:
+        """Test a streaming resolver with provided messages.
+        Supports testing both deployed resolvers and static/undeployed resolvers.
+        Returns a signed URL to download the test results.
+        """
 
 def add_SimpleStreamingServiceServicer_to_server(servicer: SimpleStreamingServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/utils/v1/field_change_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/utils/v1/field_change.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()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n!chalk/utils/v1/field_change.proto\x12\x0e\x63halk.utils.v1"\xe0\x01\n\x0b\x46ieldChange\x12\x1d\n\nfield_name\x18\x01 \x01(\tR\tfieldName\x12*\n\x0eprevious_value\x18\x02 \x01(\tH\x00R\rpreviousValue\x88\x01\x01\x12 \n\tnew_value\x18\x03 \x01(\tH\x01R\x08newValue\x88\x01\x01\x12\x43\n\x0enested_changes\x18\x04 \x03(\x0b\x32\x1c.chalk.utils.v1.NestedChangeR\rnestedChangesB\x11\n\x0f_previous_valueB\x0c\n\n_new_value"\x91\x01\n\x0cNestedChange\x12\x12\n\x04path\x18\x01 \x01(\tR\x04path\x12*\n\x0eprevious_value\x18\x02 \x01(\tH\x00R\rpreviousValue\x88\x01\x01\x12 \n\tnew_value\x18\x03 \x01(\tH\x01R\x08newValue\x88\x01\x01\x42\x11\n\x0f_previous_valueB\x0c\n\n_new_valueB\x80\x01\n\x12\x63om.chalk.utils.v1B\x10\x46ieldChangeProtoP\x01\xa2\x02\x03\x43UX\xaa\x02\x0e\x43halk.Utils.V1\xca\x02\x0e\x43halk\\Utils\\V1\xe2\x02\x1a\x43halk\\Utils\\V1\\GPBMetadata\xea\x02\x10\x43halk::Utils::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.utils.v1.field_change_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\022com.chalk.utils.v1B\020FieldChangeProtoP\001\242\002\003CUX\252\002\016Chalk.Utils.V1\312\002\016Chalk\\Utils\\V1\342\002\032Chalk\\Utils\\V1\\GPBMetadata\352\002\020Chalk::Utils::V1"
+    _globals["_FIELDCHANGE"]._serialized_start = 54
+    _globals["_FIELDCHANGE"]._serialized_end = 278
+    _globals["_NESTEDCHANGE"]._serialized_start = 281
+    _globals["_NESTEDCHANGE"]._serialized_end = 426
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/utils/v1/field_change_pb2.pyi

@@ -0,0 +1,42 @@
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class FieldChange(_message.Message):
+    __slots__ = ("field_name", "previous_value", "new_value", "nested_changes")
+    FIELD_NAME_FIELD_NUMBER: _ClassVar[int]
+    PREVIOUS_VALUE_FIELD_NUMBER: _ClassVar[int]
+    NEW_VALUE_FIELD_NUMBER: _ClassVar[int]
+    NESTED_CHANGES_FIELD_NUMBER: _ClassVar[int]
+    field_name: str
+    previous_value: str
+    new_value: str
+    nested_changes: _containers.RepeatedCompositeFieldContainer[NestedChange]
+    def __init__(
+        self,
+        field_name: _Optional[str] = ...,
+        previous_value: _Optional[str] = ...,
+        new_value: _Optional[str] = ...,
+        nested_changes: _Optional[_Iterable[_Union[NestedChange, _Mapping]]] = ...,
+    ) -> None: ...
+
+class NestedChange(_message.Message):
+    __slots__ = ("path", "previous_value", "new_value")
+    PATH_FIELD_NUMBER: _ClassVar[int]
+    PREVIOUS_VALUE_FIELD_NUMBER: _ClassVar[int]
+    NEW_VALUE_FIELD_NUMBER: _ClassVar[int]
+    path: str
+    previous_value: str
+    new_value: str
+    def __init__(
+        self, path: _Optional[str] = ..., previous_value: _Optional[str] = ..., new_value: _Optional[str] = ...
+    ) -> None: ...

chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+
+import grpc

chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi

@@ -0,0 +1,4 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""

chalk/_lsp/error_builder.py

@@ -186,6 +186,17 @@ class LSPErrorBuilder:
         Returns whether the exception was promoted."""
         if id(e) in cls._exception_map:
             uri, diagnostic = cls._exception_map[id(e)]
+
+            # Check if this diagnostic already exists (deduplication)
+            # Compare by message, range, and uri to detect duplicates
+            for existing in cls.all_errors[uri]:
+                if existing.message == diagnostic.message and existing.range == diagnostic.range:
+                    # Already exists, don't add duplicate
+                    del cls._exception_map[id(e)]
+                    del cls._strong_refs[id(e)]
+                    return False  # Not promoted, already exists
+
+            # Not a duplicate, add it
             cls.all_errors[uri].append(diagnostic)
             del cls._exception_map[id(e)]
             del cls._strong_refs[id(e)]

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.90.1"
+__version__ = "2.95.3"