chalkpy 2.94.8__py3-none-any.whl → 2.95.0__py3-none-any.whl

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

@@ -140,17 +140,26 @@ class DeployKubeComponentsResponse(_message.Message):
     def __init__(self, nonfatal_errors: _Optional[_Iterable[str]] = ...) -> None: ...
 
 class RebuildDeploymentRequest(_message.Message):
-    __slots__ = ("existing_deployment_id", "new_image_tag", "base_image_override", "enable_profiling", "build_profile")
+    __slots__ = (
+        "existing_deployment_id",
+        "new_image_tag",
+        "base_image_override",
+        "enable_profiling",
+        "build_profile",
+        "force_rebuild_dockerfile",
+    )
     EXISTING_DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
     NEW_IMAGE_TAG_FIELD_NUMBER: _ClassVar[int]
     BASE_IMAGE_OVERRIDE_FIELD_NUMBER: _ClassVar[int]
     ENABLE_PROFILING_FIELD_NUMBER: _ClassVar[int]
     BUILD_PROFILE_FIELD_NUMBER: _ClassVar[int]
+    FORCE_REBUILD_DOCKERFILE_FIELD_NUMBER: _ClassVar[int]
     existing_deployment_id: str
     new_image_tag: str
     base_image_override: str
     enable_profiling: bool
     build_profile: _environment_pb2.DeploymentBuildProfile
+    force_rebuild_dockerfile: bool
     def __init__(
         self,
         existing_deployment_id: _Optional[str] = ...,
@@ -158,6 +167,7 @@ class RebuildDeploymentRequest(_message.Message):
         base_image_override: _Optional[str] = ...,
         enable_profiling: bool = ...,
         build_profile: _Optional[_Union[_environment_pb2.DeploymentBuildProfile, str]] = ...,
+        force_rebuild_dockerfile: bool = ...,
     ) -> None: ...
 
 class RebuildDeploymentResponse(_message.Message):
@@ -235,6 +245,7 @@ class UploadSourceRequest(_message.Message):
         "use_grpc",
         "enable_profiling",
         "build_profile",
+        "force_rebuild_dockerfile",
     )
     DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
     ARCHIVE_FIELD_NUMBER: _ClassVar[int]
@@ -244,6 +255,7 @@ class UploadSourceRequest(_message.Message):
     USE_GRPC_FIELD_NUMBER: _ClassVar[int]
     ENABLE_PROFILING_FIELD_NUMBER: _ClassVar[int]
     BUILD_PROFILE_FIELD_NUMBER: _ClassVar[int]
+    FORCE_REBUILD_DOCKERFILE_FIELD_NUMBER: _ClassVar[int]
     deployment_id: str
     archive: bytes
     no_promote: bool
@@ -252,6 +264,7 @@ class UploadSourceRequest(_message.Message):
     use_grpc: bool
     enable_profiling: bool
     build_profile: _environment_pb2.DeploymentBuildProfile
+    force_rebuild_dockerfile: bool
     def __init__(
         self,
         deployment_id: _Optional[str] = ...,
@@ -262,6 +275,7 @@ class UploadSourceRequest(_message.Message):
         use_grpc: bool = ...,
         enable_profiling: bool = ...,
         build_profile: _Optional[_Union[_environment_pb2.DeploymentBuildProfile, str]] = ...,
+        force_rebuild_dockerfile: bool = ...,
     ) -> None: ...
 
 class UploadSourceResponse(_message.Message):

chalk/_gen/chalk/server/v1/clickhouse_pb2.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/server/v1/clickhouse.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.utils.v1 import sensitive_pb2 as chalk_dot_utils_dot_v1_dot_sensitive__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n chalk/server/v1/clickhouse.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1e\x63halk/utils/v1/sensitive.proto"\x19\n\x17GetClickhouseUriRequest"2\n\x18GetClickhouseUriResponse\x12\x16\n\x03uri\x18\x01 \x01(\tB\x04\xd8\xa1\'\x01R\x03uri2\x84\x01\n\x11\x43lickhouseService\x12o\n\x10GetClickhouseUri\x12(.chalk.server.v1.GetClickhouseUriRequest\x1a).chalk.server.v1.GetClickhouseUriResponse"\x06\x90\x02\x01\x80}\x0b\x42\x98\x01\n\x13\x63om.chalk.server.v1B\x0f\x43lickhouseProtoP\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.clickhouse_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.server.v1B\017ClickhouseProtoP\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["_GETCLICKHOUSEURIRESPONSE"].fields_by_name["uri"]._options = None
+    _globals["_GETCLICKHOUSEURIRESPONSE"].fields_by_name["uri"]._serialized_options = b"\330\241'\001"
+    _globals["_CLICKHOUSESERVICE"].methods_by_name["GetClickhouseUri"]._options = None
+    _globals["_CLICKHOUSESERVICE"].methods_by_name["GetClickhouseUri"]._serialized_options = b"\220\002\001\200}\013"
+    _globals["_GETCLICKHOUSEURIREQUEST"]._serialized_start = 118
+    _globals["_GETCLICKHOUSEURIREQUEST"]._serialized_end = 143
+    _globals["_GETCLICKHOUSEURIRESPONSE"]._serialized_start = 145
+    _globals["_GETCLICKHOUSEURIRESPONSE"]._serialized_end = 195
+    _globals["_CLICKHOUSESERVICE"]._serialized_start = 198
+    _globals["_CLICKHOUSESERVICE"]._serialized_end = 330
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi

@@ -0,0 +1,17 @@
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+from chalk._gen.chalk.utils.v1 import sensitive_pb2 as _sensitive_pb2
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Optional as _Optional
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class GetClickhouseUriRequest(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class GetClickhouseUriResponse(_message.Message):
+    __slots__ = ("uri",)
+    URI_FIELD_NUMBER: _ClassVar[int]
+    uri: str
+    def __init__(self, uri: _Optional[str] = ...) -> None: ...

chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py

@@ -0,0 +1,78 @@
+# 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 clickhouse_pb2 as chalk_dot_server_dot_v1_dot_clickhouse__pb2
+
+
+class ClickhouseServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetClickhouseUri = channel.unary_unary(
+            "/chalk.server.v1.ClickhouseService/GetClickhouseUri",
+            request_serializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriResponse.FromString,
+        )
+
+
+class ClickhouseServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetClickhouseUri(self, request, context):
+        """Get the Clickhouse connection URI for an environment"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+
+def add_ClickhouseServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "GetClickhouseUri": grpc.unary_unary_rpc_method_handler(
+            servicer.GetClickhouseUri,
+            request_deserializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.ClickhouseService", rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class ClickhouseService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetClickhouseUri(
+        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.ClickhouseService/GetClickhouseUri",
+            chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_clickhouse__pb2.GetClickhouseUriResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi

@@ -0,0 +1,38 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+from abc import (
+    ABCMeta,
+    abstractmethod,
+)
+from chalk._gen.chalk.server.v1.clickhouse_pb2 import (
+    GetClickhouseUriRequest,
+    GetClickhouseUriResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class ClickhouseServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    GetClickhouseUri: UnaryUnaryMultiCallable[
+        GetClickhouseUriRequest,
+        GetClickhouseUriResponse,
+    ]
+    """Get the Clickhouse connection URI for an environment"""
+
+class ClickhouseServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def GetClickhouseUri(
+        self,
+        request: GetClickhouseUriRequest,
+        context: ServicerContext,
+    ) -> GetClickhouseUriResponse:
+        """Get the Clickhouse connection URI for an environment"""
+
+def add_ClickhouseServiceServicer_to_server(servicer: ClickhouseServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/deploy_pb2.py

@@ -24,7 +24,7 @@ from google.protobuf import field_mask_pb2 as google_dot_protobuf_dot_field__mas
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1c\x63halk/server/v1/deploy.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/artifacts/v1/export.proto\x1a\x19\x63halk/auth/v1/audit.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a chalk/server/v1/deployment.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto"\x97\x01\n\x13\x44\x65ployBranchRequest\x12\x1f\n\x0b\x62ranch_name\x18\x01 \x01(\tR\nbranchName\x12!\n\x0creset_branch\x18\x02 \x01(\x08R\x0bresetBranch\x12\x18\n\x07\x61rchive\x18\x03 \x01(\x0cR\x07\x61rchive\x12"\n\ris_hot_deploy\x18\x04 \x01(\x08R\x0bisHotDeploy"\x89\x02\n\x14\x44\x65ployBranchResponse\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x34\n\x05graph\x18\x02 \x01(\x0b\x32\x15.chalk.graph.v1.GraphB\x02\x18\x01H\x00R\x05graph\x88\x01\x01\x12H\n\x11\x64\x65ployment_errors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x10\x64\x65ploymentErrors\x12\x37\n\x06\x65xport\x18\x04 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportH\x01R\x06\x65xport\x88\x01\x01\x42\x08\n\x06_graphB\t\n\x07_export"\x92\x02\n\'CreateBranchFromSourceDeploymentRequest\x12\x1f\n\x0b\x62ranch_name\x18\x01 \x01(\tR\nbranchName\x12.\n\x12source_branch_name\x18\x02 \x01(\tH\x00R\x10sourceBranchName\x12\x32\n\x14source_deployment_id\x18\x03 \x01(\tH\x00R\x12sourceDeploymentId\x12X\n\x1b\x63urrent_mainline_deployment\x18\x04 \x01(\x0b\x32\x16.google.protobuf.EmptyH\x00R\x19\x63urrentMainlineDeploymentB\x08\n\x06source"\x91\x02\n(CreateBranchFromSourceDeploymentResponse\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12H\n\x11\x64\x65ployment_errors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x10\x64\x65ploymentErrors\x12\x37\n\x06\x65xport\x18\x03 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportH\x00R\x06\x65xport\x88\x01\x01\x12\x32\n\x15\x62ranch_already_exists\x18\x04 \x01(\x08R\x13\x62ranchAlreadyExistsB\t\n\x07_export"t\n\x14GetDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x37\n\tread_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x08readMask"\x98\x01\n\x15GetDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment\x12\x37\n\x06\x65xport\x18\x02 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportH\x00R\x06\x65xport\x88\x01\x01\x42\t\n\x07_export"\xda\x01\n\x16ListDeploymentsRequest\x12\x1b\n\x06\x63ursor\x18\x01 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x12*\n\x0einclude_branch\x18\x03 \x01(\x08H\x02R\rincludeBranch\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x04 \x01(\tH\x03R\nbranchName\x88\x01\x01\x42\t\n\x07_cursorB\x08\n\x06_limitB\x11\n\x0f_include_branchB\x0e\n\x0c_branch_name"\x80\x01\n\x17ListDeploymentsResponse\x12=\n\x0b\x64\x65ployments\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.DeploymentR\x0b\x64\x65ployments\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x42\t\n\x07_cursor"?\n\x18SuspendDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId"X\n\x19SuspendDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment"v\n\x16ScaleDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x37\n\x06sizing\x18\x02 \x01(\x0b\x32\x1f.chalk.server.v1.InstanceSizingR\x06sizing"V\n\x17ScaleDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment"\x89\x01\n\x14TagDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x10\n\x03tag\x18\x02 \x01(\tR\x03tag\x12(\n\rmirror_weight\x18\x03 \x01(\x05H\x00R\x0cmirrorWeight\x88\x01\x01\x42\x10\n\x0e_mirror_weight"\xaa\x01\n\x15TagDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment\x12\x39\n\x16untagged_deployment_id\x18\x02 \x01(\tH\x00R\x14untaggedDeploymentId\x88\x01\x01\x42\x19\n\x17_untagged_deployment_id"\x1d\n\x1bGetActiveDeploymentsRequest"]\n\x1cGetActiveDeploymentsResponse\x12=\n\x0b\x64\x65ployments\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.DeploymentR\x0b\x64\x65ployments2\xad\x07\n\rDeployService\x12`\n\x0c\x44\x65ployBranch\x12$.chalk.server.v1.DeployBranchRequest\x1a%.chalk.server.v1.DeployBranchResponse"\x03\x80}\r\x12\x9c\x01\n CreateBranchFromSourceDeployment\x12\x38.chalk.server.v1.CreateBranchFromSourceDeploymentRequest\x1a\x39.chalk.server.v1.CreateBranchFromSourceDeploymentResponse"\x03\x80}\r\x12\x63\n\rGetDeployment\x12%.chalk.server.v1.GetDeploymentRequest\x1a&.chalk.server.v1.GetDeploymentResponse"\x03\x80}\x0b\x12i\n\x0fListDeployments\x12\'.chalk.server.v1.ListDeploymentsRequest\x1a(.chalk.server.v1.ListDeploymentsResponse"\x03\x80}\x0b\x12x\n\x14GetActiveDeployments\x12,.chalk.server.v1.GetActiveDeploymentsRequest\x1a-.chalk.server.v1.GetActiveDeploymentsResponse"\x03\x80}\x02\x12u\n\x11SuspendDeployment\x12).chalk.server.v1.SuspendDeploymentRequest\x1a*.chalk.server.v1.SuspendDeploymentResponse"\t\x80}\x0c\x8a\xd3\x0e\x02\x08\x02\x12o\n\x0fScaleDeployment\x12\'.chalk.server.v1.ScaleDeploymentRequest\x1a(.chalk.server.v1.ScaleDeploymentResponse"\t\x80}\x0c\x8a\xd3\x0e\x02\x08\x02\x12i\n\rTagDeployment\x12%.chalk.server.v1.TagDeploymentRequest\x1a&.chalk.server.v1.TagDeploymentResponse"\t\x80}\x0c\x8a\xd3\x0e\x02\x08\x02\x42\x94\x01\n\x13\x63om.chalk.server.v1B\x0b\x44\x65ployProtoP\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\x1c\x63halk/server/v1/deploy.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/artifacts/v1/export.proto\x1a\x19\x63halk/auth/v1/audit.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a chalk/server/v1/deployment.proto\x1a\x1bgoogle/protobuf/empty.proto\x1a google/protobuf/field_mask.proto"\x97\x01\n\x13\x44\x65ployBranchRequest\x12\x1f\n\x0b\x62ranch_name\x18\x01 \x01(\tR\nbranchName\x12!\n\x0creset_branch\x18\x02 \x01(\x08R\x0bresetBranch\x12\x18\n\x07\x61rchive\x18\x03 \x01(\x0cR\x07\x61rchive\x12"\n\ris_hot_deploy\x18\x04 \x01(\x08R\x0bisHotDeploy"\x89\x02\n\x14\x44\x65ployBranchResponse\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x34\n\x05graph\x18\x02 \x01(\x0b\x32\x15.chalk.graph.v1.GraphB\x02\x18\x01H\x00R\x05graph\x88\x01\x01\x12H\n\x11\x64\x65ployment_errors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x10\x64\x65ploymentErrors\x12\x37\n\x06\x65xport\x18\x04 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportH\x01R\x06\x65xport\x88\x01\x01\x42\x08\n\x06_graphB\t\n\x07_export"\x92\x02\n\'CreateBranchFromSourceDeploymentRequest\x12\x1f\n\x0b\x62ranch_name\x18\x01 \x01(\tR\nbranchName\x12.\n\x12source_branch_name\x18\x02 \x01(\tH\x00R\x10sourceBranchName\x12\x32\n\x14source_deployment_id\x18\x03 \x01(\tH\x00R\x12sourceDeploymentId\x12X\n\x1b\x63urrent_mainline_deployment\x18\x04 \x01(\x0b\x32\x16.google.protobuf.EmptyH\x00R\x19\x63urrentMainlineDeploymentB\x08\n\x06source"\x91\x02\n(CreateBranchFromSourceDeploymentResponse\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12H\n\x11\x64\x65ployment_errors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x10\x64\x65ploymentErrors\x12\x37\n\x06\x65xport\x18\x03 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportH\x00R\x06\x65xport\x88\x01\x01\x12\x32\n\x15\x62ranch_already_exists\x18\x04 \x01(\x08R\x13\x62ranchAlreadyExistsB\t\n\x07_export"t\n\x14GetDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x37\n\tread_mask\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x08readMask"\x98\x01\n\x15GetDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment\x12\x37\n\x06\x65xport\x18\x02 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportH\x00R\x06\x65xport\x88\x01\x01\x42\t\n\x07_export"\xda\x01\n\x16ListDeploymentsRequest\x12\x1b\n\x06\x63ursor\x18\x01 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x02 \x01(\x05H\x01R\x05limit\x88\x01\x01\x12*\n\x0einclude_branch\x18\x03 \x01(\x08H\x02R\rincludeBranch\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x04 \x01(\tH\x03R\nbranchName\x88\x01\x01\x42\t\n\x07_cursorB\x08\n\x06_limitB\x11\n\x0f_include_branchB\x0e\n\x0c_branch_name"\x80\x01\n\x17ListDeploymentsResponse\x12=\n\x0b\x64\x65ployments\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.DeploymentR\x0b\x64\x65ployments\x12\x1b\n\x06\x63ursor\x18\x02 \x01(\tH\x00R\x06\x63ursor\x88\x01\x01\x42\t\n\x07_cursor"?\n\x18SuspendDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId"X\n\x19SuspendDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment"v\n\x16ScaleDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x37\n\x06sizing\x18\x02 \x01(\x0b\x32\x1f.chalk.server.v1.InstanceSizingR\x06sizing"V\n\x17ScaleDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment"\x89\x01\n\x14TagDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x10\n\x03tag\x18\x02 \x01(\tR\x03tag\x12(\n\rmirror_weight\x18\x03 \x01(\x05H\x00R\x0cmirrorWeight\x88\x01\x01\x42\x10\n\x0e_mirror_weight"\xaa\x01\n\x15TagDeploymentResponse\x12;\n\ndeployment\x18\x01 \x01(\x0b\x32\x1b.chalk.server.v1.DeploymentR\ndeployment\x12\x39\n\x16untagged_deployment_id\x18\x02 \x01(\tH\x00R\x14untaggedDeploymentId\x88\x01\x01\x42\x19\n\x17_untagged_deployment_id"\x1d\n\x1bGetActiveDeploymentsRequest"]\n\x1cGetActiveDeploymentsResponse\x12=\n\x0b\x64\x65ployments\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.DeploymentR\x0b\x64\x65ployments"A\n\x1aGetDeploymentSourceRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId"<\n\x1bGetDeploymentSourceResponse\x12\x1d\n\nsigned_url\x18\x01 \x01(\tR\tsignedUrl2\xa4\x08\n\rDeployService\x12`\n\x0c\x44\x65ployBranch\x12$.chalk.server.v1.DeployBranchRequest\x1a%.chalk.server.v1.DeployBranchResponse"\x03\x80}\r\x12\x9c\x01\n CreateBranchFromSourceDeployment\x12\x38.chalk.server.v1.CreateBranchFromSourceDeploymentRequest\x1a\x39.chalk.server.v1.CreateBranchFromSourceDeploymentResponse"\x03\x80}\r\x12\x63\n\rGetDeployment\x12%.chalk.server.v1.GetDeploymentRequest\x1a&.chalk.server.v1.GetDeploymentResponse"\x03\x80}\x0b\x12i\n\x0fListDeployments\x12\'.chalk.server.v1.ListDeploymentsRequest\x1a(.chalk.server.v1.ListDeploymentsResponse"\x03\x80}\x0b\x12x\n\x14GetActiveDeployments\x12,.chalk.server.v1.GetActiveDeploymentsRequest\x1a-.chalk.server.v1.GetActiveDeploymentsResponse"\x03\x80}\x02\x12u\n\x11SuspendDeployment\x12).chalk.server.v1.SuspendDeploymentRequest\x1a*.chalk.server.v1.SuspendDeploymentResponse"\t\x80}\x0c\x8a\xd3\x0e\x02\x08\x02\x12o\n\x0fScaleDeployment\x12\'.chalk.server.v1.ScaleDeploymentRequest\x1a(.chalk.server.v1.ScaleDeploymentResponse"\t\x80}\x0c\x8a\xd3\x0e\x02\x08\x02\x12i\n\rTagDeployment\x12%.chalk.server.v1.TagDeploymentRequest\x1a&.chalk.server.v1.TagDeploymentResponse"\t\x80}\x0c\x8a\xd3\x0e\x02\x08\x02\x12u\n\x13GetDeploymentSource\x12+.chalk.server.v1.GetDeploymentSourceRequest\x1a,.chalk.server.v1.GetDeploymentSourceResponse"\x03\x80}\x0b\x42\x94\x01\n\x13\x63om.chalk.server.v1B\x0b\x44\x65ployProtoP\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()
@@ -59,6 +59,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_DEPLOYSERVICE"].methods_by_name[
         "TagDeployment"
     ]._serialized_options = b"\200}\014\212\323\016\002\010\002"
+    _globals["_DEPLOYSERVICE"].methods_by_name["GetDeploymentSource"]._options = None
+    _globals["_DEPLOYSERVICE"].methods_by_name["GetDeploymentSource"]._serialized_options = b"\200}\013"
     _globals["_DEPLOYBRANCHREQUEST"]._serialized_start = 303
     _globals["_DEPLOYBRANCHREQUEST"]._serialized_end = 454
     _globals["_DEPLOYBRANCHRESPONSE"]._serialized_start = 457
@@ -91,6 +93,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_GETACTIVEDEPLOYMENTSREQUEST"]._serialized_end = 2607
     _globals["_GETACTIVEDEPLOYMENTSRESPONSE"]._serialized_start = 2609
     _globals["_GETACTIVEDEPLOYMENTSRESPONSE"]._serialized_end = 2702
-    _globals["_DEPLOYSERVICE"]._serialized_start = 2705
-    _globals["_DEPLOYSERVICE"]._serialized_end = 3646
+    _globals["_GETDEPLOYMENTSOURCEREQUEST"]._serialized_start = 2704
+    _globals["_GETDEPLOYMENTSOURCEREQUEST"]._serialized_end = 2769
+    _globals["_GETDEPLOYMENTSOURCERESPONSE"]._serialized_start = 2771
+    _globals["_GETDEPLOYMENTSOURCERESPONSE"]._serialized_end = 2831
+    _globals["_DEPLOYSERVICE"]._serialized_start = 2834
+    _globals["_DEPLOYSERVICE"]._serialized_end = 3894
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/deploy_pb2.pyi

@@ -210,3 +210,15 @@ class GetActiveDeploymentsResponse(_message.Message):
     def __init__(
         self, deployments: _Optional[_Iterable[_Union[_deployment_pb2.Deployment, _Mapping]]] = ...
     ) -> None: ...
+
+class GetDeploymentSourceRequest(_message.Message):
+    __slots__ = ("deployment_id",)
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    deployment_id: str
+    def __init__(self, deployment_id: _Optional[str] = ...) -> None: ...
+
+class GetDeploymentSourceResponse(_message.Message):
+    __slots__ = ("signed_url",)
+    SIGNED_URL_FIELD_NUMBER: _ClassVar[int]
+    signed_url: str
+    def __init__(self, signed_url: _Optional[str] = ...) -> None: ...

chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py

@@ -55,6 +55,11 @@ class DeployServiceStub(object):
             request_serializer=chalk_dot_server_dot_v1_dot_deploy__pb2.TagDeploymentRequest.SerializeToString,
             response_deserializer=chalk_dot_server_dot_v1_dot_deploy__pb2.TagDeploymentResponse.FromString,
         )
+        self.GetDeploymentSource = channel.unary_unary(
+            "/chalk.server.v1.DeployService/GetDeploymentSource",
+            request_serializer=chalk_dot_server_dot_v1_dot_deploy__pb2.GetDeploymentSourceRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_deploy__pb2.GetDeploymentSourceResponse.FromString,
+        )
 
 
 class DeployServiceServicer(object):
@@ -108,6 +113,12 @@ class DeployServiceServicer(object):
         context.set_details("Method not implemented!")
         raise NotImplementedError("Method not implemented!")
 
+    def GetDeploymentSource(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_DeployServiceServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -151,6 +162,11 @@ def add_DeployServiceServicer_to_server(servicer, server):
             request_deserializer=chalk_dot_server_dot_v1_dot_deploy__pb2.TagDeploymentRequest.FromString,
             response_serializer=chalk_dot_server_dot_v1_dot_deploy__pb2.TagDeploymentResponse.SerializeToString,
         ),
+        "GetDeploymentSource": grpc.unary_unary_rpc_method_handler(
+            servicer.GetDeploymentSource,
+            request_deserializer=chalk_dot_server_dot_v1_dot_deploy__pb2.GetDeploymentSourceRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_deploy__pb2.GetDeploymentSourceResponse.SerializeToString,
+        ),
     }
     generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.DeployService", rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
@@ -391,3 +407,32 @@ class DeployService(object):
             timeout,
             metadata,
         )
+
+    @staticmethod
+    def GetDeploymentSource(
+        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.DeployService/GetDeploymentSource",
+            chalk_dot_server_dot_v1_dot_deploy__pb2.GetDeploymentSourceRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_deploy__pb2.GetDeploymentSourceResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi

@@ -16,6 +16,8 @@ from chalk._gen.chalk.server.v1.deploy_pb2 import (
     GetActiveDeploymentsResponse,
     GetDeploymentRequest,
     GetDeploymentResponse,
+    GetDeploymentSourceRequest,
+    GetDeploymentSourceResponse,
     ListDeploymentsRequest,
     ListDeploymentsResponse,
     ScaleDeploymentRequest,
@@ -66,6 +68,10 @@ class DeployServiceStub:
         TagDeploymentRequest,
         TagDeploymentResponse,
     ]
+    GetDeploymentSource: UnaryUnaryMultiCallable[
+        GetDeploymentSourceRequest,
+        GetDeploymentSourceResponse,
+    ]
 
 class DeployServiceServicer(metaclass=ABCMeta):
     @abstractmethod
@@ -116,5 +122,11 @@ class DeployServiceServicer(metaclass=ABCMeta):
         request: TagDeploymentRequest,
         context: ServicerContext,
     ) -> TagDeploymentResponse: ...
+    @abstractmethod
+    def GetDeploymentSource(
+        self,
+        request: GetDeploymentSourceRequest,
+        context: ServicerContext,
+    ) -> GetDeploymentSourceResponse: ...
 
 def add_DeployServiceServicer_to_server(servicer: DeployServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/scheduler_pb2.pyi

@@ -45,6 +45,7 @@ class CronResolverRun(_message.Message):
         "lower_bound",
         "upper_bound",
         "max_samples",
+        "used_job_queue",
     )
     ID_FIELD_NUMBER: _ClassVar[int]
     ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
@@ -60,6 +61,7 @@ class CronResolverRun(_message.Message):
     LOWER_BOUND_FIELD_NUMBER: _ClassVar[int]
     UPPER_BOUND_FIELD_NUMBER: _ClassVar[int]
     MAX_SAMPLES_FIELD_NUMBER: _ClassVar[int]
+    USED_JOB_QUEUE_FIELD_NUMBER: _ClassVar[int]
     id: str
     environment_id: str
     resolver_fqn: str
@@ -74,6 +76,7 @@ class CronResolverRun(_message.Message):
     lower_bound: _timestamp_pb2.Timestamp
     upper_bound: _timestamp_pb2.Timestamp
     max_samples: int
+    used_job_queue: bool
     def __init__(
         self,
         id: _Optional[str] = ...,
@@ -90,6 +93,7 @@ class CronResolverRun(_message.Message):
         lower_bound: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
         upper_bound: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
         max_samples: _Optional[int] = ...,
+        used_job_queue: bool = ...,
     ) -> None: ...
 
 class ManualTriggerCronResolverRequest(_message.Message):

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.94.8"
+__version__ = "2.95.0"

chalk/client/client.py

@@ -2157,21 +2157,20 @@ class ChalkClient:
         name: str,
         version: Optional[int] = None,
     ) -> Union[GetRegisteredModelResponse, GetRegisteredModelVersionResponse]:
-        """
-        Retrieve a registered model from the Chalk model registry.
+        """Retrieve a registered model from the Chalk model registry.
 
         Parameters
         ----------
-        name : str
-           Name of the model to retrieve
-        version : int, optional
-           Specific version number to retrieve. If not provided, returns
-           information about all versions of the model
+        name
+            Name of the model to retrieve.
+        version
+            Specific version number to retrieve. If not provided, returns
+            information about all versions of the model.
 
         Returns
         -------
-        GetRegisteredModelResponse or GetRegisteredModelVersionResponse
-           Model information including metadata, versions, and configuration details
+        Union[GetRegisteredModelResponse, GetRegisteredModelVersionResponse]
+            Model information including metadata, versions, and configuration details.
 
         Examples
         --------
@@ -2248,8 +2247,7 @@ class ChalkClient:
         source_config: Optional[SourceConfig] = None,
         dependencies: Optional[List[str]] = None,
     ) -> RegisterModelVersionResponse:
-        """
-        Register a model in the Chalk model registry.
+        """Register a model in the Chalk model registry.
 
         Parameters
         ----------
@@ -2291,6 +2289,7 @@ class ChalkClient:
         dependencies : List[str], optional
             List of package dependencies needed to run this model.
             e.g. ["torch==2.7.1", "numpy==1.26.4"]
+
         Returns
         -------
         ModelVersion

chalk/client/client_grpc.py

@@ -1647,49 +1647,49 @@ class ChalkGRPCClient:
         source_config: Optional[SourceConfig] = None,
         dependencies: Optional[List[str]] = None,
     ) -> RegisterModelVersionResponse:
-        """
-        Register a model in the Chalk model registry.
+        """Register a model in the Chalk model registry.
 
         Parameters
         ----------
-        name : str
-           Unique name for the model
-        aliases : list of str, optional
-           List of version aliases (e.g., ["v1.0", "latest"])
-        model : object, optional
-           Python model object (for object-based registration)
-        model_paths : list of str, optional
-           Paths to model files (for file-based registration)
-        additional_files : List[str], optional
-           Additional files needed for inference (tokenizers, configs, etc.)
-        model_type : ModelType, optional
-           Type of model framework
-        model_encoding : ModelEncoding, optional
-           Serialization format
-        input_schema : dict, list, or Any
-           Definition of the input schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        output_schema : dict, list, or Any
-           Definition of the output schema. Can be:
-           - dict: Dictionary mapping column names to dtypes for tabular data
-           - list: List of (shape, dtype) tuples for tensor data
-        metadata : dict, optional
-           Additional metadata dictionary containing framework info,
-           training details, performance metrics, etc.
-        input_features : FeatureReference, str, optional
+        name
+            Unique name for the model
+        aliases
+            List of version aliases (e.g., `["v1.0", "latest"]`)
+        model
+            Python model object (for object-based registration)
+        model_paths
+            Paths to model files (for file-based registration)
+        additional_files
+            Additional files needed for inference (tokenizers, configs, etc.)
+        model_type
+            Type of model framework
+        model_encoding
+            Serialization format
+        input_schema
+            Definition of the input schema. Can be:
+            - dict: Dictionary mapping column names to dtypes for tabular data
+            - list: List of `(shape, dtype)` tuples for tensor data
+        output_schema
+            Definition of the output schema. Can be:
+            - dict: Dictionary mapping column names to dtypes for tabular data
+            - list: List of `(shape, dtype)` tuples for tensor data
+        metadata
+            Additional metadata dictionary containing framework info,
+            training details, performance metrics, etc.
+        input_features
             The features to be used as inputs to the model.
             For example, `[User.message]`. Features can also be expressed as snakecased strings,
             e.g. `["user.message"]`
-        output_features : FeatureReference, str, optional
+        output_features
             The features to be used as outputs to the model.
             For example, `[User.is_spam]`. Features can also be expressed as snakecased strings,
             e.g. `["user.is_spam"]`
-        source_config : LocalSourceConfig, S3SourceConfig, HFSourceConfig, optional
+        source_config
             Config to pass credentials to access files from a remote source.
-        dependencies : List[str], optional
+        dependencies
             List of package dependencies needed to run this model.
-            e.g. ["torch==2.7.1", "numpy==1.26.4"]
+            e.g. `["torch==2.7.1", "numpy==1.26.4"]`
+
         Returns
         -------
         ModelVersion

chalk/features/feature_field.py

@@ -2199,6 +2199,7 @@ def has_many(
         The maximum number of items to cache for the joined feature. The
         items in the joined feature aggregate, storing the latest values
         of the joined feature for each primary key in the joined feature.
+
     Examples
     --------
     >>> from chalk.features import DataFrame, features, has_many

chalk/features/feature_set_decorator.py

@@ -141,6 +141,7 @@ def features(
         The `cache_nulls` and `cache_defaults` options can be used together on the same feature with the
         following exceptions: if `cache_nulls=False`, then `cache_defaults` cannot be `"evict_defaults"`, and if
         `cache_nulls="evict_defaults"`, then `cache_defaults` cannot be `False`.
+
     Other Parameters
     ----------------
     cls

chalk/features/resolver.py

@@ -2829,6 +2829,8 @@ class StreamResolver(Resolver[P, T]):
         sql_settings: SQLResolverSettings | None,
         feature_expressions: dict[Feature, Underscore] | None,
         message_producer_parsed: StreamResolverMessageProducerParsed | None,
+        skip_online: bool = False,
+        skip_offline: bool = False,
     ):
         super().__init__(
             function_definition=function_definition,
@@ -2897,6 +2899,8 @@ class StreamResolver(Resolver[P, T]):
 
         self.feature_expressions: dict[Feature, Underscore] | None = feature_expressions
         self.message_producer_parsed: StreamResolverMessageProducerParsed | None = message_producer_parsed
+        self.skip_online = skip_online
+        self.skip_offline = skip_offline
 
     @property
     def output_features(self) -> Sequence[Feature]:
@@ -3808,6 +3812,8 @@ def make_stream_resolver(
     owner: Optional[str] = None,
     doc: str | None = None,
     sink: Sink | None = None,
+    skip_online: bool = False,
+    skip_offline: bool = False,
 ) -> StreamResolver:
     """Constructs a streaming resolver that, instead of a Python function,
     defines its output features as column projections on an input message.
@@ -3840,6 +3846,14 @@ def make_stream_resolver(
     sink
         An optional message producer configuration that specifies where to send messages.
         Read more at https://docs.chalk.ai/api-docs#Sink
+    skip_online
+        If True, skip online persistence (no writes to Redis/DynamoDB/etc).
+        Results will still be processed but not stored in online stores.
+        Note: Only applies to native streaming. Default: False
+    skip_offline
+        If True, skip offline persistence (no result bus publishing for offline storage).
+        Results will still be processed but not stored in offline stores (S3/BigQuery/etc).
+        Note: Only applies to native streaming. Default: False
 
     Returns
     -------
@@ -4017,6 +4031,8 @@ def make_stream_resolver(
         sql_settings=None,
         feature_expressions={unwrap_feature(x): u for x, u in output_features.items()},
         message_producer_parsed=message_producer_parsed,
+        skip_online=skip_online,
+        skip_offline=skip_offline,
     )
     resolver.add_to_registry(override=False)
     return resolver