chalkpy 2.93.1__tar.gz → 2.93.3__tar.gz

{chalkpy-2.93.1 → chalkpy-2.93.3}/PKG-INFO

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

{chalkpy-2.93.1 → chalkpy-2.93.3}/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)

{chalkpy-2.93.1 → chalkpy-2.93.3}/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: ...

{chalkpy-2.93.1 → chalkpy-2.93.3}/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)

{chalkpy-2.93.1 → chalkpy-2.93.3}/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):

chalkpy-2.93.3/chalk/_version.py

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

{chalkpy-2.93.1 → chalkpy-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,
@@ -924,6 +925,8 @@ class ChalkClient:
         max_retries: int | None = None,
         query_name: str | None = None,
         query_name_version: str | None = None,
+        *,
+        input_sql: str | None = None,
     ) -> Dataset:
         """Compute feature values from the offline store or by running offline/online resolvers.
         See `Dataset` for more information.
@@ -944,6 +947,9 @@ class ChalkClient:
             times, the list must match the length of the `input` lists. Each element of input_time corresponds with the
             feature values at the same index of the `input` lists.
             See https://docs.chalk.ai/docs/temporal-consistency for more information.
+        input_sql
+            An alternative to `input`: a ChalkSQL query that returns values
+            to use as inputs.
         output
             The features that you'd like to sample, if they exist.
             If an output feature was never computed for a sample (row) in
@@ -1080,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],

{chalkpy-2.93.1 → chalkpy-2.93.3}/chalk/client/client_async.py

@@ -646,6 +646,8 @@ class AsyncChalkClient:
         max_retries: int | None = None,
         query_name: str | None = None,
         query_name_version: str | None = None,
+        *,
+        input_sql: str | None = None,
     ) -> Dataset:
         """Compute feature values from the offline store or by running offline/online resolvers.
         See `Dataset` for more information.
@@ -660,6 +662,9 @@ class AsyncChalkClient:
             an observation in line with the timestamp in `input_times`.
         input_times
             A list of the times of the observations from `input`.
+        input_sql
+            An alternative to `input`: a ChalkSQL query that returns values
+            to use as inputs.
         output
             The features that you'd like to sample, if they exist.
             If an output feature was never computed for a sample (row) in

{chalkpy-2.93.1 → chalkpy-2.93.3}/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.