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

chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py

@@ -18,7 +18,7 @@ from chalk._gen.chalk.expression.v1 import expression_pb2 as chalk_dot_expressio
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n"chalk/dataframe/v1/dataframe.proto\x12\x12\x63halk.dataframe.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto\x1a$chalk/expression/v1/expression.proto"\xe7\x04\n\x10\x44\x61taFrameOperand\x12#\n\x0cvalue_string\x18\x01 \x01(\tH\x00R\x0bvalueString\x12\x1d\n\tvalue_int\x18\x02 \x01(\x03H\x00R\x08valueInt\x12\x1f\n\nvalue_bool\x18\x03 \x01(\x08H\x00R\tvalueBool\x12;\n\nvalue_none\x18\x04 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyNoneH\x00R\tvalueNone\x12;\n\nvalue_list\x18\x05 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyListH\x00R\tvalueList\x12;\n\nvalue_dict\x18\x06 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyDictH\x00R\tvalueDict\x12X\n\x15value_dataframe_index\x18\x07 \x01(\x0b\x32".chalk.dataframe.v1.DataFrameIndexH\x00R\x13valueDataframeIndex\x12;\n\x0c\x61rrow_schema\x18\x08 \x01(\x0b\x32\x16.chalk.arrow.v1.SchemaH\x00R\x0b\x61rrowSchema\x12\x44\n\x0b\x61rrow_table\x18\t \x01(\x0b\x32!.chalk.arrow.v1.TableParquetBytesH\x00R\narrowTable\x12O\n\x0funderscore_expr\x18\n \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\x0eunderscoreExprB\t\n\x07operand">\n\x0e\x44\x61taFrameIndex\x12,\n\x12\x64\x61taframe_op_index\x18\x01 \x01(\x05R\x10\x64\x61taframeOpIndex"\x08\n\x06PyNone"M\n\x06PyList\x12\x43\n\nlist_items\x18\x01 \x03(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\tlistItems"\x97\x01\n\x0bPyDictEntry\x12\x41\n\tentry_key\x18\x01 \x01(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\x08\x65ntryKey\x12\x45\n\x0b\x65ntry_value\x18\x02 \x01(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\nentryValue"L\n\x06PyDict\x12\x42\n\x0c\x64ict_entries\x18\x01 \x03(\x0b\x32\x1f.chalk.dataframe.v1.PyDictEntryR\x0b\x64ictEntries"\xfc\x01\n\x14\x44\x61taFrameConstructor\x12J\n\x0cself_operand\x18\x01 \x01(\x0b\x32".chalk.dataframe.v1.DataFrameIndexH\x00R\x0bselfOperand\x88\x01\x01\x12#\n\rfunction_name\x18\x02 \x01(\tR\x0c\x66unctionName\x12.\n\x04\x61rgs\x18\x03 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyListR\x04\x61rgs\x12\x32\n\x06kwargs\x18\x04 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyDictR\x06kwargsB\x0f\n\r_self_operand"]\n\rDataFramePlan\x12L\n\x0c\x63onstructors\x18\x01 \x03(\x0b\x32(.chalk.dataframe.v1.DataFrameConstructorR\x0c\x63onstructorsB\x92\x01\n\x16\x63om.chalk.dataframe.v1B\x0e\x44\x61taframeProtoP\x01\xa2\x02\x03\x43\x44X\xaa\x02\x12\x43halk.Dataframe.V1\xca\x02\x12\x43halk\\Dataframe\\V1\xe2\x02\x1e\x43halk\\Dataframe\\V1\\GPBMetadata\xea\x02\x14\x43halk::Dataframe::V1b\x06proto3'
+    b'\n"chalk/dataframe/v1/dataframe.proto\x12\x12\x63halk.dataframe.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto\x1a$chalk/expression/v1/expression.proto"\xb4\x05\n\x10\x44\x61taFrameOperand\x12#\n\x0cvalue_string\x18\x01 \x01(\tH\x00R\x0bvalueString\x12\x1d\n\tvalue_int\x18\x02 \x01(\x03H\x00R\x08valueInt\x12\x1f\n\nvalue_bool\x18\x03 \x01(\x08H\x00R\tvalueBool\x12;\n\nvalue_none\x18\x04 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyNoneH\x00R\tvalueNone\x12;\n\nvalue_list\x18\x05 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyListH\x00R\tvalueList\x12;\n\nvalue_dict\x18\x06 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyDictH\x00R\tvalueDict\x12X\n\x15value_dataframe_index\x18\x07 \x01(\x0b\x32".chalk.dataframe.v1.DataFrameIndexH\x00R\x13valueDataframeIndex\x12;\n\x0c\x61rrow_schema\x18\x08 \x01(\x0b\x32\x16.chalk.arrow.v1.SchemaH\x00R\x0b\x61rrowSchema\x12\x44\n\x0b\x61rrow_table\x18\t \x01(\x0b\x32!.chalk.arrow.v1.TableParquetBytesH\x00R\narrowTable\x12O\n\x0funderscore_expr\x18\n \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\x0eunderscoreExpr\x12K\n\rlibchalk_expr\x18\x0b \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\x0clibchalkExprB\t\n\x07operand">\n\x0e\x44\x61taFrameIndex\x12,\n\x12\x64\x61taframe_op_index\x18\x01 \x01(\x05R\x10\x64\x61taframeOpIndex"\x08\n\x06PyNone"M\n\x06PyList\x12\x43\n\nlist_items\x18\x01 \x03(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\tlistItems"\x97\x01\n\x0bPyDictEntry\x12\x41\n\tentry_key\x18\x01 \x01(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\x08\x65ntryKey\x12\x45\n\x0b\x65ntry_value\x18\x02 \x01(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\nentryValue"L\n\x06PyDict\x12\x42\n\x0c\x64ict_entries\x18\x01 \x03(\x0b\x32\x1f.chalk.dataframe.v1.PyDictEntryR\x0b\x64ictEntries"\xfc\x01\n\x14\x44\x61taFrameConstructor\x12J\n\x0cself_operand\x18\x01 \x01(\x0b\x32".chalk.dataframe.v1.DataFrameIndexH\x00R\x0bselfOperand\x88\x01\x01\x12#\n\rfunction_name\x18\x02 \x01(\tR\x0c\x66unctionName\x12.\n\x04\x61rgs\x18\x03 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyListR\x04\x61rgs\x12\x32\n\x06kwargs\x18\x04 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyDictR\x06kwargsB\x0f\n\r_self_operand"]\n\rDataFramePlan\x12L\n\x0c\x63onstructors\x18\x01 \x03(\x0b\x32(.chalk.dataframe.v1.DataFrameConstructorR\x0c\x63onstructorsB\x92\x01\n\x16\x63om.chalk.dataframe.v1B\x0e\x44\x61taframeProtoP\x01\xa2\x02\x03\x43\x44X\xaa\x02\x12\x43halk.Dataframe.V1\xca\x02\x12\x43halk\\Dataframe\\V1\xe2\x02\x1e\x43halk\\Dataframe\\V1\\GPBMetadata\xea\x02\x14\x43halk::Dataframe::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -30,19 +30,19 @@ if _descriptor._USE_C_DESCRIPTORS == False:
         "DESCRIPTOR"
     ]._serialized_options = b"\n\026com.chalk.dataframe.v1B\016DataframeProtoP\001\242\002\003CDX\252\002\022Chalk.Dataframe.V1\312\002\022Chalk\\Dataframe\\V1\342\002\036Chalk\\Dataframe\\V1\\GPBMetadata\352\002\024Chalk::Dataframe::V1"
     _globals["_DATAFRAMEOPERAND"]._serialized_start = 125
-    _globals["_DATAFRAMEOPERAND"]._serialized_end = 740
-    _globals["_DATAFRAMEINDEX"]._serialized_start = 742
-    _globals["_DATAFRAMEINDEX"]._serialized_end = 804
-    _globals["_PYNONE"]._serialized_start = 806
-    _globals["_PYNONE"]._serialized_end = 814
-    _globals["_PYLIST"]._serialized_start = 816
-    _globals["_PYLIST"]._serialized_end = 893
-    _globals["_PYDICTENTRY"]._serialized_start = 896
-    _globals["_PYDICTENTRY"]._serialized_end = 1047
-    _globals["_PYDICT"]._serialized_start = 1049
-    _globals["_PYDICT"]._serialized_end = 1125
-    _globals["_DATAFRAMECONSTRUCTOR"]._serialized_start = 1128
-    _globals["_DATAFRAMECONSTRUCTOR"]._serialized_end = 1380
-    _globals["_DATAFRAMEPLAN"]._serialized_start = 1382
-    _globals["_DATAFRAMEPLAN"]._serialized_end = 1475
+    _globals["_DATAFRAMEOPERAND"]._serialized_end = 817
+    _globals["_DATAFRAMEINDEX"]._serialized_start = 819
+    _globals["_DATAFRAMEINDEX"]._serialized_end = 881
+    _globals["_PYNONE"]._serialized_start = 883
+    _globals["_PYNONE"]._serialized_end = 891
+    _globals["_PYLIST"]._serialized_start = 893
+    _globals["_PYLIST"]._serialized_end = 970
+    _globals["_PYDICTENTRY"]._serialized_start = 973
+    _globals["_PYDICTENTRY"]._serialized_end = 1124
+    _globals["_PYDICT"]._serialized_start = 1126
+    _globals["_PYDICT"]._serialized_end = 1202
+    _globals["_DATAFRAMECONSTRUCTOR"]._serialized_start = 1205
+    _globals["_DATAFRAMECONSTRUCTOR"]._serialized_end = 1457
+    _globals["_DATAFRAMEPLAN"]._serialized_start = 1459
+    _globals["_DATAFRAMEPLAN"]._serialized_end = 1552
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi

@@ -25,6 +25,7 @@ class DataFrameOperand(_message.Message):
         "arrow_schema",
         "arrow_table",
         "underscore_expr",
+        "libchalk_expr",
     )
     VALUE_STRING_FIELD_NUMBER: _ClassVar[int]
     VALUE_INT_FIELD_NUMBER: _ClassVar[int]
@@ -36,6 +37,7 @@ class DataFrameOperand(_message.Message):
     ARROW_SCHEMA_FIELD_NUMBER: _ClassVar[int]
     ARROW_TABLE_FIELD_NUMBER: _ClassVar[int]
     UNDERSCORE_EXPR_FIELD_NUMBER: _ClassVar[int]
+    LIBCHALK_EXPR_FIELD_NUMBER: _ClassVar[int]
     value_string: str
     value_int: int
     value_bool: bool
@@ -46,6 +48,7 @@ class DataFrameOperand(_message.Message):
     arrow_schema: _arrow_pb2.Schema
     arrow_table: _arrow_pb2.TableParquetBytes
     underscore_expr: _expression_pb2.LogicalExprNode
+    libchalk_expr: _expression_pb2.LogicalExprNode
     def __init__(
         self,
         value_string: _Optional[str] = ...,
@@ -58,6 +61,7 @@ class DataFrameOperand(_message.Message):
         arrow_schema: _Optional[_Union[_arrow_pb2.Schema, _Mapping]] = ...,
         arrow_table: _Optional[_Union[_arrow_pb2.TableParquetBytes, _Mapping]] = ...,
         underscore_expr: _Optional[_Union[_expression_pb2.LogicalExprNode, _Mapping]] = ...,
+        libchalk_expr: _Optional[_Union[_expression_pb2.LogicalExprNode, _Mapping]] = ...,
     ) -> None: ...
 
 class DataFrameIndex(_message.Message):

chalk/_gen/chalk/server/v1/indexing_job_pb2.py

@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/server/v1/indexing_job.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.artifacts.v1 import export_pb2 as chalk_dot_artifacts_dot_v1_dot_export__pb2
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as chalk_dot_auth_dot_v1_dot_permissions__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n"chalk/server/v1/indexing_job.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/artifacts/v1/export.proto\x1a\x1f\x63halk/auth/v1/permissions.proto"\x99\x01\n\x1bGetIndexingJobStatusRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12U\n\x15\x64irectory_prefix_enum\x18\x02 \x01(\x0e\x32!.chalk.server.v1.DirectoryOptionsR\x13\x64irectoryPrefixEnum"R\n\x1cGetIndexingJobStatusResponse\x12\x32\n\x06\x65xport\x18\x01 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\x06\x65xport*\xb2\x01\n\x10\x44irectoryOptions\x12!\n\x1d\x44IRECTORY_OPTIONS_UNSPECIFIED\x10\x00\x12\x1a\n\x16\x44IRECTORY_OPTIONS_MAIN\x10\x01\x12\x1c\n\x18\x44IRECTORY_OPTIONS_SHADOW\x10\x02\x12\x1d\n\x19\x44IRECTORY_OPTIONS_DRY_RUN\x10\x03\x12"\n\x1e\x44IRECTORY_OPTIONS_INDEXING_JOB\x10\x04\x32\x91\x01\n\x12IndexingJobService\x12{\n\x14GetIndexingJobStatus\x12,.chalk.server.v1.GetIndexingJobStatusRequest\x1a-.chalk.server.v1.GetIndexingJobStatusResponse"\x06\x90\x02\x01\x80}\x0b\x42\x99\x01\n\x13\x63om.chalk.server.v1B\x10IndexingJobProtoP\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.indexing_job_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\023com.chalk.server.v1B\020IndexingJobProtoP\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["_INDEXINGJOBSERVICE"].methods_by_name["GetIndexingJobStatus"]._options = None
+    _globals["_INDEXINGJOBSERVICE"].methods_by_name[
+        "GetIndexingJobStatus"
+    ]._serialized_options = b"\220\002\001\200}\013"
+    _globals["_DIRECTORYOPTIONS"]._serialized_start = 362
+    _globals["_DIRECTORYOPTIONS"]._serialized_end = 540
+    _globals["_GETINDEXINGJOBSTATUSREQUEST"]._serialized_start = 122
+    _globals["_GETINDEXINGJOBSTATUSREQUEST"]._serialized_end = 275
+    _globals["_GETINDEXINGJOBSTATUSRESPONSE"]._serialized_start = 277
+    _globals["_GETINDEXINGJOBSTATUSRESPONSE"]._serialized_end = 359
+    _globals["_INDEXINGJOBSERVICE"]._serialized_start = 543
+    _globals["_INDEXINGJOBSERVICE"]._serialized_end = 688
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi

@@ -0,0 +1,38 @@
+from chalk._gen.chalk.artifacts.v1 import export_pb2 as _export_pb2
+from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
+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, Mapping as _Mapping, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class DirectoryOptions(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    DIRECTORY_OPTIONS_UNSPECIFIED: _ClassVar[DirectoryOptions]
+    DIRECTORY_OPTIONS_MAIN: _ClassVar[DirectoryOptions]
+    DIRECTORY_OPTIONS_SHADOW: _ClassVar[DirectoryOptions]
+    DIRECTORY_OPTIONS_DRY_RUN: _ClassVar[DirectoryOptions]
+    DIRECTORY_OPTIONS_INDEXING_JOB: _ClassVar[DirectoryOptions]
+
+DIRECTORY_OPTIONS_UNSPECIFIED: DirectoryOptions
+DIRECTORY_OPTIONS_MAIN: DirectoryOptions
+DIRECTORY_OPTIONS_SHADOW: DirectoryOptions
+DIRECTORY_OPTIONS_DRY_RUN: DirectoryOptions
+DIRECTORY_OPTIONS_INDEXING_JOB: DirectoryOptions
+
+class GetIndexingJobStatusRequest(_message.Message):
+    __slots__ = ("deployment_id", "directory_prefix_enum")
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    DIRECTORY_PREFIX_ENUM_FIELD_NUMBER: _ClassVar[int]
+    deployment_id: str
+    directory_prefix_enum: DirectoryOptions
+    def __init__(
+        self, deployment_id: _Optional[str] = ..., directory_prefix_enum: _Optional[_Union[DirectoryOptions, str]] = ...
+    ) -> None: ...
+
+class GetIndexingJobStatusResponse(_message.Message):
+    __slots__ = ("export",)
+    EXPORT_FIELD_NUMBER: _ClassVar[int]
+    export: _export_pb2.Export
+    def __init__(self, export: _Optional[_Union[_export_pb2.Export, _Mapping]] = ...) -> None: ...

chalk/_gen/chalk/server/v1/indexing_job_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 indexing_job_pb2 as chalk_dot_server_dot_v1_dot_indexing__job__pb2
+
+
+class IndexingJobServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.GetIndexingJobStatus = channel.unary_unary(
+            "/chalk.server.v1.IndexingJobService/GetIndexingJobStatus",
+            request_serializer=chalk_dot_server_dot_v1_dot_indexing__job__pb2.GetIndexingJobStatusRequest.SerializeToString,
+            response_deserializer=chalk_dot_server_dot_v1_dot_indexing__job__pb2.GetIndexingJobStatusResponse.FromString,
+        )
+
+
+class IndexingJobServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def GetIndexingJobStatus(self, request, context):
+        """GetIndexingStatus checks if the indexing job has completed successfully for a deployment"""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details("Method not implemented!")
+        raise NotImplementedError("Method not implemented!")
+
+
+def add_IndexingJobServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+        "GetIndexingJobStatus": grpc.unary_unary_rpc_method_handler(
+            servicer.GetIndexingJobStatus,
+            request_deserializer=chalk_dot_server_dot_v1_dot_indexing__job__pb2.GetIndexingJobStatusRequest.FromString,
+            response_serializer=chalk_dot_server_dot_v1_dot_indexing__job__pb2.GetIndexingJobStatusResponse.SerializeToString,
+        ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler("chalk.server.v1.IndexingJobService", rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+# This class is part of an EXPERIMENTAL API.
+class IndexingJobService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def GetIndexingJobStatus(
+        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.IndexingJobService/GetIndexingJobStatus",
+            chalk_dot_server_dot_v1_dot_indexing__job__pb2.GetIndexingJobStatusRequest.SerializeToString,
+            chalk_dot_server_dot_v1_dot_indexing__job__pb2.GetIndexingJobStatusResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+        )

chalk/_gen/chalk/server/v1/indexing_job_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.indexing_job_pb2 import (
+    GetIndexingJobStatusRequest,
+    GetIndexingJobStatusResponse,
+)
+from grpc import (
+    Channel,
+    Server,
+    ServicerContext,
+    UnaryUnaryMultiCallable,
+)
+
+class IndexingJobServiceStub:
+    def __init__(self, channel: Channel) -> None: ...
+    GetIndexingJobStatus: UnaryUnaryMultiCallable[
+        GetIndexingJobStatusRequest,
+        GetIndexingJobStatusResponse,
+    ]
+    """GetIndexingStatus checks if the indexing job has completed successfully for a deployment"""
+
+class IndexingJobServiceServicer(metaclass=ABCMeta):
+    @abstractmethod
+    def GetIndexingJobStatus(
+        self,
+        request: GetIndexingJobStatusRequest,
+        context: ServicerContext,
+    ) -> GetIndexingJobStatusResponse:
+        """GetIndexingStatus checks if the indexing job has completed successfully for a deployment"""
+
+def add_IndexingJobServiceServicer_to_server(servicer: IndexingJobServiceServicer, server: Server) -> None: ...

chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py

@@ -19,7 +19,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n)chalk/server/v1/scheduled_query_run.proto\x12\x0f\x63halk.server.v1\x1a%chalk/server/v1/offline_queries.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xd1\x05\n\x11ScheduledQueryRun\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1a\n\x06run_id\x18\x04 \x01(\tH\x01R\x05runId\x88\x01\x01\x12"\n\rcron_query_id\x18\x05 \x01(\x03R\x0b\x63ronQueryId\x12\x33\n\x16\x63ron_query_schedule_id\x18\x06 \x01(\x03R\x13\x63ronQueryScheduleId\x12\x1b\n\tcron_name\x18\x07 \x01(\tR\x08\x63ronName\x12-\n\x10gcr_execution_id\x18\x08 \x01(\tH\x02R\x0egcrExecutionId\x88\x01\x01\x12%\n\x0cgcr_job_name\x18\t \x01(\tH\x03R\ngcrJobName\x88\x01\x01\x12(\n\x10offline_query_id\x18\n \x01(\tR\x0eofflineQueryId\x12\x39\n\ncreated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12@\n\x06status\x18\r \x01(\x0e\x32(.chalk.server.v1.ScheduledQueryRunStatusR\x06status\x12\x35\n\x14\x62locker_operation_id\x18\x0e \x01(\tH\x04R\x12\x62lockerOperationId\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_run_idB\x13\n\x11_gcr_execution_idB\x0f\n\r_gcr_job_nameB\x17\n\x15_blocker_operation_id"\xa7\x01\n\x1bGetScheduledQueryRunRequest\x12\x17\n\x06run_id\x18\x01 \x01(\x03H\x00R\x05runId\x12*\n\x10offline_query_id\x18\x02 \x01(\tH\x00R\x0eofflineQueryId\x12\x35\n\x08get_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x07getMaskB\x0c\n\nidentifier"\xd1\x01\n\x1cGetScheduledQueryRunResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun\x12K\n\roffline_query\x18\x02 \x01(\x0b\x32!.chalk.server.v1.OfflineQueryMetaH\x00R\x0cofflineQuery\x88\x01\x01\x42\x10\n\x0e_offline_query"p\n\x1cGetScheduledQueryRunsRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor\x12\x14\n\x05limit\x18\x03 \x01(\x05R\x05limit"o\n\x1dGetScheduledQueryRunsResponse\x12\x36\n\x04runs\x18\x01 \x03(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x04runs\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor*\x8f\x03\n\x17ScheduledQueryRunStatus\x12*\n&SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED\x10\x00\x12+\n\'SCHEDULED_QUERY_RUN_STATUS_INITIALIZING\x10\x01\x12*\n&SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED\x10\x02\x12&\n"SCHEDULED_QUERY_RUN_STATUS_SKIPPED\x10\x03\x12%\n!SCHEDULED_QUERY_RUN_STATUS_QUEUED\x10\x04\x12&\n"SCHEDULED_QUERY_RUN_STATUS_WORKING\x10\x05\x12(\n$SCHEDULED_QUERY_RUN_STATUS_COMPLETED\x10\x06\x12%\n!SCHEDULED_QUERY_RUN_STATUS_FAILED\x10\x07\x12\'\n#SCHEDULED_QUERY_RUN_STATUS_CANCELED\x10\x08\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16ScheduledQueryRunProtoP\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)chalk/server/v1/scheduled_query_run.proto\x12\x0f\x63halk.server.v1\x1a%chalk/server/v1/offline_queries.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xa4\x06\n\x11ScheduledQueryRun\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1a\n\x06run_id\x18\x04 \x01(\tH\x01R\x05runId\x88\x01\x01\x12"\n\rcron_query_id\x18\x05 \x01(\x03R\x0b\x63ronQueryId\x12\x33\n\x16\x63ron_query_schedule_id\x18\x06 \x01(\x03R\x13\x63ronQueryScheduleId\x12\x1b\n\tcron_name\x18\x07 \x01(\tR\x08\x63ronName\x12-\n\x10gcr_execution_id\x18\x08 \x01(\tH\x02R\x0egcrExecutionId\x88\x01\x01\x12%\n\x0cgcr_job_name\x18\t \x01(\tH\x03R\ngcrJobName\x88\x01\x01\x12(\n\x10offline_query_id\x18\n \x01(\tR\x0eofflineQueryId\x12\x39\n\ncreated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12@\n\x06status\x18\r \x01(\x0e\x32(.chalk.server.v1.ScheduledQueryRunStatusR\x06status\x12\x35\n\x14\x62locker_operation_id\x18\x0e \x01(\tH\x04R\x12\x62lockerOperationId\x88\x01\x01\x12\x37\n\x15workflow_execution_id\x18\x0f \x01(\tH\x05R\x13workflowExecutionId\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_run_idB\x13\n\x11_gcr_execution_idB\x0f\n\r_gcr_job_nameB\x17\n\x15_blocker_operation_idB\x18\n\x16_workflow_execution_id"\xa7\x01\n\x1bGetScheduledQueryRunRequest\x12\x17\n\x06run_id\x18\x01 \x01(\x03H\x00R\x05runId\x12*\n\x10offline_query_id\x18\x02 \x01(\tH\x00R\x0eofflineQueryId\x12\x35\n\x08get_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x07getMaskB\x0c\n\nidentifier"\xd1\x01\n\x1cGetScheduledQueryRunResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun\x12K\n\roffline_query\x18\x02 \x01(\x0b\x32!.chalk.server.v1.OfflineQueryMetaH\x00R\x0cofflineQuery\x88\x01\x01\x42\x10\n\x0e_offline_query"\x9d\x01\n\x1cGetScheduledQueryRunsRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12\x1d\n\tcron_name\x18\x04 \x01(\tH\x00R\x08\x63ronName\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor\x12\x14\n\x05limit\x18\x03 \x01(\x05R\x05limitB\x0c\n\nidentifier"o\n\x1dGetScheduledQueryRunsResponse\x12\x36\n\x04runs\x18\x01 \x03(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x04runs\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor*\x8f\x03\n\x17ScheduledQueryRunStatus\x12*\n&SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED\x10\x00\x12+\n\'SCHEDULED_QUERY_RUN_STATUS_INITIALIZING\x10\x01\x12*\n&SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED\x10\x02\x12&\n"SCHEDULED_QUERY_RUN_STATUS_SKIPPED\x10\x03\x12%\n!SCHEDULED_QUERY_RUN_STATUS_QUEUED\x10\x04\x12&\n"SCHEDULED_QUERY_RUN_STATUS_WORKING\x10\x05\x12(\n$SCHEDULED_QUERY_RUN_STATUS_COMPLETED\x10\x06\x12%\n!SCHEDULED_QUERY_RUN_STATUS_FAILED\x10\x07\x12\'\n#SCHEDULED_QUERY_RUN_STATUS_CANCELED\x10\x08\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16ScheduledQueryRunProtoP\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()
@@ -30,16 +30,16 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals[
         "DESCRIPTOR"
     ]._serialized_options = b"\n\023com.chalk.server.v1B\026ScheduledQueryRunProtoP\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["_SCHEDULEDQUERYRUNSTATUS"]._serialized_start = 1502
-    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 1901
+    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_start = 1631
+    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 2030
     _globals["_SCHEDULEDQUERYRUN"]._serialized_start = 169
-    _globals["_SCHEDULEDQUERYRUN"]._serialized_end = 890
-    _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_start = 893
-    _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_end = 1060
-    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_start = 1063
-    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_end = 1272
-    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_start = 1274
-    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_end = 1386
-    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_start = 1388
-    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_end = 1499
+    _globals["_SCHEDULEDQUERYRUN"]._serialized_end = 973
+    _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_start = 976
+    _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_end = 1143
+    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_start = 1146
+    _globals["_GETSCHEDULEDQUERYRUNRESPONSE"]._serialized_end = 1355
+    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_start = 1358
+    _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_end = 1515
+    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_start = 1517
+    _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_end = 1628
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi

@@ -53,6 +53,7 @@ class ScheduledQueryRun(_message.Message):
         "updated_at",
         "status",
         "blocker_operation_id",
+        "workflow_execution_id",
     )
     ID_FIELD_NUMBER: _ClassVar[int]
     ENVIRONMENT_ID_FIELD_NUMBER: _ClassVar[int]
@@ -68,6 +69,7 @@ class ScheduledQueryRun(_message.Message):
     UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
     STATUS_FIELD_NUMBER: _ClassVar[int]
     BLOCKER_OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    WORKFLOW_EXECUTION_ID_FIELD_NUMBER: _ClassVar[int]
     id: int
     environment_id: str
     deployment_id: str
@@ -82,6 +84,7 @@ class ScheduledQueryRun(_message.Message):
     updated_at: _timestamp_pb2.Timestamp
     status: ScheduledQueryRunStatus
     blocker_operation_id: str
+    workflow_execution_id: str
     def __init__(
         self,
         id: _Optional[int] = ...,
@@ -98,6 +101,7 @@ class ScheduledQueryRun(_message.Message):
         updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
         status: _Optional[_Union[ScheduledQueryRunStatus, str]] = ...,
         blocker_operation_id: _Optional[str] = ...,
+        workflow_execution_id: _Optional[str] = ...,
     ) -> None: ...
 
 class GetScheduledQueryRunRequest(_message.Message):
@@ -128,15 +132,21 @@ class GetScheduledQueryRunResponse(_message.Message):
     ) -> None: ...
 
 class GetScheduledQueryRunsRequest(_message.Message):
-    __slots__ = ("cron_query_id", "cursor", "limit")
+    __slots__ = ("cron_query_id", "cron_name", "cursor", "limit")
     CRON_QUERY_ID_FIELD_NUMBER: _ClassVar[int]
+    CRON_NAME_FIELD_NUMBER: _ClassVar[int]
     CURSOR_FIELD_NUMBER: _ClassVar[int]
     LIMIT_FIELD_NUMBER: _ClassVar[int]
     cron_query_id: int
+    cron_name: str
     cursor: str
     limit: int
     def __init__(
-        self, cron_query_id: _Optional[int] = ..., cursor: _Optional[str] = ..., limit: _Optional[int] = ...
+        self,
+        cron_query_id: _Optional[int] = ...,
+        cron_name: _Optional[str] = ...,
+        cursor: _Optional[str] = ...,
+        limit: _Optional[int] = ...,
     ) -> None: ...
 
 class GetScheduledQueryRunsResponse(_message.Message):

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.93.7"
+__version__ = "2.93.9"

chalk/client/client.py

@@ -19,7 +19,6 @@ from typing import (
 
 import requests
 
-from chalk._gen.chalk.server.v1.scheduler_pb2 import ManualTriggerScheduledQueryResponse
 from chalk.client.models import (
     BranchDeployResponse,
     BranchIdParam,
@@ -33,6 +32,7 @@ from chalk.client.models import (
     GetIncrementalProgressResponse,
     GetRegisteredModelResponse,
     GetRegisteredModelVersionResponse,
+    ManualTriggerScheduledQueryResponse,
     OfflineQueryInputUri,
     OnlineQuery,
     OnlineQueryContext,
@@ -41,6 +41,7 @@ from chalk.client.models import (
     RegisterModelVersionResponse,
     ResolverRunResponse,
     ResourceRequests,
+    ScheduledQueryRun,
     StreamResolverTestResponse,
     WhoAmIResponse,
 )
@@ -1132,6 +1133,36 @@ class ChalkClient:
         """
         ...
 
+    def get_scheduled_query_run_history(
+        self,
+        name: str,
+        limit: int = 10,
+    ) -> List[ScheduledQueryRun]:
+        """
+        Get the run history for a scheduled query.
+
+        Parameters
+        ----------
+        name
+            The name of the scheduled query.
+        limit
+            The maximum number of runs to return. Defaults to 10.
+
+        Returns
+        -------
+        list[ScheduledQueryRun]
+            A response message containing the list of scheduled query runs.
+
+        Examples
+        --------
+        >>> from chalk.client import ChalkClient
+        >>> ChalkClient().get_scheduled_query_run_history(
+        ...     name="my_scheduled_query",
+        ...     limit=20,
+        ... )
+        """
+        ...
+
     def prompt_evaluation(
         self,
         prompts: list[Prompt | str],

chalk/client/client_grpc.py

@@ -74,10 +74,9 @@ from chalk._gen.chalk.server.v1.model_registry_pb2 import (
 )
 from chalk._gen.chalk.server.v1.model_registry_pb2_grpc import ModelRegistryServiceStub
 from chalk._gen.chalk.server.v1.offline_queries_pb2_grpc import OfflineQueryMetadataServiceStub
-from chalk._gen.chalk.server.v1.scheduler_pb2 import (
-    ManualTriggerScheduledQueryRequest,
-    ManualTriggerScheduledQueryResponse,
-)
+from chalk._gen.chalk.server.v1.scheduled_query_pb2_grpc import ScheduledQueryServiceStub
+from chalk._gen.chalk.server.v1.scheduled_query_run_pb2 import GetScheduledQueryRunsRequest
+from chalk._gen.chalk.server.v1.scheduler_pb2 import ManualTriggerScheduledQueryRequest
 from chalk._gen.chalk.server.v1.scheduler_pb2_grpc import SchedulerServiceStub
 from chalk._gen.chalk.server.v1.script_tasks_pb2 import CreateScriptTaskRequest, CreateScriptTaskResponse
 from chalk._gen.chalk.server.v1.script_tasks_pb2_grpc import ScriptTaskServiceStub
@@ -97,6 +96,9 @@ from chalk.client.models import (
     CreateBranchResponse,
     GetRegisteredModelResponse,
     GetRegisteredModelVersionResponse,
+)
+from chalk.client.models import ManualTriggerScheduledQueryResponse as ManualTriggerScheduledQueryResponseDataclass
+from chalk.client.models import (
     ModelUploadUrlResponse,
     OnlineQuery,
     OnlineQueryResponse,
@@ -104,6 +106,7 @@ from chalk.client.models import (
     RegisterModelResponse,
     RegisterModelVersionResponse,
     ResourceRequests,
+    ScheduledQueryRun,
     UploadFeaturesResponse,
 )
 from chalk.client.serialization.model_serialization import ModelSerializer
@@ -292,6 +295,14 @@ class StubProvider:
             )
         return SchedulerServiceStub(self._server_channel)
 
+    @cached_property
+    def scheduled_query_run_stub(self) -> ScheduledQueryServiceStub:
+        if self._server_channel is None:
+            raise ValueError(
+                "The GRPC engine service is not available. If you would like to set up a GRPC service, please contact Chalk."
+            )
+        return ScheduledQueryServiceStub(self._server_channel)
+
     @cached_property
     def sql_stub(self) -> SqlServiceStub:
         if self._engine_channel is None:
@@ -539,6 +550,9 @@ class StubRefresher:
     def call_scheduled_query_stub(self, fn: Callable[[SchedulerServiceStub], T]) -> T:
         return self._retry_callable(fn, lambda: self._stub.scheduled_query_stub)
 
+    def call_scheduled_query_run_stub(self, fn: Callable[[ScheduledQueryServiceStub], T]) -> T:
+        return self._retry_callable(fn, lambda: self._stub.scheduled_query_run_stub)
+
     def call_sql_stub(self, fn: Callable[[SqlServiceStub], T]) -> T:
         return self._retry_callable(fn, lambda: self._stub.sql_stub)
 
@@ -1252,7 +1266,7 @@ class ChalkGRPCClient:
         incremental_resolvers: Optional[Sequence[str]],
         max_samples: Optional[int],
         env_overrides: Optional[Mapping[str, str]],
-    ) -> ManualTriggerScheduledQueryResponse:
+    ) -> ManualTriggerScheduledQueryResponseDataclass:
         """
         Manually trigger a scheduled query request.
 
@@ -1289,7 +1303,7 @@ class ChalkGRPCClient:
         ...     name="my_scheduled_query",
         ... )
         """
-        return self._stub_refresher.call_scheduled_query_stub(
+        proto_resp = self._stub_refresher.call_scheduled_query_stub(
             lambda x: x.ManualTriggerScheduledQuery(
                 request=ManualTriggerScheduledQueryRequest(
                     cron_query_name=name,
@@ -1300,6 +1314,45 @@ class ChalkGRPCClient:
                 ),
             )
         )
+        return ManualTriggerScheduledQueryResponseDataclass.from_proto(proto_resp)
+
+    def get_scheduled_query_run_history(
+        self,
+        name: str,
+        limit: int = 10,
+    ) -> List[ScheduledQueryRun]:
+        """
+        Get the run history for a scheduled query.
+
+        Parameters
+        ----------
+        name
+            The name of the scheduled query.
+        limit
+            The maximum number of runs to return. Defaults to 10.
+
+        Returns
+        -------
+        list[ScheduledQueryRun]
+            A response message containing the list of scheduled query runs.
+
+        Examples
+        --------
+        >>> from chalk.client.client_grpc import ChalkGRPCClient
+        >>> ChalkGRPCClient().get_scheduled_query_run_history(
+        ...     name="my_scheduled_query",
+        ...     limit=20,
+        ... )
+        """
+        proto_resp = self._stub_refresher.call_scheduled_query_run_stub(
+            lambda x: x.GetScheduledQueryRuns(
+                GetScheduledQueryRunsRequest(
+                    cron_name=name,
+                    limit=limit,
+                )
+            )
+        )
+        return [ScheduledQueryRun.from_proto(run) for run in proto_resp.runs]
 
     def get_graph(self, deployment: DeploymentId | None = None) -> Graph:
         """Get the graph for a given deployment.

chalk/client/client_impl.py

@@ -51,7 +51,6 @@ from typing_extensions import NoReturn, override
 from urllib3 import Retry
 
 import chalk._repr.utils as repr_utils
-from chalk._gen.chalk.server.v1.scheduler_pb2 import ManualTriggerScheduledQueryResponse
 from chalk._reporting.models import BatchReport, BatchReportResponse
 from chalk._reporting.progress import ProgressService
 from chalk._upload_features.utils import to_multi_upload_inputs
@@ -100,6 +99,7 @@ from chalk.client.models import (
     GetRegisteredModelResponse,
     GetRegisteredModelVersionResponse,
     IngestDatasetRequest,
+    ManualTriggerScheduledQueryResponse,
     MultiUploadFeaturesRequest,
     MultiUploadFeaturesResponse,
     OfflineQueryContext,
@@ -126,6 +126,7 @@ from chalk.client.models import (
     ResolverReplayResponse,
     ResolverRunResponse,
     ResourceRequests,
+    ScheduledQueryRun,
     SetDatasetRevisionMetadataRequest,
     SetDatasetRevisionMetadataResponse,
     SetIncrementalProgressRequest,
@@ -2473,6 +2474,48 @@ https://docs.chalk.ai/cli/apply
 
         return resp
 
+    def get_scheduled_query_run_history(
+        self,
+        name: str,
+        limit: int = 10,
+    ) -> List[ScheduledQueryRun]:
+        """
+        Get the run history for a scheduled query.
+
+        Parameters
+        ----------
+        name
+            The name of the scheduled query.
+        limit
+            The maximum number of runs to return. Defaults to 10.
+
+        Returns
+        -------
+        list[ScheduledQueryRun]
+            A response message containing the list of scheduled query runs.
+
+        Examples
+        --------
+        >>> from chalk.client import ChalkClient
+        >>> ChalkClient().get_scheduled_query_run_history(
+        ...     name="my_scheduled_query",
+        ...     limit=20,
+        ... )
+        """
+        from chalk.client.client_grpc import ChalkGRPCClient
+
+        client_grpc = ChalkGRPCClient(
+            client_id=self._client_id,
+            client_secret=self._client_secret,
+            environment=self._primary_environment,
+            api_server=self._api_server,
+        )
+
+        return client_grpc.get_scheduled_query_run_history(
+            name=name,
+            limit=limit,
+        )
+
     def prompt_evaluation(
         self,
         prompts: list[Prompt | str],

chalk/client/models.py

@@ -1793,3 +1793,90 @@ class GetRegisteredModelVersionResponse(BaseModel):
 
 class CreateModelTrainingJobResponse(BaseModel):
     success: bool
+
+
+class ScheduledQueryRunStatus(str, Enum):
+    """Status of a scheduled query run."""
+
+    UNSPECIFIED = "UNSPECIFIED"
+    INITIALIZING = "INITIALIZING"
+    INIT_FAILED = "INIT_FAILED"
+    SKIPPED = "SKIPPED"
+    QUEUED = "QUEUED"
+    WORKING = "WORKING"
+    COMPLETED = "COMPLETED"
+    FAILED = "FAILED"
+    CANCELED = "CANCELED"
+
+
+@dataclasses.dataclass
+class ScheduledQueryRun:
+    """A single scheduled query run."""
+
+    id: int
+    environment_id: str
+    deployment_id: str
+    run_id: str
+    cron_query_id: int
+    cron_query_schedule_id: int
+    cron_name: str
+    gcr_execution_id: str
+    gcr_job_name: str
+    offline_query_id: str
+    created_at: datetime
+    updated_at: datetime
+    status: ScheduledQueryRunStatus
+    blocker_operation_id: str
+
+    @staticmethod
+    def from_proto(proto_run: Any) -> "ScheduledQueryRun":
+        """Convert a proto ScheduledQueryRun to the dataclass version."""
+        from datetime import timezone
+
+        # Map proto status enum to our enum
+        status_map = {
+            0: ScheduledQueryRunStatus.UNSPECIFIED,
+            1: ScheduledQueryRunStatus.INITIALIZING,
+            2: ScheduledQueryRunStatus.INIT_FAILED,
+            3: ScheduledQueryRunStatus.SKIPPED,
+            4: ScheduledQueryRunStatus.QUEUED,
+            5: ScheduledQueryRunStatus.WORKING,
+            6: ScheduledQueryRunStatus.COMPLETED,
+            7: ScheduledQueryRunStatus.FAILED,
+            8: ScheduledQueryRunStatus.CANCELED,
+        }
+
+        # Helper to convert proto Timestamp to datetime
+        def _timestamp_to_datetime(ts: Any) -> datetime:
+            return datetime.fromtimestamp(ts.seconds + ts.nanos / 1e9, tz=timezone.utc)
+
+        return ScheduledQueryRun(
+            id=proto_run.id,
+            environment_id=proto_run.environment_id,
+            deployment_id=proto_run.deployment_id,
+            run_id=proto_run.run_id,
+            cron_query_id=proto_run.cron_query_id,
+            cron_query_schedule_id=proto_run.cron_query_schedule_id,
+            cron_name=proto_run.cron_name,
+            gcr_execution_id=proto_run.gcr_execution_id,
+            gcr_job_name=proto_run.gcr_job_name,
+            offline_query_id=proto_run.offline_query_id,
+            created_at=_timestamp_to_datetime(proto_run.created_at),
+            updated_at=_timestamp_to_datetime(proto_run.updated_at),
+            status=status_map.get(proto_run.status, ScheduledQueryRunStatus.UNSPECIFIED),
+            blocker_operation_id=proto_run.blocker_operation_id,
+        )
+
+
+@dataclasses.dataclass
+class ManualTriggerScheduledQueryResponse:
+    """Response from manually triggering a scheduled query."""
+
+    scheduled_query_run: ScheduledQueryRun
+
+    @staticmethod
+    def from_proto(proto_response: Any) -> "ManualTriggerScheduledQueryResponse":
+        """Convert a proto ManualTriggerScheduledQueryResponse to the dataclass version."""
+        return ManualTriggerScheduledQueryResponse(
+            scheduled_query_run=ScheduledQueryRun.from_proto(proto_response.scheduled_query_run),
+        )

chalk/df/LazyFramePlaceholder.py

@@ -16,6 +16,7 @@ from typing import TYPE_CHECKING, Any, Optional, TypeAlias
 import pyarrow
 
 import chalk._gen.chalk.dataframe.v1.dataframe_pb2 as dataframe_pb2
+import chalk._gen.chalk.expression.v1.expression_pb2 as expression_pb2
 from chalk.features._encoding.converter import PrimitiveFeatureConverter
 from chalk.features.underscore import (
     Underscore,
@@ -1049,6 +1050,19 @@ def _convert_to_dataframe_proto(
                 arrow_table=PrimitiveFeatureConverter.convert_arrow_table_to_proto(value),
             )
 
+        # If libchalk.chalktable is available in the current environment, then we might encounter
+        # a libchalk.chalktable.Expr value which needs to be proto-serialized.
+        LibchalkExpr = None
+        try:
+            from libchalk.chalktable import Expr as LibchalkExpr  # pyright: ignore
+        except ImportError:
+            pass
+        if LibchalkExpr and isinstance(value, LibchalkExpr):
+            value_expr_encoded = value.to_proto_bytes()
+            return dataframe_pb2.DataFrameOperand(
+                libchalk_expr=expression_pb2.LogicalExprNode.FromString(value_expr_encoded),
+            )
+
         raise ValueError(f"LazyFramePlaceholder function operand is of unsupported type {type(value)}")
 
     _convert_arg(lazy_frame)
@@ -1115,6 +1129,15 @@ def _convert_from_dataframe_proto(
             return PrimitiveFeatureConverter.convert_arrow_table_from_proto(value.arrow_table)
         if value.HasField("underscore_expr"):
             return Underscore._from_proto(value.underscore_expr)  # pyright: ignore[reportPrivateUsage]
+        if value.HasField("libchalk_expr"):
+            # In order to decode `libchalk_expr` vlaues, `libchalk` must be available as a module.
+            try:
+                from libchalk.chalktable import Expr as LibchalkExpr  # pyright: ignore
+            except ImportError:
+                raise ValueError(
+                    "A dataframe parameter was encoded holding a libchalk.chalktable.Expr value, but the `libchalk` module is not available in the current environment. To decode this dataframe expression, import libchalk."
+                )
+            return LibchalkExpr.from_proto_bytes(value.libchalk_expr.SerializeToString())
 
         raise ValueError(f"DataFrame operand expression {value} does not have any value set")
 

{chalkpy-2.93.7.dist-info → chalkpy-2.93.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.93.7
+Version: 2.93.9
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

{chalkpy-2.93.7.dist-info → chalkpy-2.93.9.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=9YxwkAt3Im0OCMfpmnIB_4PKjZfBCcRmwATLXdHNRm4,2609
-chalk/_version.py,sha256=dhRKIv2zUq6x8xflLPE2dsInFRTqWqvB0s27C4StTFo,23
+chalk/_version.py,sha256=-Jg09nMIyHtBExF977CnRQ9XDbLyN5M_V0P6uwzeUEw,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=hCEo7eqSfXZWklkFB2geeipGhiD0qNjPBpQJvOBW6N0,63083
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -154,8 +154,8 @@ chalk/_gen/chalk/common/v2/table_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesin
 chalk/_gen/chalk/common/v2/table_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
 chalk/_gen/chalk/dataframe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/_gen/chalk/dataframe/v1/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py,sha256=d8F_kC3GuWfcDVE16hvgOP305WNnSuISW14zsf090Z8,4626
-chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi,sha256=C8dGdriuweYuA7A4hx4qzWMfuTXyzo3prM6PBs64xR0,4605
+chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py,sha256=_CfvLT4FIKPK4RGhT5TOQsEm3be7cLBm_z1YkeBpG7E,4729
+chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi,sha256=_0wCIVdyEVtEbw9jBFmSDioJCHfv6pPugQAj3S5vhFY,4819
 chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
 chalk/_gen/chalk/engine/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -412,6 +412,10 @@ chalk/_gen/chalk/server/v1/incident_pb2.py,sha256=4-uejdKheTLIk90OqClCAxb6esH8hM
 chalk/_gen/chalk/server/v1/incident_pb2.pyi,sha256=eP3FEl3eHkIRXfXLr_2cn-_GlHHirKQZv87DBg2jLaU,6861
 chalk/_gen/chalk/server/v1/incident_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/server/v1/incident_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
+chalk/_gen/chalk/server/v1/indexing_job_pb2.py,sha256=n4Ton8_7Ut_XI9jFHqPWw-9knZnC3ft_hBQ8fh8M7Hc,3338
+chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi,sha256=Pd9_NlR8zWPydV3w774swd3EB1OSKrq0fs7k8V2jNfs,1776
+chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py,sha256=JJrXQ7mZHAzCv2EuJTuZCz0cybxqZWzGrrTXRRUrK6s,2989
+chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi,sha256=mJ-rMUy8LlS33MB83EgWMRe89HjeVVrCZ-TC9wm4oH8,1122
 chalk/_gen/chalk/server/v1/integrations_pb2.py,sha256=BSgvuWniq6oyJhoBz1Ptcov3Q9G1u8NdpQp_ogRWbDE,14216
 chalk/_gen/chalk/server/v1/integrations_pb2.pyi,sha256=_EJVXKyY75KI9xu32ZmHkew2APzNf8IWHrc2cAayfSM,11977
 chalk/_gen/chalk/server/v1/integrations_pb2_grpc.py,sha256=bWarXrUn1O7t6RhGCCKo0X0IIXaA9SKln7AQDVEY8vM,16185
@@ -480,8 +484,8 @@ chalk/_gen/chalk/server/v1/scheduled_query_pb2.py,sha256=aU7YjTJ2H9dp_ZRNmh_91NC
 chalk/_gen/chalk/server/v1/scheduled_query_pb2.pyi,sha256=4o__cQ3plq1LucDRH03pMf9LBSjk1eE7XY51gyv435c,299
 chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.py,sha256=Pb0WD51FjkvUmPcYINjkSgYX7Z2HCmakFb5wp_Ulqxw,5037
 chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.pyi,sha256=w98vRRLSiFl6Z_8c06DDAVH2593_dAgbhdlI-vY0tJw,1337
-chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=ykS7SrO0bu74O55qu6EFrnC118gV-iekJVHnNLck8I0,5330
-chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=M9JNVrg74MI-ay5bOzJU-BdC8h-jUkrAmlssiCFTln4,6113
+chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=ga7XCuptDqkJFNmMkwq7UBnGoLlhgqQKWbgqEbUhn20,5539
+chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=P7BSBTV5xIHPcz3sPVtHWBQxZUzQUAzxwvSE9dSxC1I,6426
 chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
 chalk/_gen/chalk/server/v1/scheduler_pb2.py,sha256=51WGM0OLvEzZBvgy7gubBGkUH3NSDwEHRZ1P1enkz0c,8061
@@ -580,14 +584,14 @@ chalk/_validation/validation.py,sha256=9cCMfZa9-1wxkXLme_ylmD5vIA1qExJD6aqbYvbmK
 chalk/byte_transmit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/byte_transmit/model.py,sha256=LFX8pj9X_CWXeap7fDnMl9YmXsYTgq7jBAbEWkxoYoE,13048
 chalk/client/__init__.py,sha256=wu3WQVzE5gRj6noQQDOdYJUgDaz_9QtbjXH4KuoIlXQ,1782
-chalk/client/client.py,sha256=aIOIB7OTLZ6jWfesAVRqeT6mFqtsjeAk0e8UE29bzRI,102940
+chalk/client/client.py,sha256=gdGHrC73xyC4boE8z54BDD3-7FMJ4WCv6o2I_ii8obI,103680
 chalk/client/client_async.py,sha256=nFFTWJbdBlb7zksyjOMBY566tZTAyNXQhCnq06LHWl0,50803
 chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
-chalk/client/client_grpc.py,sha256=86W32GXemHb0xt78sUKhgorY2Me3ud53YNzEr09NcDg,93683
-chalk/client/client_impl.py,sha256=smNVo-vItnjbuN-8beFNb986uOVDaPTCkd6JTfT76WY,210503
+chalk/client/client_grpc.py,sha256=7DzSUdYf7P2Y2lAo6AdLALEwwQ5YUNSeJU2rfk3KOec,95763
+chalk/client/client_impl.py,sha256=R2ZUiEQf662q3O1eo_NjzYqLGB71B5mWzcx8opC3Os0,211639
 chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
 chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
-chalk/client/models.py,sha256=CqJ8vdMAPrjOo--GWQVr2WF3zfP5bfhULho0_YL0p6o,60883
+chalk/client/models.py,sha256=FDgEqIeBJq25T9AH38jdWkeyRLiAXfLXfXHpomFJM08,63923
 chalk/client/response.py,sha256=m8sQCOj7YVv3mZSZMIC1rIMzFMQ9rfMdBRLg5NRmOOE,53257
 chalk/client/_internal_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/client/_internal_models/check.py,sha256=3Xfo4Ws4rvwjeVg0-5-kejfRfRBJeqHmnRhW-WEz784,917
@@ -604,7 +608,7 @@ chalk/config/_validator.py,sha256=QC1y52m704_bV_TYjq0sdZJ-km8iSkDX1V4sHgw4RJk,13
 chalk/config/auth_config.py,sha256=HAALkQrvDD0i7gaZK5iufS8vDDVbzLIpHLOpcJO1kmw,4498
 chalk/config/project_config.py,sha256=YHB3upvtBJu-tWfNOchBRSc9xGebDbrIpCVmKbBzJ8Q,7217
 chalk/df/ChalkDataFrameImpl.py,sha256=BRwnjQcie3gxaKhKau8HG17NWzS1zdr8SnNVurxF8QY,133
-chalk/df/LazyFramePlaceholder.py,sha256=4veohGgt6WvKdbE4Qn3qVobfsJsB1yDuWnIhCfq4How,37728
+chalk/df/LazyFramePlaceholder.py,sha256=-qPtf-xqhIhrF6Mbx-WcXxPqGchWgmf0FbSQ5VpJCkg,39067
 chalk/df/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/df/ast_parser.py,sha256=t-DwUxd2hN8LECRSBx85DIv9FtILgMiHyGyCTINfgQw,11199
 chalk/features/__init__.py,sha256=5doD7bFwRthzwdmizbRaPVHiCABiNpiOiAJVFlwqNnA,6943
@@ -795,8 +799,8 @@ chalk/utils/tracing.py,sha256=Glx8YrtjWy0zE5YbpgfgcsLDshAKnnYm9poiWNeCxXs,11075
 chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
 chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
-chalkpy-2.93.7.dist-info/METADATA,sha256=Uh4WHtrop4ydmzZR9uN5Cr4uQMad-JTFwSQ1IOy9BJs,27494
-chalkpy-2.93.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chalkpy-2.93.7.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.93.7.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.93.7.dist-info/RECORD,,
+chalkpy-2.93.9.dist-info/METADATA,sha256=Bldsmcc9CWN5Zho29uYPwc2AUYUFMbZGqXXM4cVJzSk,27494
+chalkpy-2.93.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chalkpy-2.93.9.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.93.9.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.93.9.dist-info/RECORD,,