chalkpy 2.93.8__tar.gz → 2.94.0__tar.gz

{chalkpy-2.93.8 → chalkpy-2.94.0}/PKG-INFO

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

chalkpy-2.94.0/chalk/_gen/chalk/protosql/v1/sql_service_pb2.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/protosql/v1/sql_service.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.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v1_dot_chalk__error__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n#chalk/protosql/v1/sql_service.proto\x12\x11\x63halk.protosql.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"#\n!ExecuteSqlSyncQueryRequestOptions"$\n"ExecuteSqlAsyncQueryRequestOptions"\xb1\x02\n\x16\x45xecuteSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query\x12*\n\x0e\x63orrelation_id\x18\x02 \x01(\tH\x01R\rcorrelationId\x88\x01\x01\x12Y\n\x0csync_options\x18\n \x01(\x0b\x32\x34.chalk.protosql.v1.ExecuteSqlSyncQueryRequestOptionsH\x00R\x0bsyncOptions\x12\\\n\rasync_options\x18\x0b \x01(\x0b\x32\x35.chalk.protosql.v1.ExecuteSqlAsyncQueryRequestOptionsH\x00R\x0c\x61syncOptionsB\t\n\x07optionsB\x11\n\x0f_correlation_id"O\n"ExecuteSqlSyncQueryResponsePayload\x12)\n\x10parquet_response\x18\x01 \x01(\x0cR\x0fparquetResponse"\xa0\x01\n#ExecuteSqlAsyncQueryResponsePayload\x12!\n\x0coperation_id\x18\x01 \x01(\tR\x0boperationId\x12V\n\x0e\x65xecution_mode\x18\x02 \x01(\x0e\x32/.chalk.protosql.v1.ExecuteSqlAsyncExecutionModeR\rexecutionMode"\xcb\x02\n\x17\x45xecuteSqlQueryResponse\x12\x19\n\x08query_id\x18\x01 \x01(\tR\x07queryId\x12\x1a\n\x07parquet\x18\x02 \x01(\x0cH\x00R\x07parquet\x12Z\n\x0csync_payload\x18\n \x01(\x0b\x32\x35.chalk.protosql.v1.ExecuteSqlSyncQueryResponsePayloadH\x00R\x0bsyncPayload\x12]\n\rasync_payload\x18\x0b \x01(\x0b\x32\x36.chalk.protosql.v1.ExecuteSqlAsyncQueryResponsePayloadH\x00R\x0c\x61syncPayload\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\t\n\x07payload"+\n\x13PlanSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query"n\n\x14PlanSqlQueryResponse\x12!\n\x0clogical_plan\x18\x01 \x01(\tR\x0blogicalPlan\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x16\n\x14GetDbCatalogsRequest"q\n\x15GetDbCatalogsResponse\x12#\n\rcatalog_names\x18\x01 \x03(\tR\x0c\x63\x61talogNames\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\xd0\x01\n\x13GetDbSchemasRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_pattern"W\n\x0c\x44\x62SchemaInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName"\x86\x01\n\x14GetDbSchemasResponse\x12\x39\n\x07schemas\x18\x01 \x03(\x0b\x32\x1f.chalk.protosql.v1.DbSchemaInfoR\x07schemas\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x9f\x02\n\x10GetTablesRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12>\n\x19table_name_filter_pattern\x18\x03 \x01(\tH\x02R\x16tableNameFilterPattern\x88\x01\x01\x12\'\n\x0finclude_schemas\x18\n \x01(\x08R\x0eincludeSchemasB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_patternB\x1c\n\x1a_table_name_filter_pattern"\xbd\x01\n\tTableInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName\x12\x1d\n\ntable_name\x18\x03 \x01(\tR\ttableName\x12\x31\n\x12table_arrow_schema\x18\x04 \x01(\x0cH\x00R\x10tableArrowSchema\x88\x01\x01\x42\x15\n\x13_table_arrow_schema"~\n\x11GetTablesResponse\x12\x34\n\x06tables\x18\x01 \x03(\x0b\x32\x1c.chalk.protosql.v1.TableInfoR\x06tables\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors*\xad\x01\n\x1c\x45xecuteSqlAsyncExecutionMode\x12\x30\n,EXECUTE_SQL_ASYNC_EXECUTION_MODE_UNSPECIFIED\x10\x00\x12/\n+EXECUTE_SQL_ASYNC_EXECUTION_MODE_IN_PROCESS\x10\x01\x12*\n&EXECUTE_SQL_ASYNC_EXECUTION_MODE_ASYNC\x10\x02\x32\x8d\x04\n\nSqlService\x12m\n\x0f\x45xecuteSqlQuery\x12).chalk.protosql.v1.ExecuteSqlQueryRequest\x1a*.chalk.protosql.v1.ExecuteSqlQueryResponse"\x03\x80}\x03\x12\x64\n\x0cPlanSqlQuery\x12&.chalk.protosql.v1.PlanSqlQueryRequest\x1a\'.chalk.protosql.v1.PlanSqlQueryResponse"\x03\x80}\x03\x12g\n\rGetDbCatalogs\x12\'.chalk.protosql.v1.GetDbCatalogsRequest\x1a(.chalk.protosql.v1.GetDbCatalogsResponse"\x03\x80}\x0b\x12\x64\n\x0cGetDbSchemas\x12&.chalk.protosql.v1.GetDbSchemasRequest\x1a\'.chalk.protosql.v1.GetDbSchemasResponse"\x03\x80}\x0b\x12[\n\tGetTables\x12#.chalk.protosql.v1.GetTablesRequest\x1a$.chalk.protosql.v1.GetTablesResponse"\x03\x80}\x0b\x42\x8e\x01\n\x15\x63om.chalk.protosql.v1B\x0fSqlServiceProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x11\x43halk.Protosql.V1\xca\x02\x11\x43halk\\Protosql\\V1\xe2\x02\x1d\x43halk\\Protosql\\V1\\GPBMetadata\xea\x02\x13\x43halk::Protosql::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.protosql.v1.sql_service_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\025com.chalk.protosql.v1B\017SqlServiceProtoP\001\242\002\003CPX\252\002\021Chalk.Protosql.V1\312\002\021Chalk\\Protosql\\V1\342\002\035Chalk\\Protosql\\V1\\GPBMetadata\352\002\023Chalk::Protosql::V1"
+    _globals["_SQLSERVICE"].methods_by_name["ExecuteSqlQuery"]._options = None
+    _globals["_SQLSERVICE"].methods_by_name["ExecuteSqlQuery"]._serialized_options = b"\200}\003"
+    _globals["_SQLSERVICE"].methods_by_name["PlanSqlQuery"]._options = None
+    _globals["_SQLSERVICE"].methods_by_name["PlanSqlQuery"]._serialized_options = b"\200}\003"
+    _globals["_SQLSERVICE"].methods_by_name["GetDbCatalogs"]._options = None
+    _globals["_SQLSERVICE"].methods_by_name["GetDbCatalogs"]._serialized_options = b"\200}\013"
+    _globals["_SQLSERVICE"].methods_by_name["GetDbSchemas"]._options = None
+    _globals["_SQLSERVICE"].methods_by_name["GetDbSchemas"]._serialized_options = b"\200}\013"
+    _globals["_SQLSERVICE"].methods_by_name["GetTables"]._options = None
+    _globals["_SQLSERVICE"].methods_by_name["GetTables"]._serialized_options = b"\200}\013"
+    _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_start = 2431
+    _globals["_EXECUTESQLASYNCEXECUTIONMODE"]._serialized_end = 2604
+    _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_start = 126
+    _globals["_EXECUTESQLSYNCQUERYREQUESTOPTIONS"]._serialized_end = 161
+    _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_start = 163
+    _globals["_EXECUTESQLASYNCQUERYREQUESTOPTIONS"]._serialized_end = 199
+    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 202
+    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 507
+    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 509
+    _globals["_EXECUTESQLSYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 588
+    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_start = 591
+    _globals["_EXECUTESQLASYNCQUERYRESPONSEPAYLOAD"]._serialized_end = 751
+    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 754
+    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 1085
+    _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 1087
+    _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 1130
+    _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 1132
+    _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 1242
+    _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 1244
+    _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 1266
+    _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 1268
+    _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 1381
+    _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 1384
+    _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 1592
+    _globals["_DBSCHEMAINFO"]._serialized_start = 1594
+    _globals["_DBSCHEMAINFO"]._serialized_end = 1681
+    _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 1684
+    _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 1818
+    _globals["_GETTABLESREQUEST"]._serialized_start = 1821
+    _globals["_GETTABLESREQUEST"]._serialized_end = 2108
+    _globals["_TABLEINFO"]._serialized_start = 2111
+    _globals["_TABLEINFO"]._serialized_end = 2300
+    _globals["_GETTABLESRESPONSE"]._serialized_start = 2302
+    _globals["_GETTABLESRESPONSE"]._serialized_end = 2428
+    _globals["_SQLSERVICE"]._serialized_start = 2607
+    _globals["_SQLSERVICE"]._serialized_end = 3132
+# @@protoc_insertion_point(module_scope)

{chalkpy-2.93.8 → chalkpy-2.94.0}/chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi

@@ -1,6 +1,7 @@
 from chalk._gen.chalk.auth.v1 import permissions_pb2 as _permissions_pb2
 from chalk._gen.chalk.common.v1 import chalk_error_pb2 as _chalk_error_pb2
 from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import message as _message
 from typing import (
@@ -13,24 +14,78 @@ from typing import (
 
 DESCRIPTOR: _descriptor.FileDescriptor
 
+class ExecuteSqlAsyncExecutionMode(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    EXECUTE_SQL_ASYNC_EXECUTION_MODE_UNSPECIFIED: _ClassVar[ExecuteSqlAsyncExecutionMode]
+    EXECUTE_SQL_ASYNC_EXECUTION_MODE_IN_PROCESS: _ClassVar[ExecuteSqlAsyncExecutionMode]
+    EXECUTE_SQL_ASYNC_EXECUTION_MODE_ASYNC: _ClassVar[ExecuteSqlAsyncExecutionMode]
+
+EXECUTE_SQL_ASYNC_EXECUTION_MODE_UNSPECIFIED: ExecuteSqlAsyncExecutionMode
+EXECUTE_SQL_ASYNC_EXECUTION_MODE_IN_PROCESS: ExecuteSqlAsyncExecutionMode
+EXECUTE_SQL_ASYNC_EXECUTION_MODE_ASYNC: ExecuteSqlAsyncExecutionMode
+
+class ExecuteSqlSyncQueryRequestOptions(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class ExecuteSqlAsyncQueryRequestOptions(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
 class ExecuteSqlQueryRequest(_message.Message):
-    __slots__ = ("query",)
+    __slots__ = ("query", "correlation_id", "sync_options", "async_options")
     QUERY_FIELD_NUMBER: _ClassVar[int]
+    CORRELATION_ID_FIELD_NUMBER: _ClassVar[int]
+    SYNC_OPTIONS_FIELD_NUMBER: _ClassVar[int]
+    ASYNC_OPTIONS_FIELD_NUMBER: _ClassVar[int]
     query: str
-    def __init__(self, query: _Optional[str] = ...) -> None: ...
+    correlation_id: str
+    sync_options: ExecuteSqlSyncQueryRequestOptions
+    async_options: ExecuteSqlAsyncQueryRequestOptions
+    def __init__(
+        self,
+        query: _Optional[str] = ...,
+        correlation_id: _Optional[str] = ...,
+        sync_options: _Optional[_Union[ExecuteSqlSyncQueryRequestOptions, _Mapping]] = ...,
+        async_options: _Optional[_Union[ExecuteSqlAsyncQueryRequestOptions, _Mapping]] = ...,
+    ) -> None: ...
+
+class ExecuteSqlSyncQueryResponsePayload(_message.Message):
+    __slots__ = ("parquet_response",)
+    PARQUET_RESPONSE_FIELD_NUMBER: _ClassVar[int]
+    parquet_response: bytes
+    def __init__(self, parquet_response: _Optional[bytes] = ...) -> None: ...
+
+class ExecuteSqlAsyncQueryResponsePayload(_message.Message):
+    __slots__ = ("operation_id", "execution_mode")
+    OPERATION_ID_FIELD_NUMBER: _ClassVar[int]
+    EXECUTION_MODE_FIELD_NUMBER: _ClassVar[int]
+    operation_id: str
+    execution_mode: ExecuteSqlAsyncExecutionMode
+    def __init__(
+        self,
+        operation_id: _Optional[str] = ...,
+        execution_mode: _Optional[_Union[ExecuteSqlAsyncExecutionMode, str]] = ...,
+    ) -> None: ...
 
 class ExecuteSqlQueryResponse(_message.Message):
-    __slots__ = ("query_id", "parquet", "errors")
+    __slots__ = ("query_id", "parquet", "sync_payload", "async_payload", "errors")
     QUERY_ID_FIELD_NUMBER: _ClassVar[int]
     PARQUET_FIELD_NUMBER: _ClassVar[int]
+    SYNC_PAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    ASYNC_PAYLOAD_FIELD_NUMBER: _ClassVar[int]
     ERRORS_FIELD_NUMBER: _ClassVar[int]
     query_id: str
     parquet: bytes
+    sync_payload: ExecuteSqlSyncQueryResponsePayload
+    async_payload: ExecuteSqlAsyncQueryResponsePayload
     errors: _containers.RepeatedCompositeFieldContainer[_chalk_error_pb2.ChalkError]
     def __init__(
         self,
         query_id: _Optional[str] = ...,
         parquet: _Optional[bytes] = ...,
+        sync_payload: _Optional[_Union[ExecuteSqlSyncQueryResponsePayload, _Mapping]] = ...,
+        async_payload: _Optional[_Union[ExecuteSqlAsyncQueryResponsePayload, _Mapping]] = ...,
         errors: _Optional[_Iterable[_Union[_chalk_error_pb2.ChalkError, _Mapping]]] = ...,
     ) -> None: ...
 

chalkpy-2.94.0/chalk/_version.py

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

{chalkpy-2.93.8 → chalkpy-2.94.0}/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.8 → chalkpy-2.94.0}/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.8 → chalkpy-2.94.0}/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],

{chalkpy-2.93.8 → chalkpy-2.94.0}/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),
+        )

{chalkpy-2.93.8 → chalkpy-2.94.0}/chalk/streams/_windows.py

@@ -80,9 +80,14 @@ JsonValue = Any
 
 def get_name_with_duration(name_or_fqn: str, duration: Union[str, int, timedelta]) -> str:
     duration_secs = parse_chalk_duration_s(duration)
+    name_or_fqn_components = name_or_fqn.split("@")
+    assert len(name_or_fqn_components) <= 2, f"Received invalid fqn format.\nfqn={name_or_fqn}"
+    unversioned_fqn = name_or_fqn_components[0]
+    version = None if len(name_or_fqn_components) != 2 else name_or_fqn_components[1]
+
     if duration_secs >= CHALK_MAX_TIMEDELTA.total_seconds():
-        return f"{name_or_fqn}__all__"
-    return f"{name_or_fqn}__{duration_secs}__"
+        return f"{unversioned_fqn}__all__" + ("" if version is None else f"@{version}")
+    return f"{unversioned_fqn}__{duration_secs}__" + ("" if version is None else f"@{version}")
 
 
 if TYPE_CHECKING:

chalkpy-2.93.8/chalk/_gen/chalk/protosql/v1/sql_service_pb2.py

@@ -1,68 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: chalk/protosql/v1/sql_service.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.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v1_dot_chalk__error__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n#chalk/protosql/v1/sql_service.proto\x12\x11\x63halk.protosql.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto".\n\x16\x45xecuteSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query"\x90\x01\n\x17\x45xecuteSqlQueryResponse\x12\x19\n\x08query_id\x18\x01 \x01(\tR\x07queryId\x12\x1a\n\x07parquet\x18\x02 \x01(\x0cH\x00R\x07parquet\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\t\n\x07payload"+\n\x13PlanSqlQueryRequest\x12\x14\n\x05query\x18\x01 \x01(\tR\x05query"n\n\x14PlanSqlQueryResponse\x12!\n\x0clogical_plan\x18\x01 \x01(\tR\x0blogicalPlan\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x16\n\x14GetDbCatalogsRequest"q\n\x15GetDbCatalogsResponse\x12#\n\rcatalog_names\x18\x01 \x03(\tR\x0c\x63\x61talogNames\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\xd0\x01\n\x13GetDbSchemasRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrorsB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_pattern"W\n\x0c\x44\x62SchemaInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName"\x86\x01\n\x14GetDbSchemasResponse\x12\x39\n\x07schemas\x18\x01 \x03(\x0b\x32\x1f.chalk.protosql.v1.DbSchemaInfoR\x07schemas\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\x9f\x02\n\x10GetTablesRequest\x12\x1d\n\x07\x63\x61talog\x18\x01 \x01(\tH\x00R\x07\x63\x61talog\x88\x01\x01\x12<\n\x18\x64\x62_schema_filter_pattern\x18\x02 \x01(\tH\x01R\x15\x64\x62SchemaFilterPattern\x88\x01\x01\x12>\n\x19table_name_filter_pattern\x18\x03 \x01(\tH\x02R\x16tableNameFilterPattern\x88\x01\x01\x12\'\n\x0finclude_schemas\x18\n \x01(\x08R\x0eincludeSchemasB\n\n\x08_catalogB\x1b\n\x19_db_schema_filter_patternB\x1c\n\x1a_table_name_filter_pattern"\xbd\x01\n\tTableInfo\x12!\n\x0c\x63\x61talog_name\x18\x01 \x01(\tR\x0b\x63\x61talogName\x12$\n\x0e\x64\x62_schema_name\x18\x02 \x01(\tR\x0c\x64\x62SchemaName\x12\x1d\n\ntable_name\x18\x03 \x01(\tR\ttableName\x12\x31\n\x12table_arrow_schema\x18\x04 \x01(\x0cH\x00R\x10tableArrowSchema\x88\x01\x01\x42\x15\n\x13_table_arrow_schema"~\n\x11GetTablesResponse\x12\x34\n\x06tables\x18\x01 \x03(\x0b\x32\x1c.chalk.protosql.v1.TableInfoR\x06tables\x12\x33\n\x06\x65rrors\x18\x05 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors2\x8d\x04\n\nSqlService\x12m\n\x0f\x45xecuteSqlQuery\x12).chalk.protosql.v1.ExecuteSqlQueryRequest\x1a*.chalk.protosql.v1.ExecuteSqlQueryResponse"\x03\x80}\x03\x12\x64\n\x0cPlanSqlQuery\x12&.chalk.protosql.v1.PlanSqlQueryRequest\x1a\'.chalk.protosql.v1.PlanSqlQueryResponse"\x03\x80}\x03\x12g\n\rGetDbCatalogs\x12\'.chalk.protosql.v1.GetDbCatalogsRequest\x1a(.chalk.protosql.v1.GetDbCatalogsResponse"\x03\x80}\x0b\x12\x64\n\x0cGetDbSchemas\x12&.chalk.protosql.v1.GetDbSchemasRequest\x1a\'.chalk.protosql.v1.GetDbSchemasResponse"\x03\x80}\x0b\x12[\n\tGetTables\x12#.chalk.protosql.v1.GetTablesRequest\x1a$.chalk.protosql.v1.GetTablesResponse"\x03\x80}\x0b\x42\x8e\x01\n\x15\x63om.chalk.protosql.v1B\x0fSqlServiceProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x11\x43halk.Protosql.V1\xca\x02\x11\x43halk\\Protosql\\V1\xe2\x02\x1d\x43halk\\Protosql\\V1\\GPBMetadata\xea\x02\x13\x43halk::Protosql::V1b\x06proto3'
-)
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.protosql.v1.sql_service_pb2", _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-    _globals["DESCRIPTOR"]._options = None
-    _globals[
-        "DESCRIPTOR"
-    ]._serialized_options = b"\n\025com.chalk.protosql.v1B\017SqlServiceProtoP\001\242\002\003CPX\252\002\021Chalk.Protosql.V1\312\002\021Chalk\\Protosql\\V1\342\002\035Chalk\\Protosql\\V1\\GPBMetadata\352\002\023Chalk::Protosql::V1"
-    _globals["_SQLSERVICE"].methods_by_name["ExecuteSqlQuery"]._options = None
-    _globals["_SQLSERVICE"].methods_by_name["ExecuteSqlQuery"]._serialized_options = b"\200}\003"
-    _globals["_SQLSERVICE"].methods_by_name["PlanSqlQuery"]._options = None
-    _globals["_SQLSERVICE"].methods_by_name["PlanSqlQuery"]._serialized_options = b"\200}\003"
-    _globals["_SQLSERVICE"].methods_by_name["GetDbCatalogs"]._options = None
-    _globals["_SQLSERVICE"].methods_by_name["GetDbCatalogs"]._serialized_options = b"\200}\013"
-    _globals["_SQLSERVICE"].methods_by_name["GetDbSchemas"]._options = None
-    _globals["_SQLSERVICE"].methods_by_name["GetDbSchemas"]._serialized_options = b"\200}\013"
-    _globals["_SQLSERVICE"].methods_by_name["GetTables"]._options = None
-    _globals["_SQLSERVICE"].methods_by_name["GetTables"]._serialized_options = b"\200}\013"
-    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_start = 126
-    _globals["_EXECUTESQLQUERYREQUEST"]._serialized_end = 172
-    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_start = 175
-    _globals["_EXECUTESQLQUERYRESPONSE"]._serialized_end = 319
-    _globals["_PLANSQLQUERYREQUEST"]._serialized_start = 321
-    _globals["_PLANSQLQUERYREQUEST"]._serialized_end = 364
-    _globals["_PLANSQLQUERYRESPONSE"]._serialized_start = 366
-    _globals["_PLANSQLQUERYRESPONSE"]._serialized_end = 476
-    _globals["_GETDBCATALOGSREQUEST"]._serialized_start = 478
-    _globals["_GETDBCATALOGSREQUEST"]._serialized_end = 500
-    _globals["_GETDBCATALOGSRESPONSE"]._serialized_start = 502
-    _globals["_GETDBCATALOGSRESPONSE"]._serialized_end = 615
-    _globals["_GETDBSCHEMASREQUEST"]._serialized_start = 618
-    _globals["_GETDBSCHEMASREQUEST"]._serialized_end = 826
-    _globals["_DBSCHEMAINFO"]._serialized_start = 828
-    _globals["_DBSCHEMAINFO"]._serialized_end = 915
-    _globals["_GETDBSCHEMASRESPONSE"]._serialized_start = 918
-    _globals["_GETDBSCHEMASRESPONSE"]._serialized_end = 1052
-    _globals["_GETTABLESREQUEST"]._serialized_start = 1055
-    _globals["_GETTABLESREQUEST"]._serialized_end = 1342
-    _globals["_TABLEINFO"]._serialized_start = 1345
-    _globals["_TABLEINFO"]._serialized_end = 1534
-    _globals["_GETTABLESRESPONSE"]._serialized_start = 1536
-    _globals["_GETTABLESRESPONSE"]._serialized_end = 1662
-    _globals["_SQLSERVICE"]._serialized_start = 1665
-    _globals["_SQLSERVICE"]._serialized_end = 2190
-# @@protoc_insertion_point(module_scope)

chalkpy-2.93.8/chalk/_version.py

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