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

chalk/_gen/chalk/server/v1/builder_pb2_grpc.py

@@ -150,6 +150,11 @@ class BuilderServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchProfileRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchProfileResponse.FromString,
         )
+        self.GetBranchServerStatus = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/GetBranchServerStatus",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchServerStatusRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchServerStatusResponse.FromString,
+        )
         self.GetNodepools = channel.unary_unary(
             "/chalk.server.v1.BuilderService/GetNodepools",
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetNodepoolsRequest.SerializeToString,
@@ -210,6 +215,11 @@ class BuilderServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentResponse.FromString,
         )
+        self.PrepareDeployment = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/PrepareDeployment",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentResponse.FromString,
+        )
         self.GetTelemetryDeployment = channel.unary_unary(
             "/chalk.server.v1.BuilderService/GetTelemetryDeployment",
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetTelemetryDeploymentRequest.SerializeToString,
@@ -220,11 +230,21 @@ class BuilderServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateTelemetryDeploymentRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateTelemetryDeploymentResponse.FromString,
         )
+        self.UpdateTelemetryDeployment = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/UpdateTelemetryDeployment",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.UpdateTelemetryDeploymentRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.UpdateTelemetryDeploymentResponse.FromString,
+        )
         self.DeleteTelemetryDeployment = channel.unary_unary(
             "/chalk.server.v1.BuilderService/DeleteTelemetryDeployment",
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.DeleteTelemetryDeploymentRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.DeleteTelemetryDeploymentResponse.FromString,
         )
+        self.MigrateTelemetryDeployment = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/MigrateTelemetryDeployment",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.MigrateTelemetryDeploymentRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.MigrateTelemetryDeploymentResponse.FromString,
+        )
         self.GetEnvironmentKubeClusters = channel.unary_unary(
             "/chalk.server.v1.BuilderService/GetEnvironmentKubeClusters",
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetEnvironmentKubeClustersRequest.SerializeToString,
@@ -240,6 +260,26 @@ class BuilderServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentResponse.FromString,
         )
+        self.SuspendClusterGateway = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/SuspendClusterGateway",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayResponse.FromString,
+        )
+        self.ResumeClusterGateway = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/ResumeClusterGateway",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayResponse.FromString,
+        )
+        self.SuspendClusterBackgroundPersistence = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/SuspendClusterBackgroundPersistence",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceResponse.FromString,
+        )
+        self.ResumeClusterBackgroundPersistence = channel.unary_unary(
+            "/chalk.server.v1.BuilderService/ResumeClusterBackgroundPersistence",
+            request_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceResponse.FromString,
+        )
 
 
 class BuilderServiceServicer(object):
@@ -417,6 +457,12 @@ class BuilderServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def GetBranchServerStatus(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 GetNodepools(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -489,6 +535,12 @@ class BuilderServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def PrepareDeployment(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 GetTelemetryDeployment(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -501,12 +553,24 @@ class BuilderServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def UpdateTelemetryDeployment(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 DeleteTelemetryDeployment(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 MigrateTelemetryDeployment(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 GetEnvironmentKubeClusters(self, request, context):
         """Missing associated documentation comment in .proto file."""
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -525,6 +589,30 @@ class BuilderServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def SuspendClusterGateway(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 ResumeClusterGateway(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 SuspendClusterBackgroundPersistence(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 ResumeClusterBackgroundPersistence(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_BuilderServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -663,6 +751,11 @@ def add_BuilderServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchProfileRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchProfileResponse.SerializeToString,
         ),
+        "GetBranchServerStatus": grpc.unary_unary_rpc_method_handler(
+            servicer.GetBranchServerStatus,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchServerStatusRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchServerStatusResponse.SerializeToString,
+        ),
         "GetNodepools": grpc.unary_unary_rpc_method_handler(
             servicer.GetNodepools,
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetNodepoolsRequest.FromString,
@@ -723,6 +816,11 @@ def add_BuilderServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateDeploymentResponse.SerializeToString,
         ),
+        "PrepareDeployment": grpc.unary_unary_rpc_method_handler(
+            servicer.PrepareDeployment,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentResponse.SerializeToString,
+        ),
         "GetTelemetryDeployment": grpc.unary_unary_rpc_method_handler(
             servicer.GetTelemetryDeployment,
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetTelemetryDeploymentRequest.FromString,
@@ -733,11 +831,21 @@ def add_BuilderServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateTelemetryDeploymentRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.CreateTelemetryDeploymentResponse.SerializeToString,
         ),
+        "UpdateTelemetryDeployment": grpc.unary_unary_rpc_method_handler(
+            servicer.UpdateTelemetryDeployment,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.UpdateTelemetryDeploymentRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.UpdateTelemetryDeploymentResponse.SerializeToString,
+        ),
         "DeleteTelemetryDeployment": grpc.unary_unary_rpc_method_handler(
             servicer.DeleteTelemetryDeployment,
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.DeleteTelemetryDeploymentRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.DeleteTelemetryDeploymentResponse.SerializeToString,
         ),
+        "MigrateTelemetryDeployment": grpc.unary_unary_rpc_method_handler(
+            servicer.MigrateTelemetryDeployment,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.MigrateTelemetryDeploymentRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.MigrateTelemetryDeploymentResponse.SerializeToString,
+        ),
         "GetEnvironmentKubeClusters": grpc.unary_unary_rpc_method_handler(
             servicer.GetEnvironmentKubeClusters,
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.GetEnvironmentKubeClustersRequest.FromString,
@@ -753,6 +861,26 @@ def add_BuilderServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeEnvironmentResponse.SerializeToString,
         ),
+        "SuspendClusterGateway": grpc.unary_unary_rpc_method_handler(
+            servicer.SuspendClusterGateway,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayResponse.SerializeToString,
+        ),
+        "ResumeClusterGateway": grpc.unary_unary_rpc_method_handler(
+            servicer.ResumeClusterGateway,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayResponse.SerializeToString,
+        ),
+        "SuspendClusterBackgroundPersistence": grpc.unary_unary_rpc_method_handler(
+            servicer.SuspendClusterBackgroundPersistence,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceResponse.SerializeToString,
+        ),
+        "ResumeClusterBackgroundPersistence": grpc.unary_unary_rpc_method_handler(
+            servicer.ResumeClusterBackgroundPersistence,
+            request_deserializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.BuilderService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -1545,6 +1673,35 @@ class BuilderService(object):
             metadata,
         )
 
+    @staticmethod
+    def GetBranchServerStatus(
+        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.BuilderService/GetBranchServerStatus",
+            chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchServerStatusRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.GetBranchServerStatusResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
     @staticmethod
     def GetNodepools(
         request,
@@ -1893,6 +2050,35 @@ class BuilderService(object):
             metadata,
         )
 
+    @staticmethod
+    def PrepareDeployment(
+        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.BuilderService/PrepareDeployment",
+            chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.PrepareDeploymentResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
     @staticmethod
     def GetTelemetryDeployment(
         request,
@@ -1951,6 +2137,35 @@ class BuilderService(object):
             metadata,
         )
 
+    @staticmethod
+    def UpdateTelemetryDeployment(
+        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.BuilderService/UpdateTelemetryDeployment",
+            chalk_dot_server_dot_v1_dot_builder__pb2.UpdateTelemetryDeploymentRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.UpdateTelemetryDeploymentResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
     @staticmethod
     def DeleteTelemetryDeployment(
         request,
@@ -1980,6 +2195,35 @@ class BuilderService(object):
             metadata,
         )
 
+    @staticmethod
+    def MigrateTelemetryDeployment(
+        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.BuilderService/MigrateTelemetryDeployment",
+            chalk_dot_server_dot_v1_dot_builder__pb2.MigrateTelemetryDeploymentRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.MigrateTelemetryDeploymentResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
     @staticmethod
     def GetEnvironmentKubeClusters(
         request,
@@ -2067,6 +2311,122 @@ class BuilderService(object):
             metadata,
         )
 
+    @staticmethod
+    def SuspendClusterGateway(
+        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.BuilderService/SuspendClusterGateway",
+            chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterGatewayResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def ResumeClusterGateway(
+        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.BuilderService/ResumeClusterGateway",
+            chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterGatewayResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def SuspendClusterBackgroundPersistence(
+        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.BuilderService/SuspendClusterBackgroundPersistence",
+            chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.SuspendClusterBackgroundPersistenceResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
+    @staticmethod
+    def ResumeClusterBackgroundPersistence(
+        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.BuilderService/ResumeClusterBackgroundPersistence",
+            chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_builder__pb2.ResumeClusterBackgroundPersistenceResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )
+
 
 class ClusterBuilderServiceStub(object):
     """Missing associated documentation comment in .proto file."""

chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi

@@ -42,6 +42,8 @@ from chalk._gen.chalk.server.v1.builder_pb2 import (
     DeployKubeComponentsResponse,
     GetBranchProfileRequest,
     GetBranchProfileResponse,
+    GetBranchServerStatusRequest,
+    GetBranchServerStatusResponse,
     GetClusterBackgroundPersistenceRequest,
     GetClusterBackgroundPersistenceResponse,
     GetClusterGatewayDefaultRequest,
@@ -80,10 +82,18 @@ from chalk._gen.chalk.server.v1.builder_pb2 import (
     LintSourceResponse,
     MigrateClusterTimescaleDBRequest,
     MigrateClusterTimescaleDBResponse,
+    MigrateTelemetryDeploymentRequest,
+    MigrateTelemetryDeploymentResponse,
+    PrepareDeploymentRequest,
+    PrepareDeploymentResponse,
     RebuildDeploymentRequest,
     RebuildDeploymentResponse,
     RedeployDeploymentRequest,
     RedeployDeploymentResponse,
+    ResumeClusterBackgroundPersistenceRequest,
+    ResumeClusterBackgroundPersistenceResponse,
+    ResumeClusterGatewayRequest,
+    ResumeClusterGatewayResponse,
     ResumeEnvironmentRequest,
     ResumeEnvironmentResponse,
     ScaleBranchRequest,
@@ -92,6 +102,10 @@ from chalk._gen.chalk.server.v1.builder_pb2 import (
     SetTagWeightsResponse,
     StartBranchRequest,
     StartBranchResponse,
+    SuspendClusterBackgroundPersistenceRequest,
+    SuspendClusterBackgroundPersistenceResponse,
+    SuspendClusterGatewayRequest,
+    SuspendClusterGatewayResponse,
     SuspendEnvironmentRequest,
     SuspendEnvironmentResponse,
     UpdateEnvironmentVariablesRequest,
@@ -100,6 +114,8 @@ from chalk._gen.chalk.server.v1.builder_pb2 import (
     UpdateKarpenterNodepoolResponse,
     UpdateNodepoolRequest,
     UpdateNodepoolResponse,
+    UpdateTelemetryDeploymentRequest,
+    UpdateTelemetryDeploymentResponse,
     UploadSourceRequest,
     UploadSourceResponse,
 )
@@ -236,6 +252,10 @@ class BuilderServiceStub:
         GetBranchProfileRequest,
         GetBranchProfileResponse,
     ]
+    GetBranchServerStatus: UnaryUnaryMultiCallable[
+        GetBranchServerStatusRequest,
+        GetBranchServerStatusResponse,
+    ]
     GetNodepools: UnaryUnaryMultiCallable[
         GetNodepoolsRequest,
         GetNodepoolsResponse,
@@ -284,6 +304,10 @@ class BuilderServiceStub:
         CreateDeploymentRequest,
         CreateDeploymentResponse,
     ]
+    PrepareDeployment: UnaryUnaryMultiCallable[
+        PrepareDeploymentRequest,
+        PrepareDeploymentResponse,
+    ]
     GetTelemetryDeployment: UnaryUnaryMultiCallable[
         GetTelemetryDeploymentRequest,
         GetTelemetryDeploymentResponse,
@@ -292,10 +316,18 @@ class BuilderServiceStub:
         CreateTelemetryDeploymentRequest,
         CreateTelemetryDeploymentResponse,
     ]
+    UpdateTelemetryDeployment: UnaryUnaryMultiCallable[
+        UpdateTelemetryDeploymentRequest,
+        UpdateTelemetryDeploymentResponse,
+    ]
     DeleteTelemetryDeployment: UnaryUnaryMultiCallable[
         DeleteTelemetryDeploymentRequest,
         DeleteTelemetryDeploymentResponse,
     ]
+    MigrateTelemetryDeployment: UnaryUnaryMultiCallable[
+        MigrateTelemetryDeploymentRequest,
+        MigrateTelemetryDeploymentResponse,
+    ]
     GetEnvironmentKubeClusters: UnaryUnaryMultiCallable[
         GetEnvironmentKubeClustersRequest,
         GetEnvironmentKubeClustersResponse,
@@ -308,6 +340,22 @@ class BuilderServiceStub:
         ResumeEnvironmentRequest,
         ResumeEnvironmentResponse,
     ]
+    SuspendClusterGateway: UnaryUnaryMultiCallable[
+        SuspendClusterGatewayRequest,
+        SuspendClusterGatewayResponse,
+    ]
+    ResumeClusterGateway: UnaryUnaryMultiCallable[
+        ResumeClusterGatewayRequest,
+        ResumeClusterGatewayResponse,
+    ]
+    SuspendClusterBackgroundPersistence: UnaryUnaryMultiCallable[
+        SuspendClusterBackgroundPersistenceRequest,
+        SuspendClusterBackgroundPersistenceResponse,
+    ]
+    ResumeClusterBackgroundPersistence: UnaryUnaryMultiCallable[
+        ResumeClusterBackgroundPersistenceRequest,
+        ResumeClusterBackgroundPersistenceResponse,
+    ]
 
 class BuilderServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -489,6 +537,12 @@ class BuilderServiceServicer(metaclass=ABCMeta):
         context: ServicerContext,
     ) -> GetBranchProfileResponse: ...
     @abstractmethod
+    def GetBranchServerStatus(
+        self,
+        request: GetBranchServerStatusRequest,
+        context: ServicerContext,
+    ) -> GetBranchServerStatusResponse: ...
+    @abstractmethod
     def GetNodepools(
         self,
         request: GetNodepoolsRequest,
@@ -561,6 +615,12 @@ class BuilderServiceServicer(metaclass=ABCMeta):
         context: ServicerContext,
     ) -> CreateDeploymentResponse: ...
     @abstractmethod
+    def PrepareDeployment(
+        self,
+        request: PrepareDeploymentRequest,
+        context: ServicerContext,
+    ) -> PrepareDeploymentResponse: ...
+    @abstractmethod
     def GetTelemetryDeployment(
         self,
         request: GetTelemetryDeploymentRequest,
@@ -573,12 +633,24 @@ class BuilderServiceServicer(metaclass=ABCMeta):
         context: ServicerContext,
     ) -> CreateTelemetryDeploymentResponse: ...
     @abstractmethod
+    def UpdateTelemetryDeployment(
+        self,
+        request: UpdateTelemetryDeploymentRequest,
+        context: ServicerContext,
+    ) -> UpdateTelemetryDeploymentResponse: ...
+    @abstractmethod
     def DeleteTelemetryDeployment(
         self,
         request: DeleteTelemetryDeploymentRequest,
         context: ServicerContext,
     ) -> DeleteTelemetryDeploymentResponse: ...
     @abstractmethod
+    def MigrateTelemetryDeployment(
+        self,
+        request: MigrateTelemetryDeploymentRequest,
+        context: ServicerContext,
+    ) -> MigrateTelemetryDeploymentResponse: ...
+    @abstractmethod
     def GetEnvironmentKubeClusters(
         self,
         request: GetEnvironmentKubeClustersRequest,
@@ -596,6 +668,30 @@ class BuilderServiceServicer(metaclass=ABCMeta):
         request: ResumeEnvironmentRequest,
         context: ServicerContext,
     ) -> ResumeEnvironmentResponse: ...
+    @abstractmethod
+    def SuspendClusterGateway(
+        self,
+        request: SuspendClusterGatewayRequest,
+        context: ServicerContext,
+    ) -> SuspendClusterGatewayResponse: ...
+    @abstractmethod
+    def ResumeClusterGateway(
+        self,
+        request: ResumeClusterGatewayRequest,
+        context: ServicerContext,
+    ) -> ResumeClusterGatewayResponse: ...
+    @abstractmethod
+    def SuspendClusterBackgroundPersistence(
+        self,
+        request: SuspendClusterBackgroundPersistenceRequest,
+        context: ServicerContext,
+    ) -> SuspendClusterBackgroundPersistenceResponse: ...
+    @abstractmethod
+    def ResumeClusterBackgroundPersistence(
+        self,
+        request: ResumeClusterBackgroundPersistenceRequest,
+        context: ServicerContext,
+    ) -> ResumeClusterBackgroundPersistenceResponse: ...
 
 def add_BuilderServiceServicer_to_server(servicer: BuilderServiceServicer, server: Server) -> None: ...