chalkpy 2.93.2__py3-none-any.whl → 2.93.3__py3-none-any.whl

chalk/_gen/chalk/server/v1/integrations_pb2.py

@@ -20,7 +20,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n"chalk/server/v1/integrations.proto\x12\x0f\x63halk.server.v1\x1a\x19\x63halk/auth/v1/audit.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a)chalk/server/v1/environment_secrets.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x92\x02\n\x0bIntegration\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x17\n\x04name\x18\x02 \x01(\tH\x00R\x04name\x88\x01\x01\x12\x34\n\x04kind\x18\x03 \x01(\x0e\x32 .chalk.server.v1.IntegrationKindR\x04kind\x12%\n\x0e\x65nvironment_id\x18\x04 \x01(\tR\renvironmentId\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\tupdatedAtB\x07\n\x05_name"\x94\x01\n\x16IntegrationWithSecrets\x12>\n\x0bintegration\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0bintegration\x12:\n\x07secrets\x18\x02 \x03(\x0b\x32 .chalk.server.v1.SecretWithValueR\x07secrets"\x19\n\x17ListIntegrationsRequest"\\\n\x18ListIntegrationsResponse\x12@\n\x0cintegrations\x18\x01 \x03(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0cintegrations"=\n!ListIntegrationsAndSecretsRequest\x12\x18\n\x07\x64\x65\x63rypt\x18\x01 \x01(\x08R\x07\x64\x65\x63rypt"\xba\x01\n"ListIntegrationsAndSecretsResponse\x12K\n\x0cintegrations\x18\x01 \x03(\x0b\x32\'.chalk.server.v1.IntegrationWithSecretsR\x0cintegrations\x12G\n\x0e\x63ustom_secrets\x18\x02 \x03(\x0b\x32 .chalk.server.v1.SecretWithValueR\rcustomSecrets"d\n\x1aGetIntegrationValueRequest\x12%\n\x0eintegration_id\x18\x01 \x01(\tR\rintegrationId\x12\x1f\n\x0bsecret_name\x18\x02 \x01(\tR\nsecretName"r\n\x1bGetIntegrationValueResponse\x12\x43\n\x0bsecretvalue\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.SecretValueH\x00R\x0bsecretvalue\x88\x01\x01\x42\x0e\n\x0c_secretvalue">\n\x15GetIntegrationRequest\x12%\n\x0eintegration_id\x18\x01 \x01(\tR\rintegrationId"\x9d\x01\n\x16GetIntegrationResponse\x12\x66\n\x18integration_with_secrets\x18\x01 \x01(\x0b\x32\'.chalk.server.v1.IntegrationWithSecretsH\x00R\x16integrationWithSecrets\x88\x01\x01\x42\x1b\n\x19_integration_with_secrets"\xbe\x02\n\x18InsertIntegrationRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12K\n\x10integration_kind\x18\x02 \x01(\x0e\x32 .chalk.server.v1.IntegrationKindR\x0fintegrationKind\x12x\n\x15\x65nvironment_variables\x18\x03 \x03(\x0b\x32\x43.chalk.server.v1.InsertIntegrationRequest.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"[\n\x19InsertIntegrationResponse\x12>\n\x0bintegration\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0bintegration"\x98\x02\n\x18UpdateIntegrationRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12%\n\x0eintegration_id\x18\x02 \x01(\tR\rintegrationId\x12x\n\x15\x65nvironment_variables\x18\x03 \x03(\x0b\x32\x43.chalk.server.v1.UpdateIntegrationRequest.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"[\n\x19UpdateIntegrationResponse\x12>\n\x0bintegration\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0bintegration"*\n\x18\x44\x65leteIntegrationRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"\x1b\n\x19\x44\x65leteIntegrationResponse"\xce\x02\n\x16TestIntegrationRequest\x12\x34\n\x04kind\x18\x01 \x01(\x0e\x32 .chalk.server.v1.IntegrationKindR\x04kind\x12v\n\x15\x65nvironment_variables\x18\x02 \x03(\x0b\x32\x41.chalk.server.v1.TestIntegrationRequest.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x12*\n\x0eintegration_id\x18\x03 \x01(\tH\x00R\rintegrationId\x88\x01\x01\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_integration_id"\xa3\x01\n\x17TestIntegrationResponse\x12\x12\n\x04kind\x18\x01 \x01(\tR\x04kind\x12\x18\n\x07success\x18\x02 \x01(\x08R\x07success\x12\x18\n\x07message\x18\x03 \x01(\tR\x07message\x12,\n\x0flatency_seconds\x18\x04 \x01(\x01H\x00R\x0elatencySeconds\x88\x01\x01\x42\x12\n\x10_latency_seconds*\xe7\x04\n\x0fIntegrationKind\x12 \n\x1cINTEGRATION_KIND_UNSPECIFIED\x10\x00\x12\x1b\n\x17INTEGRATION_KIND_ATHENA\x10\x01\x12\x18\n\x14INTEGRATION_KIND_AWS\x10\x02\x12\x1d\n\x19INTEGRATION_KIND_BIGQUERY\x10\x03\x12\x1f\n\x1bINTEGRATION_KIND_CLICKHOUSE\x10\x04\x12\x1b\n\x17INTEGRATION_KIND_COHERE\x10\x05\x12\x1f\n\x1bINTEGRATION_KIND_DATABRICKS\x10\x06\x12\x1d\n\x19INTEGRATION_KIND_DYNAMODB\x10\x07\x12\x18\n\x14INTEGRATION_KIND_GCP\x10\x08\x12\x1a\n\x16INTEGRATION_KIND_KAFKA\x10\t\x12\x1c\n\x18INTEGRATION_KIND_KINESIS\x10\n\x12\x1a\n\x16INTEGRATION_KIND_MYSQL\x10\x0b\x12\x1b\n\x17INTEGRATION_KIND_OPENAI\x10\x0c\x12\x1f\n\x1bINTEGRATION_KIND_POSTGRESQL\x10\r\x12\x1b\n\x17INTEGRATION_KIND_PUBSUB\x10\x0e\x12\x1d\n\x19INTEGRATION_KIND_REDSHIFT\x10\x0f\x12\x1e\n\x1aINTEGRATION_KIND_SNOWFLAKE\x10\x10\x12\x1c\n\x18INTEGRATION_KIND_SPANNER\x10\x11\x12\x1a\n\x16INTEGRATION_KIND_TRINO\x10\x12\x12\x1a\n\x16INTEGRATION_KIND_MSSQL\x10\x13\x32\xbf\x07\n\x13IntegrationsService\x12l\n\x10ListIntegrations\x12(.chalk.server.v1.ListIntegrationsRequest\x1a).chalk.server.v1.ListIntegrationsResponse"\x03\x80}\x14\x12\x8a\x01\n\x1aListIntegrationsAndSecrets\x12\x32.chalk.server.v1.ListIntegrationsAndSecretsRequest\x1a\x33.chalk.server.v1.ListIntegrationsAndSecretsResponse"\x03\x80}\x13\x12u\n\x13GetIntegrationValue\x12+.chalk.server.v1.GetIntegrationValueRequest\x1a,.chalk.server.v1.GetIntegrationValueResponse"\x03\x80}\x13\x12\x66\n\x0eGetIntegration\x12&.chalk.server.v1.GetIntegrationRequest\x1a\'.chalk.server.v1.GetIntegrationResponse"\x03\x80}\x13\x12u\n\x11InsertIntegration\x12).chalk.server.v1.InsertIntegrationRequest\x1a*.chalk.server.v1.InsertIntegrationResponse"\t\x80}\x12\x8a\xd3\x0e\x02\x08\x02\x12u\n\x11UpdateIntegration\x12).chalk.server.v1.UpdateIntegrationRequest\x1a*.chalk.server.v1.UpdateIntegrationResponse"\t\x80}\x12\x8a\xd3\x0e\x02\x08\x02\x12u\n\x11\x44\x65leteIntegration\x12).chalk.server.v1.DeleteIntegrationRequest\x1a*.chalk.server.v1.DeleteIntegrationResponse"\t\x80}\x12\x8a\xd3\x0e\x02\x08\x02\x12i\n\x0fTestIntegration\x12\'.chalk.server.v1.TestIntegrationRequest\x1a(.chalk.server.v1.TestIntegrationResponse"\x03\x80}\x14\x42\x9a\x01\n\x13\x63om.chalk.server.v1B\x11IntegrationsProtoP\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/integrations.proto\x12\x0f\x63halk.server.v1\x1a\x19\x63halk/auth/v1/audit.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a)chalk/server/v1/environment_secrets.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x92\x02\n\x0bIntegration\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x17\n\x04name\x18\x02 \x01(\tH\x00R\x04name\x88\x01\x01\x12\x34\n\x04kind\x18\x03 \x01(\x0e\x32 .chalk.server.v1.IntegrationKindR\x04kind\x12%\n\x0e\x65nvironment_id\x18\x04 \x01(\tR\renvironmentId\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\tupdatedAtB\x07\n\x05_name"\x94\x01\n\x16IntegrationWithSecrets\x12>\n\x0bintegration\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0bintegration\x12:\n\x07secrets\x18\x02 \x03(\x0b\x32 .chalk.server.v1.SecretWithValueR\x07secrets"\x19\n\x17ListIntegrationsRequest"\\\n\x18ListIntegrationsResponse\x12@\n\x0cintegrations\x18\x01 \x03(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0cintegrations"=\n!ListIntegrationsAndSecretsRequest\x12\x18\n\x07\x64\x65\x63rypt\x18\x01 \x01(\x08R\x07\x64\x65\x63rypt"\xba\x01\n"ListIntegrationsAndSecretsResponse\x12K\n\x0cintegrations\x18\x01 \x03(\x0b\x32\'.chalk.server.v1.IntegrationWithSecretsR\x0cintegrations\x12G\n\x0e\x63ustom_secrets\x18\x02 \x03(\x0b\x32 .chalk.server.v1.SecretWithValueR\rcustomSecrets"d\n\x1aGetIntegrationValueRequest\x12%\n\x0eintegration_id\x18\x01 \x01(\tR\rintegrationId\x12\x1f\n\x0bsecret_name\x18\x02 \x01(\tR\nsecretName"r\n\x1bGetIntegrationValueResponse\x12\x43\n\x0bsecretvalue\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.SecretValueH\x00R\x0bsecretvalue\x88\x01\x01\x42\x0e\n\x0c_secretvalue">\n\x15GetIntegrationRequest\x12%\n\x0eintegration_id\x18\x01 \x01(\tR\rintegrationId"\x9d\x01\n\x16GetIntegrationResponse\x12\x66\n\x18integration_with_secrets\x18\x01 \x01(\x0b\x32\'.chalk.server.v1.IntegrationWithSecretsH\x00R\x16integrationWithSecrets\x88\x01\x01\x42\x1b\n\x19_integration_with_secrets"\xbe\x02\n\x18InsertIntegrationRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12K\n\x10integration_kind\x18\x02 \x01(\x0e\x32 .chalk.server.v1.IntegrationKindR\x0fintegrationKind\x12x\n\x15\x65nvironment_variables\x18\x03 \x03(\x0b\x32\x43.chalk.server.v1.InsertIntegrationRequest.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"[\n\x19InsertIntegrationResponse\x12>\n\x0bintegration\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0bintegration"\x98\x02\n\x18UpdateIntegrationRequest\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12%\n\x0eintegration_id\x18\x02 \x01(\tR\rintegrationId\x12x\n\x15\x65nvironment_variables\x18\x03 \x03(\x0b\x32\x43.chalk.server.v1.UpdateIntegrationRequest.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"[\n\x19UpdateIntegrationResponse\x12>\n\x0bintegration\x18\x01 \x01(\x0b\x32\x1c.chalk.server.v1.IntegrationR\x0bintegration"*\n\x18\x44\x65leteIntegrationRequest\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id"\x1b\n\x19\x44\x65leteIntegrationResponse"\xd7\x01\n\x10PreviewedMessage\x12!\n\x0cvalue_base64\x18\x01 \x01(\tR\x0bvalueBase64\x12"\n\nkey_base64\x18\x02 \x01(\tH\x00R\tkeyBase64\x88\x01\x01\x12\x14\n\x05topic\x18\x03 \x01(\tR\x05topic\x12\x1c\n\tpartition\x18\x04 \x01(\tR\tpartition\x12\x16\n\x06offset\x18\x05 \x01(\tR\x06offset\x12!\n\x0ctimestamp_ms\x18\x06 \x01(\x03R\x0btimestampMsB\r\n\x0b_key_base64"\x90\x03\n\x16TestIntegrationRequest\x12\x34\n\x04kind\x18\x01 \x01(\x0e\x32 .chalk.server.v1.IntegrationKindR\x04kind\x12v\n\x15\x65nvironment_variables\x18\x02 \x03(\x0b\x32\x41.chalk.server.v1.TestIntegrationRequest.EnvironmentVariablesEntryR\x14\x65nvironmentVariables\x12*\n\x0eintegration_id\x18\x03 \x01(\tH\x00R\rintegrationId\x88\x01\x01\x12,\n\x0finclude_preview\x18\x04 \x01(\x08H\x01R\x0eincludePreview\x88\x01\x01\x1aG\n\x19\x45nvironmentVariablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_integration_idB\x12\n\x10_include_preview"\xf1\x01\n\x17TestIntegrationResponse\x12\x12\n\x04kind\x18\x01 \x01(\tR\x04kind\x12\x18\n\x07success\x18\x02 \x01(\x08R\x07success\x12\x18\n\x07message\x18\x03 \x01(\tR\x07message\x12,\n\x0flatency_seconds\x18\x04 \x01(\x01H\x00R\x0elatencySeconds\x88\x01\x01\x12L\n\x10preview_messages\x18\x05 \x03(\x0b\x32!.chalk.server.v1.PreviewedMessageR\x0fpreviewMessagesB\x12\n\x10_latency_seconds*\xe7\x04\n\x0fIntegrationKind\x12 \n\x1cINTEGRATION_KIND_UNSPECIFIED\x10\x00\x12\x1b\n\x17INTEGRATION_KIND_ATHENA\x10\x01\x12\x18\n\x14INTEGRATION_KIND_AWS\x10\x02\x12\x1d\n\x19INTEGRATION_KIND_BIGQUERY\x10\x03\x12\x1f\n\x1bINTEGRATION_KIND_CLICKHOUSE\x10\x04\x12\x1b\n\x17INTEGRATION_KIND_COHERE\x10\x05\x12\x1f\n\x1bINTEGRATION_KIND_DATABRICKS\x10\x06\x12\x1d\n\x19INTEGRATION_KIND_DYNAMODB\x10\x07\x12\x18\n\x14INTEGRATION_KIND_GCP\x10\x08\x12\x1a\n\x16INTEGRATION_KIND_KAFKA\x10\t\x12\x1c\n\x18INTEGRATION_KIND_KINESIS\x10\n\x12\x1a\n\x16INTEGRATION_KIND_MYSQL\x10\x0b\x12\x1b\n\x17INTEGRATION_KIND_OPENAI\x10\x0c\x12\x1f\n\x1bINTEGRATION_KIND_POSTGRESQL\x10\r\x12\x1b\n\x17INTEGRATION_KIND_PUBSUB\x10\x0e\x12\x1d\n\x19INTEGRATION_KIND_REDSHIFT\x10\x0f\x12\x1e\n\x1aINTEGRATION_KIND_SNOWFLAKE\x10\x10\x12\x1c\n\x18INTEGRATION_KIND_SPANNER\x10\x11\x12\x1a\n\x16INTEGRATION_KIND_TRINO\x10\x12\x12\x1a\n\x16INTEGRATION_KIND_MSSQL\x10\x13\x32\xbf\x07\n\x13IntegrationsService\x12l\n\x10ListIntegrations\x12(.chalk.server.v1.ListIntegrationsRequest\x1a).chalk.server.v1.ListIntegrationsResponse"\x03\x80}\x14\x12\x8a\x01\n\x1aListIntegrationsAndSecrets\x12\x32.chalk.server.v1.ListIntegrationsAndSecretsRequest\x1a\x33.chalk.server.v1.ListIntegrationsAndSecretsResponse"\x03\x80}\x13\x12u\n\x13GetIntegrationValue\x12+.chalk.server.v1.GetIntegrationValueRequest\x1a,.chalk.server.v1.GetIntegrationValueResponse"\x03\x80}\x13\x12\x66\n\x0eGetIntegration\x12&.chalk.server.v1.GetIntegrationRequest\x1a\'.chalk.server.v1.GetIntegrationResponse"\x03\x80}\x13\x12u\n\x11InsertIntegration\x12).chalk.server.v1.InsertIntegrationRequest\x1a*.chalk.server.v1.InsertIntegrationResponse"\t\x80}\x12\x8a\xd3\x0e\x02\x08\x02\x12u\n\x11UpdateIntegration\x12).chalk.server.v1.UpdateIntegrationRequest\x1a*.chalk.server.v1.UpdateIntegrationResponse"\t\x80}\x12\x8a\xd3\x0e\x02\x08\x02\x12u\n\x11\x44\x65leteIntegration\x12).chalk.server.v1.DeleteIntegrationRequest\x1a*.chalk.server.v1.DeleteIntegrationResponse"\t\x80}\x12\x8a\xd3\x0e\x02\x08\x02\x12i\n\x0fTestIntegration\x12\'.chalk.server.v1.TestIntegrationRequest\x1a(.chalk.server.v1.TestIntegrationResponse"\x03\x80}\x14\x42\x9a\x01\n\x13\x63om.chalk.server.v1B\x11IntegrationsProtoP\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()
@@ -59,8 +59,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     ]._serialized_options = b"\200}\022\212\323\016\002\010\002"
     _globals["_INTEGRATIONSSERVICE"].methods_by_name["TestIntegration"]._options = None
     _globals["_INTEGRATIONSSERVICE"].methods_by_name["TestIntegration"]._serialized_options = b"\200}\024"
-    _globals["_INTEGRATIONKIND"]._serialized_start = 2801
-    _globals["_INTEGRATIONKIND"]._serialized_end = 3416
+    _globals["_INTEGRATIONKIND"]._serialized_start = 3163
+    _globals["_INTEGRATIONKIND"]._serialized_end = 3778
     _globals["_INTEGRATION"]._serialized_start = 192
     _globals["_INTEGRATION"]._serialized_end = 466
     _globals["_INTEGRATIONWITHSECRETS"]._serialized_start = 469
@@ -97,12 +97,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_DELETEINTEGRATIONREQUEST"]._serialized_end = 2266
     _globals["_DELETEINTEGRATIONRESPONSE"]._serialized_start = 2268
     _globals["_DELETEINTEGRATIONRESPONSE"]._serialized_end = 2295
-    _globals["_TESTINTEGRATIONREQUEST"]._serialized_start = 2298
-    _globals["_TESTINTEGRATIONREQUEST"]._serialized_end = 2632
+    _globals["_PREVIEWEDMESSAGE"]._serialized_start = 2298
+    _globals["_PREVIEWEDMESSAGE"]._serialized_end = 2513
+    _globals["_TESTINTEGRATIONREQUEST"]._serialized_start = 2516
+    _globals["_TESTINTEGRATIONREQUEST"]._serialized_end = 2916
     _globals["_TESTINTEGRATIONREQUEST_ENVIRONMENTVARIABLESENTRY"]._serialized_start = 1682
     _globals["_TESTINTEGRATIONREQUEST_ENVIRONMENTVARIABLESENTRY"]._serialized_end = 1753
-    _globals["_TESTINTEGRATIONRESPONSE"]._serialized_start = 2635
-    _globals["_TESTINTEGRATIONRESPONSE"]._serialized_end = 2798
-    _globals["_INTEGRATIONSSERVICE"]._serialized_start = 3419
-    _globals["_INTEGRATIONSSERVICE"]._serialized_end = 4378
+    _globals["_TESTINTEGRATIONRESPONSE"]._serialized_start = 2919
+    _globals["_TESTINTEGRATIONRESPONSE"]._serialized_end = 3160
+    _globals["_INTEGRATIONSSERVICE"]._serialized_start = 3781
+    _globals["_INTEGRATIONSSERVICE"]._serialized_end = 4740
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/server/v1/integrations_pb2.pyi

@@ -220,8 +220,32 @@ class DeleteIntegrationResponse(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
 
+class PreviewedMessage(_message.Message):
+    __slots__ = ("value_base64", "key_base64", "topic", "partition", "offset", "timestamp_ms")
+    VALUE_BASE64_FIELD_NUMBER: _ClassVar[int]
+    KEY_BASE64_FIELD_NUMBER: _ClassVar[int]
+    TOPIC_FIELD_NUMBER: _ClassVar[int]
+    PARTITION_FIELD_NUMBER: _ClassVar[int]
+    OFFSET_FIELD_NUMBER: _ClassVar[int]
+    TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int]
+    value_base64: str
+    key_base64: str
+    topic: str
+    partition: str
+    offset: str
+    timestamp_ms: int
+    def __init__(
+        self,
+        value_base64: _Optional[str] = ...,
+        key_base64: _Optional[str] = ...,
+        topic: _Optional[str] = ...,
+        partition: _Optional[str] = ...,
+        offset: _Optional[str] = ...,
+        timestamp_ms: _Optional[int] = ...,
+    ) -> None: ...
+
 class TestIntegrationRequest(_message.Message):
-    __slots__ = ("kind", "environment_variables", "integration_id")
+    __slots__ = ("kind", "environment_variables", "integration_id", "include_preview")
     class EnvironmentVariablesEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -233,30 +257,36 @@ class TestIntegrationRequest(_message.Message):
     KIND_FIELD_NUMBER: _ClassVar[int]
     ENVIRONMENT_VARIABLES_FIELD_NUMBER: _ClassVar[int]
     INTEGRATION_ID_FIELD_NUMBER: _ClassVar[int]
+    INCLUDE_PREVIEW_FIELD_NUMBER: _ClassVar[int]
     kind: IntegrationKind
     environment_variables: _containers.ScalarMap[str, str]
     integration_id: str
+    include_preview: bool
     def __init__(
         self,
         kind: _Optional[_Union[IntegrationKind, str]] = ...,
         environment_variables: _Optional[_Mapping[str, str]] = ...,
         integration_id: _Optional[str] = ...,
+        include_preview: bool = ...,
     ) -> None: ...
 
 class TestIntegrationResponse(_message.Message):
-    __slots__ = ("kind", "success", "message", "latency_seconds")
+    __slots__ = ("kind", "success", "message", "latency_seconds", "preview_messages")
     KIND_FIELD_NUMBER: _ClassVar[int]
     SUCCESS_FIELD_NUMBER: _ClassVar[int]
     MESSAGE_FIELD_NUMBER: _ClassVar[int]
     LATENCY_SECONDS_FIELD_NUMBER: _ClassVar[int]
+    PREVIEW_MESSAGES_FIELD_NUMBER: _ClassVar[int]
     kind: str
     success: bool
     message: str
     latency_seconds: float
+    preview_messages: _containers.RepeatedCompositeFieldContainer[PreviewedMessage]
     def __init__(
         self,
         kind: _Optional[str] = ...,
         success: bool = ...,
         message: _Optional[str] = ...,
         latency_seconds: _Optional[float] = ...,
+        preview_messages: _Optional[_Iterable[_Union[PreviewedMessage, _Mapping]]] = ...,
     ) -> None: ...

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"\xfe\x05\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\x42\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"\xad\x04\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\x00R\nmaxSamples\x88\x01\x01\x12j\n\renv_overrides\x18\x05 \x03(\x0b\x32\x45.chalk.server.v1.ManualTriggerScheduledQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\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\x0e\n\x0c_max_samples"y\n#ManualTriggerScheduledQueryResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun*\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\xac\x02\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\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"\xfe\x05\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\x42\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"\xf0\x04\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\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*\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\xac\x02\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\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()
@@ -40,8 +40,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerCronResolver"]._serialized_options = b"\200}\021"
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerScheduledQuery"]._options = None
     _globals["_SCHEDULERSERVICE"].methods_by_name["ManualTriggerScheduledQuery"]._serialized_options = b"\200}\021"
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 2204
-    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 2360
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_start = 2271
+    _globals["_CRONRUNTRIGGERKIND"]._serialized_end = 2427
     _globals["_CRONRESOLVERRUN"]._serialized_start = 221
     _globals["_CRONRESOLVERRUN"]._serialized_end = 987
     _globals["_MANUALTRIGGERCRONRESOLVERREQUEST"]._serialized_start = 990
@@ -49,13 +49,13 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_MANUALTRIGGERCRONRESOLVERRESPONSE"]._serialized_start = 1405
     _globals["_MANUALTRIGGERCRONRESOLVERRESPONSE"]._serialized_end = 1518
     _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST"]._serialized_start = 1521
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST"]._serialized_end = 2078
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_start = 1908
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_end = 1997
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 1999
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 2062
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_start = 2080
-    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_end = 2201
-    _globals["_SCHEDULERSERVICE"]._serialized_start = 2363
-    _globals["_SCHEDULERSERVICE"]._serialized_end = 2663
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST"]._serialized_end = 2145
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_start = 1950
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_end = 2039
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 2041
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 2104
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_start = 2147
+    _globals["_MANUALTRIGGERSCHEDULEDQUERYRESPONSE"]._serialized_end = 2268
+    _globals["_SCHEDULERSERVICE"]._serialized_start = 2430
+    _globals["_SCHEDULERSERVICE"]._serialized_end = 2730
 # @@protoc_insertion_point(module_scope)

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

@@ -123,7 +123,14 @@ class ManualTriggerCronResolverResponse(_message.Message):
     def __init__(self, cron_resolver_run: _Optional[_Union[CronResolverRun, _Mapping]] = ...) -> None: ...
 
 class ManualTriggerScheduledQueryRequest(_message.Message):
-    __slots__ = ("cron_query_id", "planner_options", "incremental_resolvers", "max_samples", "env_overrides")
+    __slots__ = (
+        "cron_query_id",
+        "planner_options",
+        "incremental_resolvers",
+        "max_samples",
+        "env_overrides",
+        "cron_query_name",
+    )
     class PlannerOptionsEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -147,11 +154,13 @@ class ManualTriggerScheduledQueryRequest(_message.Message):
     INCREMENTAL_RESOLVERS_FIELD_NUMBER: _ClassVar[int]
     MAX_SAMPLES_FIELD_NUMBER: _ClassVar[int]
     ENV_OVERRIDES_FIELD_NUMBER: _ClassVar[int]
+    CRON_QUERY_NAME_FIELD_NUMBER: _ClassVar[int]
     cron_query_id: int
     planner_options: _containers.MessageMap[str, _struct_pb2.Value]
     incremental_resolvers: _containers.RepeatedScalarFieldContainer[str]
     max_samples: int
     env_overrides: _containers.ScalarMap[str, str]
+    cron_query_name: str
     def __init__(
         self,
         cron_query_id: _Optional[int] = ...,
@@ -159,6 +168,7 @@ class ManualTriggerScheduledQueryRequest(_message.Message):
         incremental_resolvers: _Optional[_Iterable[str]] = ...,
         max_samples: _Optional[int] = ...,
         env_overrides: _Optional[_Mapping[str, str]] = ...,
+        cron_query_name: _Optional[str] = ...,
     ) -> None: ...
 
 class ManualTriggerScheduledQueryResponse(_message.Message):

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.93.2"
+__version__ = "2.93.3"

chalk/client/client.py

@@ -19,6 +19,7 @@ from typing import (
 
 import requests
 
+from chalk._gen.chalk.server.v1.scheduler_pb2 import ManualTriggerScheduledQueryResponse
 from chalk.client.models import (
     BranchDeployResponse,
     BranchIdParam,
@@ -1085,6 +1086,52 @@ class ChalkClient:
         """
         ...
 
+    def run_scheduled_query(
+        self,
+        name: str,
+        planner_options: Optional[Mapping[str, Any]],
+        incremental_resolvers: Optional[Sequence[str]],
+        max_samples: Optional[int],
+        env_overrides: Optional[Mapping[str, str]],
+    ) -> ManualTriggerScheduledQueryResponse:
+        """
+        Manually trigger a scheduled query request.
+
+        Parameters
+        ----------
+        name
+            The name of the scheduled query to be triggered.
+        incremental_resolvers
+            If set to None, Chalk will incrementalize resolvers in the query's root namespaces.
+            If set to a list of resolvers, this set will be used for incrementalization.
+            Incremental resolvers must return a feature time in its output, and must return a `DataFrame`.
+            Most commonly, this will be the name of a SQL file resolver. Chalk will ingest all new data
+            from these resolvers and propagate changes to values in the root namespace.
+        max_samples
+            The maximum number of samples to compute.
+        env_overrides:
+            A dictionary of environment values to override during this specific triggered query.
+
+        Other Parameters
+        ----------------
+        planner_options
+            A dictionary of options to pass to the planner.
+            These are typically provided by Chalk Support for specific use cases.
+
+        Returns
+        -------
+        ManualTriggerScheduledQueryResponse
+            A response message containing metadata around the triggered run.
+
+        Examples
+        --------
+        >>> from chalk.client.client_grpc import ChalkGRPCClient
+        >>> ChalkGRPCClient().run_scheduled_query(
+        ...     name="my_scheduled_query",
+        ... )
+        """
+        ...
+
     def prompt_evaluation(
         self,
         prompts: list[Prompt | str],

chalk/client/client_grpc.py

@@ -74,6 +74,11 @@ from chalk._gen.chalk.server.v1.model_registry_pb2 import (
 )
 from chalk._gen.chalk.server.v1.model_registry_pb2_grpc import ModelRegistryServiceStub
 from chalk._gen.chalk.server.v1.offline_queries_pb2_grpc import OfflineQueryMetadataServiceStub
+from chalk._gen.chalk.server.v1.scheduler_pb2 import (
+    ManualTriggerScheduledQueryRequest,
+    ManualTriggerScheduledQueryResponse,
+)
+from chalk._gen.chalk.server.v1.scheduler_pb2_grpc import SchedulerServiceStub
 from chalk._gen.chalk.server.v1.script_tasks_pb2 import CreateScriptTaskRequest, CreateScriptTaskResponse
 from chalk._gen.chalk.server.v1.script_tasks_pb2_grpc import ScriptTaskServiceStub
 from chalk._gen.chalk.server.v1.team_pb2 import (
@@ -279,6 +284,14 @@ class StubProvider:
             )
         return OfflineQueryMetadataServiceStub(self._server_channel)
 
+    @cached_property
+    def scheduled_query_stub(self) -> SchedulerServiceStub:
+        if self._server_channel is None:
+            raise ValueError(
+                "The GRPC engine service is not available. If you would like to set up a GRPC service, please contact Chalk."
+            )
+        return SchedulerServiceStub(self._server_channel)
+
     @cached_property
     def sql_stub(self) -> SqlServiceStub:
         if self._engine_channel is None:
@@ -523,6 +536,9 @@ class StubRefresher:
     def call_offline_query_stub(self, fn: Callable[[OfflineQueryMetadataServiceStub], T]) -> T:
         return self._retry_callable(fn, lambda: self._stub.offline_query_stub)
 
+    def call_scheduled_query_stub(self, fn: Callable[[SchedulerServiceStub], T]) -> T:
+        return self._retry_callable(fn, lambda: self._stub.scheduled_query_stub)
+
     def call_sql_stub(self, fn: Callable[[SqlServiceStub], T]) -> T:
         return self._retry_callable(fn, lambda: self._stub.sql_stub)
 
@@ -1229,6 +1245,62 @@ class ChalkGRPCClient:
             body_type=online_query_pb2.FEATHER_BODY_TYPE_RECORD_BATCHES,
         )
 
+    def run_scheduled_query(
+        self,
+        name: str,
+        planner_options: Optional[Mapping[str, Any]],
+        incremental_resolvers: Optional[Sequence[str]],
+        max_samples: Optional[int],
+        env_overrides: Optional[Mapping[str, str]],
+    ) -> ManualTriggerScheduledQueryResponse:
+        """
+        Manually trigger a scheduled query request.
+
+        Parameters
+        ----------
+        name
+            The name of the scheduled query to be triggered.
+        incremental_resolvers
+            If set to None, Chalk will incrementalize resolvers in the query's root namespaces.
+            If set to a list of resolvers, this set will be used for incrementalization.
+            Incremental resolvers must return a feature time in its output, and must return a `DataFrame`.
+            Most commonly, this will be the name of a SQL file resolver. Chalk will ingest all new data
+            from these resolvers and propagate changes to values in the root namespace.
+        max_samples
+            The maximum number of samples to compute.
+        env_overrides:
+            A dictionary of environment values to override during this specific triggered query.
+
+        Other Parameters
+        ----------------
+        planner_options
+            A dictionary of options to pass to the planner.
+            These are typically provided by Chalk Support for specific use cases.
+
+        Returns
+        -------
+        ManualTriggerScheduledQueryResponse
+            A response message containing metadata around the triggered run.
+
+        Examples
+        --------
+        >>> from chalk.client.client_grpc import ChalkGRPCClient
+        >>> ChalkGRPCClient().run_scheduled_query(
+        ...     name="my_scheduled_query",
+        ... )
+        """
+        return self._stub_refresher.call_scheduled_query_stub(
+            lambda x: x.ManualTriggerScheduledQuery(
+                request=ManualTriggerScheduledQueryRequest(
+                    cron_query_name=name,
+                    planner_options=planner_options or {},
+                    incremental_resolvers=incremental_resolvers or (),
+                    max_samples=max_samples,
+                    env_overrides=env_overrides or {},
+                ),
+            )
+        )
+
     def get_graph(self, deployment: DeploymentId | None = None) -> Graph:
         """Get the graph for a given deployment.
 

chalk/client/client_impl.py

@@ -51,6 +51,7 @@ from typing_extensions import NoReturn, override
 from urllib3 import Retry
 
 import chalk._repr.utils as repr_utils
+from chalk._gen.chalk.server.v1.scheduler_pb2 import ManualTriggerScheduledQueryResponse
 from chalk._reporting.models import BatchReport, BatchReportResponse
 from chalk._reporting.progress import ProgressService
 from chalk._upload_features.utils import to_multi_upload_inputs
@@ -2409,6 +2410,69 @@ https://docs.chalk.ai/cli/apply
         initialized_dataset.is_finished = True
         return initialized_dataset
 
+    def run_scheduled_query(
+        self,
+        name: str,
+        planner_options: Optional[Mapping[str, Any]] = None,
+        incremental_resolvers: Optional[Sequence[str]] = None,
+        max_samples: Optional[int] = None,
+        env_overrides: Optional[Mapping[str, str]] = None,
+    ) -> ManualTriggerScheduledQueryResponse:
+        """
+        Manually trigger a scheduled query request.
+
+        Parameters
+        ----------
+        name
+            The name of the scheduled query to be triggered.
+        incremental_resolvers
+            If set to None, Chalk will incrementalize resolvers in the query's root namespaces.
+            If set to a list of resolvers, this set will be used for incrementalization.
+            Incremental resolvers must return a feature time in its output, and must return a `DataFrame`.
+            Most commonly, this will be the name of a SQL file resolver. Chalk will ingest all new data
+            from these resolvers and propagate changes to values in the root namespace.
+        max_samples
+            The maximum number of samples to compute.
+        env_overrides:
+            A dictionary of environment values to override during this specific triggered query.
+
+        Other Parameters
+        ----------------
+        planner_options
+            A dictionary of options to pass to the planner.
+            These are typically provided by Chalk Support for specific use cases.
+
+        Returns
+        -------
+        ManualTriggerScheduledQueryResponse
+            A response message containing metadata around the triggered run.
+
+        Examples
+        --------
+        >>> from chalk.client.client_grpc import ChalkGRPCClient
+        >>> ChalkGRPCClient().run_scheduled_query(
+        ...     name="my_scheduled_query",
+        ... )
+        """
+        from chalk.client.client_grpc import ChalkGRPCClient
+
+        client_grpc = ChalkGRPCClient(
+            client_id=self._client_id,
+            client_secret=self._client_secret,
+            environment=self._primary_environment,
+            api_server=self._api_server,
+        )
+
+        resp = client_grpc.run_scheduled_query(
+            name=name,
+            planner_options=planner_options,
+            incremental_resolvers=incremental_resolvers,
+            max_samples=max_samples,
+            env_overrides=env_overrides,
+        )
+
+        return resp
+
     def prompt_evaluation(
         self,
         prompts: list[Prompt | str],

chalk/features/_class_property.py

@@ -5,6 +5,8 @@ import functools
 from typing import Any, Callable, List, Type, TypeVar, cast
 
 from chalk._lsp.error_builder import FeatureClassErrorBuilder
+from chalk.features.feature_wrapper import UnresolvedFeature
+from chalk.utils.notebook import is_notebook
 
 T = TypeVar("T")
 V = TypeVar("V")
@@ -54,6 +56,11 @@ def classproperty_support(cls: Type[T]) -> Type[T]:
             if (res := self.__chalk_notebook_feature_expressions__.get(item)) is not None:
                 return res
 
+            # If in notebook, fallback to constructing FQN string instead of raising error
+            if is_notebook():
+                fqn = f"{self.namespace}.{item}"
+                return UnresolvedFeature(fqn)
+
             builder: FeatureClassErrorBuilder = self.__chalk_error_builder__
             builder.invalid_attribute(
                 root_feature_str=self.namespace,

chalk/features/_embedding/embedding.py

@@ -25,6 +25,7 @@ from chalk.utils.collections import ensure_tuple
 SUPPORTED_LOCAL_MODELS = {
     "all-MiniLM-L6-v2",  # https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2
     "sample-bert",  # For internal Chalk use
+    "sample-linear-nn",  # For internal Chalk use
 }
 
 # This will eventually be included in SUPPORTED_LOCAL_MODELS

chalk/features/_encoding/converter.py

@@ -50,6 +50,7 @@ from chalk.features._encoding.pyarrow import (
     rich_to_pyarrow,
 )
 from chalk.features._encoding.rich import structure_primitive_to_rich, unstructure_rich_to_primitive
+from chalk.features.feature_wrapper import UnresolvedFeature
 from chalk.utils.collections import unwrap_annotated_if_needed, unwrap_optional_and_annotated_if_needed
 from chalk.utils.df_utils import pa_array_to_pl_series
 from chalk.utils.json import JSON, TJSON, is_pyarrow_json_type, pyarrow_json_type
@@ -1234,8 +1235,14 @@ class FeatureConverter(PrimitiveFeatureConverter[_TPrim], Generic[_TPrim, _TRich
         # because it is also used for error handling inside of `from_rich_to_primitive`.
         self._name = name
         if rich_default != ...:
-            # The missing value strategy doesn't really matter because rich_default is not missing
-            primitive_default = self.from_rich_to_primitive(rich_default, missing_value_strategy="allow")
+            # In notebook environments, UnresolvedFeature may be used as a placeholder
+            # for features that can't be resolved due to a stale registry.
+            # Treat these as missing defaults since they're not concrete values.
+            if isinstance(rich_default, UnresolvedFeature):
+                rich_default = ...
+            else:
+                # The missing value strategy doesn't really matter because rich_default is not missing
+                primitive_default = self.from_rich_to_primitive(rich_default, missing_value_strategy="allow")
         super().__init__(
             name, is_nullable=is_nullable, pyarrow_dtype=pyarrow_dtype, primitive_default=primitive_default
         )

chalk/features/feature_wrapper.py

@@ -11,6 +11,7 @@ from chalk.features._chalkop import op, Aggregation
 from chalk.features.filter import Filter
 from chalk.serialization.parsed_annotation import ParsedAnnotation
 from chalk.utils.collections import ensure_tuple
+from chalk.utils.notebook import is_notebook
 
 if TYPE_CHECKING:
     from chalk.features.feature_field import Feature
@@ -22,16 +23,43 @@ class NearestNeighborException(ValueError):
     ...
 
 
+class UnresolvedFeature:
+    """Fallback for features that can't be resolved in notebook environments.
+
+    This allows notebooks to work even when the feature registry is stale or incomplete.
+    The server will validate the feature exists when the query is executed.
+    """
+    __slots__ = ("fqn",)
+
+    def __init__(self, fqn: str):
+        self.fqn = fqn
+        super().__init__()
+
+    def __str__(self):
+        return self.fqn
+
+    def __repr__(self):
+        return f"UnresolvedFeature({self.fqn!r})"
+
+    def __hash__(self):
+        return hash(self.fqn)
+
+    def __eq__(self, other: object):
+        if isinstance(other, UnresolvedFeature):
+            return self.fqn == other.fqn
+        return False
+
+
 class _MarkedUnderlyingFeature:
     __slots__ = ("_fn", "_source", "_debug_info")
 
-    def __init__(self, fn: Callable[[], Feature | Filter | type[DataFrame] | FeatureWrapper | Aggregation],
+    def __init__(self, fn: Callable[[], Feature | Filter | type[DataFrame] | FeatureWrapper | Aggregation | UnresolvedFeature],
                  debug_info: Any = None) -> None:
         super().__init__()
         self._fn = fn
         self._debug_info = debug_info
 
-    def __call__(self, *args: Any, **kwds: Any) -> Feature | Filter | type[DataFrame] | FeatureWrapper | Aggregation:
+    def __call__(self, *args: Any, **kwds: Any) -> Feature | Filter | type[DataFrame] | FeatureWrapper | Aggregation | UnresolvedFeature:
         return self._fn()
 
 
@@ -51,7 +79,7 @@ class FeatureWrapper:
         super().__init__()
         self._chalk_underlying = underlying
 
-    def _chalk_get_underlying(self) -> Feature | Aggregation | Filter | type[DataFrame]:
+    def _chalk_get_underlying(self) -> Feature | Aggregation | Filter | type[DataFrame] | UnresolvedFeature:
         if isinstance(self._chalk_underlying, _MarkedUnderlyingFeature):
             self._chalk_underlying = self._chalk_underlying()
         if isinstance(self._chalk_underlying, FeatureWrapper):
@@ -303,6 +331,12 @@ class FeatureWrapper:
                     if f.attribute_name == item:
                         return FeatureWrapper(underlying.copy_with_path(f))
 
+                if is_notebook():
+                    # Construct FQN by preserving the path from the underlying feature
+                    # If underlying has a path, we need to include it in the FQN
+                    fqn = f"{underlying.root_fqn}.{item}"
+                    return UnresolvedFeature(fqn)
+
                 assert underlying.features_cls is not None
                 underlying.features_cls.__chalk_error_builder__.invalid_attribute(
                     root_feature_str=joined_class.namespace,
@@ -314,6 +348,11 @@ class FeatureWrapper:
                 )
                 assert False, "unreachable"
 
+            # If in notebook, fallback to constructing FQN string instead of raising error
+            if is_notebook():
+                fqn = f"{underlying.fqn}.{item}"
+                return UnresolvedFeature(fqn)
+
             assert underlying.features_cls is not None
             underlying.features_cls.__chalk_error_builder__.invalid_attribute(
                 root_feature_str=underlying.fqn,

chalk/features/resolver.py

@@ -555,28 +555,18 @@ class ResolverRegistry:
         short_name = resolver.name
         if short_name in self._short_name_to_resolver:
             if not override and not notebook.is_notebook():
-                if resolver.fqn == self._short_name_to_resolver[short_name]:
-                    # Same resolver was redefined
-                    resolver.lsp_builder.add_diagnostic(
-                        message=f"Duplicate resolver '{resolver.fqn}'. Multiple resolvers cannot have the same name.",
-                        code="71",
-                        label="duplicate name",
-                        range=resolver.lsp_builder.function_name(),
-                        raise_error=ValueError,
-                    )
-                else:
-                    # Same short name was reused
-                    resolver.lsp_builder.add_diagnostic(
-                        message=(
-                            f"Another resolver with the same function name '{resolver.name}' in module "
-                            f"'{self._short_name_to_resolver[short_name].__module__}' exists. "
-                            f"Resolver function names must be unique. Please rename this resolver in module '{resolver.__module__}'."
-                        ),
-                        label="duplicate resolver shortname",
-                        code="71",
-                        range=resolver.lsp_builder.function_name(),
-                        raise_error=None,
-                    )
+                # Same short name was reused
+                resolver.lsp_builder.add_diagnostic(
+                    message=(
+                        f"Another resolver with the same function name '{resolver.name}' in module "
+                        f"'{self._short_name_to_resolver[short_name].__module__}' exists. "
+                        f"Resolver function names must be unique. Please rename this resolver in module '{resolver.__module__}'."
+                    ),
+                    label="duplicate resolver shortname",
+                    code="71",
+                    range=resolver.lsp_builder.function_name(),
+                    raise_error=None,
+                )
                 return
             existing_resolver = self._short_name_to_resolver[short_name]
             # Need to remove the resolver from the typed registry

chalk/parsed/duplicate_input_gql.py

@@ -280,6 +280,8 @@ class UpsertCronQueryGQL:
     resourceGroup: Optional[str] = None
     plannerOptions: Optional[Dict[str, str]] = None
     completionDeadline: Optional[str] = None
+    numShards: Optional[int] = None
+    numWorkers: Optional[int] = None
 
 
 @dataclasses_json.dataclass_json

chalk/parsed/json_conversions.py

@@ -429,6 +429,8 @@ def convert_type_to_gql(
             resourceGroup=t.resource_group,
             plannerOptions=t.planner_options,
             completionDeadline=None if t.completion_deadline is None else timedelta_to_duration(t.completion_deadline),
+            numShards=t.num_shards,
+            numWorkers=t.num_workers,
         )
 
     if isinstance(t, NamedQuery):

chalk/queries/scheduled_query.py

@@ -29,6 +29,8 @@ class ScheduledQuery:
         planner_options: dict[str, str] | None = None,
         resource_group: str | None = None,
         completion_deadline: timedelta | None = None,
+        num_shards: int | None = None,
+        num_workers: int | None = None,
     ):
         """Create an offline query which runs on a schedule.
 
@@ -162,6 +164,9 @@ class ScheduledQuery:
 
         self.completion_deadline = completion_deadline
 
+        self.num_shards = num_shards
+        self.num_workers = num_workers
+
         CRON_QUERY_REGISTRY[name] = self
 
 

chalk/sql/_internal/sql_file_resolver.py

@@ -321,7 +321,10 @@ def get_sql_file_resolvers(
                     continue
                 _filepath_to_sql_string[filepath] = sql_string_result.sql_string
             yield get_sql_file_resolver(sources, sql_string_result, has_import_errors)
-    for sql_string_result in _GENERATED_SQL_FILE_RESOLVER_REGISTRY.get_generated_sql_file_resolvers():
+    # Only yield generated resolvers whose filepath is under the directory being scanned
+    for sql_string_result in _GENERATED_SQL_FILE_RESOLVER_REGISTRY.get_generated_sql_file_resolvers(
+        filter_by_directory=sql_file_resolve_location
+    ):
         yield get_sql_file_resolver(sources, sql_string_result, has_import_errors)
 
 
@@ -344,8 +347,12 @@ def get_sql_file_resolvers_from_paths(
             sql_string_result=sql_string_result,
             has_import_errors=has_import_errors,
         )
+    # Only yield generated resolvers whose filepath is in the paths list
+    # If paths is empty, yield all generated resolvers (no filtering)
     for sql_string_result in _GENERATED_SQL_FILE_RESOLVER_REGISTRY.get_generated_sql_file_resolvers():
-        yield get_sql_file_resolver(sources, sql_string_result, has_import_errors)
+        # Check if this generated resolver's filepath is in the provided paths
+        if not paths or sql_string_result.path in paths:
+            yield get_sql_file_resolver(sources, sql_string_result, has_import_errors)
 
 
 def get_sql_file_resolver(
@@ -1601,8 +1608,29 @@ class GeneratedSQLFileResolverRegistry:
             filepath=filepath, sql_string=sql_string, comment_dict=comment_dict, postprocessing_expr=postprocessing_expr
         )
 
-    def get_generated_sql_file_resolvers(self) -> Iterable[SQLStringResult]:
+    def get_generated_sql_file_resolvers(self, filter_by_directory: Path | None = None) -> Iterable[SQLStringResult]:
+        """
+        Yield generated SQL file resolvers, optionally filtered by directory.
+
+        Args:
+            filter_by_directory: If provided, only yield resolvers whose filepath is under this directory.
+                                 If None, yield all generated resolvers (legacy behavior).
+        """
         for name, generated_info in self.resolver_name_to_generated_infos.items():
+            # If filtering by directory is requested, check if the resolver's filepath is under that directory
+            if filter_by_directory is not None:
+                # Special case: notebook resolvers (filepath == "<notebook>") should never be auto-yielded
+                # when scanning directories, only when explicitly requested
+                if generated_info.filepath == "<notebook>":
+                    continue
+
+                # Convert to absolute paths for comparison and check if resolver path is under filter directory
+                resolver_path = Path(generated_info.filepath).resolve()
+                filter_path = Path(filter_by_directory).resolve()
+
+                if not resolver_path.is_relative_to(filter_path):
+                    continue
+
             yield SQLStringResult(
                 path=generated_info.filepath,
                 sql_string=generated_info.sql_string,

{chalkpy-2.93.2.dist-info → chalkpy-2.93.3.dist-info}/METADATA

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

{chalkpy-2.93.2.dist-info → chalkpy-2.93.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=9YxwkAt3Im0OCMfpmnIB_4PKjZfBCcRmwATLXdHNRm4,2609
-chalk/_version.py,sha256=t2ehNNV4nh6yMJVBg9EP9Y9nTxjmn2UHhlYfxUMfc_Q,23
+chalk/_version.py,sha256=a8M8W7TWOUNnBbncydFQJ2EtW4CRYKZRaSSP5FM-rG0,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=hCEo7eqSfXZWklkFB2geeipGhiD0qNjPBpQJvOBW6N0,63083
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -390,8 +390,8 @@ chalk/_gen/chalk/server/v1/incident_pb2.py,sha256=4-uejdKheTLIk90OqClCAxb6esH8hM
 chalk/_gen/chalk/server/v1/incident_pb2.pyi,sha256=eP3FEl3eHkIRXfXLr_2cn-_GlHHirKQZv87DBg2jLaU,6861
 chalk/_gen/chalk/server/v1/incident_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
 chalk/_gen/chalk/server/v1/incident_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
-chalk/_gen/chalk/server/v1/integrations_pb2.py,sha256=bG6SuQ5B98TVKrVSQNtozTMm6nOzsyW4PDWlawsg3WM,13525
-chalk/_gen/chalk/server/v1/integrations_pb2.pyi,sha256=hI38Gbw5FGkCG_DktsM8w-F2hQsN1-t9C8nL5--Ep6w,10795
+chalk/_gen/chalk/server/v1/integrations_pb2.py,sha256=BSgvuWniq6oyJhoBz1Ptcov3Q9G1u8NdpQp_ogRWbDE,14216
+chalk/_gen/chalk/server/v1/integrations_pb2.pyi,sha256=_EJVXKyY75KI9xu32ZmHkew2APzNf8IWHrc2cAayfSM,11977
 chalk/_gen/chalk/server/v1/integrations_pb2_grpc.py,sha256=bWarXrUn1O7t6RhGCCKo0X0IIXaA9SKln7AQDVEY8vM,16185
 chalk/_gen/chalk/server/v1/integrations_pb2_grpc.pyi,sha256=uV38oDuzdnQObTbdFjrVSVH0C92FTB4Wx_Cd4v1-wr0,3478
 chalk/_gen/chalk/server/v1/kube_pb2.py,sha256=5cV8jL3Nwal2FTX3-6dh0lrMxKduzjanC3hV8lLU-CI,6791
@@ -462,8 +462,8 @@ chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=ykS7SrO0bu74O55qu6E
 chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=M9JNVrg74MI-ay5bOzJU-BdC8h-jUkrAmlssiCFTln4,6113
 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=UxKSxKlZmbexO8Jfrj9elvPseFVI7zUpMeA-3rQpHBc,7960
-chalk/_gen/chalk/server/v1/scheduler_pb2.pyi,sha256=NlmOPvY3NNQUKlKD-QQTjbPUZETZ8X7vsG4x8UYSWqM,6959
+chalk/_gen/chalk/server/v1/scheduler_pb2.py,sha256=51WGM0OLvEzZBvgy7gubBGkUH3NSDwEHRZ1P1enkz0c,8061
+chalk/_gen/chalk/server/v1/scheduler_pb2.pyi,sha256=G--OoEXULE2nRhp2pQ9rXYRicqeY7J47FWDwBdNvSx0,7154
 chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py,sha256=LEfa6Htuud5PHvl9_daN-Bs0KfBPkvrS9YHjM8EjY08,4969
 chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi,sha256=tDMzDHvI-TNcmQzCmlKebNl8KECl7p41zgPmZOreC1o,1395
 chalk/_gen/chalk/server/v1/script_tasks_pb2.py,sha256=MfrJdphkoDV-aNWYF10pG_cvPFlVO72-LK_RUmTQYX8,7534
@@ -550,11 +550,11 @@ 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=xq3_AgshEA0KR2tZ-tJ4lxU_UGmEL_1WG_o6mRQgmZE,101056
+chalk/client/client.py,sha256=aIOIB7OTLZ6jWfesAVRqeT6mFqtsjeAk0e8UE29bzRI,102940
 chalk/client/client_async.py,sha256=nFFTWJbdBlb7zksyjOMBY566tZTAyNXQhCnq06LHWl0,50803
 chalk/client/client_async_impl.py,sha256=ZphhgTB49JBWHCGXe-dI0wWWKc9zPcOczy02q_gFy50,6925
-chalk/client/client_grpc.py,sha256=nwURtE99LFHWXsHiJwOA_xTocYl9XUmThkjtm1Eujkg,90672
-chalk/client/client_impl.py,sha256=Od07ybvm3or4GL5WpqR86ZSiVNcwvUPRLwRc7dnWlIc,208031
+chalk/client/client_grpc.py,sha256=86W32GXemHb0xt78sUKhgorY2Me3ud53YNzEr09NcDg,93683
+chalk/client/client_impl.py,sha256=smNVo-vItnjbuN-8beFNb986uOVDaPTCkd6JTfT76WY,210503
 chalk/client/dataset.py,sha256=LneWwaAOHCjtj7gaJjsSeVNruj-QJ51hjRi62zrFNVE,77561
 chalk/client/exc.py,sha256=kZJ80YbSeSRDmTLTh240j_eRdJFZBa7IaDsNSRoDroU,4145
 chalk/client/models.py,sha256=CqJ8vdMAPrjOo--GWQVr2WF3zfP5bfhULho0_YL0p6o,60883
@@ -578,7 +578,7 @@ 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
 chalk/features/_chalkop.py,sha256=N229Tsoc3fgAiyoH6ZtygrD6uCdkBk_JSOXQSpDd6QA,18383
-chalk/features/_class_property.py,sha256=LwpYGzkbzx6ETg1S5mrCUlhc-BweqLBMhJiuNYaBx5U,3256
+chalk/features/_class_property.py,sha256=YR_GjTcdDKYEyQ4uyGLpt-CYvs4XuVnLEI1N2QEyTNA,3579
 chalk/features/_document.py,sha256=ck3J958tNEIIKU4s_uoLObM-H2PG0Oqs4DtKtouGJVE,368
 chalk/features/_geospatial.py,sha256=OF3lasbFstw5gj8vrAqkU8815bvZeHoFRppS9ZfB5rU,4382
 chalk/features/_last.py,sha256=IyYe0PVAKBu8FRhCZiK6Dy5afKbwDHEUTES5qVXbaZU,1163
@@ -589,7 +589,7 @@ chalk/features/feature_field.py,sha256=tLP9iybcYa7eM2eJJ0iSRks-ptyabA0l_SAJ83t6n
 chalk/features/feature_set.py,sha256=yNi0_J4CylAVkVp1Y67qV6i8vHMdE0p99DnyLPABPHI,12406
 chalk/features/feature_set_decorator.py,sha256=HMGnQdXQevpsTCNGIG2njViHDnm1ltFoZDhABFkp3Dw,65059
 chalk/features/feature_time.py,sha256=iUk8NDelig81jP7QT3tguyzx5eOZ-YC84OVgJRRKVwo,1639
-chalk/features/feature_wrapper.py,sha256=WXTArqpL5TtXFv64zwVZjRps31iMgf3qCnf3cz38y70,17910
+chalk/features/feature_wrapper.py,sha256=OolNWGGX67IAEMHCObFvOCpH5EmwjbMvMygRSBJJtu0,19259
 chalk/features/filter.py,sha256=2ldMbqvXC-nJ0jc-OZ36qHtrej-Jkx4TNQ1W_NZodAs,11177
 chalk/features/hooks.py,sha256=KPJnudnhosEH7UptQT8W8sumIYPcZVMVjaJJJhPsOyk,5489
 chalk/features/inference.py,sha256=hQ6htPh6Q-XzfRjpEgMHKaCGwpKlnGG0wcxoGqwPQfo,4912
@@ -597,20 +597,20 @@ chalk/features/live_updates.py,sha256=8ZbiDjcLqfFruSL15_aycwzSqJ0TbKNhas06KfZLyL
 chalk/features/namespace_context.py,sha256=fL-nPohqtNiyPDS1uQTAaHLns4aivuBL2Flf50DajU4,1813
 chalk/features/primary.py,sha256=BZ8mrMmKfRNy_wnKGDJt2cdnejP_CZb6xBsD9Ljgajc,5209
 chalk/features/pseudofeatures.py,sha256=50Pe_Xi8ttYWtgNNRpgkhBxP8xoCZCYwyLb0aWUQ-PI,2147
-chalk/features/resolver.py,sha256=oMSveVkEhlAbjS9XS4HLMsVGZBJQ2uAHuR1Q3YGakaY,182357
+chalk/features/resolver.py,sha256=uH9bpYB_a8ihARHaUTvGe6-S2qZmwnB-YczW99DtLLQ,181763
 chalk/features/tag.py,sha256=LRmKRA8ANCOvmaIAk-L5j1QW2U0aah2SeASy8Uydkmk,2675
 chalk/features/underscore.py,sha256=4xnfQV3bfvVn0PNEtkT4J-k7hW4ebtH9KBe4_BvGjY4,26763
 chalk/features/underscore_features.py,sha256=PlVCoaDDffOgtiSMaxPSWCoj8IjscbkOzDLA471HsJ4,13005
 chalk/features/_embedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/features/_embedding/cohere.py,sha256=lOPUBgxgalKRC-2VsfvvIWiDkWULL0BZouSNmvDOyVA,3678
-chalk/features/_embedding/embedding.py,sha256=HHwcbSG-HhhOmOvrMnkEsfR5tTLZXXMQmtf4h8F9wZY,18284
+chalk/features/_embedding/embedding.py,sha256=sHZ48AaiNPPlSuFvmpZIE8PVPSKbthGFTqAC-3BBFdg,18334
 chalk/features/_embedding/embedding_provider.py,sha256=4bpla6qs9AkDzVWX4u_VV3lg1b8m4d35VMztRaUcgLE,1855
 chalk/features/_embedding/openai.py,sha256=9St_ks5oavMoKlaw-Q3Flvj32XAIApgScwX8aPiZP6E,6286
 chalk/features/_embedding/sentence_transformer.py,sha256=hNYuT9D-16C49lkhIVK_KXZTYYD1mJsPs_mfBrMd3DA,4811
 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=zqdbilY1kMXX5-izsQuAk-CpSBOjpWpKBKCyI4S5D-I,64967
+chalk/features/_encoding/converter.py,sha256=Emb48g_d0njx0Ix5-IcoByBaS__E6aNTmLBOdK-88TY,65392
 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
@@ -661,9 +661,9 @@ chalk/parsed/_graph_validation.py,sha256=XHVPw4cf-nkPdHrqGXevsFuOrMtWtibUTRkTuiz
 chalk/parsed/_metadata.py,sha256=TCjKr20Fs4Yguo97yKjATurwOdV-DoFQeiGxbX_s1hs,6749
 chalk/parsed/branch_state.py,sha256=g3sCyjOWlcc3rJwTF-ZVoWMbFWmzJQVDjnQQ2x7mrtQ,8306
 chalk/parsed/branch_state_rich.py,sha256=X9st1vRMRMOfnEy3hwitsAyNcTzZlubXJt9jI1MlNoo,3911
-chalk/parsed/duplicate_input_gql.py,sha256=Jp7uB0DUreUULfH9cl0JdqG0ng3BlG4SZQRnJkQiYKE,20123
+chalk/parsed/duplicate_input_gql.py,sha256=IbVRKDCS-M7f9k127LpeVjRlXvzp1tuIek-bOqTULC4,20196
 chalk/parsed/expressions.py,sha256=A8U54v9jacLX4UVtLqSUYUjL-lOnI-kWvG2hHZvfYE0,554
-chalk/parsed/json_conversions.py,sha256=6CRPR4YMHapxgHfheXvOkF2IY8AbRh5ptdrCWCm2R3Y,30385
+chalk/parsed/json_conversions.py,sha256=GRBg2KvwahP1tXwr7n--s7ueULN6prEDPO2I9B9MWzk,30459
 chalk/parsed/to_proto.py,sha256=Ajq20zLvHlnGjP1YhvMbsEFo5Jw5q9NUT9GqlM4y4no,75025
 chalk/parsed/user_types_to_json.py,sha256=ZJWdYFqyhr5InvItQybtHadXQjW3vjHrv8hjMGtL3Bc,13318
 chalk/parsed/validation_from_registries.py,sha256=nfiAj1tvWRu0RrkhkGtElhAsL8V7ayEKUisrKQF-wYc,7900
@@ -675,7 +675,7 @@ chalk/parsed/_proto/validation.py,sha256=D6dOnCoeARHOLPHQb8zYGXDc0eWJblSQ4tO4rb_
 chalk/queries/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/queries/named_query.py,sha256=Yc_g7QYCmEbgKneGPQ8FPsKeic6B3MY2LeX4fPEHyWc,8537
 chalk/queries/query_context.py,sha256=O8ZVXq2cD3YnhIzRHOtewlRl4ur-Pzsz4SricO3H7Fo,2989
-chalk/queries/scheduled_query.py,sha256=w7_SVkxOC4sDTQAaW9FOdGaXjd4tusv6v3Gg7fxKtpY,6989
+chalk/queries/scheduled_query.py,sha256=VJ4rIXpVIwhmSsaOHuJ6TUsafSGIjLZfb-mz0S7dAjY,7145
 chalk/serialization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/serialization/parsed_annotation.py,sha256=1-xwevPMsZ9E407SuRWZdRQKMuR2TligWprkmdb5dxM,25526
 chalk/sink/__init__.py,sha256=R58HP9J4hSJbYiKk7xg-uTN3hmpLlt7knLHkco2JdlU,101
@@ -688,7 +688,7 @@ chalk/sql/_internal/chalk_query.py,sha256=O2Mv0JHWvSrP3iDfkooy22yoDuWAH57m1i8qcy
 chalk/sql/_internal/incremental.py,sha256=xqnohrZIoqE3y92edUl_oEkHQoFZw9GN8GIOUynyyEI,2900
 chalk/sql/_internal/query_execution_parameters.py,sha256=FT0GixOryGeKR1x7UrQNoGFtySd_8-B4Xrp38nFi0FU,3679
 chalk/sql/_internal/query_registry.py,sha256=jOk2x9NvW2pZocI_iTvgd9DyOlr4-ajz8rTi0kkLSaI,3005
-chalk/sql/_internal/sql_file_resolver.py,sha256=AqOjsd9iKj74yKfhCMjfTvKCXY3vc3eAJnX5v_e7o20,76251
+chalk/sql/_internal/sql_file_resolver.py,sha256=BCGTpe3tVLQTWNSrfyUFsg8VfO-pupeN8yxGpiSmy5E,77817
 chalk/sql/_internal/sql_settings.py,sha256=9lcpHNrmEhr1Zxl_Ct7U0p0AbLUvlpSayxHEPmyqu8E,543
 chalk/sql/_internal/sql_source.py,sha256=D5GEYIBVyGpsb66qcXasJaldM6uhE4sZSaaTbYqd3Q4,42302
 chalk/sql/_internal/sql_source_group.py,sha256=RIMEjEG3oSGwign37w9avpCRdVfVQepR8SPVv6JoOx0,5562
@@ -764,8 +764,8 @@ chalk/utils/tracing.py,sha256=Glx8YrtjWy0zE5YbpgfgcsLDshAKnnYm9poiWNeCxXs,11075
 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.93.2.dist-info/METADATA,sha256=MJ-YYfA49s9AzRQLShAVA47L5kLMpWOXE0V4VtowCEM,27494
-chalkpy-2.93.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chalkpy-2.93.2.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.93.2.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.93.2.dist-info/RECORD,,
+chalkpy-2.93.3.dist-info/METADATA,sha256=RSXT5f7CE9lqU59166jt7xmFAmch5JkwBKp18rgDgqs,27494
+chalkpy-2.93.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chalkpy-2.93.3.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.93.3.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.93.3.dist-info/RECORD,,