chalkpy 2.93.5__py3-none-any.whl → 2.93.7__py3-none-any.whl

chalk/_gen/chalk/server/v1/builder_pb2.pyi

@@ -575,6 +575,7 @@ class ClusterTimescaleSpecs(_message.Message):
         "dns_hostname",
         "bootstrap_cloud_resources",
         "suspended",
+        "ip_allowlist",
     )
     class PostgresParametersEntry(_message.Message):
         __slots__ = ("key", "value")
@@ -618,6 +619,7 @@ class ClusterTimescaleSpecs(_message.Message):
     DNS_HOSTNAME_FIELD_NUMBER: _ClassVar[int]
     BOOTSTRAP_CLOUD_RESOURCES_FIELD_NUMBER: _ClassVar[int]
     SUSPENDED_FIELD_NUMBER: _ClassVar[int]
+    IP_ALLOWLIST_FIELD_NUMBER: _ClassVar[int]
     timescale_image: str
     database_name: str
     database_replicas: int
@@ -644,6 +646,7 @@ class ClusterTimescaleSpecs(_message.Message):
     dns_hostname: str
     bootstrap_cloud_resources: bool
     suspended: bool
+    ip_allowlist: _containers.RepeatedScalarFieldContainer[str]
     def __init__(
         self,
         timescale_image: _Optional[str] = ...,
@@ -672,6 +675,7 @@ class ClusterTimescaleSpecs(_message.Message):
         dns_hostname: _Optional[str] = ...,
         bootstrap_cloud_resources: bool = ...,
         suspended: bool = ...,
+        ip_allowlist: _Optional[_Iterable[str]] = ...,
     ) -> None: ...
 
 class CreateClusterTimescaleDBResponse(_message.Message):
@@ -1326,20 +1330,73 @@ class ClickHouseSpec(_message.Message):
         instance_type: _Optional[str] = ...,
     ) -> None: ...
 
+class ZombieKillerSpec(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class CoreDumpCollectorSpec(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class PySpyStackTraceCollectorSpec(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class ObservabilityDaemonSpec(_message.Message):
+    __slots__ = (
+        "keep_running_when_suspended",
+        "request",
+        "limit",
+        "zombie_killer",
+        "core_dump_collector",
+        "py_spy_stack_trace_collector",
+    )
+    KEEP_RUNNING_WHEN_SUSPENDED_FIELD_NUMBER: _ClassVar[int]
+    REQUEST_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    ZOMBIE_KILLER_FIELD_NUMBER: _ClassVar[int]
+    CORE_DUMP_COLLECTOR_FIELD_NUMBER: _ClassVar[int]
+    PY_SPY_STACK_TRACE_COLLECTOR_FIELD_NUMBER: _ClassVar[int]
+    keep_running_when_suspended: bool
+    request: KubeResourceConfig
+    limit: KubeResourceConfig
+    zombie_killer: ZombieKillerSpec
+    core_dump_collector: CoreDumpCollectorSpec
+    py_spy_stack_trace_collector: PySpyStackTraceCollectorSpec
+    def __init__(
+        self,
+        keep_running_when_suspended: bool = ...,
+        request: _Optional[_Union[KubeResourceConfig, _Mapping]] = ...,
+        limit: _Optional[_Union[KubeResourceConfig, _Mapping]] = ...,
+        zombie_killer: _Optional[_Union[ZombieKillerSpec, _Mapping]] = ...,
+        core_dump_collector: _Optional[_Union[CoreDumpCollectorSpec, _Mapping]] = ...,
+        py_spy_stack_trace_collector: _Optional[_Union[PySpyStackTraceCollectorSpec, _Mapping]] = ...,
+    ) -> None: ...
+
 class TelemetryDeploymentSpec(_message.Message):
-    __slots__ = ("namespace", "click_house", "otel", "node_selectors", "dns_name_override", "aggregator")
+    __slots__ = (
+        "namespace",
+        "click_house",
+        "otel",
+        "node_selectors",
+        "dns_name_override",
+        "aggregator",
+        "observability_daemons",
+    )
     NAMESPACE_FIELD_NUMBER: _ClassVar[int]
     CLICK_HOUSE_FIELD_NUMBER: _ClassVar[int]
     OTEL_FIELD_NUMBER: _ClassVar[int]
     NODE_SELECTORS_FIELD_NUMBER: _ClassVar[int]
     DNS_NAME_OVERRIDE_FIELD_NUMBER: _ClassVar[int]
     AGGREGATOR_FIELD_NUMBER: _ClassVar[int]
+    OBSERVABILITY_DAEMONS_FIELD_NUMBER: _ClassVar[int]
     namespace: str
     click_house: ClickHouseSpec
     otel: OtelCollectorSpec
     node_selectors: _containers.RepeatedCompositeFieldContainer[KubeNodeSelector]
     dns_name_override: str
     aggregator: AggregatorSpec
+    observability_daemons: _containers.RepeatedCompositeFieldContainer[ObservabilityDaemonSpec]
     def __init__(
         self,
         namespace: _Optional[str] = ...,
@@ -1348,6 +1405,7 @@ class TelemetryDeploymentSpec(_message.Message):
         node_selectors: _Optional[_Iterable[_Union[KubeNodeSelector, _Mapping]]] = ...,
         dns_name_override: _Optional[str] = ...,
         aggregator: _Optional[_Union[AggregatorSpec, _Mapping]] = ...,
+        observability_daemons: _Optional[_Iterable[_Union[ObservabilityDaemonSpec, _Mapping]]] = ...,
     ) -> None: ...
 
 class TelemetryDeployment(_message.Message):

chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py

@@ -0,0 +1,70 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/server/v1/dataplaneworkflows.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 chalk._gen.chalk.server.v1 import dataplanejobqueue_pb2 as chalk_dot_server_dot_v1_dot_dataplanejobqueue__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(chalk/server/v1/dataplaneworkflows.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\'chalk/server/v1/dataplanejobqueue.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xc5\x05\n\x11WorkflowExecution\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x39\n\ncreated_at\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x04 \x01(\tR\x0c\x64\x65ploymentId\x12\x31\n\x04kind\x18\x05 \x01(\x0e\x32\x1d.chalk.server.v1.WorkflowKindR\x04kind\x12@\n\x06status\x18\x06 \x01(\x0e\x32(.chalk.server.v1.WorkflowExecutionStatusR\x06status\x12$\n\x0b\x62ranch_name\x18\x07 \x01(\tH\x00R\nbranchName\x88\x01\x01\x12\x39\n\x16mainline_deployment_id\x18\x08 \x01(\tH\x01R\x14mainlineDeploymentId\x88\x01\x01\x12\x1e\n\x08\x61gent_id\x18\t \x01(\tH\x02R\x07\x61gentId\x88\x01\x01\x12\x39\n\tmeta_data\x18\n \x01(\x0b\x32\x17.google.protobuf.StructH\x03R\x08metaData\x88\x01\x01\x12>\n\nupdated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\tupdatedAt\x88\x01\x01\x12\x42\n\x0c\x66inalized_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x05R\x0b\x66inalizedAt\x88\x01\x01\x42\x0e\n\x0c_branch_nameB\x19\n\x17_mainline_deployment_idB\x0b\n\t_agent_idB\x0c\n\n_meta_dataB\r\n\x0b_updated_atB\x0f\n\r_finalized_at"-\n\x1bGetDataPlaneWorkflowRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"^\n\x1cGetDataPlaneWorkflowResponse\x12>\n\x08workflow\x18\x01 \x01(\x0b\x32".chalk.server.v1.WorkflowExecutionR\x08workflow"\x8e\x02\n\x1dListDataPlaneWorkflowsRequest\x12%\n\x0e\x65nvironment_id\x18\x01 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId\x12\x31\n\x04kind\x18\x03 \x01(\x0e\x32\x1d.chalk.server.v1.WorkflowKindR\x04kind\x12@\n\x06status\x18\x04 \x01(\x0e\x32(.chalk.server.v1.WorkflowExecutionStatusR\x06status\x12\x14\n\x05limit\x18\x05 \x01(\x05R\x05limit\x12\x16\n\x06offset\x18\x06 \x01(\x05R\x06offset"x\n\x1eListDataPlaneWorkflowsResponse\x12@\n\tworkflows\x18\x01 \x03(\x0b\x32".chalk.server.v1.WorkflowExecutionR\tworkflows\x12\x14\n\x05total\x18\x02 \x01(\x05R\x05total"M\n\x17GetWorkflowGraphRequest\x12\x32\n\x15workflow_execution_id\x18\x01 \x01(\tR\x13workflowExecutionId"\xc3\x03\n\x11WorkflowGraphNode\x12 \n\x0cjob_queue_id\x18\x01 \x01(\x05R\njobQueueId\x12!\n\x0coperation_id\x18\x02 \x01(\tR\x0boperationId\x12\x19\n\x08job_name\x18\x03 \x01(\tR\x07jobName\x12 \n\tjob_index\x18\x04 \x01(\x03H\x00R\x08jobIndex\x88\x01\x01\x12\x34\n\x05state\x18\x05 \x01(\x0e\x32\x1e.chalk.server.v1.JobQueueStateR\x05state\x12\x31\n\x04kind\x18\x06 \x01(\x0e\x32\x1d.chalk.server.v1.JobQueueKindR\x04kind\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x42\n\x0c\x66inalized_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x0b\x66inalizedAt\x88\x01\x01\x12%\n\x0eresource_group\x18\t \x01(\tR\rresourceGroupB\x0c\n\n_job_indexB\x0f\n\r_finalized_at"\xc8\x01\n\x11WorkflowGraphEdge\x12.\n\x13source_operation_id\x18\x01 \x01(\tR\x11sourceOperationId\x12.\n\x13target_operation_id\x18\x02 \x01(\tR\x11targetOperationId\x12\x42\n\x0csatisfied_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x0bsatisfiedAt\x88\x01\x01\x42\x0f\n\r_satisfied_at"\x8e\x01\n\x18GetWorkflowGraphResponse\x12\x38\n\x05nodes\x18\x01 \x03(\x0b\x32".chalk.server.v1.WorkflowGraphNodeR\x05nodes\x12\x38\n\x05\x65\x64ges\x18\x02 \x03(\x0b\x32".chalk.server.v1.WorkflowGraphEdgeR\x05\x65\x64ges*_\n\x0cWorkflowKind\x12\x1d\n\x19WORKFLOW_KIND_UNSPECIFIED\x10\x00\x12\x18\n\x14WORKFLOW_KIND_MANUAL\x10\x01\x12\x16\n\x12WORKFLOW_KIND_CRON\x10\x02*\x88\x02\n\x17WorkflowExecutionStatus\x12)\n%WORKFLOW_EXECUTION_STATUS_UNSPECIFIED\x10\x00\x12$\n WORKFLOW_EXECUTION_STATUS_QUEUED\x10\x01\x12%\n!WORKFLOW_EXECUTION_STATUS_WORKING\x10\x02\x12\'\n#WORKFLOW_EXECUTION_STATUS_COMPLETED\x10\x03\x12$\n WORKFLOW_EXECUTION_STATUS_FAILED\x10\x04\x12&\n"WORKFLOW_EXECUTION_STATUS_CANCELED\x10\x05\x32\x8d\x03\n\x19\x44\x61taPlaneWorkflowsService\x12{\n\x14GetDataPlaneWorkflow\x12,.chalk.server.v1.GetDataPlaneWorkflowRequest\x1a-.chalk.server.v1.GetDataPlaneWorkflowResponse"\x06\x90\x02\x01\x80}\x10\x12\x81\x01\n\x16ListDataPlaneWorkflows\x12..chalk.server.v1.ListDataPlaneWorkflowsRequest\x1a/.chalk.server.v1.ListDataPlaneWorkflowsResponse"\x06\x90\x02\x01\x80}\x10\x12o\n\x10GetWorkflowGraph\x12(.chalk.server.v1.GetWorkflowGraphRequest\x1a).chalk.server.v1.GetWorkflowGraphResponse"\x06\x90\x02\x01\x80}\x10\x42\xa0\x01\n\x13\x63om.chalk.server.v1B\x17\x44\x61taplaneworkflowsProtoP\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.dataplaneworkflows_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.server.v1B\027DataplaneworkflowsProtoP\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["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name["GetDataPlaneWorkflow"]._options = None
+    _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name[
+        "GetDataPlaneWorkflow"
+    ]._serialized_options = b"\220\002\001\200}\020"
+    _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name["ListDataPlaneWorkflows"]._options = None
+    _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name[
+        "ListDataPlaneWorkflows"
+    ]._serialized_options = b"\220\002\001\200}\020"
+    _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name["GetWorkflowGraph"]._options = None
+    _globals["_DATAPLANEWORKFLOWSSERVICE"].methods_by_name[
+        "GetWorkflowGraph"
+    ]._serialized_options = b"\220\002\001\200}\020"
+    _globals["_WORKFLOWKIND"]._serialized_start = 2329
+    _globals["_WORKFLOWKIND"]._serialized_end = 2424
+    _globals["_WORKFLOWEXECUTIONSTATUS"]._serialized_start = 2427
+    _globals["_WORKFLOWEXECUTIONSTATUS"]._serialized_end = 2691
+    _globals["_WORKFLOWEXECUTION"]._serialized_start = 199
+    _globals["_WORKFLOWEXECUTION"]._serialized_end = 908
+    _globals["_GETDATAPLANEWORKFLOWREQUEST"]._serialized_start = 910
+    _globals["_GETDATAPLANEWORKFLOWREQUEST"]._serialized_end = 955
+    _globals["_GETDATAPLANEWORKFLOWRESPONSE"]._serialized_start = 957
+    _globals["_GETDATAPLANEWORKFLOWRESPONSE"]._serialized_end = 1051
+    _globals["_LISTDATAPLANEWORKFLOWSREQUEST"]._serialized_start = 1054
+    _globals["_LISTDATAPLANEWORKFLOWSREQUEST"]._serialized_end = 1324
+    _globals["_LISTDATAPLANEWORKFLOWSRESPONSE"]._serialized_start = 1326
+    _globals["_LISTDATAPLANEWORKFLOWSRESPONSE"]._serialized_end = 1446
+    _globals["_GETWORKFLOWGRAPHREQUEST"]._serialized_start = 1448
+    _globals["_GETWORKFLOWGRAPHREQUEST"]._serialized_end = 1525
+    _globals["_WORKFLOWGRAPHNODE"]._serialized_start = 1528
+    _globals["_WORKFLOWGRAPHNODE"]._serialized_end = 1979
+    _globals["_WORKFLOWGRAPHEDGE"]._serialized_start = 1982
+    _globals["_WORKFLOWGRAPHEDGE"]._serialized_end = 2182
+    _globals["_GETWORKFLOWGRAPHRESPONSE"]._serialized_start = 2185
+    _globals["_GETWORKFLOWGRAPHRESPONSE"]._serialized_end = 2327
+    _globals["_DATAPLANEWORKFLOWSSERVICE"]._serialized_start = 2694
+    _globals["_DATAPLANEWORKFLOWSSERVICE"]._serialized_end = 3091
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi

@@ -0,0 +1,219 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from chalk._gen.chalk.server.v1 import dataplanejobqueue_pb2 as _dataplanejobqueue_pb2
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf import timestamp_pb2 as _timestamp_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+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 WorkflowKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    WORKFLOW_KIND_UNSPECIFIED: _ClassVar[WorkflowKind]
+    WORKFLOW_KIND_MANUAL: _ClassVar[WorkflowKind]
+    WORKFLOW_KIND_CRON: _ClassVar[WorkflowKind]
+
+class WorkflowExecutionStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    WORKFLOW_EXECUTION_STATUS_UNSPECIFIED: _ClassVar[WorkflowExecutionStatus]
+    WORKFLOW_EXECUTION_STATUS_QUEUED: _ClassVar[WorkflowExecutionStatus]
+    WORKFLOW_EXECUTION_STATUS_WORKING: _ClassVar[WorkflowExecutionStatus]
+    WORKFLOW_EXECUTION_STATUS_COMPLETED: _ClassVar[WorkflowExecutionStatus]
+    WORKFLOW_EXECUTION_STATUS_FAILED: _ClassVar[WorkflowExecutionStatus]
+    WORKFLOW_EXECUTION_STATUS_CANCELED: _ClassVar[WorkflowExecutionStatus]
+
+WORKFLOW_KIND_UNSPECIFIED: WorkflowKind
+WORKFLOW_KIND_MANUAL: WorkflowKind
+WORKFLOW_KIND_CRON: WorkflowKind
+WORKFLOW_EXECUTION_STATUS_UNSPECIFIED: WorkflowExecutionStatus
+WORKFLOW_EXECUTION_STATUS_QUEUED: WorkflowExecutionStatus
+WORKFLOW_EXECUTION_STATUS_WORKING: WorkflowExecutionStatus
+WORKFLOW_EXECUTION_STATUS_COMPLETED: WorkflowExecutionStatus
+WORKFLOW_EXECUTION_STATUS_FAILED: WorkflowExecutionStatus
+WORKFLOW_EXECUTION_STATUS_CANCELED: WorkflowExecutionStatus
+
+class WorkflowExecution(_message.Message):
+    __slots__ = (
+        "id",
+        "created_at",
+        "environment_id",
+        "deployment_id",
+        "kind",
+        "status",
+        "branch_name",
+        "mainline_deployment_id",
+        "agent_id",
+        "meta_data",
+        "updated_at",
+        "finalized_at",
+    )
+    ID_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    BRANCH_NAME_FIELD_NUMBER: _ClassVar[int]
+    MAINLINE_DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    META_DATA_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    FINALIZED_AT_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    created_at: _timestamp_pb2.Timestamp
+    environment_id: str
+    deployment_id: str
+    kind: WorkflowKind
+    status: WorkflowExecutionStatus
+    branch_name: str
+    mainline_deployment_id: str
+    agent_id: str
+    meta_data: _struct_pb2.Struct
+    updated_at: _timestamp_pb2.Timestamp
+    finalized_at: _timestamp_pb2.Timestamp
+    def __init__(
+        self,
+        id: _Optional[str] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        environment_id: _Optional[str] = ...,
+        deployment_id: _Optional[str] = ...,
+        kind: _Optional[_Union[WorkflowKind, str]] = ...,
+        status: _Optional[_Union[WorkflowExecutionStatus, str]] = ...,
+        branch_name: _Optional[str] = ...,
+        mainline_deployment_id: _Optional[str] = ...,
+        agent_id: _Optional[str] = ...,
+        meta_data: _Optional[_Union[_struct_pb2.Struct, _Mapping]] = ...,
+        updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        finalized_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+    ) -> None: ...
+
+class GetDataPlaneWorkflowRequest(_message.Message):
+    __slots__ = ("id",)
+    ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    def __init__(self, id: _Optional[str] = ...) -> None: ...
+
+class GetDataPlaneWorkflowResponse(_message.Message):
+    __slots__ = ("workflow",)
+    WORKFLOW_FIELD_NUMBER: _ClassVar[int]
+    workflow: WorkflowExecution
+    def __init__(self, workflow: _Optional[_Union[WorkflowExecution, _Mapping]] = ...) -> None: ...
+
+class ListDataPlaneWorkflowsRequest(_message.Message):
+    __slots__ = ("environment_id", "deployment_id", "kind", "status", "limit", "offset")
+    ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
+    environment_id: str
+    deployment_id: str
+    kind: WorkflowKind
+    status: WorkflowExecutionStatus
+    limit: int
+    offset: int
+    def __init__(
+        self,
+        environment_id: _Optional[str] = ...,
+        deployment_id: _Optional[str] = ...,
+        kind: _Optional[_Union[WorkflowKind, str]] = ...,
+        status: _Optional[_Union[WorkflowExecutionStatus, str]] = ...,
+        limit: _Optional[int] = ...,
+        offset: _Optional[int] = ...,
+    ) -> None: ...
+
+class ListDataPlaneWorkflowsResponse(_message.Message):
+    __slots__ = ("workflows", "total")
+    WORKFLOWS_FIELD_NUMBER: _ClassVar[int]
+    TOTAL_FIELD_NUMBER: _ClassVar[int]
+    workflows: _containers.RepeatedCompositeFieldContainer[WorkflowExecution]
+    total: int
+    def __init__(
+        self, workflows: _Optional[_Iterable[_Union[WorkflowExecution, _Mapping]]] = ..., total: _Optional[int] = ...
+    ) -> None: ...
+
+class GetWorkflowGraphRequest(_message.Message):
+    __slots__ = ("workflow_execution_id",)
+    WORKFLOW_EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
+    workflow_execution_id: str
+    def __init__(self, workflow_execution_id: _Optional[str] = ...) -> None: ...
+
+class WorkflowGraphNode(_message.Message):
+    __slots__ = (
+        "job_queue_id",
+        "operation_id",
+        "job_name",
+        "job_index",
+        "state",
+        "kind",
+        "created_at",
+        "finalized_at",
+        "resource_group",
+    )
+    JOB_QUEUE_ID_FIELD_NUMBER: _ClassVar[int]
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    JOB_NAME_FIELD_NUMBER: _ClassVar[int]
+    JOB_INDEX_FIELD_NUMBER: _ClassVar[int]
+    STATE_FIELD_NUMBER: _ClassVar[int]
+    KIND_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    FINALIZED_AT_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
+    job_queue_id: int
+    operation_id: str
+    job_name: str
+    job_index: int
+    state: _dataplanejobqueue_pb2.JobQueueState
+    kind: _dataplanejobqueue_pb2.JobQueueKind
+    created_at: _timestamp_pb2.Timestamp
+    finalized_at: _timestamp_pb2.Timestamp
+    resource_group: str
+    def __init__(
+        self,
+        job_queue_id: _Optional[int] = ...,
+        operation_id: _Optional[str] = ...,
+        job_name: _Optional[str] = ...,
+        job_index: _Optional[int] = ...,
+        state: _Optional[_Union[_dataplanejobqueue_pb2.JobQueueState, str]] = ...,
+        kind: _Optional[_Union[_dataplanejobqueue_pb2.JobQueueKind, str]] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        finalized_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        resource_group: _Optional[str] = ...,
+    ) -> None: ...
+
+class WorkflowGraphEdge(_message.Message):
+    __slots__ = ("source_operation_id", "target_operation_id", "satisfied_at")
+    SOURCE_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    TARGET_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    SATISFIED_AT_FIELD_NUMBER: _ClassVar[int]
+    source_operation_id: str
+    target_operation_id: str
+    satisfied_at: _timestamp_pb2.Timestamp
+    def __init__(
+        self,
+        source_operation_id: _Optional[str] = ...,
+        target_operation_id: _Optional[str] = ...,
+        satisfied_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+    ) -> None: ...
+
+class GetWorkflowGraphResponse(_message.Message):
+    __slots__ = ("nodes", "edges")
+    NODES_FIELD_NUMBER: _ClassVar[int]
+    EDGES_FIELD_NUMBER: _ClassVar[int]
+    nodes: _containers.RepeatedCompositeFieldContainer[WorkflowGraphNode]
+    edges: _containers.RepeatedCompositeFieldContainer[WorkflowGraphEdge]
+    def __init__(
+        self,
+        nodes: _Optional[_Iterable[_Union[WorkflowGraphNode, _Mapping]]] = ...,
+        edges: _Optional[_Iterable[_Union[WorkflowGraphEdge, _Mapping]]] = ...,
+    ) -> None: ...

chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py

@@ -0,0 +1,170 @@
+# 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 dataplaneworkflows_pb2 as chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2
+
+
+class DataPlaneWorkflowsServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetDataPlaneWorkflow = channel.unary_unary(
+            "/chalk.server.v1.DataPlaneWorkflowsService/GetDataPlaneWorkflow",
+            request_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowResponse.FromString,
+        )
+        self.ListDataPlaneWorkflows = channel.unary_unary(
+            "/chalk.server.v1.DataPlaneWorkflowsService/ListDataPlaneWorkflows",
+            request_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsResponse.FromString,
+        )
+        self.GetWorkflowGraph = channel.unary_unary(
+            "/chalk.server.v1.DataPlaneWorkflowsService/GetWorkflowGraph",
+            request_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphResponse.FromString,
+        )
+
+
+class DataPlaneWorkflowsServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetDataPlaneWorkflow(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 ListDataPlaneWorkflows(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 GetWorkflowGraph(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_DataPlaneWorkflowsServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "GetDataPlaneWorkflow": grpc.unary_unary_rpc_method_handler(
+            servicer.GetDataPlaneWorkflow,
+            request_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowResponse.SerializeToString,
+        ),
+        "ListDataPlaneWorkflows": grpc.unary_unary_rpc_method_handler(
+            servicer.ListDataPlaneWorkflows,
+            request_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsResponse.SerializeToString,
+        ),
+        "GetWorkflowGraph": grpc.unary_unary_rpc_method_handler(
+            servicer.GetWorkflowGraph,
+            request_deserializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+        "chalk.server.v1.DataPlaneWorkflowsService", rpc_method_handlers
+    )
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class DataPlaneWorkflowsService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetDataPlaneWorkflow(
+        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.DataPlaneWorkflowsService/GetDataPlaneWorkflow",
+            chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetDataPlaneWorkflowResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def ListDataPlaneWorkflows(
+        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.DataPlaneWorkflowsService/ListDataPlaneWorkflows",
+            chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.ListDataPlaneWorkflowsResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def GetWorkflowGraph(
+        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.DataPlaneWorkflowsService/GetWorkflowGraph",
+            chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_dataplaneworkflows__pb2.GetWorkflowGraphResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi

@@ -0,0 +1,62 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.server.v1.dataplaneworkflows_pb2 import (
+    GetDataPlaneWorkflowRequest,
+    GetDataPlaneWorkflowResponse,
+    GetWorkflowGraphRequest,
+    GetWorkflowGraphResponse,
+    ListDataPlaneWorkflowsRequest,
+    ListDataPlaneWorkflowsResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class DataPlaneWorkflowsServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    GetDataPlaneWorkflow: UnaryUnaryMultiCallable[
+        GetDataPlaneWorkflowRequest,
+        GetDataPlaneWorkflowResponse,
+    ]
+    ListDataPlaneWorkflows: UnaryUnaryMultiCallable[
+        ListDataPlaneWorkflowsRequest,
+        ListDataPlaneWorkflowsResponse,
+    ]
+    GetWorkflowGraph: UnaryUnaryMultiCallable[
+        GetWorkflowGraphRequest,
+        GetWorkflowGraphResponse,
+    ]
+
+class DataPlaneWorkflowsServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def GetDataPlaneWorkflow(
+        self,
+        request: GetDataPlaneWorkflowRequest,
+        context: ServicerContext,
+    ) -> GetDataPlaneWorkflowResponse: ...
+    @abstractmethod
+    def ListDataPlaneWorkflows(
+        self,
+        request: ListDataPlaneWorkflowsRequest,
+        context: ServicerContext,
+    ) -> ListDataPlaneWorkflowsResponse: ...
+    @abstractmethod
+    def GetWorkflowGraph(
+        self,
+        request: GetWorkflowGraphRequest,
+        context: ServicerContext,
+    ) -> GetWorkflowGraphResponse: ...
+
+def add_DataPlaneWorkflowsServiceServicer_to_server(
+    servicer: DataPlaneWorkflowsServiceServicer, server: Server
+) -> None: ...