chalkpy 2.93.7__tar.gz → 2.93.9__tar.gz

{chalkpy-2.93.7 → chalkpy-2.93.9}/PKG-INFO

@@ -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 → chalkpy-2.93.9}/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)

{chalkpy-2.93.7 → chalkpy-2.93.9}/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):

chalkpy-2.93.9/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)

chalkpy-2.93.9/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: ...

chalkpy-2.93.9/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,
+        )

chalkpy-2.93.9/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: ...

{chalkpy-2.93.7 → chalkpy-2.93.9}/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)

{chalkpy-2.93.7 → chalkpy-2.93.9}/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):

chalkpy-2.93.9/chalk/_version.py

@@ -0,0 +1 @@
+__version__ = "2.93.9"

{chalkpy-2.93.7 → chalkpy-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],

{chalkpy-2.93.7 → chalkpy-2.93.9}/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.

{chalkpy-2.93.7 → chalkpy-2.93.9}/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],