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

chalk/_gen/chalk/server/v1/scheduler_pb2.py

@@ -21,7 +21,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1f\x63halk/server/v1/scheduler.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1b\x63halk/server/v1/batch.proto\x1a)chalk/server/v1/scheduled_query_run.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xa4\x06\n\x0f\x43ronResolverRun\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12!\n\x0cresolver_fqn\x18\x03 \x01(\tR\x0bresolverFqn\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12+\n\x11schedule_readable\x18\x05 \x01(\tR\x10scheduleReadable\x12\x36\n\x17schedule_resolver_value\x18\x06 \x01(\tR\x15scheduleResolverValue\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12(\n\rdeployment_id\x18\x08 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x31\n\x03\x65nd\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x03\x65nd\x88\x01\x01\x12:\n\x05\x62\x61tch\x18\n \x01(\x0b\x32\x1f.chalk.server.v1.BatchOperationH\x02R\x05\x62\x61tch\x88\x01\x01\x12\x46\n\x0ctrigger_kind\x18\x0b \x01(\x0e\x32#.chalk.server.v1.CronRunTriggerKindR\x0btriggerKind\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$\n\x0bmax_samples\x18\x0e \x01(\x03H\x05R\nmaxSamples\x88\x01\x01\x12$\n\x0eused_job_queue\x18\x0f \x01(\x08R\x0cusedJobQueueB\x10\n\x0e_deployment_idB\x06\n\x04_endB\x08\n\x06_batchB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x0e\n\x0c_max_samples"\x9d\x03\n ManualTriggerCronResolverRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12$\n\x0bmax_samples\x18\x02 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12@\n\x0blower_bound\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nupperBound\x88\x01\x01\x12\x30\n\x11timestamping_mode\x18\x06 \x01(\tH\x03R\x10timestampingMode\x88\x01\x01\x12$\n\x0bjob_options\x18\x07 \x01(\tH\x04R\njobOptions\x88\x01\x01\x42\x0e\n\x0c_max_samplesB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x14\n\x12_timestamping_modeB\x0e\n\x0c_job_options"q\n!ManualTriggerCronResolverResponse\x12L\n\x11\x63ron_resolver_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x0f\x63ronResolverRun"\x9c\x05\n"ManualTriggerScheduledQueryRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x03R\x0b\x63ronQueryId\x12p\n\x0fplanner_options\x18\x02 \x03(\x0b\x32G.chalk.server.v1.ManualTriggerScheduledQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\x33\n\x15incremental_resolvers\x18\x03 \x03(\tR\x14incrementalResolvers\x12$\n\x0bmax_samples\x18\x04 \x01(\x03H\x01R\nmaxSamples\x88\x01\x01\x12j\n\renv_overrides\x18\x05 \x03(\x0b\x32\x45.chalk.server.v1.ManualTriggerScheduledQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12(\n\x0f\x63ron_query_name\x18\x06 \x01(\tH\x00R\rcronQueryName\x12*\n\x11store_plan_stages\x18\x07 \x01(\x08R\x0fstorePlanStages\x1aY\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x17\n\x15\x63ron_query_identifierB\x0e\n\x0c_max_samples"y\n#ManualTriggerScheduledQueryResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun"0\n\x1eGetScheduledResolverRunRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"U\n\x1fGetScheduledResolverRunResponse\x12\x32\n\x03run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x03run"\xa7\x03\n ListScheduledResolverRunsRequest\x12\x37\n\x06\x63ursor\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x06\x63ursor\x88\x01\x01\x12,\n\x0fresolver_filter\x18\x02 \x01(\tH\x01R\x0eresolverFilter\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x02R\x05limit\x88\x01\x01\x12\x35\n\x05start\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\x05start\x88\x01\x01\x12\x31\n\x03\x65nd\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\x03\x65nd\x88\x01\x01\x12J\n\rstatus_filter\x18\x06 \x01(\x0e\x32 .chalk.server.v1.OperationStatusH\x05R\x0cstatusFilter\x88\x01\x01\x42\t\n\x07_cursorB\x12\n\x10_resolver_filterB\x08\n\x06_limitB\x08\n\x06_startB\x06\n\x04_endB\x10\n\x0e_status_filter"Y\n!ListScheduledResolverRunsResponse\x12\x34\n\x04runs\x18\x01 \x03(\x0b\x32 .chalk.server.v1.CronResolverRunR\x04runs":\n!CancelScheduledResolverRunRequest\x12\x15\n\x06run_id\x18\x01 \x01(\tR\x05runId"a\n"CancelScheduledResolverRunResponse\x12;\n\x08\x63ron_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x07\x63ronRun*\x9c\x01\n\x12\x43ronRunTriggerKind\x12%\n!CRON_RUN_TRIGGER_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x43RON_RUN_TRIGGER_KIND_MANUAL\x10\x01\x12\x1d\n\x19\x43RON_RUN_TRIGGER_KIND_API\x10\x02\x12\x1e\n\x1a\x43RON_RUN_TRIGGER_KIND_CRON\x10\x03\x32\xc7\x05\n\x10SchedulerService\x12\x87\x01\n\x19ManualTriggerCronResolver\x12\x31.chalk.server.v1.ManualTriggerCronResolverRequest\x1a\x32.chalk.server.v1.ManualTriggerCronResolverResponse"\x03\x80}\x11\x12\x8d\x01\n\x1bManualTriggerScheduledQuery\x12\x33.chalk.server.v1.ManualTriggerScheduledQueryRequest\x1a\x34.chalk.server.v1.ManualTriggerScheduledQueryResponse"\x03\x80}\x11\x12\x81\x01\n\x17GetScheduledResolverRun\x12/.chalk.server.v1.GetScheduledResolverRunRequest\x1a\x30.chalk.server.v1.GetScheduledResolverRunResponse"\x03\x80}\x10\x12\x87\x01\n\x19ListScheduledResolverRuns\x12\x31.chalk.server.v1.ListScheduledResolverRunsRequest\x1a\x32.chalk.server.v1.ListScheduledResolverRunsResponse"\x03\x80}\x10\x12\x8a\x01\n\x1a\x43\x61ncelScheduledResolverRun\x12\x32.chalk.server.v1.CancelScheduledResolverRunRequest\x1a\x33.chalk.server.v1.CancelScheduledResolverRunResponse"\x03\x80}\x11\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0eSchedulerProtoP\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\x1f\x63halk/server/v1/scheduler.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a\x1b\x63halk/server/v1/batch.proto\x1a)chalk/server/v1/scheduled_query_run.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xa4\x06\n\x0f\x43ronResolverRun\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12!\n\x0cresolver_fqn\x18\x03 \x01(\tR\x0bresolverFqn\x12\x12\n\x04kind\x18\x04 \x01(\tR\x04kind\x12+\n\x11schedule_readable\x18\x05 \x01(\tR\x10scheduleReadable\x12\x36\n\x17schedule_resolver_value\x18\x06 \x01(\tR\x15scheduleResolverValue\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12(\n\rdeployment_id\x18\x08 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x31\n\x03\x65nd\x18\t \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x03\x65nd\x88\x01\x01\x12:\n\x05\x62\x61tch\x18\n \x01(\x0b\x32\x1f.chalk.server.v1.BatchOperationH\x02R\x05\x62\x61tch\x88\x01\x01\x12\x46\n\x0ctrigger_kind\x18\x0b \x01(\x0e\x32#.chalk.server.v1.CronRunTriggerKindR\x0btriggerKind\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$\n\x0bmax_samples\x18\x0e \x01(\x03H\x05R\nmaxSamples\x88\x01\x01\x12$\n\x0eused_job_queue\x18\x0f \x01(\x08R\x0cusedJobQueueB\x10\n\x0e_deployment_idB\x06\n\x04_endB\x08\n\x06_batchB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x0e\n\x0c_max_samples"\x9d\x03\n ManualTriggerCronResolverRequest\x12!\n\x0cresolver_fqn\x18\x01 \x01(\tR\x0bresolverFqn\x12$\n\x0bmax_samples\x18\x02 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12@\n\x0blower_bound\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x02R\nupperBound\x88\x01\x01\x12\x30\n\x11timestamping_mode\x18\x06 \x01(\tH\x03R\x10timestampingMode\x88\x01\x01\x12$\n\x0bjob_options\x18\x07 \x01(\tH\x04R\njobOptions\x88\x01\x01\x42\x0e\n\x0c_max_samplesB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x14\n\x12_timestamping_modeB\x0e\n\x0c_job_options"q\n!ManualTriggerCronResolverResponse\x12L\n\x11\x63ron_resolver_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x0f\x63ronResolverRun"\x9c\x05\n"ManualTriggerScheduledQueryRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x03R\x0b\x63ronQueryId\x12p\n\x0fplanner_options\x18\x02 \x03(\x0b\x32G.chalk.server.v1.ManualTriggerScheduledQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\x33\n\x15incremental_resolvers\x18\x03 \x03(\tR\x14incrementalResolvers\x12$\n\x0bmax_samples\x18\x04 \x01(\x03H\x01R\nmaxSamples\x88\x01\x01\x12j\n\renv_overrides\x18\x05 \x03(\x0b\x32\x45.chalk.server.v1.ManualTriggerScheduledQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12(\n\x0f\x63ron_query_name\x18\x06 \x01(\tH\x00R\rcronQueryName\x12*\n\x11store_plan_stages\x18\x07 \x01(\x08R\x0fstorePlanStages\x1aY\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x17\n\x15\x63ron_query_identifierB\x0e\n\x0c_max_samples"y\n#ManualTriggerScheduledQueryResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun"0\n\x1eGetScheduledResolverRunRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"U\n\x1fGetScheduledResolverRunResponse\x12\x32\n\x03run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x03run"\x97\x04\n ListScheduledResolverRunsRequest\x12;\n\x06\x63ursor\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.TimestampB\x02\x18\x01H\x00R\x06\x63ursor\x88\x01\x01\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x01R\x05limit\x88\x01\x01\x12"\n\npage_token\x18\x07 \x01(\tH\x02R\tpageToken\x88\x01\x01\x12,\n\x0fresolver_filter\x18\x02 \x01(\tH\x03R\x0eresolverFilter\x88\x01\x01\x12&\n\x0cresolver_fqn\x18\x08 \x01(\tH\x04R\x0bresolverFqn\x88\x01\x01\x12J\n\rstatus_filter\x18\x06 \x01(\x0e\x32 .chalk.server.v1.OperationStatusH\x05R\x0cstatusFilter\x88\x01\x01\x12\x35\n\x05start\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x06R\x05start\x88\x01\x01\x12\x31\n\x03\x65nd\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x07R\x03\x65nd\x88\x01\x01\x42\t\n\x07_cursorB\x08\n\x06_limitB\r\n\x0b_page_tokenB\x12\n\x10_resolver_filterB\x0f\n\r_resolver_fqnB\x10\n\x0e_status_filterB\x08\n\x06_startB\x06\n\x04_end"\x9a\x01\n!ListScheduledResolverRunsResponse\x12\x34\n\x04runs\x18\x01 \x03(\x0b\x32 .chalk.server.v1.CronResolverRunR\x04runs\x12+\n\x0fnext_page_token\x18\x02 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token":\n!CancelScheduledResolverRunRequest\x12\x15\n\x06run_id\x18\x01 \x01(\tR\x05runId"a\n"CancelScheduledResolverRunResponse\x12;\n\x08\x63ron_run\x18\x01 \x01(\x0b\x32 .chalk.server.v1.CronResolverRunR\x07\x63ronRun*\x9c\x01\n\x12\x43ronRunTriggerKind\x12%\n!CRON_RUN_TRIGGER_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x43RON_RUN_TRIGGER_KIND_MANUAL\x10\x01\x12\x1d\n\x19\x43RON_RUN_TRIGGER_KIND_API\x10\x02\x12\x1e\n\x1a\x43RON_RUN_TRIGGER_KIND_CRON\x10\x03\x32\xc7\x05\n\x10SchedulerService\x12\x87\x01\n\x19ManualTriggerCronResolver\x12\x31.chalk.server.v1.ManualTriggerCronResolverRequest\x1a\x32.chalk.server.v1.ManualTriggerCronResolverResponse"\x03\x80}\x11\x12\x8d\x01\n\x1bManualTriggerScheduledQuery\x12\x33.chalk.server.v1.ManualTriggerScheduledQueryRequest\x1a\x34.chalk.server.v1.ManualTriggerScheduledQueryResponse"\x03\x80}\x11\x12\x81\x01\n\x17GetScheduledResolverRun\x12/.chalk.server.v1.GetScheduledResolverRunRequest\x1a\x30.chalk.server.v1.GetScheduledResolverRunResponse"\x03\x80}\x10\x12\x87\x01\n\x19ListScheduledResolverRuns\x12\x31.chalk.server.v1.ListScheduledResolverRunsRequest\x1a\x32.chalk.server.v1.ListScheduledResolverRunsResponse"\x03\x80}\x10\x12\x8a\x01\n\x1a\x43\x61ncelScheduledResolverRun\x12\x32.chalk.server.v1.CancelScheduledResolverRunRequest\x1a\x33.chalk.server.v1.CancelScheduledResolverRunResponse"\x03\x80}\x11\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0eSchedulerProtoP\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()
@@ -36,6 +36,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_options = b"8\001"
     _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._options = None
     _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_options = b"8\001"
+    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"].fields_by_name["cursor"]._options = None
+    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"].fields_by_name["cursor"]._serialized_options = b"\030\001"
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerCronResolver"]._options = None
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerCronResolver"]._serialized_options = b"\200}\021"
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerScheduledQuery"]._options = None
@@ -46,8 +48,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SCHEDULERSERVICE"].methods_by_name["ListScheduledResolverRuns"]._serialized_options = b"\200}\020"
     _globals["_SCHEDULERSERVICE"].methods_by_name["CancelScheduledResolverRun"]._options = None
     _globals["_SCHEDULERSERVICE"].methods_by_name["CancelScheduledResolverRun"]._serialized_options = b"\200}\021"
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 3166
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 3322
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 3344
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 3500
     _globals["_CRONRESOLVERRUN"]._serialized_start = 221
     _globals["_CRONRESOLVERRUN"]._serialized_end = 1025
     _globals["_MANUALTRIGGERCRONRESOLVERREQUEST"]._serialized_start = 1028
@@ -67,13 +69,13 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_start = 2402
     _globals["_GETSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_end = 2487
     _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_start = 2490
-    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_end = 2913
-    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_start = 2915
-    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_end = 3004
-    _globals["_CANCELSCHEDULEDRESOLVERRUNREQUEST"]._serialized_start = 3006
-    _globals["_CANCELSCHEDULEDRESOLVERRUNREQUEST"]._serialized_end = 3064
-    _globals["_CANCELSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_start = 3066
-    _globals["_CANCELSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_end = 3163
-    _globals["_SCHEDULERSERVICE"]._serialized_start = 3325
-    _globals["_SCHEDULERSERVICE"]._serialized_end = 4036
+    _globals["_LISTSCHEDULEDRESOLVERRUNSREQUEST"]._serialized_end = 3025
+    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_start = 3028
+    _globals["_LISTSCHEDULEDRESOLVERRUNSRESPONSE"]._serialized_end = 3182
+    _globals["_CANCELSCHEDULEDRESOLVERRUNREQUEST"]._serialized_start = 3184
+    _globals["_CANCELSCHEDULEDRESOLVERRUNREQUEST"]._serialized_end = 3242
+    _globals["_CANCELSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_start = 3244
+    _globals["_CANCELSCHEDULEDRESOLVERRUNRESPONSE"]._serialized_end = 3341
+    _globals["_SCHEDULERSERVICE"]._serialized_start = 3503
+    _globals["_SCHEDULERSERVICE"]._serialized_end = 4214
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/scheduler_pb2.pyi

@@ -200,34 +200,44 @@ class GetScheduledResolverRunResponse(_message.Message):
     def __init__(self, run: _Optional[_Union[CronResolverRun, _Mapping]] = ...) -> None: ...
 
 class ListScheduledResolverRunsRequest(_message.Message):
-    __slots__ = ("cursor", "resolver_filter", "limit", "start", "end", "status_filter")
+    __slots__ = ("cursor", "limit", "page_token", "resolver_filter", "resolver_fqn", "status_filter", "start", "end")
     CURSOR_FIELD_NUMBER: _ClassVar[int]
-    RESOLVER_FILTER_FIELD_NUMBER: _ClassVar[int]
     LIMIT_FIELD_NUMBER: _ClassVar[int]
+    PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
+    RESOLVER_FILTER_FIELD_NUMBER: _ClassVar[int]
+    RESOLVER_FQN_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FILTER_FIELD_NUMBER: _ClassVar[int]
     START_FIELD_NUMBER: _ClassVar[int]
     END_FIELD_NUMBER: _ClassVar[int]
-    STATUS_FILTER_FIELD_NUMBER: _ClassVar[int]
     cursor: _timestamp_pb2.Timestamp
-    resolver_filter: str
     limit: int
+    page_token: str
+    resolver_filter: str
+    resolver_fqn: str
+    status_filter: _batch_pb2.OperationStatus
     start: _timestamp_pb2.Timestamp
     end: _timestamp_pb2.Timestamp
-    status_filter: _batch_pb2.OperationStatus
     def __init__(
         self,
         cursor: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
-        resolver_filter: _Optional[str] = ...,
         limit: _Optional[int] = ...,
+        page_token: _Optional[str] = ...,
+        resolver_filter: _Optional[str] = ...,
+        resolver_fqn: _Optional[str] = ...,
+        status_filter: _Optional[_Union[_batch_pb2.OperationStatus, str]] = ...,
         start: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
         end: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
-        status_filter: _Optional[_Union[_batch_pb2.OperationStatus, str]] = ...,
     ) -> None: ...
 
 class ListScheduledResolverRunsResponse(_message.Message):
-    __slots__ = ("runs",)
+    __slots__ = ("runs", "next_page_token")
     RUNS_FIELD_NUMBER: _ClassVar[int]
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
     runs: _containers.RepeatedCompositeFieldContainer[CronResolverRun]
-    def __init__(self, runs: _Optional[_Iterable[_Union[CronResolverRun, _Mapping]]] = ...) -> None: ...
+    next_page_token: str
+    def __init__(
+        self, runs: _Optional[_Iterable[_Union[CronResolverRun, _Mapping]]] = ..., next_page_token: _Optional[str] = ...
+    ) -> None: ...
 
 class CancelScheduledResolverRunRequest(_message.Message):
     __slots__ = ("run_id",)

chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py

@@ -18,7 +18,7 @@ from chalk._gen.chalk.common.v1 import chalk_error_pb2 as chalk_dot_common_dot_v
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\xd6\x01\n!SimpleStreamingUnaryInvokeRequest\x12\x34\n\x16streaming_resolver_fqn\x18\x01 \x01(\tR\x14streamingResolverFqn\x12\x1d\n\ninput_data\x18\x02 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x04 \x01(\x08H\x01R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_operation_idB\x08\n\x06_debug"\xe7\x02\n"SimpleStreamingUnaryInvokeResponse\x12(\n\x10num_rows_succeed\x18\x01 \x01(\x03R\x0enumRowsSucceed\x12&\n\x0fnum_rows_failed\x18\x02 \x01(\x03R\rnumRowsFailed\x12(\n\x10num_rows_skipped\x18\x03 \x01(\x03R\x0enumRowsSkipped\x12\x36\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12$\n\x0boutput_data\x18\x05 \x01(\x0cH\x01R\noutputData\x88\x01\x01\x12M\n\x10\x65xecution_errors\x18\x06 \x03(\x0b\x32".chalk.streaming.v1.StreamingErrorR\x0f\x65xecutionErrorsB\x08\n\x06_errorB\x0e\n\x0c_output_data"\xb5\x02\n\x1cTestStreamingResolverRequest\x12&\n\x0cresolver_fqn\x18\x01 \x01(\tH\x00R\x0bresolverFqn\x88\x01\x01\x12@\n\x1astatic_stream_resolver_b64\x18\x02 \x01(\tH\x01R\x17staticStreamResolverB64\x88\x01\x01\x12\x1d\n\ninput_data\x18\x03 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x04 \x01(\tH\x02R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x03R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_resolver_fqnB\x1d\n\x1b_static_stream_resolver_b64B\x0f\n\r_operation_idB\x08\n\x06_debug"\xe1\x01\n\x1dTestStreamingResolverResponse\x12\x44\n\x06status\x18\x01 \x01(\x0e\x32,.chalk.streaming.v1.TestStreamResolverStatusR\x06status\x12\x1e\n\x08\x64\x61ta_uri\x18\x02 \x01(\tH\x00R\x07\x64\x61taUri\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12\x18\n\x07message\x18\x04 \x01(\tR\x07messageB\x0b\n\t_data_uri*\xc5\x01\n\x0e\x45xecutionPhase\x12\x1f\n\x1b\x45XECUTION_PHASE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_PHASE_PARSE\x10\x01\x12\x1b\n\x17\x45XECUTION_PHASE_MAPPING\x10\x02\x12\x1f\n\x1b\x45XECUTION_PHASE_PERSISTENCE\x10\x03\x12\x1f\n\x1b\x45XECUTION_PHASE_AGGREGATION\x10\x04\x12\x18\n\x14\x45XECUTION_PHASE_SINK\x10\x05*\x99\x01\n\x18TestStreamResolverStatus\x12+\n\'TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED\x10\x00\x12\'\n#TEST_STREAM_RESOLVER_STATUS_SUCCESS\x10\x01\x12\'\n#TEST_STREAM_RESOLVER_STATUS_FAILURE\x10\x02\x32\xaf\x02\n\x16SimpleStreamingService\x12\x90\x01\n\x1aSimpleStreamingUnaryInvoke\x12\x35.chalk.streaming.v1.SimpleStreamingUnaryInvokeRequest\x1a\x36.chalk.streaming.v1.SimpleStreamingUnaryInvokeResponse"\x03\x80}\x02\x12\x81\x01\n\x15TestStreamingResolver\x12\x30.chalk.streaming.v1.TestStreamingResolverRequest\x1a\x31.chalk.streaming.v1.TestStreamingResolverResponse"\x03\x80}\x02\x42\x9f\x01\n\x16\x63om.chalk.streaming.v1B\x1bSimpleStreamingServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
+    b'\n1chalk/streaming/v1/simple_streaming_service.proto\x12\x12\x63halk.streaming.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto"\xa0\x01\n\x0eStreamingError\x12\x31\n\x05\x65rror\x18\x01 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x05\x65rror\x12\x38\n\x05phase\x18\x02 \x01(\x0e\x32".chalk.streaming.v1.ExecutionPhaseR\x05phase\x12!\n\x0coperation_id\x18\x03 \x01(\tR\x0boperationId"\xd6\x01\n!SimpleStreamingUnaryInvokeRequest\x12\x34\n\x16streaming_resolver_fqn\x18\x01 \x01(\tR\x14streamingResolverFqn\x12\x1d\n\ninput_data\x18\x02 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x03 \x01(\tH\x00R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x04 \x01(\x08H\x01R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_operation_idB\x08\n\x06_debug"\xe7\x02\n"SimpleStreamingUnaryInvokeResponse\x12(\n\x10num_rows_succeed\x18\x01 \x01(\x03R\x0enumRowsSucceed\x12&\n\x0fnum_rows_failed\x18\x02 \x01(\x03R\rnumRowsFailed\x12(\n\x10num_rows_skipped\x18\x03 \x01(\x03R\x0enumRowsSkipped\x12\x36\n\x05\x65rror\x18\x04 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12$\n\x0boutput_data\x18\x05 \x01(\x0cH\x01R\noutputData\x88\x01\x01\x12M\n\x10\x65xecution_errors\x18\x06 \x03(\x0b\x32".chalk.streaming.v1.StreamingErrorR\x0f\x65xecutionErrorsB\x08\n\x06_errorB\x0e\n\x0c_output_data"\xb5\x02\n\x1cTestStreamingResolverRequest\x12&\n\x0cresolver_fqn\x18\x01 \x01(\tH\x00R\x0bresolverFqn\x88\x01\x01\x12@\n\x1astatic_stream_resolver_b64\x18\x02 \x01(\tH\x01R\x17staticStreamResolverB64\x88\x01\x01\x12\x1d\n\ninput_data\x18\x03 \x01(\x0cR\tinputData\x12&\n\x0coperation_id\x18\x04 \x01(\tH\x02R\x0boperationId\x88\x01\x01\x12\x19\n\x05\x64\x65\x62ug\x18\x05 \x01(\x08H\x03R\x05\x64\x65\x62ug\x88\x01\x01\x42\x0f\n\r_resolver_fqnB\x1d\n\x1b_static_stream_resolver_b64B\x0f\n\r_operation_idB\x08\n\x06_debug"\xe1\x01\n\x1dTestStreamingResolverResponse\x12\x44\n\x06status\x18\x01 \x01(\x0e\x32,.chalk.streaming.v1.TestStreamResolverStatusR\x06status\x12\x1e\n\x08\x64\x61ta_uri\x18\x02 \x01(\tH\x00R\x07\x64\x61taUri\x88\x01\x01\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12\x18\n\x07message\x18\x04 \x01(\tR\x07messageB\x0b\n\t_data_uri*\xc5\x01\n\x0e\x45xecutionPhase\x12\x1f\n\x1b\x45XECUTION_PHASE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_PHASE_PARSE\x10\x01\x12\x1b\n\x17\x45XECUTION_PHASE_MAPPING\x10\x02\x12\x1f\n\x1b\x45XECUTION_PHASE_PERSISTENCE\x10\x03\x12\x1f\n\x1b\x45XECUTION_PHASE_AGGREGATION\x10\x04\x12\x18\n\x14\x45XECUTION_PHASE_SINK\x10\x05*\xe4\x01\n\x16StreamingMessageStatus\x12(\n$STREAMING_MESSAGE_STATUS_UNSPECIFIED\x10\x00\x12)\n%STREAMING_MESSAGE_STATUS_PARSE_FAILED\x10\x01\x12*\n&STREAMING_MESSAGE_STATUS_PARSE_SKIPPED\x10\x02\x12#\n\x1fSTREAMING_MESSAGE_STATUS_FAILED\x10\x03\x12$\n STREAMING_MESSAGE_STATUS_SUCCESS\x10\x04*\x99\x01\n\x18TestStreamResolverStatus\x12+\n\'TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED\x10\x00\x12\'\n#TEST_STREAM_RESOLVER_STATUS_SUCCESS\x10\x01\x12\'\n#TEST_STREAM_RESOLVER_STATUS_FAILURE\x10\x02\x32\xaf\x02\n\x16SimpleStreamingService\x12\x90\x01\n\x1aSimpleStreamingUnaryInvoke\x12\x35.chalk.streaming.v1.SimpleStreamingUnaryInvokeRequest\x1a\x36.chalk.streaming.v1.SimpleStreamingUnaryInvokeResponse"\x03\x80}\x02\x12\x81\x01\n\x15TestStreamingResolver\x12\x30.chalk.streaming.v1.TestStreamingResolverRequest\x1a\x31.chalk.streaming.v1.TestStreamingResolverResponse"\x03\x80}\x02\x42\x9f\x01\n\x16\x63om.chalk.streaming.v1B\x1bSimpleStreamingServiceProtoP\x01\xa2\x02\x03\x43SX\xaa\x02\x12\x43halk.Streaming.V1\xca\x02\x12\x43halk\\Streaming\\V1\xe2\x02\x1e\x43halk\\Streaming\\V1\\GPBMetadata\xea\x02\x14\x43halk::Streaming::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -35,8 +35,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SIMPLESTREAMINGSERVICE"].methods_by_name["TestStreamingResolver"]._serialized_options = b"\200}\002"
     _globals["_EXECUTIONPHASE"]._serialized_start = 1424
     _globals["_EXECUTIONPHASE"]._serialized_end = 1621
-    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_start = 1624
-    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_end = 1777
+    _globals["_STREAMINGMESSAGESTATUS"]._serialized_start = 1624
+    _globals["_STREAMINGMESSAGESTATUS"]._serialized_end = 1852
+    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_start = 1855
+    _globals["_TESTSTREAMRESOLVERSTATUS"]._serialized_end = 2008
     _globals["_STREAMINGERROR"]._serialized_start = 142
     _globals["_STREAMINGERROR"]._serialized_end = 302
     _globals["_SIMPLESTREAMINGUNARYINVOKEREQUEST"]._serialized_start = 305
@@ -47,6 +49,6 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_TESTSTREAMINGRESOLVERREQUEST"]._serialized_end = 1193
     _globals["_TESTSTREAMINGRESOLVERRESPONSE"]._serialized_start = 1196
     _globals["_TESTSTREAMINGRESOLVERRESPONSE"]._serialized_end = 1421
-    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 1780
-    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 2083
+    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_start = 2011
+    _globals["_SIMPLESTREAMINGSERVICE"]._serialized_end = 2314
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi

@@ -23,6 +23,14 @@ class ExecutionPhase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     EXECUTION_PHASE_AGGREGATION: _ClassVar[ExecutionPhase]
     EXECUTION_PHASE_SINK: _ClassVar[ExecutionPhase]
 
+class StreamingMessageStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    STREAMING_MESSAGE_STATUS_UNSPECIFIED: _ClassVar[StreamingMessageStatus]
+    STREAMING_MESSAGE_STATUS_PARSE_FAILED: _ClassVar[StreamingMessageStatus]
+    STREAMING_MESSAGE_STATUS_PARSE_SKIPPED: _ClassVar[StreamingMessageStatus]
+    STREAMING_MESSAGE_STATUS_FAILED: _ClassVar[StreamingMessageStatus]
+    STREAMING_MESSAGE_STATUS_SUCCESS: _ClassVar[StreamingMessageStatus]
+
 class TestStreamResolverStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
     TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED: _ClassVar[TestStreamResolverStatus]
@@ -35,6 +43,11 @@ EXECUTION_PHASE_MAPPING: ExecutionPhase
 EXECUTION_PHASE_PERSISTENCE: ExecutionPhase
 EXECUTION_PHASE_AGGREGATION: ExecutionPhase
 EXECUTION_PHASE_SINK: ExecutionPhase
+STREAMING_MESSAGE_STATUS_UNSPECIFIED: StreamingMessageStatus
+STREAMING_MESSAGE_STATUS_PARSE_FAILED: StreamingMessageStatus
+STREAMING_MESSAGE_STATUS_PARSE_SKIPPED: StreamingMessageStatus
+STREAMING_MESSAGE_STATUS_FAILED: StreamingMessageStatus
+STREAMING_MESSAGE_STATUS_SUCCESS: StreamingMessageStatus
 TEST_STREAM_RESOLVER_STATUS_UNSPECIFIED: TestStreamResolverStatus
 TEST_STREAM_RESOLVER_STATUS_SUCCESS: TestStreamResolverStatus
 TEST_STREAM_RESOLVER_STATUS_FAILURE: TestStreamResolverStatus

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.96.3"
+__version__ = "2.96.5"

chalk/features/resolver.py

@@ -3812,6 +3812,7 @@ def make_stream_resolver(
     owner: Optional[str] = None,
     doc: str | None = None,
     sink: Sink | None = None,
+    additional_output_features: Iterable[FeatureWrapper | str] | None = None,
     skip_online: bool = False,
     skip_offline: bool = False,
 ) -> StreamResolver:
@@ -3846,6 +3847,10 @@ def make_stream_resolver(
     sink
         An optional message producer configuration that specifies where to send messages.
         Read more at https://docs.chalk.ai/api-docs#Sink
+    additional_output_features
+        An optional iterable of additional features to compute and persist to the online store
+        without publishing to an auxiliary stream. Mutually exclusive with `sink`.
+        Use this when you want to enrich features without setting up stream publishing infrastructure.
     skip_online
         If True, skip online persistence (no writes to Redis/DynamoDB/etc).
         Results will still be processed but not stored in online stores.
@@ -3888,6 +3893,18 @@ def make_stream_resolver(
             range=error_builder.function_arg_range_by_name("source"),
         )
 
+    # Validate that sink and additional_output_features are mutually exclusive
+    if sink is not None and additional_output_features is not None:
+        error_builder.add_diagnostic(
+            message=(
+                "Cannot specify both 'sink' and 'additional_output_features'. "
+                + "If you're using Sink, add all desired features to Sink.output_features instead."
+            ),
+            code="208",
+            label="Use Sink.output_features for all features",
+            range=error_builder.function_arg_range_by_name("additional_output_features"),
+        )
+
     # Validate name is a string
     if not isinstance(name, str):  # pyright: ignore[reportUnnecessaryIsInstance]
         error_builder.add_diagnostic(
@@ -3992,12 +4009,16 @@ def make_stream_resolver(
                 f"Native streaming resolvers only support python parse functions with input bytes 'bytes'. Function {parse} has input type {parse_info.input_type}"
             )
 
-    # Validate and parse sink before creating StreamResolver
+    # Validate and parse sink or additional_output_features before creating StreamResolver
     message_producer_parsed: StreamResolverMessageProducerParsed | None = None
     if sink is not None:
         message_producer_parsed = parse_message_producer_with_lsp_errors(
             sink, error_builder, "sink", message_type, name
         )
+    elif additional_output_features is not None:
+        message_producer_parsed = parse_additional_output_features_with_lsp_errors(
+            additional_output_features, error_builder, "additional_output_features", message_type, name
+        )
 
     resolver = StreamResolver(
         function_definition=caller_source,
@@ -4038,25 +4059,22 @@ def make_stream_resolver(
     return resolver
 
 
-@dataclass
+@dataclass(kw_only=True)
 class Sink:
-    """Sends stream messages to a stream source of your choice.
-
-    Messages are arrow tables serialized in IPC stream format with the specified output columns
-    """
+    """Computes additional output features and sends them to a stream source."""
 
     send_to: StreamSource
-    """The source to send results to"""
+    """The stream source to send results to."""
 
     output_features: Iterable[FeatureWrapper | str]
     """
     The requested output features. These can include both features returned by the stream resolvers and
-    other features retrievable via online query given the stream resolver outputs as inputs. The message sent to
-    the stream will have these output features as columns
+    other features retrievable via online query given the stream resolver outputs as inputs.
+    The message sent to the stream will have these output features as columns.
     """
 
     format: Literal["json", "ipc_stream"] = "ipc_stream"
-    """Format of messages sent"""
+    """Format of messages sent."""
 
     feature_expressions: Mapping[FeatureWrapper | str, Underscore] | None = None
 
@@ -4184,10 +4202,80 @@ def parse_message_producer_with_lsp_errors(
     )
 
 
+def parse_additional_output_features_with_lsp_errors(
+    output_features_input: Iterable[FeatureWrapper | str],
+    error_builder: FunctionCallErrorBuilder,
+    param_name: str,
+    message_type: Type[BaseModel | google.protobuf.message.Message | AnyDataclass | str | bytes],
+    resolver_name: str,
+) -> StreamResolverMessageProducerParsed | None:
+    """Parse additional_output_features into StreamResolverMessageProducerParsed.
+
+    Similar to parse_message_producer_with_lsp_errors but for additional_output_features
+    parameter, which doesn't require a send_to destination.
+    """
+    # Validate output_features is iterable
+    if not isinstance(output_features_input, collections.abc.Iterable):  # pyright: ignore[reportUnnecessaryIsInstance]
+        error_builder.add_diagnostic(
+            message="Expected an iterable for argument 'additional_output_features'",
+            code="209",
+            label="Invalid additional_output_features type",
+            range=error_builder.function_arg_range_by_name(param_name),
+        )
+        return None
+
+    output_features_list = list(output_features_input)
+    if len(output_features_list) == 0:
+        error_builder.add_diagnostic(
+            message="Expected at least one feature in 'additional_output_features'",
+            code="210",
+            label="Empty additional_output_features",
+            range=error_builder.function_arg_range_by_name(param_name),
+        )
+        return None
+
+    # Validate and unwrap features (same logic as parse_message_producer_with_lsp_errors)
+    unwrapped_features: list[Feature] = []
+    for f in output_features_list:
+        if isinstance(f, FeatureWrapper):
+            unwrapped_features.append(unwrap_feature(f))
+        elif isinstance(f, str):  # pyright: ignore[reportUnnecessaryIsInstance]
+            try:
+                feature = Feature.from_root_fqn(f)
+                unwrapped_features.append(feature)
+            except Exception:
+                error_builder.add_diagnostic(
+                    message=f"Invalid feature FQN '{f}' in additional_output_features",
+                    code="211",
+                    label="Invalid feature FQN",
+                    range=error_builder.function_arg_range_by_name(param_name),
+                )
+                return None
+        else:
+            error_builder.add_diagnostic(
+                message=f"Feature '{f}' must be a Feature or string, got {type(f)} instead",
+                code="212",
+                label="Invalid feature type",
+                range=error_builder.function_arg_range_by_name(param_name),
+            )
+            return None
+
+    # Reuse existing validation
+    _validate_output_features(unwrapped_features, error_builder, resolver_name)
+
+    # Create StreamResolverMessageProducerParsed with send_to=None
+    return StreamResolverMessageProducerParsed(
+        output_features=[str(f) for f in output_features_list],
+        format="ipc_stream",
+        send_to=None,
+        feature_expressions=None,
+    )
+
+
 class StreamResolverMessageProducerParsed:
     def __init__(
         self,
-        send_to: StreamSource,
+        send_to: StreamSource | None,
         output_features: list[str],
         format: Literal["json", "ipc_stream"],
         feature_expressions: Mapping[str, Underscore] | None = None,

chalk/functions/__init__.py

@@ -391,6 +391,40 @@ def substr(expr: Underscore | Any, start: int, length: int | None = None):
     return UnderscoreFunction("substr", expr, start + 1, length)
 
 
+def str_slice(expr: Underscore | Any, start: int, end: int):
+    """
+    Extract a substring from a string using start and end indices.
+
+    This is a convenience function that wraps `substr` with Python-style slicing semantics,
+    where you provide start and end indices rather than start and length.
+
+    Parameters
+    ----------
+    expr
+        The string expression to slice.
+    start
+        The starting index (0-based, inclusive).
+    end
+        The ending index (0-based, exclusive).
+
+    Returns
+    -------
+    A substring from index `start` to `end` (exclusive).
+
+    Examples
+    --------
+    >>> import chalk.functions as F
+    >>> from chalk.features import _, features
+    >>> @features
+    ... class User:
+    ...    id: str
+    ...    name: str
+    ...    first_three: str = F.str_slice(_.name, 0, 3)  # Extract first 3 characters
+    ...    middle: str = F.str_slice(_.name, 2, 5)  # Extract characters from index 2 to 4
+    """
+    return substr(expr, start, end - start)
+
+
 def reverse(expr: Underscore | Any):
     """
     Reverse the order of a string.
@@ -926,6 +960,38 @@ def strrpos(expr: Any, substring: Any):
     return UnderscoreFunction("strrpos", expr, substring) - 1
 
 
+def str_contains(expr: Any, substring: Any):
+    """
+    Check if a string contains a substring.
+
+    This is a convenience function that checks whether `substring` appears anywhere
+    within the string `expr`. It returns a boolean value.
+
+    Parameters
+    ----------
+    expr
+        The string expression to search within.
+    substring
+        The substring to search for.
+
+    Returns
+    -------
+    Boolean indicating whether `substring` is found in `expr`.
+
+    Examples
+    --------
+    >>> import chalk.functions as F
+    >>> from chalk.features import _, features
+    >>> @features
+    ... class User:
+    ...    id: str
+    ...    email: str
+    ...    is_gmail: bool = F.str_contains(_.email, "@gmail.com")
+    ...    has_admin: bool = F.str_contains(_.email, "admin")
+    """
+    return strpos(expr, substring) >= 0
+
+
 def chr(code: Underscore | Any):
     """
     Convert Unicode code point to character.
@@ -5988,6 +6054,7 @@ __all__ = (
     "h3_cell_to_lat_lon",
     "if_then_else",
     "inference",
+    "is_in",
     "is_leap_year",
     "is_month_end",
     "is_not_null",
@@ -6078,6 +6145,8 @@ __all__ = (
     "strpos",
     "strrpos",
     "struct_pack",
+    "str_contains",
+    "str_slice",
     "substr",
     "to_base",
     "to_iso8601",

chalk/parsed/to_proto.py

@@ -1337,7 +1337,9 @@ class ToProtoConverter:
         message_producer: pb.StreamResolverMessageProducerParsed | None = None
         if r.message_producer_parsed is not None:
             message_producer = pb.StreamResolverMessageProducerParsed(
-                send_to=ToProtoConverter.create_stream_source_reference(r.message_producer_parsed.send_to),
+                send_to=ToProtoConverter.create_stream_source_reference(r.message_producer_parsed.send_to)
+                if r.message_producer_parsed.send_to is not None
+                else None,
                 output_features=r.message_producer_parsed.output_features,
                 transformations={
                     str(k): pb.FeatureExpression(underscore_expr=cls.convert_underscore(v))

{chalkpy-2.96.3.dist-info → chalkpy-2.96.5.dist-info}/METADATA

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