chalkpy 2.96.5__tar.gz → 2.96.7__tar.gz

{chalkpy-2.96.5 → chalkpy-2.96.7}/PKG-INFO

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

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py

@@ -18,7 +18,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n#chalk/artifacts/v1/cron_query.proto\x12\x12\x63halk.artifacts.v1\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"Y\n\x11RecomputeSettings\x12!\n\x0c\x66\x65\x61ture_fqns\x18\x01 \x03(\tR\x0b\x66\x65\x61tureFqns\x12!\n\x0c\x61ll_features\x18\x02 \x01(\x08R\x0b\x61llFeatures"\xe7\x06\n\tCronQuery\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04\x63ron\x18\x02 \x01(\tR\x04\x63ron\x12\x1b\n\tfile_name\x18\x03 \x01(\tR\x08\x66ileName\x12\x16\n\x06output\x18\x04 \x03(\tR\x06output\x12$\n\x0bmax_samples\x18\x05 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12\x43\n\trecompute\x18\x06 \x01(\x0b\x32%.chalk.artifacts.v1.RecomputeSettingsR\trecompute\x12;\n\x0blower_bound\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nlowerBound\x12;\n\x0bupper_bound\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nupperBound\x12\x12\n\x04tags\x18\t \x03(\tR\x04tags\x12\x34\n\x16required_resolver_tags\x18\n \x03(\tR\x14requiredResolverTags\x12!\n\x0cstore_online\x18\x0b \x01(\x08R\x0bstoreOnline\x12#\n\rstore_offline\x18\x0c \x01(\x08R\x0cstoreOffline\x12/\n\x13incremental_sources\x18\r \x03(\tR\x12incrementalSources\x12*\n\x0eresource_group\x18\x0e \x01(\tH\x01R\rresourceGroup\x88\x01\x01\x12Z\n\x0fplanner_options\x18\x0f \x03(\x0b\x32\x31.chalk.artifacts.v1.CronQuery.PlannerOptionsEntryR\x0eplannerOptions\x12O\n\x13\x63ompletion_deadline\x18\x10 \x01(\x0b\x32\x19.google.protobuf.DurationH\x02R\x12\x63ompletionDeadline\x88\x01\x01\x1a\x41\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0e\n\x0c_max_samplesB\x11\n\x0f_resource_groupB\x16\n\x14_completion_deadlineB\x92\x01\n\x16\x63om.chalk.artifacts.v1B\x0e\x43ronQueryProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x12\x43halk.Artifacts.V1\xca\x02\x12\x43halk\\Artifacts\\V1\xe2\x02\x1e\x43halk\\Artifacts\\V1\\GPBMetadata\xea\x02\x14\x43halk::Artifacts::V1b\x06proto3'
+    b'\n#chalk/artifacts/v1/cron_query.proto\x12\x12\x63halk.artifacts.v1\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"Y\n\x11RecomputeSettings\x12!\n\x0c\x66\x65\x61ture_fqns\x18\x01 \x03(\tR\x0b\x66\x65\x61tureFqns\x12!\n\x0c\x61ll_features\x18\x02 \x01(\x08R\x0b\x61llFeatures"\xd0\x07\n\tCronQuery\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04\x63ron\x18\x02 \x01(\tR\x04\x63ron\x12\x1b\n\tfile_name\x18\x03 \x01(\tR\x08\x66ileName\x12\x16\n\x06output\x18\x04 \x03(\tR\x06output\x12$\n\x0bmax_samples\x18\x05 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12\x43\n\trecompute\x18\x06 \x01(\x0b\x32%.chalk.artifacts.v1.RecomputeSettingsR\trecompute\x12;\n\x0blower_bound\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nlowerBound\x12;\n\x0bupper_bound\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nupperBound\x12\x12\n\x04tags\x18\t \x03(\tR\x04tags\x12\x34\n\x16required_resolver_tags\x18\n \x03(\tR\x14requiredResolverTags\x12!\n\x0cstore_online\x18\x0b \x01(\x08R\x0bstoreOnline\x12#\n\rstore_offline\x18\x0c \x01(\x08R\x0cstoreOffline\x12/\n\x13incremental_sources\x18\r \x03(\tR\x12incrementalSources\x12*\n\x0eresource_group\x18\x0e \x01(\tH\x01R\rresourceGroup\x88\x01\x01\x12Z\n\x0fplanner_options\x18\x0f \x03(\x0b\x32\x31.chalk.artifacts.v1.CronQuery.PlannerOptionsEntryR\x0eplannerOptions\x12O\n\x13\x63ompletion_deadline\x18\x10 \x01(\x0b\x32\x19.google.protobuf.DurationH\x02R\x12\x63ompletionDeadline\x88\x01\x01\x12"\n\nnum_shards\x18\x11 \x01(\x05H\x03R\tnumShards\x88\x01\x01\x12$\n\x0bnum_workers\x18\x12 \x01(\x05H\x04R\nnumWorkers\x88\x01\x01\x1a\x41\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0e\n\x0c_max_samplesB\x11\n\x0f_resource_groupB\x16\n\x14_completion_deadlineB\r\n\x0b_num_shardsB\x0e\n\x0c_num_workersB\x92\x01\n\x16\x63om.chalk.artifacts.v1B\x0e\x43ronQueryProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x12\x43halk.Artifacts.V1\xca\x02\x12\x43halk\\Artifacts\\V1\xe2\x02\x1e\x43halk\\Artifacts\\V1\\GPBMetadata\xea\x02\x14\x43halk::Artifacts::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -34,7 +34,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_RECOMPUTESETTINGS"]._serialized_start = 124
     _globals["_RECOMPUTESETTINGS"]._serialized_end = 213
     _globals["_CRONQUERY"]._serialized_start = 216
-    _globals["_CRONQUERY"]._serialized_end = 1087
-    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_start = 963
-    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_end = 1028
+    _globals["_CRONQUERY"]._serialized_end = 1192
+    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_start = 1037
+    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_end = 1102
 # @@protoc_insertion_point(module_scope)

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi

@@ -39,6 +39,8 @@ class CronQuery(_message.Message):
         "resource_group",
         "planner_options",
         "completion_deadline",
+        "num_shards",
+        "num_workers",
     )
     class PlannerOptionsEntry(_message.Message):
         __slots__ = ("key", "value")
@@ -64,6 +66,8 @@ class CronQuery(_message.Message):
     RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
     PLANNER_OPTIONS_FIELD_NUMBER: _ClassVar[int]
     COMPLETION_DEADLINE_FIELD_NUMBER: _ClassVar[int]
+    NUM_SHARDS_FIELD_NUMBER: _ClassVar[int]
+    NUM_WORKERS_FIELD_NUMBER: _ClassVar[int]
     name: str
     cron: str
     file_name: str
@@ -80,6 +84,8 @@ class CronQuery(_message.Message):
     resource_group: str
     planner_options: _containers.ScalarMap[str, str]
     completion_deadline: _duration_pb2.Duration
+    num_shards: int
+    num_workers: int
     def __init__(
         self,
         name: _Optional[str] = ...,
@@ -98,4 +104,6 @@ class CronQuery(_message.Message):
         resource_group: _Optional[str] = ...,
         planner_options: _Optional[_Mapping[str, str]] = ...,
         completion_deadline: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
+        num_shards: _Optional[int] = ...,
+        num_workers: _Optional[int] = ...,
     ) -> None: ...

{chalkpy-2.96.5 → chalkpy-2.96.7}/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"\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"\xdd\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\x34\n\x15workflow_execution_id\x18\x04 \x01(\tH\x00R\x13workflowExecutionId\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'
+    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"\xdd\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\x34\n\x15workflow_execution_id\x18\x04 \x01(\tH\x00R\x13workflowExecutionId\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"\xf5\x02\n\x15ScheduledQueryControl\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12&\n\x0f\x63ron_query_name\x18\x02 \x01(\tR\rcronQueryName\x12:\n\x06status\x18\x03 \x01(\x0e\x32".chalk.server.v1.CronControlStatusR\x06status\x12\x1e\n\x08\x61gent_id\x18\x04 \x01(\tH\x00R\x07\x61gentId\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12"\n\njob_config\x18\x07 \x01(\tH\x01R\tjobConfig\x88\x01\x01\x42\x0b\n\t_agent_idB\r\n\x0b_job_config"\xd4\x07\n\x16ScheduledQuerySchedule\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12"\n\rcron_query_id\x18\x02 \x01(\x05R\x0b\x63ronQueryId\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId\x12\x12\n\x04name\x18\x04 \x01(\tR\x04name\x12\x12\n\x04\x63ron\x18\x05 \x01(\tR\x04\x63ron\x12\x1a\n\x08\x66ilename\x18\x06 \x01(\tR\x08\x66ilename\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x1b\n\x06output\x18\t \x01(\tH\x00R\x06output\x88\x01\x01\x12$\n\x0bmax_samples\x18\n \x01(\x05H\x01R\nmaxSamples\x88\x01\x01\x12\x32\n\x12recompute_features\x18\x0b \x01(\tH\x02R\x11recomputeFeatures\x88\x01\x01\x12@\n\x0blower_bound\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\nupperBound\x88\x01\x01\x12\x17\n\x04tags\x18\x0e \x01(\tH\x05R\x04tags\x88\x01\x01\x12\x39\n\x16required_resolver_tags\x18\x0f \x01(\tH\x06R\x14requiredResolverTags\x88\x01\x01\x12,\n\x0fplanner_options\x18\x10 \x01(\tH\x07R\x0eplannerOptions\x88\x01\x01\x12&\n\x0c\x64\x61taset_name\x18\x11 \x01(\tH\x08R\x0b\x64\x61tasetName\x88\x01\x01\x12"\n\nnum_shards\x18\x12 \x01(\x05H\tR\tnumShards\x88\x01\x01\x12$\n\x0bnum_workers\x18\x13 \x01(\x05H\nR\nnumWorkers\x88\x01\x01\x42\t\n\x07_outputB\x0e\n\x0c_max_samplesB\x15\n\x13_recompute_featuresB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x07\n\x05_tagsB\x19\n\x17_required_resolver_tagsB\x12\n\x10_planner_optionsB\x0f\n\r_dataset_nameB\r\n\x0b_num_shardsB\x0e\n\x0c_num_workers*\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*u\n\x11\x43ronControlStatus\x12#\n\x1f\x43RON_CONTROL_STATUS_UNSPECIFIED\x10\x00\x12\x1c\n\x18\x43RON_CONTROL_STATUS_PLAY\x10\x01\x12\x1d\n\x19\x43RON_CONTROL_STATUS_PAUSE\x10\x02\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,8 +30,10 @@ 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 = 1685
-    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 2084
+    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_start = 3044
+    _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 3443
+    _globals["_CRONCONTROLSTATUS"]._serialized_start = 3445
+    _globals["_CRONCONTROLSTATUS"]._serialized_end = 3562
     _globals["_SCHEDULEDQUERYRUN"]._serialized_start = 169
     _globals["_SCHEDULEDQUERYRUN"]._serialized_end = 973
     _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_start = 976
@@ -42,4 +44,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_end = 1569
     _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_start = 1571
     _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_end = 1682
+    _globals["_SCHEDULEDQUERYCONTROL"]._serialized_start = 1685
+    _globals["_SCHEDULEDQUERYCONTROL"]._serialized_end = 2058
+    _globals["_SCHEDULEDQUERYSCHEDULE"]._serialized_start = 2061
+    _globals["_SCHEDULEDQUERYSCHEDULE"]._serialized_end = 3041
 # @@protoc_insertion_point(module_scope)

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi

@@ -27,6 +27,12 @@ class ScheduledQueryRunStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper)
     SCHEDULED_QUERY_RUN_STATUS_FAILED: _ClassVar[ScheduledQueryRunStatus]
     SCHEDULED_QUERY_RUN_STATUS_CANCELED: _ClassVar[ScheduledQueryRunStatus]
 
+class CronControlStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    CRON_CONTROL_STATUS_UNSPECIFIED: _ClassVar[CronControlStatus]
+    CRON_CONTROL_STATUS_PLAY: _ClassVar[CronControlStatus]
+    CRON_CONTROL_STATUS_PAUSE: _ClassVar[CronControlStatus]
+
 SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED: ScheduledQueryRunStatus
 SCHEDULED_QUERY_RUN_STATUS_INITIALIZING: ScheduledQueryRunStatus
 SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED: ScheduledQueryRunStatus
@@ -36,6 +42,9 @@ SCHEDULED_QUERY_RUN_STATUS_WORKING: ScheduledQueryRunStatus
 SCHEDULED_QUERY_RUN_STATUS_COMPLETED: ScheduledQueryRunStatus
 SCHEDULED_QUERY_RUN_STATUS_FAILED: ScheduledQueryRunStatus
 SCHEDULED_QUERY_RUN_STATUS_CANCELED: ScheduledQueryRunStatus
+CRON_CONTROL_STATUS_UNSPECIFIED: CronControlStatus
+CRON_CONTROL_STATUS_PLAY: CronControlStatus
+CRON_CONTROL_STATUS_PAUSE: CronControlStatus
 
 class ScheduledQueryRun(_message.Message):
     __slots__ = (
@@ -161,3 +170,113 @@ class GetScheduledQueryRunsResponse(_message.Message):
     def __init__(
         self, runs: _Optional[_Iterable[_Union[ScheduledQueryRun, _Mapping]]] = ..., cursor: _Optional[str] = ...
     ) -> None: ...
+
+class ScheduledQueryControl(_message.Message):
+    __slots__ = ("cron_query_id", "cron_query_name", "status", "agent_id", "created_at", "updated_at", "job_config")
+    CRON_QUERY_ID_FIELD_NUMBER: _ClassVar[int]
+    CRON_QUERY_NAME_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    AGENT_ID_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    JOB_CONFIG_FIELD_NUMBER: _ClassVar[int]
+    cron_query_id: int
+    cron_query_name: str
+    status: CronControlStatus
+    agent_id: str
+    created_at: _timestamp_pb2.Timestamp
+    updated_at: _timestamp_pb2.Timestamp
+    job_config: str
+    def __init__(
+        self,
+        cron_query_id: _Optional[int] = ...,
+        cron_query_name: _Optional[str] = ...,
+        status: _Optional[_Union[CronControlStatus, str]] = ...,
+        agent_id: _Optional[str] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        job_config: _Optional[str] = ...,
+    ) -> None: ...
+
+class ScheduledQuerySchedule(_message.Message):
+    __slots__ = (
+        "id",
+        "cron_query_id",
+        "deployment_id",
+        "name",
+        "cron",
+        "filename",
+        "created_at",
+        "updated_at",
+        "output",
+        "max_samples",
+        "recompute_features",
+        "lower_bound",
+        "upper_bound",
+        "tags",
+        "required_resolver_tags",
+        "planner_options",
+        "dataset_name",
+        "num_shards",
+        "num_workers",
+    )
+    ID_FIELD_NUMBER: _ClassVar[int]
+    CRON_QUERY_ID_FIELD_NUMBER: _ClassVar[int]
+    DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    CRON_FIELD_NUMBER: _ClassVar[int]
+    FILENAME_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_FIELD_NUMBER: _ClassVar[int]
+    UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
+    OUTPUT_FIELD_NUMBER: _ClassVar[int]
+    MAX_SAMPLES_FIELD_NUMBER: _ClassVar[int]
+    RECOMPUTE_FEATURES_FIELD_NUMBER: _ClassVar[int]
+    LOWER_BOUND_FIELD_NUMBER: _ClassVar[int]
+    UPPER_BOUND_FIELD_NUMBER: _ClassVar[int]
+    TAGS_FIELD_NUMBER: _ClassVar[int]
+    REQUIRED_RESOLVER_TAGS_FIELD_NUMBER: _ClassVar[int]
+    PLANNER_OPTIONS_FIELD_NUMBER: _ClassVar[int]
+    DATASET_NAME_FIELD_NUMBER: _ClassVar[int]
+    NUM_SHARDS_FIELD_NUMBER: _ClassVar[int]
+    NUM_WORKERS_FIELD_NUMBER: _ClassVar[int]
+    id: int
+    cron_query_id: int
+    deployment_id: str
+    name: str
+    cron: str
+    filename: str
+    created_at: _timestamp_pb2.Timestamp
+    updated_at: _timestamp_pb2.Timestamp
+    output: str
+    max_samples: int
+    recompute_features: str
+    lower_bound: _timestamp_pb2.Timestamp
+    upper_bound: _timestamp_pb2.Timestamp
+    tags: str
+    required_resolver_tags: str
+    planner_options: str
+    dataset_name: str
+    num_shards: int
+    num_workers: int
+    def __init__(
+        self,
+        id: _Optional[int] = ...,
+        cron_query_id: _Optional[int] = ...,
+        deployment_id: _Optional[str] = ...,
+        name: _Optional[str] = ...,
+        cron: _Optional[str] = ...,
+        filename: _Optional[str] = ...,
+        created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        output: _Optional[str] = ...,
+        max_samples: _Optional[int] = ...,
+        recompute_features: _Optional[str] = ...,
+        lower_bound: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        upper_bound: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
+        tags: _Optional[str] = ...,
+        required_resolver_tags: _Optional[str] = ...,
+        planner_options: _Optional[str] = ...,
+        dataset_name: _Optional[str] = ...,
+        num_shards: _Optional[int] = ...,
+        num_workers: _Optional[int] = ...,
+    ) -> None: ...

chalkpy-2.96.7/chalk/_version.py

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

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/client/client.py

@@ -526,6 +526,7 @@ class ChalkClient:
         explain: bool = False,
         num_input_rows: Optional[int] = None,
         headers: Mapping[str, str] | None = None,
+        planner_options: Mapping[str, str | int | bool] | None = None,
     ) -> PlanQueryResponse:
         """Plan a query without executing it.
 
@@ -575,6 +576,10 @@ class ChalkClient:
             The number of input rows that this plan will be run with. If unknown, specify `None`.
         headers
             Additional headers to provide with the request
+        planner_options
+            Dictionary of additional options to pass to the Chalk query engine.
+            Values may be provided as part of conversations with Chalk support
+            to enable or disable specific functionality.
 
         Returns
         -------

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/client/client_async.py

@@ -486,6 +486,7 @@ class AsyncChalkClient:
         explain: bool = False,
         num_input_rows: Optional[int] = None,
         headers: Mapping[str, str] | None = None,
+        planner_options: Mapping[str, str | int | bool] | None = None,
     ) -> PlanQueryResponse:
         """Plan a query without executing it.
 
@@ -535,6 +536,10 @@ class AsyncChalkClient:
             The number of input rows that this plan will be run with. If unknown, specify `None`.
         headers
             Additional headers to provide with the request
+        planner_options
+            Dictionary of additional options to pass to the Chalk query engine.
+            Values may be provided as part of conversations with Chalk support
+            to enable or disable specific functionality.
 
         Returns
         -------

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/client/client_impl.py

@@ -4438,6 +4438,7 @@ https://docs.chalk.ai/cli/apply
         explain: bool = False,
         num_input_rows: Optional[int] = None,
         headers: Mapping[str, str] | None = None,
+        planner_options: Mapping[str, str | int | bool] | None = None,
     ) -> PlanQueryResponse:
         encoded_inputs = encode_outputs(input).string_outputs
         outputs = encode_outputs(output).string_outputs
@@ -4472,6 +4473,7 @@ https://docs.chalk.ai/cli/apply
             store_plan_stages=store_plan_stages,
             explain=explain,
             num_input_rows=num_input_rows,
+            planner_options=planner_options,
         )
 
         extra_headers: dict[str, str] = {}

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/client/models.py

@@ -915,7 +915,7 @@ class CreateOfflineQueryJobRequest(BaseModel):
     @root_validator
     def _validate_multiple_computers(cls, values: Dict[str, Any]):
         if values["input"] is None or isinstance(
-            values["input"], (UploadedParquetShardedOfflineQueryInput, OfflineQueryInputUri)
+            values["input"], (UploadedParquetShardedOfflineQueryInput, OfflineQueryInputUri, OfflineQueryInputSql)
         ):
             return values
         expected_use_multiple_computers = isinstance(values["input"], tuple)
@@ -1653,6 +1653,7 @@ class PlanQueryRequest(BaseModel):
     explain: bool = False
     store_plan_stages: bool = False
     encoding_options: FeatureEncodingOptions = FeatureEncodingOptions()
+    planner_options: Mapping[str, str | int | bool | float] | None = None
 
 
 class FeatureSchema(BaseModel):

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/df/LazyFramePlaceholder.py

@@ -101,6 +101,11 @@ class LazyFramePlaceholder:
 
     __str__ = __repr__
 
+    def _is_equal(self, other: LazyFramePlaceholder) -> bool:
+        # proto equality is janky but it's hard to write a good eq method here given
+        # we have dicts and the proto round trip is slightly lossy on tuples vs lists
+        return self._to_proto() == other._to_proto()
+
     def _to_proto(self) -> dataframe_pb2.DataFramePlan:
         """
         Convert this proto plan to a dataframe.

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/features/_encoding/converter.py

@@ -940,10 +940,9 @@ class PrimitiveFeatureConverter(Generic[_TPrim]):
 
     @staticmethod
     def convert_arrow_table_from_proto(proto: pb.TableParquetBytes) -> pa.Table:
-        import pyarrow.parquet
+        import pyarrow.parquet as pq
 
-        pf = pyarrow.parquet.ParquetFile(io.BytesIO(proto.encoded_parquet_bytes))
-        return pyarrow.parquet.read_table(pf)
+        return pq.read_table(io.BytesIO(proto.encoded_parquet_bytes))
 
     @staticmethod
     def _serialize_pa_decimal_to_pb(value: Union[pa.Decimal128Scalar, pa.Decimal256Scalar]) -> pb.ScalarValue:

{chalkpy-2.96.5 → chalkpy-2.96.7}/chalk/parsed/_proto/export.py

@@ -215,6 +215,8 @@ def export_from_registry() -> export_pb.Export:
                 completion_deadline=timedelta_to_proto_duration(cron.completion_deadline)
                 if cron.completion_deadline is not None
                 else cron.completion_deadline,
+                num_shards=cron.num_shards,
+                num_workers=cron.num_workers,
             )
         )
 

chalkpy-2.96.5/chalk/_version.py

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