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

chalk/_gen/chalk/server/v1/trace_pb2_grpc.py

@@ -20,6 +20,21 @@ class TraceServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetTraceRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetTraceResponse.FromString,
         )
+        self.ListTrace = channel.unary_unary(
+            "/chalk.server.v1.TraceService/ListTrace",
+            request_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListTraceRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListTraceResponse.FromString,
+        )
+        self.GetSpan = channel.unary_unary(
+            "/chalk.server.v1.TraceService/GetSpan",
+            request_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetSpanRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetSpanResponse.FromString,
+        )
+        self.ListSpan = channel.unary_unary(
+            "/chalk.server.v1.TraceService/ListSpan",
+            request_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListSpanRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListSpanResponse.FromString,
+        )
 
 
 class TraceServiceServicer(object):
@@ -31,6 +46,24 @@ class TraceServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def ListTrace(self, request, context):
+        """ListTrace retrieves a list of traces with optional filtering"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def GetSpan(self, request, context):
+        """GetSpan retrieves a specific span by span ID and trace ID"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+    def ListSpan(self, request, context):
+        """ListSpan retrieves a list of spans for a specific trace with optional filtering"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
 
 def add_TraceServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -39,6 +72,21 @@ def add_TraceServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetTraceRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetTraceResponse.SerializeToString,
         ),
+        "ListTrace": grpc.unary_unary_rpc_method_handler(
+            servicer.ListTrace,
+            request_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListTraceRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListTraceResponse.SerializeToString,
+        ),
+        "GetSpan": grpc.unary_unary_rpc_method_handler(
+            servicer.GetSpan,
+            request_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetSpanRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.GetSpanResponse.SerializeToString,
+        ),
+        "ListSpan": grpc.unary_unary_rpc_method_handler(
+            servicer.ListSpan,
+            request_deserializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListSpanRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_trace__pb2.ListSpanResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.TraceService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -76,3 +124,90 @@ class TraceService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def ListTrace(
+        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.TraceService/ListTrace",
+            chalk_dot_server_dot_v1_dot_trace__pb2.ListTraceRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_trace__pb2.ListTraceResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetSpan(
+        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.TraceService/GetSpan",
+            chalk_dot_server_dot_v1_dot_trace__pb2.GetSpanRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_trace__pb2.GetSpanResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def ListSpan(
+        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.TraceService/ListSpan",
+            chalk_dot_server_dot_v1_dot_trace__pb2.ListSpanRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_trace__pb2.ListSpanResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/trace_pb2_grpc.pyi

@@ -8,8 +8,14 @@ from abc import (
     abstractmethod,
 )
 from chalk._gen.chalk.server.v1.trace_pb2 import (
+    GetSpanRequest,
+    GetSpanResponse,
     GetTraceRequest,
     GetTraceResponse,
+    ListSpanRequest,
+    ListSpanResponse,
+    ListTraceRequest,
+    ListTraceResponse,
 )
 from grpc import (
     Channel,
@@ -27,6 +33,21 @@ class TraceServiceStub:
         GetTraceResponse,
     ]
     """GetTrace retrieves a trace, optionally by operation ID"""
+    ListTrace: UnaryUnaryMultiCallable[
+        ListTraceRequest,
+        ListTraceResponse,
+    ]
+    """ListTrace retrieves a list of traces with optional filtering"""
+    GetSpan: UnaryUnaryMultiCallable[
+        GetSpanRequest,
+        GetSpanResponse,
+    ]
+    """GetSpan retrieves a specific span by span ID and trace ID"""
+    ListSpan: UnaryUnaryMultiCallable[
+        ListSpanRequest,
+        ListSpanResponse,
+    ]
+    """ListSpan retrieves a list of spans for a specific trace with optional filtering"""
 
 class TraceServiceServicer(metaclass=ABCMeta):
     """TraceService provides methods for retrieving trace data"""
@@ -38,5 +59,26 @@ class TraceServiceServicer(metaclass=ABCMeta):
         context: ServicerContext,
     ) -> GetTraceResponse:
         """GetTrace retrieves a trace, optionally by operation ID"""
+    @abstractmethod
+    def ListTrace(
+        self,
+        request: ListTraceRequest,
+        context: ServicerContext,
+    ) -> ListTraceResponse:
+        """ListTrace retrieves a list of traces with optional filtering"""
+    @abstractmethod
+    def GetSpan(
+        self,
+        request: GetSpanRequest,
+        context: ServicerContext,
+    ) -> GetSpanResponse:
+        """GetSpan retrieves a specific span by span ID and trace ID"""
+    @abstractmethod
+    def ListSpan(
+        self,
+        request: ListSpanRequest,
+        context: ServicerContext,
+    ) -> ListSpanResponse:
+        """ListSpan retrieves a list of spans for a specific trace with optional filtering"""
 
 def add_TraceServiceServicer_to_server(servicer: TraceServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/webhook_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\x1d\x63halk/server/v1/webhook.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x87\x03\n\x07Webhook\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12\x17\n\x07team_id\x18\x03 \x01(\tR\x06teamId\x12\x12\n\x04name\x18\x04 \x01(\tR\x04name\x12\x10\n\x03url\x18\x05 \x01(\tR\x03url\x12$\n\rsubscriptions\x18\x06 \x03(\tR\rsubscriptions\x12\x1b\n\x06secret\x18\x07 \x01(\tH\x00R\x06secret\x88\x01\x01\x12\x36\n\x07headers\x18\x08 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x07headers\x88\x01\x01\x12\x39\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAtB\t\n\x07_secretB\n\n\x08_headers"\xf5\x01\n\x14\x43reateWebhookRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x10\n\x03url\x18\x03 \x01(\tR\x03url\x12$\n\rsubscriptions\x18\x04 \x03(\tR\rsubscriptions\x12\x1b\n\x06secret\x18\x05 \x01(\tH\x00R\x06secret\x88\x01\x01\x12\x36\n\x07headers\x18\x06 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x07headers\x88\x01\x01\x42\t\n\x07_secretB\n\n\x08_headers"K\n\x15\x43reateWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"\xf9\x01\n\x14UpdateWebhookRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x17\n\x04name\x18\x02 \x01(\tH\x00R\x04name\x88\x01\x01\x12\x15\n\x03url\x18\x03 \x01(\tH\x01R\x03url\x88\x01\x01\x12$\n\rsubscriptions\x18\x04 \x03(\tR\rsubscriptions\x12\x1b\n\x06secret\x18\x05 \x01(\tH\x02R\x06secret\x88\x01\x01\x12\x36\n\x07headers\x18\x06 \x01(\x0b\x32\x17.google.protobuf.StructH\x03R\x07headers\x88\x01\x01\x42\x07\n\x05_nameB\x06\n\x04_urlB\t\n\x07_secretB\n\n\x08_headers"K\n\x15UpdateWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"&\n\x14\x44\x65leteWebhookRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"K\n\x15\x44\x65leteWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"#\n\x11GetWebhookRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"H\n\x12GetWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"<\n\x13ListWebhooksRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId"L\n\x14ListWebhooksResponse\x12\x34\n\x08webhooks\x18\x01 \x03(\x0b\x32\x18.chalk.server.v1.WebhookR\x08webhooks2\x83\x04\n\x0eWebhookService\x12\x63\n\rCreateWebhook\x12%.chalk.server.v1.CreateWebhookRequest\x1a&.chalk.server.v1.CreateWebhookResponse"\x03\x80}\x05\x12\x63\n\rUpdateWebhook\x12%.chalk.server.v1.UpdateWebhookRequest\x1a&.chalk.server.v1.UpdateWebhookResponse"\x03\x80}\x05\x12\x63\n\rDeleteWebhook\x12%.chalk.server.v1.DeleteWebhookRequest\x1a&.chalk.server.v1.DeleteWebhookResponse"\x03\x80}\x05\x12]\n\nGetWebhook\x12".chalk.server.v1.GetWebhookRequest\x1a#.chalk.server.v1.GetWebhookResponse"\x06\x90\x02\x01\x80}\x06\x12\x63\n\x0cListWebhooks\x12$.chalk.server.v1.ListWebhooksRequest\x1a%.chalk.server.v1.ListWebhooksResponse"\x06\x90\x02\x01\x80}\x06\x42\x95\x01\n\x13\x63om.chalk.server.v1B\x0cWebhookProtoP\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\x1d\x63halk/server/v1/webhook.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x87\x03\n\x07Webhook\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12\x17\n\x07team_id\x18\x03 \x01(\tR\x06teamId\x12\x12\n\x04name\x18\x04 \x01(\tR\x04name\x12\x10\n\x03url\x18\x05 \x01(\tR\x03url\x12$\n\rsubscriptions\x18\x06 \x03(\tR\rsubscriptions\x12\x1b\n\x06secret\x18\x07 \x01(\tH\x00R\x06secret\x88\x01\x01\x12\x36\n\x07headers\x18\x08 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x07headers\x88\x01\x01\x12\x39\n\ncreated_at\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\n \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAtB\t\n\x07_secretB\n\n\x08_headers"\xf5\x01\n\x14\x43reateWebhookRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x10\n\x03url\x18\x03 \x01(\tR\x03url\x12$\n\rsubscriptions\x18\x04 \x03(\tR\rsubscriptions\x12\x1b\n\x06secret\x18\x05 \x01(\tH\x00R\x06secret\x88\x01\x01\x12\x36\n\x07headers\x18\x06 \x01(\x0b\x32\x17.google.protobuf.StructH\x01R\x07headers\x88\x01\x01\x42\t\n\x07_secretB\n\n\x08_headers"K\n\x15\x43reateWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"\xf9\x01\n\x14UpdateWebhookRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x17\n\x04name\x18\x02 \x01(\tH\x00R\x04name\x88\x01\x01\x12\x15\n\x03url\x18\x03 \x01(\tH\x01R\x03url\x88\x01\x01\x12$\n\rsubscriptions\x18\x04 \x03(\tR\rsubscriptions\x12\x1b\n\x06secret\x18\x05 \x01(\tH\x02R\x06secret\x88\x01\x01\x12\x36\n\x07headers\x18\x06 \x01(\x0b\x32\x17.google.protobuf.StructH\x03R\x07headers\x88\x01\x01\x42\x07\n\x05_nameB\x06\n\x04_urlB\t\n\x07_secretB\n\n\x08_headers"K\n\x15UpdateWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"&\n\x14\x44\x65leteWebhookRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"K\n\x15\x44\x65leteWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"#\n\x11GetWebhookRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"H\n\x12GetWebhookResponse\x12\x32\n\x07webhook\x18\x01 \x01(\x0b\x32\x18.chalk.server.v1.WebhookR\x07webhook"<\n\x13ListWebhooksRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId"L\n\x14ListWebhooksResponse\x12\x34\n\x08webhooks\x18\x01 \x03(\x0b\x32\x18.chalk.server.v1.WebhookR\x08webhooks"$\n\x12TestWebhookRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"u\n\x13TestWebhookResponse\x12\x18\n\x07success\x18\x01 \x01(\x08R\x07success\x12\x1f\n\x0bstatus_code\x18\x02 \x01(\x05R\nstatusCode\x12#\n\rerror_message\x18\x03 \x01(\tR\x0c\x65rrorMessage2\xe2\x04\n\x0eWebhookService\x12\x63\n\rCreateWebhook\x12%.chalk.server.v1.CreateWebhookRequest\x1a&.chalk.server.v1.CreateWebhookResponse"\x03\x80}\x05\x12\x63\n\rUpdateWebhook\x12%.chalk.server.v1.UpdateWebhookRequest\x1a&.chalk.server.v1.UpdateWebhookResponse"\x03\x80}\x05\x12\x63\n\rDeleteWebhook\x12%.chalk.server.v1.DeleteWebhookRequest\x1a&.chalk.server.v1.DeleteWebhookResponse"\x03\x80}\x05\x12]\n\nGetWebhook\x12".chalk.server.v1.GetWebhookRequest\x1a#.chalk.server.v1.GetWebhookResponse"\x06\x90\x02\x01\x80}\x06\x12\x63\n\x0cListWebhooks\x12$.chalk.server.v1.ListWebhooksRequest\x1a%.chalk.server.v1.ListWebhooksResponse"\x06\x90\x02\x01\x80}\x06\x12]\n\x0bTestWebhook\x12#.chalk.server.v1.TestWebhookRequest\x1a$.chalk.server.v1.TestWebhookResponse"\x03\x80}\x05\x42\x95\x01\n\x13\x63om.chalk.server.v1B\x0cWebhookProtoP\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,6 +40,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_WEBHOOKSERVICE"].methods_by_name["GetWebhook"]._serialized_options = b"\220\002\001\200}\006"
     _globals["_WEBHOOKSERVICE"].methods_by_name["ListWebhooks"]._options = None
     _globals["_WEBHOOKSERVICE"].methods_by_name["ListWebhooks"]._serialized_options = b"\220\002\001\200}\006"
+    _globals["_WEBHOOKSERVICE"].methods_by_name["TestWebhook"]._options = None
+    _globals["_WEBHOOKSERVICE"].methods_by_name["TestWebhook"]._serialized_options = b"\200}\005"
     _globals["_WEBHOOK"]._serialized_start = 147
     _globals["_WEBHOOK"]._serialized_end = 538
     _globals["_CREATEWEBHOOKREQUEST"]._serialized_start = 541
@@ -62,6 +64,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_LISTWEBHOOKSREQUEST"]._serialized_end = 1482
     _globals["_LISTWEBHOOKSRESPONSE"]._serialized_start = 1484
     _globals["_LISTWEBHOOKSRESPONSE"]._serialized_end = 1560
-    _globals["_WEBHOOKSERVICE"]._serialized_start = 1563
-    _globals["_WEBHOOKSERVICE"]._serialized_end = 2078
+    _globals["_TESTWEBHOOKREQUEST"]._serialized_start = 1562
+    _globals["_TESTWEBHOOKREQUEST"]._serialized_end = 1598
+    _globals["_TESTWEBHOOKRESPONSE"]._serialized_start = 1600
+    _globals["_TESTWEBHOOKRESPONSE"]._serialized_end = 1717
+    _globals["_WEBHOOKSERVICE"]._serialized_start = 1720
+    _globals["_WEBHOOKSERVICE"]._serialized_end = 2330
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/webhook_pb2.pyi

@@ -156,3 +156,21 @@ class ListWebhooksResponse(_message.Message):
     WEBHOOKS_FIELD_NUMBER: _ClassVar[int]
     webhooks: _containers.RepeatedCompositeFieldContainer[Webhook]
     def __init__(self, webhooks: _Optional[_Iterable[_Union[Webhook, _Mapping]]] = ...) -> None: ...
+
+class TestWebhookRequest(_message.Message):
+    __slots__ = ("id",)
+    ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    def __init__(self, id: _Optional[str] = ...) -> None: ...
+
+class TestWebhookResponse(_message.Message):
+    __slots__ = ("success", "status_code", "error_message")
+    SUCCESS_FIELD_NUMBER: _ClassVar[int]
+    STATUS_CODE_FIELD_NUMBER: _ClassVar[int]
+    ERROR_MESSAGE_FIELD_NUMBER: _ClassVar[int]
+    success: bool
+    status_code: int
+    error_message: str
+    def __init__(
+        self, success: bool = ..., status_code: _Optional[int] = ..., error_message: _Optional[str] = ...
+    ) -> None: ...

chalk/_gen/chalk/server/v1/webhook_pb2_grpc.py

@@ -40,6 +40,11 @@ class WebhookServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_webhook__pb2.ListWebhooksRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_webhook__pb2.ListWebhooksResponse.FromString,
         )
+        self.TestWebhook = channel.unary_unary(
+            "/chalk.server.v1.WebhookService/TestWebhook",
+            request_serializer=chalk_dot_server_dot_v1_dot_webhook__pb2.TestWebhookRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_webhook__pb2.TestWebhookResponse.FromString,
+        )
 
 
 class WebhookServiceServicer(object):
@@ -75,6 +80,12 @@ class WebhookServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def TestWebhook(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_WebhookServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -103,6 +114,11 @@ def add_WebhookServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_webhook__pb2.ListWebhooksRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_webhook__pb2.ListWebhooksResponse.SerializeToString,
         ),
+        "TestWebhook": grpc.unary_unary_rpc_method_handler(
+            servicer.TestWebhook,
+            request_deserializer=chalk_dot_server_dot_v1_dot_webhook__pb2.TestWebhookRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_webhook__pb2.TestWebhookResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.WebhookService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -256,3 +272,32 @@ class WebhookService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def TestWebhook(
+        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.WebhookService/TestWebhook",
+            chalk_dot_server_dot_v1_dot_webhook__pb2.TestWebhookRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_webhook__pb2.TestWebhookResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/webhook_pb2_grpc.pyi

@@ -16,6 +16,8 @@ from chalk._gen.chalk.server.v1.webhook_pb2 import (
     GetWebhookResponse,
     ListWebhooksRequest,
     ListWebhooksResponse,
+    TestWebhookRequest,
+    TestWebhookResponse,
     UpdateWebhookRequest,
     UpdateWebhookResponse,
 )
@@ -48,6 +50,10 @@ class WebhookServiceStub:
         ListWebhooksRequest,
         ListWebhooksResponse,
     ]
+    TestWebhook: UnaryUnaryMultiCallable[
+        TestWebhookRequest,
+        TestWebhookResponse,
+    ]
 
 class WebhookServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -80,5 +86,11 @@ class WebhookServiceServicer(metaclass=ABCMeta):
         request: ListWebhooksRequest,
         context: ServicerContext,
     ) -> ListWebhooksResponse: ...
+    @abstractmethod
+    def TestWebhook(
+        self,
+        request: TestWebhookRequest,
+        context: ServicerContext,
+    ) -> TestWebhookResponse: ...
 
 def add_WebhookServiceServicer_to_server(servicer: WebhookServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/streaming/v1/debug_service_pb2.py

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/streaming/v1/debug_service.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/streaming/v1/debug_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1fgoogle/protobuf/timestamp.proto"`\n\x16\x45nableDebugModeRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId"n\n\x17\x45nableDebugModeResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled\x12\x39\n\nenabled_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tenabledAt"<\n\x17\x44isableDebugModeRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"4\n\x18\x44isableDebugModeResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled">\n\x19GetDebugModeStatusRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"\xe8\x01\n\x1aGetDebugModeStatusResponse\x12\x18\n\x07\x65nabled\x18\x01 \x01(\x08R\x07\x65nabled\x12>\n\nenabled_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\tenabledAt\x88\x01\x01\x12%\n\x0estorage_bucket\x18\x03 \x01(\tR\rstorageBucket\x12(\n\rdeployment_id\x18\x04 \x01(\tH\x01R\x0c\x64\x65ploymentId\x88\x01\x01\x42\r\n\x0b_enabled_atB\x10\n\x0e_deployment_id"<\n\x17GetDebugMessagesRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn"Y\n\x18GetDebugMessagesResponse\x12\x18\n\x07parquet\x18\x01 \x01(\x0cR\x07parquet\x12\x19\n\x05\x65rror\x18\x02 \x01(\tH\x00R\x05\x65rror\x88\x01\x01\x42\x08\n\x06_error2\xf0\x03\n\x15StreamingDebugService\x12o\n\x0f\x45nableDebugMode\x12*.chalk.streaming.v1.EnableDebugModeRequest\x1a+.chalk.streaming.v1.EnableDebugModeResponse"\x03\x80}\x0c\x12r\n\x10\x44isableDebugMode\x12+.chalk.streaming.v1.DisableDebugModeRequest\x1a,.chalk.streaming.v1.DisableDebugModeResponse"\x03\x80}\x0c\x12{\n\x12GetDebugModeStatus\x12-.chalk.streaming.v1.GetDebugModeStatusRequest\x1a..chalk.streaming.v1.GetDebugModeStatusResponse"\x06\x90\x02\x01\x80}\x0b\x12u\n\x10GetDebugMessages\x12+.chalk.streaming.v1.GetDebugMessagesRequest\x1a,.chalk.streaming.v1.GetDebugMessagesResponse"\x06\x90\x02\x01\x80}\x0b\x42\x95\x01\n\x16\x63om.chalk.streaming.v1B\x11\x44\x65\x62ugServiceProtoP\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()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.streaming.v1.debug_service_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\026com.chalk.streaming.v1B\021DebugServiceProtoP\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["_STREAMINGDEBUGSERVICE"].methods_by_name["EnableDebugMode"]._options = None
+    _globals["_STREAMINGDEBUGSERVICE"].methods_by_name["EnableDebugMode"]._serialized_options = b"\200}\014"
+    _globals["_STREAMINGDEBUGSERVICE"].methods_by_name["DisableDebugMode"]._options = None
+    _globals["_STREAMINGDEBUGSERVICE"].methods_by_name["DisableDebugMode"]._serialized_options = b"\200}\014"
+    _globals["_STREAMINGDEBUGSERVICE"].methods_by_name["GetDebugModeStatus"]._options = None
+    _globals["_STREAMINGDEBUGSERVICE"].methods_by_name[
+        "GetDebugModeStatus"
+    ]._serialized_options = b"\220\002\001\200}\013"
+    _globals["_STREAMINGDEBUGSERVICE"].methods_by_name["GetDebugMessages"]._options = None
+    _globals["_STREAMINGDEBUGSERVICE"].methods_by_name[
+        "GetDebugMessages"
+    ]._serialized_options = b"\220\002\001\200}\013"
+    _globals["_ENABLEDEBUGMODEREQUEST"]._serialized_start = 128
+    _globals["_ENABLEDEBUGMODEREQUEST"]._serialized_end = 224
+    _globals["_ENABLEDEBUGMODERESPONSE"]._serialized_start = 226
+    _globals["_ENABLEDEBUGMODERESPONSE"]._serialized_end = 336
+    _globals["_DISABLEDEBUGMODEREQUEST"]._serialized_start = 338
+    _globals["_DISABLEDEBUGMODEREQUEST"]._serialized_end = 398
+    _globals["_DISABLEDEBUGMODERESPONSE"]._serialized_start = 400
+    _globals["_DISABLEDEBUGMODERESPONSE"]._serialized_end = 452
+    _globals["_GETDEBUGMODESTATUSREQUEST"]._serialized_start = 454
+    _globals["_GETDEBUGMODESTATUSREQUEST"]._serialized_end = 516
+    _globals["_GETDEBUGMODESTATUSRESPONSE"]._serialized_start = 519
+    _globals["_GETDEBUGMODESTATUSRESPONSE"]._serialized_end = 751
+    _globals["_GETDEBUGMESSAGESREQUEST"]._serialized_start = 753
+    _globals["_GETDEBUGMESSAGESREQUEST"]._serialized_end = 813
+    _globals["_GETDEBUGMESSAGESRESPONSE"]._serialized_start = 815
+    _globals["_GETDEBUGMESSAGESRESPONSE"]._serialized_end = 904
+    _globals["_STREAMINGDEBUGSERVICE"]._serialized_start = 907
+    _globals["_STREAMINGDEBUGSERVICE"]._serialized_end = 1403
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi

@@ -0,0 +1,75 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_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 EnableDebugModeRequest(_message.Message):
+    __slots__ = ("resolver_fqn", "deployment_id")
+    RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    resolver_fqn: str
+    deployment_id: str
+    def __init__(self, resolver_fqn: _Optional[str] = ..., deployment_id: _Optional[str] = ...) -> None: ...
+
+class EnableDebugModeResponse(_message.Message):
+    __slots__ = ("enabled", "enabled_at")
+    ENABLED_FIELD_NUMBER: _ClassVar[int]
+    ENABLED_AT_FIELD_NUMBER: _ClassVar[int]
+    enabled: bool
+    enabled_at: _timestamp_pb2.Timestamp
+    def __init__(
+        self, enabled: bool = ..., enabled_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...
+    ) -> None: ...
+
+class DisableDebugModeRequest(_message.Message):
+    __slots__ = ("resolver_fqn",)
+    RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
+    resolver_fqn: str
+    def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
+
+class DisableDebugModeResponse(_message.Message):
+    __slots__ = ("enabled",)
+    ENABLED_FIELD_NUMBER: _ClassVar[int]
+    enabled: bool
+    def __init__(self, enabled: bool = ...) -> None: ...
+
+class GetDebugModeStatusRequest(_message.Message):
+    __slots__ = ("resolver_fqn",)
+    RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
+    resolver_fqn: str
+    def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
+
+class GetDebugModeStatusResponse(_message.Message):
+    __slots__ = ("enabled", "enabled_at", "storage_bucket", "deployment_id")
+    ENABLED_FIELD_NUMBER: _ClassVar[int]
+    ENABLED_AT_FIELD_NUMBER: _ClassVar[int]
+    STORAGE_BUCKET_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    enabled: bool
+    enabled_at: _timestamp_pb2.Timestamp
+    storage_bucket: str
+    deployment_id: str
+    def __init__(
+        self,
+        enabled: bool = ...,
+        enabled_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        storage_bucket: _Optional[str] = ...,
+        deployment_id: _Optional[str] = ...,
+    ) -> None: ...
+
+class GetDebugMessagesRequest(_message.Message):
+    __slots__ = ("resolver_fqn",)
+    RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
+    resolver_fqn: str
+    def __init__(self, resolver_fqn: _Optional[str] = ...) -> None: ...
+
+class GetDebugMessagesResponse(_message.Message):
+    __slots__ = ("parquet", "error")
+    PARQUET_FIELD_NUMBER: _ClassVar[int]
+    ERROR_FIELD_NUMBER: _ClassVar[int]
+    parquet: bytes
+    error: str
+    def __init__(self, parquet: _Optional[bytes] = ..., error: _Optional[str] = ...) -> None: ...