chalkpy 2.96.5__py3-none-any.whl → 2.96.7__py3-none-any.whl

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)

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: ...

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)

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: ...

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.96.5"
+__version__ = "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
         -------

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
         -------

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] = {}

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):

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.

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:

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.dist-info → chalkpy-2.96.7.dist-info}/METADATA

@@ -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.dist-info → chalkpy-2.96.7.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=8ZvKPpEPaNnATdFe98glfnwM5s6vGpw6tcR9AeaiP58,23
+chalk/_version.py,sha256=3_AK7WV-Cb97v3VLeqbWqWc5ckXI1N8Wm6F3FsWkhQI,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -41,8 +41,8 @@ chalk/_gen/chalk/artifacts/v1/chart_pb2.py,sha256=XOXGa60UkIn6g2I0KWZihDF0PSiHp3
 chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi,sha256=BDs1o5LFC1gqMdO67wimsCI4sVfX-42aKwDQqDb7DpQ,21653
 chalk/_gen/chalk/artifacts/v1/chart_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/artifacts/v1/chart_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
-chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py,sha256=VyJU2lT1_DqrWJG9Ce6mIiTQHJM2oDANlsOaiRMEtK8,3776
-chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi,sha256=SR2_0nsD61s9EutuTo6UrtrHa--FRi-qPle53C5Ul8A,3867
+chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py,sha256=HGcz2kkIV7V7e7DKV_fTxjSl9d0Qm36ZtdwdSp4ze_k,3956
+chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi,sha256=xlL_r_iy4WUHMYIY3uay2oiFWNY7pJq1anK15KRX04Y,4127
 chalk/_gen/chalk/artifacts/v1/cron_query_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/artifacts/v1/cron_query_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
 chalk/_gen/chalk/artifacts/v1/deployment_pb2.py,sha256=7ifnsrurEyVGcUMX41tD3XGNveBGrCfqG21GEdhrRZE,2877
@@ -504,8 +504,8 @@ chalk/_gen/chalk/server/v1/scheduled_query_pb2.py,sha256=aU7YjTJ2H9dp_ZRNmh_91NC
 chalk/_gen/chalk/server/v1/scheduled_query_pb2.pyi,sha256=4o__cQ3plq1LucDRH03pMf9LBSjk1eE7XY51gyv435c,299
 chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.py,sha256=Pb0WD51FjkvUmPcYINjkSgYX7Z2HCmakFb5wp_Ulqxw,5037
 chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.pyi,sha256=w98vRRLSiFl6Z_8c06DDAVH2593_dAgbhdlI-vY0tJw,1337
-chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=WJ-nWBvaVoAUy57jZ1kG6_Q4xvQgwpW83xoIE1K_2CI,5619
-chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=ePknNWz-VY73vA60QqzyX6ZgimWyJQ6rO_f_DUlNziY,6590
+chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=oFIhmStu504lzlE9Lrc7zGzK_T21pq_EDl54rG0_87Y,8412
+chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=b1KQf_j5joN92W2_Q2ha04wCyj5jIdqrRWEKGJpjFJA,10952
 chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
 chalk/_gen/chalk/server/v1/scheduler_pb2.py,sha256=kfmYnzdx22eNeemjrfuANsiyI1oARmfSRw3MrURaFAI,11969
@@ -612,14 +612,14 @@ chalk/_validation/validation.py,sha256=9cCMfZa9-1wxkXLme_ylmD5vIA1qExJD6aqbYvbmK
 chalk/byte_transmit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/byte_transmit/model.py,sha256=LFX8pj9X_CWXeap7fDnMl9YmXsYTgq7jBAbEWkxoYoE,13048
 chalk/client/__init__.py,sha256=wu3WQVzE5gRj6noQQDOdYJUgDaz_9QtbjXH4KuoIlXQ,1782
-chalk/client/client.py,sha256=fqw75x4yiAk3CXryGbuF_qg09WDIlluhYeFdQ-BfTG8,103337
-chalk/client/client_async.py,sha256=nFFTWJbdBlb7zksyjOMBY566tZTAyNXQhCnq06LHWl0,50803
+chalk/client/client.py,sha256=uu0S3bG0_6HG9S2seIH1oaww6Jz684fG9fIm3x2MwvI,103648
+chalk/client/client_async.py,sha256=I-1igsjYgK6qFh-KN524QBSvtxc8Yntltgt5fRHpQvA,51114
 chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
 chalk/client/client_grpc.py,sha256=skpSHYCyE8hg0T3Bcl5R8MMi92QjrFtxQRu_NxCSVlw,106952
-chalk/client/client_impl.py,sha256=rvT02yriRo7_WFqbUsHXRjm6DbRyXzw2QjcLwKaW10E,211844
+chalk/client/client_impl.py,sha256=rHkVYn0jawWwlFYPLseGuSm8sHwuLlMGY9uf1KSa8wY,211960
 chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
 chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
-chalk/client/models.py,sha256=FDgEqIeBJq25T9AH38jdWkeyRLiAXfLXfXHpomFJM08,63923
+chalk/client/models.py,sha256=X3sq-kv4F601ABUcwE3Y7X-yF4sTHwbCk1HfX7xnhO0,64019
 chalk/client/response.py,sha256=m8sQCOj7YVv3mZSZMIC1rIMzFMQ9rfMdBRLg5NRmOOE,53257
 chalk/client/_internal_models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/client/_internal_models/check.py,sha256=3Xfo4Ws4rvwjeVg0-5-kejfRfRBJeqHmnRhW-WEz784,917
@@ -636,7 +636,7 @@ chalk/config/_validator.py,sha256=QC1y52m704_bV_TYjq0sdZJ-km8iSkDX1V4sHgw4RJk,13
 chalk/config/auth_config.py,sha256=HAALkQrvDD0i7gaZK5iufS8vDDVbzLIpHLOpcJO1kmw,4498
 chalk/config/project_config.py,sha256=YHB3upvtBJu-tWfNOchBRSc9xGebDbrIpCVmKbBzJ8Q,7217
 chalk/df/ChalkDataFrameImpl.py,sha256=BRwnjQcie3gxaKhKau8HG17NWzS1zdr8SnNVurxF8QY,133
-chalk/df/LazyFramePlaceholder.py,sha256=uTinB2buEFBT8djBCqiPN2hKmr_nU7n4i_DcCLNLuWs,39111
+chalk/df/LazyFramePlaceholder.py,sha256=D4oYEGPfoxpJMmyiDJq3G9Wfl1jF3XSXD71q-GxOrmQ,39398
 chalk/df/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/df/ast_parser.py,sha256=t-DwUxd2hN8LECRSBx85DIv9FtILgMiHyGyCTINfgQw,11199
 chalk/features/__init__.py,sha256=5doD7bFwRthzwdmizbRaPVHiCABiNpiOiAJVFlwqNnA,6943
@@ -673,7 +673,7 @@ chalk/features/_embedding/sentence_transformer.py,sha256=hNYuT9D-16C49lkhIVK_KXZ
 chalk/features/_embedding/utils.py,sha256=gUYUJCt-9XLeioC3QYg42yc5f69ywAr_w_Q-xoF3o54,877
 chalk/features/_embedding/vertexai.py,sha256=PUaMmd7qbK9vXrlce_8BQHg_OM9L982k6LuyM_PU-5w,5362
 chalk/features/_encoding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chalk/features/_encoding/converter.py,sha256=YMIFZLAiN7z8JhDj4hsrH4qQTZ0pFBIwaJN6ULVsjd8,66221
+chalk/features/_encoding/converter.py,sha256=YLNX4hxoT5GlP_n_cL5anQlIzfMW0TfBuuK1TBccaiI,66169
 chalk/features/_encoding/http.py,sha256=MVmkgv90O4sOnImc-FrRPVwYR2U7zOGduNkHO4umoe4,2227
 chalk/features/_encoding/inputs.py,sha256=4K4kOCAl0MuobAC_lVJE70toQVrPt2ZlzrlwYALuLHE,10151
 chalk/features/_encoding/json.py,sha256=lHSBWIjNKMDmt_AmiFtf4ZsW5QPIEC6r7NvDW0IdqPI,13804
@@ -731,7 +731,7 @@ chalk/parsed/to_proto.py,sha256=ZSwEYkuwr3UhMVabt6FM02fBN_oBy_mSMjDYS_crJw0,7553
 chalk/parsed/user_types_to_json.py,sha256=ZJWdYFqyhr5InvItQybtHadXQjW3vjHrv8hjMGtL3Bc,13318
 chalk/parsed/validation_from_registries.py,sha256=nfiAj1tvWRu0RrkhkGtElhAsL8V7ayEKUisrKQF-wYc,7900
 chalk/parsed/_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-chalk/parsed/_proto/export.py,sha256=hsHKIEHl6-Bd-lMPU1zMzu8MEA1lal8FNWBFX4bRFPw,11676
+chalk/parsed/_proto/export.py,sha256=mRPDyXNOqHLE9GcpmADjQgzHTCeYpZW5x8qp41UIlMQ,11766
 chalk/parsed/_proto/lsp.py,sha256=hAv20bo8edVDhn2EgJRo9PmzoIi2JlR6bBcvkUPF4YM,3711
 chalk/parsed/_proto/utils.py,sha256=X6vZp3OAFMxWS5BaWVeKuNhF5ob4Ws9yAz4RiNalCx4,10680
 chalk/parsed/_proto/validation.py,sha256=D6dOnCoeARHOLPHQb8zYGXDc0eWJblSQ4tO4rb_HKNU,35367
@@ -827,8 +827,8 @@ chalk/utils/tracing.py,sha256=NiiM-9dbuJhSCv6R1npR1uYNKWlkqTR6Ygm0Voi2NrY,13078
 chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
 chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
-chalkpy-2.96.5.dist-info/METADATA,sha256=dYguQl-353EIvqauFcbZX5hG1RrMj45QqCLsi2la9Fw,27754
-chalkpy-2.96.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-chalkpy-2.96.5.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.96.5.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.96.5.dist-info/RECORD,,
+chalkpy-2.96.7.dist-info/METADATA,sha256=QWqtSzsETPKu6gG6oHi0s_nhv7SfLSuca2-SjNFKAKU,27754
+chalkpy-2.96.7.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+chalkpy-2.96.7.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.96.7.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.96.7.dist-info/RECORD,,