chalkpy 2.154.2__tar.gz → 2.154.4__tar.gz

{chalkpy-2.154.2 → chalkpy-2.154.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.154.2
+Version: 2.154.4
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai
@@ -77,6 +77,8 @@ Provides-Extra: vertexai
 Requires-Dist: google-cloud-aiplatform<2; extra == "vertexai"
 Provides-Extra: sentence-transformers
 Requires-Dist: sentence-transformers==5.3.0; extra == "sentence-transformers"
+Provides-Extra: compute
+Requires-Dist: chalkcompute>=2.3.2; extra == "compute"
 Provides-Extra: sql
 Requires-Dist: sqlalchemy[asyncio]<2,>=1.4.26; extra == "sql"
 Requires-Dist: adlfs; extra == "sql"

{chalkpy-2.154.2 → chalkpy-2.154.4}/chalk/_gen/chalk/planner/v1/batch_operator_pb2.py

@@ -26,7 +26,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n%chalk/planner/v1/batch_operator.proto\x12\x10\x63halk.planner.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto\x1a$chalk/expression/v1/expression.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a$chalk/planner/v1/feature_types.proto\x1a%chalk/planner/v1/symbolic_value.proto\x1a,chalk/symbolic_value/v1/symbolic_value.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x8c\x03\n\tBatchPlan\x12=\n\toperators\x18\x01 \x03(\x0b\x32\x1f.chalk.planner.v1.BatchOperatorR\toperators\x12L\n\x0fsymbolic_values\x18\x02 \x03(\x0b\x32\x1f.chalk.planner.v1.SymbolicValueB\x02\x18\x01R\x0esymbolicValues\x12T\n\x10\x66\x65\x61ture_ref_info\x18\x03 \x01(\x0b\x32&.chalk.planner.v1.FeatureReferenceInfoB\x02\x18\x01R\x0e\x66\x65\x61tureRefInfo\x12\x46\n\x0e\x61uxiliary_info\x18\x04 \x01(\x0b\x32\x1f.chalk.planner.v1.AuxiliaryInfoR\rauxiliaryInfo\x12T\n\x12symbolic_values_v2\x18\x05 \x03(\x0b\x32&.chalk.symbolic_value.v1.SymbolicValueR\x10symbolicValuesV2"\xb8\x02\n\rBatchOperator\x12\x43\n\roperator_type\x18\x01 \x01(\x0e\x32\x1e.chalk.planner.v1.OperatorTypeR\x0coperatorType\x12:\n\ngraph_info\x18\x02 \x01(\x0b\x32\x1b.chalk.planner.v1.GraphInfoR\tgraphInfo\x12L\n\targuments\x18\x03 \x03(\x0b\x32..chalk.planner.v1.BatchOperator.ArgumentsEntryR\targuments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"\x85\x02\n\x14\x46\x65\x61tureReferenceInfo\x12\x45\n\x0c\x66\x65\x61ture_refs\x18\x01 \x03(\x0b\x32".chalk.planner.v1.FeatureReferenceR\x0b\x66\x65\x61tureRefs\x12I\n\x10\x64\x61ta_frame_types\x18\x02 \x03(\x0b\x32\x1f.chalk.planner.v1.DataFrameTypeR\x0e\x64\x61taFrameTypes\x12W\n\x12\x66ilter_expressions\x18\x03 \x03(\x0b\x32(.chalk.planner.v1.FilterExpressionParsedR\x11\x66ilterExpressions:\x02\x18\x01",\n\tGraphInfo\x12\x1f\n\x0boperator_id\x18\x01 \x01(\x04R\noperatorId"\x99\x0e\n\x08\x41rgument\x12,\n\x04none\x18\x01 \x01(\x0b\x32\x16.chalk.planner.v1.VoidH\x00R\x04none\x12\x1f\n\nbool_value\x18\x02 \x01(\x08H\x00R\tboolValue\x12#\n\x0c\x64ouble_value\x18\x03 \x01(\x01H\x00R\x0b\x64oubleValue\x12\x1d\n\tint_value\x18\x04 \x01(\x03H\x00R\x08intValue\x12#\n\x0cstring_value\x18\x05 \x01(\tH\x00R\x0bstringValue\x12:\n\ttimestamp\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\ttimestamp\x12\x37\n\x08\x64uration\x18\x07 \x01(\x0b\x32\x19.google.protobuf.DurationH\x00R\x08\x64uration\x12!\n\x0boperator_id\x18\x08 \x01(\x04H\x00R\noperatorId\x12G\n\x0b\x66\x65\x61ture_ref\x18\t \x01(\x0b\x32 .chalk.graph.v1.FeatureReferenceB\x02\x18\x01H\x00R\nfeatureRef\x12!\n\x0b\x62ytes_value\x18\n \x01(\x0cH\x00R\nbytesValue\x12\x36\n\x05tuple\x18\x0e \x01(\x0b\x32\x1e.chalk.planner.v1.ArgumentListH\x00R\x05tuple\x12\x37\n\x06submap\x18\x0f \x01(\x0b\x32\x1d.chalk.planner.v1.ArgumentMapH\x00R\x06submap\x12:\n\narrow_type\x18\x10 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeH\x00R\tarrowType\x12O\n\x0funderscore_expr\x18\x11 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\x0eunderscoreExpr\x12G\n\x0b\x66ilter_expr\x18\x12 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\nfilterExpr\x12n\n\x1c\x64\x65tached_column_feature_type\x18\x13 \x01(\x0b\x32+.chalk.planner.v1.DetachedColumnFeatureTypeH\x00R\x19\x64\x65tachedColumnFeatureType\x12t\n\x1eoutput_underscore_feature_type\x18\x14 \x01(\x0b\x32-.chalk.planner.v1.OutputUnderscoreFeatureTypeH\x00R\x1boutputUnderscoreFeatureType\x12M\n\x0f\x64\x61ta_frame_type\x18\x15 \x01(\x0b\x32\x1f.chalk.planner.v1.DataFrameTypeB\x02\x18\x01H\x00R\rdataFrameType\x12U\n\x11\x66\x65\x61ture_reference\x18\x16 \x01(\x0b\x32".chalk.planner.v1.FeatureReferenceB\x02\x18\x01H\x00R\x10\x66\x65\x61tureReference\x12L\n\x0esymbolic_value\x18\x17 \x01(\x0b\x32\x1f.chalk.planner.v1.SymbolicValueB\x02\x18\x01H\x00R\rsymbolicValue\x12T\n\x12\x64\x61ta_frame_type_id\x18\x18 \x01(\x0b\x32!.chalk.planner.v1.DataFrameTypeIdB\x02\x18\x01H\x00R\x0f\x64\x61taFrameTypeId\x12P\n\x0e\x66\x65\x61ture_ref_id\x18\x19 \x01(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdB\x02\x18\x01H\x00R\x0c\x66\x65\x61tureRefId\x12W\n\x15\x64\x61ta_frame_type_id_v2\x18\x1a \x01(\x0b\x32#.chalk.planner.v1.DataFrameTypeIdV2H\x00R\x11\x64\x61taFrameTypeIdV2\x12S\n\x11\x66\x65\x61ture_ref_id_v2\x18\x1b \x01(\x0b\x32&.chalk.planner.v1.FeatureReferenceIdV2H\x00R\x0e\x66\x65\x61tureRefIdV2\x12S\n\x11underscore_parsed\x18\x1c \x01(\x0b\x32$.chalk.planner.v1.UnderscoreParsedIdH\x00R\x10underscoreParsed\x12(\n\x0fipc_arrow_table\x18\x1d \x01(\x0cH\x00R\ripcArrowTable\x12T\n\x11symbolic_value_v2\x18\x1e \x01(\x0b\x32&.chalk.symbolic_value.v1.SymbolicValueH\x00R\x0fsymbolicValueV2B\x05\n\x03\x61rg"t\n\x12\x41rgumentMapElement\x12,\n\x03key\x18\x01 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value"\xf6\x02\n\x0b\x41rgumentMap\x12N\n\targuments\x18\x01 \x03(\x0b\x32,.chalk.planner.v1.ArgumentMap.ArgumentsEntryB\x02\x18\x01R\targuments\x12\x32\n\x04keys\x18\x02 \x03(\x0b\x32\x1a.chalk.planner.v1.ArgumentB\x02\x18\x01R\x04keys\x12\x36\n\x06values\x18\x03 \x03(\x0b\x32\x1a.chalk.planner.v1.ArgumentB\x02\x18\x01R\x06values\x12Q\n\x11ordered_arguments\x18\x04 \x03(\x0b\x32$.chalk.planner.v1.ArgumentMapElementR\x10orderedArguments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"B\n\x0c\x41rgumentList\x12\x32\n\x06values\x18\x01 \x03(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x06values"\x06\n\x04Void"\xcf\x01\n\x19\x44\x65tachedColumnFeatureType\x12X\n\targuments\x18\x01 \x03(\x0b\x32:.chalk.planner.v1.DetachedColumnFeatureType.ArgumentsEntryR\targuments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"\xd3\x01\n\x1bOutputUnderscoreFeatureType\x12Z\n\targuments\x18\x01 \x03(\x0b\x32<.chalk.planner.v1.OutputUnderscoreFeatureType.ArgumentsEntryR\targuments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"\xef\x01\n\x16\x46ilterExpressionParsed\x12\x43\n\x07this_id\x18\x01 \x01(\x0b\x32*.chalk.planner.v1.FilterExpressionParsedIdR\x06thisId\x12U\n\x11\x66ilter_expression\x18\x02 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeB\x02\x18\x01R\x10\x66ilterExpression\x12\x35\n\x04\x65xpr\x18\x03 \x01(\x0b\x32!.chalk.planner.v1.UnderscoreValueR\x04\x65xpr:\x02\x18\x01".\n\x18\x46ilterExpressionParsedId\x12\x0e\n\x02id\x18\x01 \x01(\x04R\x02id:\x02\x18\x01"\xef\x04\n\rDataFrameType\x12:\n\x07this_id\x18\x04 \x01(\x0b\x32!.chalk.planner.v1.DataFrameTypeIdR\x06thisId\x12-\n\x02\x64\x66\x18\x01 \x01(\x0b\x32\x1d.chalk.graph.v1.DataFrameTypeR\x02\x64\x66\x12Z\n\x11\x66ilter_expression\x18\x02 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeB\x02\x18\x01H\x00R\x10\x66ilterExpression\x88\x01\x01\x12\x61\n\x14\x66ilter_expression_id\x18\x05 \x01(\x0b\x32*.chalk.planner.v1.FilterExpressionParsedIdH\x01R\x12\x66ilterExpressionId\x88\x01\x01\x12Q\n\x10optional_columns\x18\x03 \x03(\x0b\x32".chalk.planner.v1.FeatureReferenceB\x02\x18\x01R\x0foptionalColumns\x12V\n\x14optional_column_refs\x18\x06 \x03(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x12optionalColumnRefs\x12V\n\x14required_column_refs\x18\x07 \x03(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x12requiredColumnRefs:\x02\x18\x01\x42\x14\n\x12_filter_expressionB\x17\n\x15_filter_expression_id"%\n\x0f\x44\x61taFrameTypeId\x12\x0e\n\x02id\x18\x01 \x01(\x04R\x02id:\x02\x18\x01"\x9d\x03\n\x10\x46\x65\x61tureReference\x12=\n\x07this_id\x18\x05 \x01(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x06thisId\x12\x41\n\x0b\x66\x65\x61ture_ref\x18\x01 \x01(\x0b\x32 .chalk.graph.v1.FeatureReferenceR\nfeatureRef\x12:\n\x04path\x18\x02 \x03(\x0b\x32".chalk.planner.v1.FeatureReferenceB\x02\x18\x01R\x04path\x12?\n\x08path_ids\x18\x04 \x03(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x07pathIds\x12\x38\n\x02\x64\x66\x18\x03 \x01(\x0b\x32\x1f.chalk.planner.v1.DataFrameTypeB\x02\x18\x01H\x00R\x02\x64\x66\x88\x01\x01\x12;\n\x05\x64\x66_id\x18\x06 \x01(\x0b\x32!.chalk.planner.v1.DataFrameTypeIdH\x01R\x04\x64\x66Id\x88\x01\x01:\x02\x18\x01\x42\x05\n\x03_dfB\x08\n\x06_df_id"(\n\x12\x46\x65\x61tureReferenceId\x12\x0e\n\x02id\x18\x01 \x01(\x04R\x02id:\x02\x18\x01"\xb7\x01\n\x0fUnderscoreValue\x12U\n\x13original_underscore\x18\x01 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeR\x12originalUnderscore\x12I\n\ncodec_info\x18\x02 \x01(\x0b\x32*.chalk.planner.v1.UnderscoreValueCodecInfoR\tcodecInfo:\x02\x18\x01"\x85\x02\n\x18UnderscoreValueCodecInfo\x12J\n\x0b\x66or_feature\x18\x01 \x01(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdH\x00R\nforFeature\x88\x01\x01\x12%\n\x0eroot_namespace\x18\x02 \x01(\tR\rrootNamespace\x12\x62\n\x18root_underscore_behavior\x18\x03 \x01(\x0b\x32(.chalk.planner.v1.RootUnderscoreBehaviorR\x16rootUnderscoreBehavior:\x02\x18\x01\x42\x0e\n\x0c_for_feature"\xd7\x01\n\x16RootUnderscoreBehavior\x12N\n\x08resolver\x18\x01 \x01(\x0b\x32\x30.chalk.planner.v1.ResolverRootUnderscoreBehaviorH\x00R\x08resolver\x12\x61\n\x0fstream_resolver\x18\x02 \x01(\x0b\x32\x36.chalk.planner.v1.StreamResolverRootUnderscoreBehaviorH\x00R\x0estreamResolverB\n\n\x08\x62\x65havior" \n\x1eResolverRootUnderscoreBehavior"f\n$StreamResolverRootUnderscoreBehavior\x12>\n\rmessage_dtype\x18\x01 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\x0cmessageDtype*\xbd\x17\n\x0cOperatorType\x12\x1d\n\x19OPERATOR_TYPE_UNSPECIFIED\x10\x00\x12%\n!OPERATOR_TYPE_ADD_CHILD_INDEX_COL\x10\x01\x12\x1f\n\x1bOPERATOR_TYPE_ADD_INDEX_COL\x10\x02\x12$\n OPERATOR_TYPE_BATCH_AGG_OPERATOR\x10\x03\x12\'\n#OPERATOR_TYPE_BATCH_DWHAGG_OPERATOR\x10\x04\x12\x1e\n\x1aOPERATOR_TYPE_BATCH_RENAME\x10\x05\x12\'\n#OPERATOR_TYPE_BIGTABLE_CACHE_LOOKUP\x10\x06\x12\x1e\n\x1aOPERATOR_TYPE_CACHE_LOOKUP\x10\x07\x12\'\n#OPERATOR_TYPE_CACHE_LOOKUP_HAS_MANY\x10\x08\x12-\n)OPERATOR_TYPE_COMPUTE_CRON_OUTPUT_SUMMARY\x10\t\x12+\n\'OPERATOR_TYPE_COMPUTE_RELEVANT_ROW_MASK\x10\x44\x12"\n\x1eOPERATOR_TYPE_CPP_CACHE_LOOKUP\x10\n\x12*\n&OPERATOR_TYPE_DATAFRAME_MERGE_OPERATOR\x10\x0b\x12\x1e\n\x1aOPERATOR_TYPE_DATASET_SCAN\x10\x0c\x12(\n$OPERATOR_TYPE_DEBUG_COMMENT_OPERATOR\x10\r\x12"\n\x1eOPERATOR_TYPE_DEFAULT_INJECTOR\x10\x0e\x12\x1d\n\x19OPERATOR_TYPE_DISTINCT_ON\x10\x0f\x12&\n"OPERATOR_TYPE_DROP_CHILD_INDEX_COL\x10\x10\x12\x1e\n\x1aOPERATOR_TYPE_DROP_COLUMNS\x10\x11\x12\'\n#OPERATOR_TYPE_DYNAMO_DBCACHE_LOOKUP\x10\x12\x12+\n\'OPERATOR_TYPE_EXPLODE_HAS_MANY_OPERATOR\x10\x13\x12)\n%OPERATOR_TYPE_EXTRA_TABLES_TO_SCALARS\x10\x14\x12 \n\x1cOPERATOR_TYPE_FILTER_MISSING\x10\x15\x12&\n"OPERATOR_TYPE_FILTER_RELEVANT_ROWS\x10\x45\x12\x1f\n\x1bOPERATOR_TYPE_FINAL_PROJECT\x10\x16\x12\x1d\n\x19OPERATOR_TYPE_GIVENS_SCAN\x10\x17\x12)\n%OPERATOR_TYPE_HAS_MANY_INPUT_OPERATOR\x10\x18\x12/\n+OPERATOR_TYPE_HAS_MANY_OUTPUT_JOIN_OPERATOR\x10\x19\x12\x1f\n\x1bOPERATOR_TYPE_IGNORE_OUTPUT\x10\x1a\x12@\n<OPERATOR_TYPE_INTERMEDIATE_FEATURE_METRIC_RECORDING_OPERATOR\x10\x1b\x12*\n&OPERATOR_TYPE_JOIN_SINGLETONS_OPERATOR\x10\x1c\x12-\n)OPERATOR_TYPE_LATERAL_JOIN_INPUT_OPERATOR\x10\x1d\x12.\n*OPERATOR_TYPE_LATERAL_JOIN_OUTPUT_OPERATOR\x10\x1e\x12\'\n#OPERATOR_TYPE_LATERAL_JOIN_RESOLVER\x10\x1f\x12&\n"OPERATOR_TYPE_LIFT_RESULT_TO_GROUP\x10 \x12.\n*OPERATOR_TYPE_LIGHTNING_REDIS_CACHE_LOOKUP\x10!\x12)\n!OPERATOR_TYPE_MERGE_JOIN_OPERATOR\x10"\x1a\x02\x08\x01\x12#\n\x1fOPERATOR_TYPE_METRICS_PUBLISHER\x10#\x12*\n&OPERATOR_TYPE_NARY_MERGE_JOIN_OPERATOR\x10\x46\x12\x30\n,OPERATOR_TYPE_NEAREST_NEIGHBOR_SUB_PLAN_JOIN\x10$\x12\x31\n-OPERATOR_TYPE_NEW_DATAFRAME_RESOLVER_OPERATOR\x10%\x12*\n&OPERATOR_TYPE_NEW_OFFLINE_CACHE_LOOKUP\x10&\x12*\n&OPERATOR_TYPE_NON_BUS_PERSIST_OPERATOR\x10\'\x12\'\n#OPERATOR_TYPE_OFFLINE_CACHE_SAMPLER\x10(\x12,\n(OPERATOR_TYPE_ONE_TO_ONE_SCALAR_RESOLVER\x10)\x12)\n%OPERATOR_TYPE_ONLINE_STORE_AGG_WRITER\x10*\x12&\n"OPERATOR_TYPE_ONLINE_VECTOR_SEARCH\x10+\x12!\n\x1dOPERATOR_TYPE_OPTIMISTIC_LOAD\x10,\x12,\n(OPERATOR_TYPE_OUTPUT_UNDERSCORE_OPERATOR\x10-\x12*\n&OPERATOR_TYPE_PACK_GROUPS_INTO_STRUCTS\x10.\x12 \n\x1cOPERATOR_TYPE_PARQUET_WRITER\x10/\x12*\n&OPERATOR_TYPE_PRELOADED_TABLE_OPERATOR\x10H\x12\x19\n\x15OPERATOR_TYPE_PROJECT\x10\x30\x12\x1f\n\x1bOPERATOR_TYPE_PROMOTE_SPINE\x10\x31\x12"\n\x1eOPERATOR_TYPE_PUSH_DFTO_RESULT\x10\x32\x12)\n%OPERATOR_TYPE_PUSH_HAS_MANY_TO_RESULT\x10\x33\x12\x30\n,OPERATOR_TYPE_REMOVE_LARGE_LISTS_FROM_SCHEMA\x10\x34\x12.\n*OPERATOR_TYPE_RENAME_INDEX_COLUMN_OPERATOR\x10\x35\x12\x18\n\x14OPERATOR_TYPE_REPLAY\x10\x36\x12\x30\n,OPERATOR_TYPE_RESULT_BUS_PERSIST_OPERATOR_V2\x10\x37\x12#\n\x1fOPERATOR_TYPE_RUN_IN_BACKGROUND\x10\x38\x12\x30\n,OPERATOR_TYPE_SQLDATAFRAME_RESOLVER_OPERATOR\x10\x39\x12\x33\n/OPERATOR_TYPE_SQLLATERAL_JOIN_RESOLVER_OPERATOR\x10:\x12-\n)OPERATOR_TYPE_SQLSCALAR_RESOLVER_OPERATOR\x10;\x12!\n\x1dOPERATOR_TYPE_STATS_COLLECTOR\x10<\x12.\n*OPERATOR_TYPE_STREAMING_AGG_WRITE_OPERATOR\x10=\x12!\n\x1dOPERATOR_TYPE_SUB_PLAN_FILTER\x10>\x12\x1f\n\x1bOPERATOR_TYPE_SUB_PLAN_JOIN\x10?\x12.\n*OPERATOR_TYPE_SYMBOLIC_EXPRESSION_FALLBACK\x10@\x12\x35\n1OPERATOR_TYPE_UNDERSCORE_SCALAR_RESOLVER_OPERATOR\x10\x41\x12 \n\x1cOPERATOR_TYPE_UNION_OPERATOR\x10\x42\x12(\n$OPERATOR_TYPE_UNLOADED_RESOLVER_SCAN\x10G\x12"\n\x1eOPERATOR_TYPE_VALUES_PERSISTER\x10\x43\x42\x8c\x01\n\x14\x63om.chalk.planner.v1B\x12\x42\x61tchOperatorProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x10\x43halk.Planner.V1\xca\x02\x10\x43halk\\Planner\\V1\xe2\x02\x1c\x43halk\\Planner\\V1\\GPBMetadata\xea\x02\x12\x43halk::Planner::V1b\x06proto3'
+    b'\n%chalk/planner/v1/batch_operator.proto\x12\x10\x63halk.planner.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto\x1a$chalk/expression/v1/expression.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a$chalk/planner/v1/feature_types.proto\x1a%chalk/planner/v1/symbolic_value.proto\x1a,chalk/symbolic_value/v1/symbolic_value.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\x8c\x03\n\tBatchPlan\x12=\n\toperators\x18\x01 \x03(\x0b\x32\x1f.chalk.planner.v1.BatchOperatorR\toperators\x12L\n\x0fsymbolic_values\x18\x02 \x03(\x0b\x32\x1f.chalk.planner.v1.SymbolicValueB\x02\x18\x01R\x0esymbolicValues\x12T\n\x10\x66\x65\x61ture_ref_info\x18\x03 \x01(\x0b\x32&.chalk.planner.v1.FeatureReferenceInfoB\x02\x18\x01R\x0e\x66\x65\x61tureRefInfo\x12\x46\n\x0e\x61uxiliary_info\x18\x04 \x01(\x0b\x32\x1f.chalk.planner.v1.AuxiliaryInfoR\rauxiliaryInfo\x12T\n\x12symbolic_values_v2\x18\x05 \x03(\x0b\x32&.chalk.symbolic_value.v1.SymbolicValueR\x10symbolicValuesV2"\xb8\x02\n\rBatchOperator\x12\x43\n\roperator_type\x18\x01 \x01(\x0e\x32\x1e.chalk.planner.v1.OperatorTypeR\x0coperatorType\x12:\n\ngraph_info\x18\x02 \x01(\x0b\x32\x1b.chalk.planner.v1.GraphInfoR\tgraphInfo\x12L\n\targuments\x18\x03 \x03(\x0b\x32..chalk.planner.v1.BatchOperator.ArgumentsEntryR\targuments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"\x85\x02\n\x14\x46\x65\x61tureReferenceInfo\x12\x45\n\x0c\x66\x65\x61ture_refs\x18\x01 \x03(\x0b\x32".chalk.planner.v1.FeatureReferenceR\x0b\x66\x65\x61tureRefs\x12I\n\x10\x64\x61ta_frame_types\x18\x02 \x03(\x0b\x32\x1f.chalk.planner.v1.DataFrameTypeR\x0e\x64\x61taFrameTypes\x12W\n\x12\x66ilter_expressions\x18\x03 \x03(\x0b\x32(.chalk.planner.v1.FilterExpressionParsedR\x11\x66ilterExpressions:\x02\x18\x01",\n\tGraphInfo\x12\x1f\n\x0boperator_id\x18\x01 \x01(\x04R\noperatorId"\x99\x0e\n\x08\x41rgument\x12,\n\x04none\x18\x01 \x01(\x0b\x32\x16.chalk.planner.v1.VoidH\x00R\x04none\x12\x1f\n\nbool_value\x18\x02 \x01(\x08H\x00R\tboolValue\x12#\n\x0c\x64ouble_value\x18\x03 \x01(\x01H\x00R\x0b\x64oubleValue\x12\x1d\n\tint_value\x18\x04 \x01(\x03H\x00R\x08intValue\x12#\n\x0cstring_value\x18\x05 \x01(\tH\x00R\x0bstringValue\x12:\n\ttimestamp\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\ttimestamp\x12\x37\n\x08\x64uration\x18\x07 \x01(\x0b\x32\x19.google.protobuf.DurationH\x00R\x08\x64uration\x12!\n\x0boperator_id\x18\x08 \x01(\x04H\x00R\noperatorId\x12G\n\x0b\x66\x65\x61ture_ref\x18\t \x01(\x0b\x32 .chalk.graph.v1.FeatureReferenceB\x02\x18\x01H\x00R\nfeatureRef\x12!\n\x0b\x62ytes_value\x18\n \x01(\x0cH\x00R\nbytesValue\x12\x36\n\x05tuple\x18\x0e \x01(\x0b\x32\x1e.chalk.planner.v1.ArgumentListH\x00R\x05tuple\x12\x37\n\x06submap\x18\x0f \x01(\x0b\x32\x1d.chalk.planner.v1.ArgumentMapH\x00R\x06submap\x12:\n\narrow_type\x18\x10 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeH\x00R\tarrowType\x12O\n\x0funderscore_expr\x18\x11 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\x0eunderscoreExpr\x12G\n\x0b\x66ilter_expr\x18\x12 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\nfilterExpr\x12n\n\x1c\x64\x65tached_column_feature_type\x18\x13 \x01(\x0b\x32+.chalk.planner.v1.DetachedColumnFeatureTypeH\x00R\x19\x64\x65tachedColumnFeatureType\x12t\n\x1eoutput_underscore_feature_type\x18\x14 \x01(\x0b\x32-.chalk.planner.v1.OutputUnderscoreFeatureTypeH\x00R\x1boutputUnderscoreFeatureType\x12M\n\x0f\x64\x61ta_frame_type\x18\x15 \x01(\x0b\x32\x1f.chalk.planner.v1.DataFrameTypeB\x02\x18\x01H\x00R\rdataFrameType\x12U\n\x11\x66\x65\x61ture_reference\x18\x16 \x01(\x0b\x32".chalk.planner.v1.FeatureReferenceB\x02\x18\x01H\x00R\x10\x66\x65\x61tureReference\x12L\n\x0esymbolic_value\x18\x17 \x01(\x0b\x32\x1f.chalk.planner.v1.SymbolicValueB\x02\x18\x01H\x00R\rsymbolicValue\x12T\n\x12\x64\x61ta_frame_type_id\x18\x18 \x01(\x0b\x32!.chalk.planner.v1.DataFrameTypeIdB\x02\x18\x01H\x00R\x0f\x64\x61taFrameTypeId\x12P\n\x0e\x66\x65\x61ture_ref_id\x18\x19 \x01(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdB\x02\x18\x01H\x00R\x0c\x66\x65\x61tureRefId\x12W\n\x15\x64\x61ta_frame_type_id_v2\x18\x1a \x01(\x0b\x32#.chalk.planner.v1.DataFrameTypeIdV2H\x00R\x11\x64\x61taFrameTypeIdV2\x12S\n\x11\x66\x65\x61ture_ref_id_v2\x18\x1b \x01(\x0b\x32&.chalk.planner.v1.FeatureReferenceIdV2H\x00R\x0e\x66\x65\x61tureRefIdV2\x12S\n\x11underscore_parsed\x18\x1c \x01(\x0b\x32$.chalk.planner.v1.UnderscoreParsedIdH\x00R\x10underscoreParsed\x12(\n\x0fipc_arrow_table\x18\x1d \x01(\x0cH\x00R\ripcArrowTable\x12T\n\x11symbolic_value_v2\x18\x1e \x01(\x0b\x32&.chalk.symbolic_value.v1.SymbolicValueH\x00R\x0fsymbolicValueV2B\x05\n\x03\x61rg"t\n\x12\x41rgumentMapElement\x12,\n\x03key\x18\x01 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value"\xf6\x02\n\x0b\x41rgumentMap\x12N\n\targuments\x18\x01 \x03(\x0b\x32,.chalk.planner.v1.ArgumentMap.ArgumentsEntryB\x02\x18\x01R\targuments\x12\x32\n\x04keys\x18\x02 \x03(\x0b\x32\x1a.chalk.planner.v1.ArgumentB\x02\x18\x01R\x04keys\x12\x36\n\x06values\x18\x03 \x03(\x0b\x32\x1a.chalk.planner.v1.ArgumentB\x02\x18\x01R\x06values\x12Q\n\x11ordered_arguments\x18\x04 \x03(\x0b\x32$.chalk.planner.v1.ArgumentMapElementR\x10orderedArguments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"B\n\x0c\x41rgumentList\x12\x32\n\x06values\x18\x01 \x03(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x06values"\x06\n\x04Void"\xcf\x01\n\x19\x44\x65tachedColumnFeatureType\x12X\n\targuments\x18\x01 \x03(\x0b\x32:.chalk.planner.v1.DetachedColumnFeatureType.ArgumentsEntryR\targuments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"\xd3\x01\n\x1bOutputUnderscoreFeatureType\x12Z\n\targuments\x18\x01 \x03(\x0b\x32<.chalk.planner.v1.OutputUnderscoreFeatureType.ArgumentsEntryR\targuments\x1aX\n\x0e\x41rgumentsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x30\n\x05value\x18\x02 \x01(\x0b\x32\x1a.chalk.planner.v1.ArgumentR\x05value:\x02\x38\x01"\xef\x01\n\x16\x46ilterExpressionParsed\x12\x43\n\x07this_id\x18\x01 \x01(\x0b\x32*.chalk.planner.v1.FilterExpressionParsedIdR\x06thisId\x12U\n\x11\x66ilter_expression\x18\x02 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeB\x02\x18\x01R\x10\x66ilterExpression\x12\x35\n\x04\x65xpr\x18\x03 \x01(\x0b\x32!.chalk.planner.v1.UnderscoreValueR\x04\x65xpr:\x02\x18\x01".\n\x18\x46ilterExpressionParsedId\x12\x0e\n\x02id\x18\x01 \x01(\x04R\x02id:\x02\x18\x01"\xef\x04\n\rDataFrameType\x12:\n\x07this_id\x18\x04 \x01(\x0b\x32!.chalk.planner.v1.DataFrameTypeIdR\x06thisId\x12-\n\x02\x64\x66\x18\x01 \x01(\x0b\x32\x1d.chalk.graph.v1.DataFrameTypeR\x02\x64\x66\x12Z\n\x11\x66ilter_expression\x18\x02 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeB\x02\x18\x01H\x00R\x10\x66ilterExpression\x88\x01\x01\x12\x61\n\x14\x66ilter_expression_id\x18\x05 \x01(\x0b\x32*.chalk.planner.v1.FilterExpressionParsedIdH\x01R\x12\x66ilterExpressionId\x88\x01\x01\x12Q\n\x10optional_columns\x18\x03 \x03(\x0b\x32".chalk.planner.v1.FeatureReferenceB\x02\x18\x01R\x0foptionalColumns\x12V\n\x14optional_column_refs\x18\x06 \x03(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x12optionalColumnRefs\x12V\n\x14required_column_refs\x18\x07 \x03(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x12requiredColumnRefs:\x02\x18\x01\x42\x14\n\x12_filter_expressionB\x17\n\x15_filter_expression_id"%\n\x0f\x44\x61taFrameTypeId\x12\x0e\n\x02id\x18\x01 \x01(\x04R\x02id:\x02\x18\x01"\x9d\x03\n\x10\x46\x65\x61tureReference\x12=\n\x07this_id\x18\x05 \x01(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x06thisId\x12\x41\n\x0b\x66\x65\x61ture_ref\x18\x01 \x01(\x0b\x32 .chalk.graph.v1.FeatureReferenceR\nfeatureRef\x12:\n\x04path\x18\x02 \x03(\x0b\x32".chalk.planner.v1.FeatureReferenceB\x02\x18\x01R\x04path\x12?\n\x08path_ids\x18\x04 \x03(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdR\x07pathIds\x12\x38\n\x02\x64\x66\x18\x03 \x01(\x0b\x32\x1f.chalk.planner.v1.DataFrameTypeB\x02\x18\x01H\x00R\x02\x64\x66\x88\x01\x01\x12;\n\x05\x64\x66_id\x18\x06 \x01(\x0b\x32!.chalk.planner.v1.DataFrameTypeIdH\x01R\x04\x64\x66Id\x88\x01\x01:\x02\x18\x01\x42\x05\n\x03_dfB\x08\n\x06_df_id"(\n\x12\x46\x65\x61tureReferenceId\x12\x0e\n\x02id\x18\x01 \x01(\x04R\x02id:\x02\x18\x01"\xb7\x01\n\x0fUnderscoreValue\x12U\n\x13original_underscore\x18\x01 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeR\x12originalUnderscore\x12I\n\ncodec_info\x18\x02 \x01(\x0b\x32*.chalk.planner.v1.UnderscoreValueCodecInfoR\tcodecInfo:\x02\x18\x01"\x85\x02\n\x18UnderscoreValueCodecInfo\x12J\n\x0b\x66or_feature\x18\x01 \x01(\x0b\x32$.chalk.planner.v1.FeatureReferenceIdH\x00R\nforFeature\x88\x01\x01\x12%\n\x0eroot_namespace\x18\x02 \x01(\tR\rrootNamespace\x12\x62\n\x18root_underscore_behavior\x18\x03 \x01(\x0b\x32(.chalk.planner.v1.RootUnderscoreBehaviorR\x16rootUnderscoreBehavior:\x02\x18\x01\x42\x0e\n\x0c_for_feature"\xd7\x01\n\x16RootUnderscoreBehavior\x12N\n\x08resolver\x18\x01 \x01(\x0b\x32\x30.chalk.planner.v1.ResolverRootUnderscoreBehaviorH\x00R\x08resolver\x12\x61\n\x0fstream_resolver\x18\x02 \x01(\x0b\x32\x36.chalk.planner.v1.StreamResolverRootUnderscoreBehaviorH\x00R\x0estreamResolverB\n\n\x08\x62\x65havior" \n\x1eResolverRootUnderscoreBehavior"f\n$StreamResolverRootUnderscoreBehavior\x12>\n\rmessage_dtype\x18\x01 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\x0cmessageDtype*\xad\x19\n\x0cOperatorType\x12\x1d\n\x19OPERATOR_TYPE_UNSPECIFIED\x10\x00\x12%\n!OPERATOR_TYPE_ADD_CHILD_INDEX_COL\x10\x01\x12\x1f\n\x1bOPERATOR_TYPE_ADD_INDEX_COL\x10\x02\x12$\n OPERATOR_TYPE_BATCH_AGG_OPERATOR\x10\x03\x12\'\n#OPERATOR_TYPE_BATCH_DWHAGG_OPERATOR\x10\x04\x12\x1e\n\x1aOPERATOR_TYPE_BATCH_RENAME\x10\x05\x12\'\n#OPERATOR_TYPE_BIGTABLE_CACHE_LOOKUP\x10\x06\x12/\n+OPERATOR_TYPE_BLOOM_FILTER_PERSIST_OPERATOR\x10L\x12\x1e\n\x1aOPERATOR_TYPE_CACHE_LOOKUP\x10\x07\x12\'\n#OPERATOR_TYPE_CACHE_LOOKUP_HAS_MANY\x10\x08\x12-\n)OPERATOR_TYPE_COMPUTE_CRON_OUTPUT_SUMMARY\x10\t\x12+\n\'OPERATOR_TYPE_COMPUTE_RELEVANT_ROW_MASK\x10\x44\x12"\n\x1eOPERATOR_TYPE_CPP_CACHE_LOOKUP\x10\n\x12*\n&OPERATOR_TYPE_DATAFRAME_MERGE_OPERATOR\x10\x0b\x12\x1e\n\x1aOPERATOR_TYPE_DATASET_SCAN\x10\x0c\x12(\n$OPERATOR_TYPE_DEBUG_COMMENT_OPERATOR\x10\r\x12"\n\x1eOPERATOR_TYPE_DEFAULT_INJECTOR\x10\x0e\x12\x1d\n\x19OPERATOR_TYPE_DISTINCT_ON\x10\x0f\x12&\n"OPERATOR_TYPE_DROP_CHILD_INDEX_COL\x10\x10\x12\x1e\n\x1aOPERATOR_TYPE_DROP_COLUMNS\x10\x11\x12\'\n#OPERATOR_TYPE_DYNAMO_DBCACHE_LOOKUP\x10\x12\x12+\n\'OPERATOR_TYPE_EXPLODE_HAS_MANY_OPERATOR\x10\x13\x12)\n%OPERATOR_TYPE_EXTRA_TABLES_TO_SCALARS\x10\x14\x12 \n\x1cOPERATOR_TYPE_FILTER_MISSING\x10\x15\x12&\n"OPERATOR_TYPE_FILTER_RELEVANT_ROWS\x10\x45\x12\x1f\n\x1bOPERATOR_TYPE_FINAL_PROJECT\x10\x16\x12\x1d\n\x19OPERATOR_TYPE_GIVENS_SCAN\x10\x17\x12)\n%OPERATOR_TYPE_HAS_MANY_INPUT_OPERATOR\x10\x18\x12/\n+OPERATOR_TYPE_HAS_MANY_OUTPUT_JOIN_OPERATOR\x10\x19\x12\x1f\n\x1bOPERATOR_TYPE_IGNORE_OUTPUT\x10\x1a\x12@\n<OPERATOR_TYPE_INTERMEDIATE_FEATURE_METRIC_RECORDING_OPERATOR\x10\x1b\x12*\n&OPERATOR_TYPE_JOIN_SINGLETONS_OPERATOR\x10\x1c\x12-\n)OPERATOR_TYPE_LATERAL_JOIN_INPUT_OPERATOR\x10\x1d\x12.\n*OPERATOR_TYPE_LATERAL_JOIN_OUTPUT_OPERATOR\x10\x1e\x12\'\n#OPERATOR_TYPE_LATERAL_JOIN_RESOLVER\x10\x1f\x12&\n"OPERATOR_TYPE_LIFT_RESULT_TO_GROUP\x10 \x12.\n*OPERATOR_TYPE_LIGHTNING_REDIS_CACHE_LOOKUP\x10!\x12)\n!OPERATOR_TYPE_MERGE_JOIN_OPERATOR\x10"\x1a\x02\x08\x01\x12#\n\x1fOPERATOR_TYPE_METRICS_PUBLISHER\x10#\x12*\n&OPERATOR_TYPE_NARY_MERGE_JOIN_OPERATOR\x10\x46\x12\x30\n,OPERATOR_TYPE_NEAREST_NEIGHBOR_SUB_PLAN_JOIN\x10$\x12\x31\n-OPERATOR_TYPE_NEW_DATAFRAME_RESOLVER_OPERATOR\x10%\x12*\n&OPERATOR_TYPE_NEW_OFFLINE_CACHE_LOOKUP\x10&\x12*\n&OPERATOR_TYPE_NON_BUS_PERSIST_OPERATOR\x10\'\x12\'\n#OPERATOR_TYPE_OFFLINE_CACHE_SAMPLER\x10(\x12,\n(OPERATOR_TYPE_ONE_TO_ONE_SCALAR_RESOLVER\x10)\x12)\n%OPERATOR_TYPE_ONLINE_PERSIST_OPERATOR\x10I\x12)\n%OPERATOR_TYPE_ONLINE_STORE_AGG_WRITER\x10*\x12&\n"OPERATOR_TYPE_ONLINE_VECTOR_SEARCH\x10+\x12!\n\x1dOPERATOR_TYPE_OPTIMISTIC_LOAD\x10,\x12,\n(OPERATOR_TYPE_OUTPUT_UNDERSCORE_OPERATOR\x10-\x12*\n&OPERATOR_TYPE_PACK_GROUPS_INTO_STRUCTS\x10.\x12 \n\x1cOPERATOR_TYPE_PARQUET_WRITER\x10/\x12*\n&OPERATOR_TYPE_PRELOADED_TABLE_OPERATOR\x10H\x12\x19\n\x15OPERATOR_TYPE_PROJECT\x10\x30\x12\x1f\n\x1bOPERATOR_TYPE_PROMOTE_SPINE\x10\x31\x12"\n\x1eOPERATOR_TYPE_PUSH_DFTO_RESULT\x10\x32\x12)\n%OPERATOR_TYPE_PUSH_HAS_MANY_TO_RESULT\x10\x33\x12\x30\n,OPERATOR_TYPE_REMOVE_LARGE_LISTS_FROM_SCHEMA\x10\x34\x12.\n*OPERATOR_TYPE_RENAME_INDEX_COLUMN_OPERATOR\x10\x35\x12\x18\n\x14OPERATOR_TYPE_REPLAY\x10\x36\x12\x30\n,OPERATOR_TYPE_RESULT_BUS_PERSIST_OPERATOR_V2\x10\x37\x12#\n\x1fOPERATOR_TYPE_RUN_IN_BACKGROUND\x10\x38\x12\x30\n,OPERATOR_TYPE_SQLDATAFRAME_RESOLVER_OPERATOR\x10\x39\x12\x33\n/OPERATOR_TYPE_SQLLATERAL_JOIN_RESOLVER_OPERATOR\x10:\x12-\n)OPERATOR_TYPE_SQLSCALAR_RESOLVER_OPERATOR\x10;\x12!\n\x1dOPERATOR_TYPE_STATS_COLLECTOR\x10<\x12.\n*OPERATOR_TYPE_STREAMING_AGG_WRITE_OPERATOR\x10=\x12!\n\x1dOPERATOR_TYPE_SUB_PLAN_FILTER\x10>\x12\x1f\n\x1bOPERATOR_TYPE_SUB_PLAN_JOIN\x10?\x12.\n*OPERATOR_TYPE_SYMBOLIC_EXPRESSION_FALLBACK\x10@\x12\x36\n2OPERATOR_TYPE_TABLE_WRITE_OFFLINE_PERSIST_OPERATOR\x10K\x12\x35\n1OPERATOR_TYPE_UNDERSCORE_SCALAR_RESOLVER_OPERATOR\x10\x41\x12 \n\x1cOPERATOR_TYPE_UNION_OPERATOR\x10\x42\x12(\n$OPERATOR_TYPE_UNLOADED_RESOLVER_SCAN\x10G\x12"\n\x1eOPERATOR_TYPE_VALUES_PERSISTER\x10\x43\x12)\n%OPERATOR_TYPE_VECTOR_PERSIST_OPERATOR\x10J\x12/\n+OPERATOR_TYPE_UNPACK_HAS_MANY_STRUCT_COLUMN\x10MB\x8c\x01\n\x14\x63om.chalk.planner.v1B\x12\x42\x61tchOperatorProtoP\x01\xa2\x02\x03\x43PX\xaa\x02\x10\x43halk.Planner.V1\xca\x02\x10\x43halk\\Planner\\V1\xe2\x02\x1c\x43halk\\Planner\\V1\\GPBMetadata\xea\x02\x12\x43halk::Planner::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -98,7 +98,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_UNDERSCOREVALUECODECINFO"]._options = None
     _globals["_UNDERSCOREVALUECODECINFO"]._serialized_options = b"\030\001"
     _globals["_OPERATORTYPE"]._serialized_start = 6400
-    _globals["_OPERATORTYPE"]._serialized_end = 9405
+    _globals["_OPERATORTYPE"]._serialized_end = 9645
     _globals["_BATCHPLAN"]._serialized_start = 342
     _globals["_BATCHPLAN"]._serialized_end = 738
     _globals["_BATCHOPERATOR"]._serialized_start = 741

{chalkpy-2.154.2 → chalkpy-2.154.4}/chalk/_gen/chalk/planner/v1/batch_operator_pb2.pyi

@@ -29,6 +29,7 @@ class OperatorType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     OPERATOR_TYPE_BATCH_DWHAGG_OPERATOR: _ClassVar[OperatorType]
     OPERATOR_TYPE_BATCH_RENAME: _ClassVar[OperatorType]
     OPERATOR_TYPE_BIGTABLE_CACHE_LOOKUP: _ClassVar[OperatorType]
+    OPERATOR_TYPE_BLOOM_FILTER_PERSIST_OPERATOR: _ClassVar[OperatorType]
     OPERATOR_TYPE_CACHE_LOOKUP: _ClassVar[OperatorType]
     OPERATOR_TYPE_CACHE_LOOKUP_HAS_MANY: _ClassVar[OperatorType]
     OPERATOR_TYPE_COMPUTE_CRON_OUTPUT_SUMMARY: _ClassVar[OperatorType]
@@ -67,6 +68,7 @@ class OperatorType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     OPERATOR_TYPE_NON_BUS_PERSIST_OPERATOR: _ClassVar[OperatorType]
     OPERATOR_TYPE_OFFLINE_CACHE_SAMPLER: _ClassVar[OperatorType]
     OPERATOR_TYPE_ONE_TO_ONE_SCALAR_RESOLVER: _ClassVar[OperatorType]
+    OPERATOR_TYPE_ONLINE_PERSIST_OPERATOR: _ClassVar[OperatorType]
     OPERATOR_TYPE_ONLINE_STORE_AGG_WRITER: _ClassVar[OperatorType]
     OPERATOR_TYPE_ONLINE_VECTOR_SEARCH: _ClassVar[OperatorType]
     OPERATOR_TYPE_OPTIMISTIC_LOAD: _ClassVar[OperatorType]
@@ -91,10 +93,13 @@ class OperatorType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     OPERATOR_TYPE_SUB_PLAN_FILTER: _ClassVar[OperatorType]
     OPERATOR_TYPE_SUB_PLAN_JOIN: _ClassVar[OperatorType]
     OPERATOR_TYPE_SYMBOLIC_EXPRESSION_FALLBACK: _ClassVar[OperatorType]
+    OPERATOR_TYPE_TABLE_WRITE_OFFLINE_PERSIST_OPERATOR: _ClassVar[OperatorType]
     OPERATOR_TYPE_UNDERSCORE_SCALAR_RESOLVER_OPERATOR: _ClassVar[OperatorType]
     OPERATOR_TYPE_UNION_OPERATOR: _ClassVar[OperatorType]
     OPERATOR_TYPE_UNLOADED_RESOLVER_SCAN: _ClassVar[OperatorType]
     OPERATOR_TYPE_VALUES_PERSISTER: _ClassVar[OperatorType]
+    OPERATOR_TYPE_VECTOR_PERSIST_OPERATOR: _ClassVar[OperatorType]
+    OPERATOR_TYPE_UNPACK_HAS_MANY_STRUCT_COLUMN: _ClassVar[OperatorType]
 
 OPERATOR_TYPE_UNSPECIFIED: OperatorType
 OPERATOR_TYPE_ADD_CHILD_INDEX_COL: OperatorType
@@ -103,6 +108,7 @@ OPERATOR_TYPE_BATCH_AGG_OPERATOR: OperatorType
 OPERATOR_TYPE_BATCH_DWHAGG_OPERATOR: OperatorType
 OPERATOR_TYPE_BATCH_RENAME: OperatorType
 OPERATOR_TYPE_BIGTABLE_CACHE_LOOKUP: OperatorType
+OPERATOR_TYPE_BLOOM_FILTER_PERSIST_OPERATOR: OperatorType
 OPERATOR_TYPE_CACHE_LOOKUP: OperatorType
 OPERATOR_TYPE_CACHE_LOOKUP_HAS_MANY: OperatorType
 OPERATOR_TYPE_COMPUTE_CRON_OUTPUT_SUMMARY: OperatorType
@@ -141,6 +147,7 @@ OPERATOR_TYPE_NEW_OFFLINE_CACHE_LOOKUP: OperatorType
 OPERATOR_TYPE_NON_BUS_PERSIST_OPERATOR: OperatorType
 OPERATOR_TYPE_OFFLINE_CACHE_SAMPLER: OperatorType
 OPERATOR_TYPE_ONE_TO_ONE_SCALAR_RESOLVER: OperatorType
+OPERATOR_TYPE_ONLINE_PERSIST_OPERATOR: OperatorType
 OPERATOR_TYPE_ONLINE_STORE_AGG_WRITER: OperatorType
 OPERATOR_TYPE_ONLINE_VECTOR_SEARCH: OperatorType
 OPERATOR_TYPE_OPTIMISTIC_LOAD: OperatorType
@@ -165,10 +172,13 @@ OPERATOR_TYPE_STREAMING_AGG_WRITE_OPERATOR: OperatorType
 OPERATOR_TYPE_SUB_PLAN_FILTER: OperatorType
 OPERATOR_TYPE_SUB_PLAN_JOIN: OperatorType
 OPERATOR_TYPE_SYMBOLIC_EXPRESSION_FALLBACK: OperatorType
+OPERATOR_TYPE_TABLE_WRITE_OFFLINE_PERSIST_OPERATOR: OperatorType
 OPERATOR_TYPE_UNDERSCORE_SCALAR_RESOLVER_OPERATOR: OperatorType
 OPERATOR_TYPE_UNION_OPERATOR: OperatorType
 OPERATOR_TYPE_UNLOADED_RESOLVER_SCAN: OperatorType
 OPERATOR_TYPE_VALUES_PERSISTER: OperatorType
+OPERATOR_TYPE_VECTOR_PERSIST_OPERATOR: OperatorType
+OPERATOR_TYPE_UNPACK_HAS_MANY_STRUCT_COLUMN: OperatorType
 
 class BatchPlan(_message.Message):
     __slots__ = ("operators", "symbolic_values", "feature_ref_info", "auxiliary_info", "symbolic_values_v2")

{chalkpy-2.154.2 → chalkpy-2.154.4}/chalk/_gen/chalk/server/v1/graph_pb2.py

@@ -22,7 +22,7 @@ from chalk._gen.chalk.graph.v1 import graph_pb2 as chalk_dot_graph_dot_v1_dot_gr
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1b\x63halk/server/v1/graph.proto\x12\x0f\x63halk.server.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto\x1a\x1d\x63halk/artifacts/v1/diff.proto\x1a\x1f\x63halk/artifacts/v1/export.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto\x1a\x1a\x63halk/graph/v1/graph.proto"\xd8\x04\n\nFeatureSQL\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x10\n\x03\x66qn\x18\x04 \x01(\tR\x03\x66qn\x12\x12\n\x04name\x18\x05 \x01(\tR\x04name\x12\x1c\n\tnamespace\x18\x06 \x01(\tR\tnamespace\x12(\n\rmax_staleness\x18\x07 \x01(\tH\x01R\x0cmaxStaleness\x88\x01\x01\x12\x31\n\x15\x65tl_offline_to_online\x18\x08 \x01(\x08R\x12\x65tlOfflineToOnline\x12%\n\x0b\x64\x65scription\x18\t \x01(\tH\x02R\x0b\x64\x65scription\x88\x01\x01\x12\x19\n\x05owner\x18\n \x01(\tH\x03R\x05owner\x88\x01\x01\x12\x12\n\x04tags\x18\x0b \x03(\tR\x04tags\x12\x1b\n\tkind_enum\x18\x0c \x01(\tR\x08kindEnum\x12\x12\n\x04kind\x18\r \x01(\tR\x04kind\x12\x1b\n\twas_reset\x18\x0e \x01(\x08R\x08wasReset\x12.\n\x10internal_version\x18\x0f \x01(\x03H\x04R\x0finternalVersion\x88\x01\x01\x12!\n\x0cis_singleton\x18\x10 \x01(\x08R\x0bisSingletonB\x10\n\x0e_deployment_idB\x10\n\x0e_max_stalenessB\x0e\n\x0c_descriptionB\x08\n\x06_ownerB\x13\n\x11_internal_version"P\n\x15GetFeatureSQLResponse\x12\x37\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.FeatureSQLR\x08\x66\x65\x61tures";\n\x14GetFeatureSQLRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId"\x86\x03\n\x0f\x46\x65\x61tureMetadata\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12%\n\x0b\x64\x65scription\x18\x04 \x01(\tH\x00R\x0b\x64\x65scription\x88\x01\x01\x12\x19\n\x05owner\x18\x05 \x01(\tH\x01R\x05owner\x88\x01\x01\x12\x12\n\x04tags\x18\x06 \x03(\tR\x04tags\x12(\n\rmax_staleness\x18\x07 \x01(\tH\x02R\x0cmaxStaleness\x88\x01\x01\x12\x31\n\x15\x65tl_offline_to_online\x18\x08 \x01(\x08R\x12\x65tlOfflineToOnline\x12\x34\n\x08pa_dtype\x18\t \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\x07paDtype\x12\x1a\n\x08nullable\x18\n \x01(\x08R\x08nullableB\x0e\n\x0c_descriptionB\x08\n\x06_ownerB\x10\n\x0e_max_staleness"\xa7\x01\n\x1bGetFeaturesMetadataResponse\x12<\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32 .chalk.server.v1.FeatureMetadataR\x08\x66\x65\x61tures\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId"=\n\x1aGetFeaturesMetadataRequest\x12\x1f\n\x0b\x66qns_filter\x18\x01 \x03(\tR\nfqnsFilter"\xe6\x01\n\x12UpdateGraphRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12/\n\x05graph\x18\x02 \x01(\x0b\x32\x15.chalk.graph.v1.GraphB\x02\x18\x01R\x05graph\x12\'\n\x0f\x63halkpy_version\x18\x03 \x01(\tR\x0e\x63halkpyVersion\x12\x15\n\x03tag\x18\x04 \x01(\tH\x00R\x03tag\x88\x01\x01\x12\x32\n\x06\x65xport\x18\x05 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\x06\x65xportB\x06\n\x04_tag"\x15\n\x13UpdateGraphResponse"6\n\x0fGetGraphRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId"\xe4\x01\n\x10GetGraphResponse\x12/\n\x05graph\x18\x01 \x01(\x0b\x32\x15.chalk.graph.v1.GraphB\x02\x18\x01R\x05graph\x12\'\n\x0f\x63halkpy_version\x18\x02 \x01(\tR\x0e\x63halkpyVersion\x12\x15\n\x03tag\x18\x03 \x01(\tH\x00R\x03tag\x88\x01\x01\x12\x32\n\x06\x65xport\x18\x04 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\x06\x65xport\x12#\n\rdeployment_id\x18\x05 \x01(\tR\x0c\x64\x65ploymentIdB\x06\n\x04_tag"Q\n\rPythonVersion\x12\x14\n\x05major\x18\x01 \x01(\x03R\x05major\x12\x14\n\x05minor\x18\x02 \x01(\x03R\x05minor\x12\x14\n\x05patch\x18\x03 \x01(\x03R\x05patch"\xe7\x01\n"GetCodegenFeaturesFromGraphRequest\x12(\n\rdeployment_id\x18\x01 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1b\n\x06\x62ranch\x18\x02 \x01(\tH\x01R\x06\x62ranch\x88\x01\x01\x12J\n\x0epython_version\x18\x03 \x01(\x0b\x32\x1e.chalk.server.v1.PythonVersionH\x02R\rpythonVersion\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_branchB\x11\n\x0f_python_version"t\n#GetCodegenFeaturesFromGraphResponse\x12\x18\n\x07\x63odegen\x18\x01 \x01(\tR\x07\x63odegen\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\xe7\x07\n\rGraphMutation\x12]\n\x16\x61\x64\x64_streaming_resolver\x18\x01 \x01(\x0b\x32%.chalk.server.v1.AddStreamingResolverH\x00R\x14\x61\x64\x64StreamingResolver\x12\x66\n\x19update_streaming_resolver\x18\x02 \x01(\x0b\x32(.chalk.server.v1.UpdateStreamingResolverH\x00R\x17updateStreamingResolver\x12\x66\n\x19\x64\x65lete_streaming_resolver\x18\x03 \x01(\x0b\x32(.chalk.server.v1.DeleteStreamingResolverH\x00R\x17\x64\x65leteStreamingResolver\x12>\n\x0b\x61\x64\x64_feature\x18\x04 \x01(\x0b\x32\x1b.chalk.server.v1.AddFeatureH\x00R\naddFeature\x12G\n\x0eupdate_feature\x18\x05 \x01(\x0b\x32\x1e.chalk.server.v1.UpdateFeatureH\x00R\rupdateFeature\x12G\n\x0e\x64\x65lete_feature\x18\x06 \x01(\x0b\x32\x1e.chalk.server.v1.DeleteFeatureH\x00R\rdeleteFeature\x12H\n\x0f\x61\x64\x64_feature_set\x18\x07 \x01(\x0b\x32\x1e.chalk.server.v1.AddFeatureSetH\x00R\raddFeatureSet\x12Q\n\x12update_feature_set\x18\x08 \x01(\x0b\x32!.chalk.server.v1.UpdateFeatureSetH\x00R\x10updateFeatureSet\x12Q\n\x12\x64\x65lete_feature_set\x18\t \x01(\x0b\x32!.chalk.server.v1.DeleteFeatureSetH\x00R\x10\x64\x65leteFeatureSet\x12\x41\n\x0c\x61\x64\x64_resolver\x18\n \x01(\x0b\x32\x1c.chalk.server.v1.AddResolverH\x00R\x0b\x61\x64\x64Resolver\x12J\n\x0fupdate_resolver\x18\x0b \x01(\x0b\x32\x1f.chalk.server.v1.UpdateResolverH\x00R\x0eupdateResolver\x12J\n\x0f\x64\x65lete_resolver\x18\x0c \x01(\x0b\x32\x1f.chalk.server.v1.DeleteResolverH\x00R\x0e\x64\x65leteResolverB\n\n\x08mutation"R\n\x14\x41\x64\x64StreamingResolver\x12:\n\x08resolver\x18\x01 \x01(\x0b\x32\x1e.chalk.graph.v1.StreamResolverR\x08resolver"z\n\x17UpdateStreamingResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName\x12:\n\x08resolver\x18\x02 \x01(\x0b\x32\x1e.chalk.graph.v1.StreamResolverR\x08resolver">\n\x17\x44\x65leteStreamingResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName"C\n\x0b\x41\x64\x64Resolver\x12\x34\n\x08resolver\x18\x01 \x01(\x0b\x32\x18.chalk.graph.v1.ResolverR\x08resolver"k\n\x0eUpdateResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName\x12\x34\n\x08resolver\x18\x02 \x01(\x0b\x32\x18.chalk.graph.v1.ResolverR\x08resolver"5\n\x0e\x44\x65leteResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName"m\n\nAddFeature\x12(\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\tR\x0e\x66\x65\x61tureSetName\x12\x35\n\x07\x66\x65\x61ture\x18\x02 \x01(\x0b\x32\x1b.chalk.graph.v1.FeatureTypeR\x07\x66\x65\x61ture"X\n\rUpdateFeature\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn\x12\x35\n\x07\x66\x65\x61ture\x18\x02 \x01(\x0b\x32\x1b.chalk.graph.v1.FeatureTypeR\x07\x66\x65\x61ture"!\n\rDeleteFeature\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn"L\n\rAddFeatureSet\x12;\n\x0b\x66\x65\x61ture_set\x18\x01 \x01(\x0b\x32\x1a.chalk.graph.v1.FeatureSetR\nfeatureSet"y\n\x10UpdateFeatureSet\x12(\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\tR\x0e\x66\x65\x61tureSetName\x12;\n\x0b\x66\x65\x61ture_set\x18\x02 \x01(\x0b\x32\x1a.chalk.graph.v1.FeatureSetR\nfeatureSet"<\n\x10\x44\x65leteFeatureSet\x12(\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\tR\x0e\x66\x65\x61tureSetName"}\n\x18\x41pplyGraphUpdatesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\tmutations\x18\x02 \x03(\x0b\x32\x1e.chalk.server.v1.GraphMutationR\tmutations"\x1b\n\x19\x41pplyGraphUpdatesResponse"~\n\x19TestGraphMutationsRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\tmutations\x18\x02 \x03(\x0b\x32\x1e.chalk.server.v1.GraphMutationR\tmutations"\x85\x01\n\x1aTestGraphMutationsResponse\x12\x32\n\x06\x65xport\x18\x01 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\x06\x65xport\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"&\n\nColumnList\x12\x18\n\x07\x63olumns\x18\x01 \x03(\tR\x07\x63olumns"\xb1\x01\n\x0cTableLineage\x12G\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32+.chalk.server.v1.TableLineage.FeaturesEntryR\x08\x66\x65\x61tures\x1aX\n\rFeaturesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x31\n\x05value\x18\x02 \x01(\x0b\x32\x1b.chalk.server.v1.ColumnListR\x05value:\x02\x38\x01"\xb5\x01\n\x11\x44\x61taSourceLineage\x12\x46\n\x06tables\x18\x01 \x03(\x0b\x32..chalk.server.v1.DataSourceLineage.TablesEntryR\x06tables\x1aX\n\x0bTablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x33\n\x05value\x18\x02 \x01(\x0b\x32\x1d.chalk.server.v1.TableLineageR\x05value:\x02\x38\x01"\xd3\x01\n\x13ResolverDataLineage\x12X\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32\x35.chalk.server.v1.ResolverDataLineage.DataSourcesEntryR\x0b\x64\x61taSources\x1a\x62\n\x10\x44\x61taSourcesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32".chalk.server.v1.DataSourceLineageR\x05value:\x02\x38\x01"\x1c\n\x1aGetDataLineageIndexRequest"\x86\x02\n\x1bGetDataLineageIndexResponse\x12y\n\x15resolver_data_lineage\x18\x01 \x03(\x0b\x32\x45.chalk.server.v1.GetDataLineageIndexResponse.ResolverDataLineageEntryR\x13resolverDataLineage\x1al\n\x18ResolverDataLineageEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12:\n\x05value\x18\x02 \x01(\x0b\x32$.chalk.server.v1.ResolverDataLineageR\x05value:\x02\x38\x01"j\n\x0cOfflineTable\x12)\n\x10internal_version\x18\x01 \x01(\x05R\x0finternalVersion\x12\x1d\n\ntable_name\x18\x02 \x01(\tR\ttableName\x12\x10\n\x03\x66qn\x18\x03 \x01(\tR\x03\x66qn"_\n\x1bGetOfflineStoreTableRequest\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn\x12 \n\tbranch_id\x18\x02 \x01(\tH\x00R\x08\x62ranchId\x88\x01\x01\x42\x0c\n\n_branch_id"U\n\x1cGetOfflineStoreTableResponse\x12\x35\n\x06tables\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.OfflineTableR\x06tables"v\n\x1fGetAllOfflineStoreTablesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12 \n\tbranch_id\x18\x02 \x01(\tH\x00R\x08\x62ranchId\x88\x01\x01\x42\x0c\n\n_branch_id"Y\n GetAllOfflineStoreTablesResponse\x12\x35\n\x06tables\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.OfflineTableR\x06tables"\xd0\x01\n\x16\x44iffDeploymentsRequest\x12\x35\n\x14\x64\x65ployment_id_before\x18\x01 \x01(\tH\x00R\x12\x64\x65ploymentIdBefore\x88\x01\x01\x12.\n\x13\x64\x65ployment_id_after\x18\x02 \x01(\tR\x11\x64\x65ploymentIdAfter\x12\x36\n\tdiff_mode\x18\x03 \x01(\x0e\x32\x19.chalk.server.v1.DiffModeR\x08\x64iffModeB\x17\n\x15_deployment_id_before"\xb9\x01\n\x17\x44iffDeploymentsResponse\x12-\n\x10\x64\x65ploy_id_before\x18\x01 \x01(\tH\x00R\x0e\x64\x65ployIdBefore\x88\x01\x01\x12&\n\x0f\x64\x65ploy_id_after\x18\x02 \x01(\tR\rdeployIdAfter\x12\x32\n\x04\x64iff\x18\x03 \x01(\x0b\x32\x1e.chalk.artifacts.v1.ExportDiffR\x04\x64iffB\x13\n\x11_deploy_id_before"y\n\x1aSmartDiffDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x36\n\tdiff_mode\x18\x02 \x01(\x0e\x32\x19.chalk.server.v1.DiffModeR\x08\x64iffMode"\xbd\x01\n\x1bSmartDiffDeploymentResponse\x12-\n\x10\x64\x65ploy_id_before\x18\x01 \x01(\tH\x00R\x0e\x64\x65ployIdBefore\x88\x01\x01\x12&\n\x0f\x64\x65ploy_id_after\x18\x02 \x01(\tR\rdeployIdAfter\x12\x32\n\x04\x64iff\x18\x03 \x01(\x0b\x32\x1e.chalk.artifacts.v1.ExportDiffR\x04\x64iffB\x13\n\x11_deploy_id_before"\x88\x01\n\x14\x44iffCandidateRequest\x12\x38\n\tcandidate\x18\x01 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\tcandidate\x12\x36\n\tdiff_mode\x18\x02 \x01(\x0e\x32\x19.chalk.server.v1.DiffModeR\x08\x64iffMode"\x8f\x01\n\x15\x44iffCandidateResponse\x12-\n\x10\x64\x65ploy_id_before\x18\x01 \x01(\tH\x00R\x0e\x64\x65ployIdBefore\x88\x01\x01\x12\x32\n\x04\x64iff\x18\x02 \x01(\x0b\x32\x1e.chalk.artifacts.v1.ExportDiffR\x04\x64iffB\x13\n\x11_deploy_id_before*O\n\x08\x44iffMode\x12\x19\n\x15\x44IFF_MODE_UNSPECIFIED\x10\x00\x12\x12\n\x0e\x44IFF_MODE_FULL\x10\x01\x12\x14\n\x10\x44IFF_MODE_SIMPLE\x10\x02\x32\xf4\x0b\n\x0cGraphService\x12\x66\n\rGetFeatureSQL\x12%.chalk.server.v1.GetFeatureSQLRequest\x1a&.chalk.server.v1.GetFeatureSQLResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13GetFeaturesMetadata\x12+.chalk.server.v1.GetFeaturesMetadataRequest\x1a,.chalk.server.v1.GetFeaturesMetadataResponse"\x06\x90\x02\x01\x80}\x0b\x12W\n\x08GetGraph\x12 .chalk.server.v1.GetGraphRequest\x1a!.chalk.server.v1.GetGraphResponse"\x06\x90\x02\x01\x80}\x0b\x12]\n\x0bUpdateGraph\x12#.chalk.server.v1.UpdateGraphRequest\x1a$.chalk.server.v1.UpdateGraphResponse"\x03\x80}\x0c\x12\x90\x01\n\x1bGetCodegenFeaturesFromGraph\x12\x33.chalk.server.v1.GetCodegenFeaturesFromGraphRequest\x1a\x34.chalk.server.v1.GetCodegenFeaturesFromGraphResponse"\x06\x90\x02\x01\x80}\x0b\x12o\n\x11\x41pplyGraphUpdates\x12).chalk.server.v1.ApplyGraphUpdatesRequest\x1a*.chalk.server.v1.ApplyGraphUpdatesResponse"\x03\x80}\x0c\x12u\n\x12TestGraphMutations\x12*.chalk.server.v1.TestGraphMutationsRequest\x1a+.chalk.server.v1.TestGraphMutationsResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13GetDataLineageIndex\x12+.chalk.server.v1.GetDataLineageIndexRequest\x1a,.chalk.server.v1.GetDataLineageIndexResponse"\x06\x90\x02\x01\x80}\x0b\x12{\n\x14GetOfflineStoreTable\x12,.chalk.server.v1.GetOfflineStoreTableRequest\x1a-.chalk.server.v1.GetOfflineStoreTableResponse"\x06\x90\x02\x01\x80}\x0b\x12\x87\x01\n\x18GetAllOfflineStoreTables\x12\x30.chalk.server.v1.GetAllOfflineStoreTablesRequest\x1a\x31.chalk.server.v1.GetAllOfflineStoreTablesResponse"\x06\x90\x02\x01\x80}\x0b\x12l\n\x0f\x44iffDeployments\x12\'.chalk.server.v1.DiffDeploymentsRequest\x1a(.chalk.server.v1.DiffDeploymentsResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13SmartDiffDeployment\x12+.chalk.server.v1.SmartDiffDeploymentRequest\x1a,.chalk.server.v1.SmartDiffDeploymentResponse"\x06\x90\x02\x01\x80}\x0b\x12\x66\n\rDiffCandidate\x12%.chalk.server.v1.DiffCandidateRequest\x1a&.chalk.server.v1.DiffCandidateResponse"\x06\x90\x02\x01\x80}\x0b\x42\x93\x01\n\x13\x63om.chalk.server.v1B\nGraphProtoP\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\x1b\x63halk/server/v1/graph.proto\x12\x0f\x63halk.server.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto\x1a\x1d\x63halk/artifacts/v1/diff.proto\x1a\x1f\x63halk/artifacts/v1/export.proto\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a!chalk/common/v1/chalk_error.proto\x1a\x1a\x63halk/graph/v1/graph.proto"\xd8\x04\n\nFeatureSQL\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x10\n\x03\x66qn\x18\x04 \x01(\tR\x03\x66qn\x12\x12\n\x04name\x18\x05 \x01(\tR\x04name\x12\x1c\n\tnamespace\x18\x06 \x01(\tR\tnamespace\x12(\n\rmax_staleness\x18\x07 \x01(\tH\x01R\x0cmaxStaleness\x88\x01\x01\x12\x31\n\x15\x65tl_offline_to_online\x18\x08 \x01(\x08R\x12\x65tlOfflineToOnline\x12%\n\x0b\x64\x65scription\x18\t \x01(\tH\x02R\x0b\x64\x65scription\x88\x01\x01\x12\x19\n\x05owner\x18\n \x01(\tH\x03R\x05owner\x88\x01\x01\x12\x12\n\x04tags\x18\x0b \x03(\tR\x04tags\x12\x1b\n\tkind_enum\x18\x0c \x01(\tR\x08kindEnum\x12\x12\n\x04kind\x18\r \x01(\tR\x04kind\x12\x1b\n\twas_reset\x18\x0e \x01(\x08R\x08wasReset\x12.\n\x10internal_version\x18\x0f \x01(\x03H\x04R\x0finternalVersion\x88\x01\x01\x12!\n\x0cis_singleton\x18\x10 \x01(\x08R\x0bisSingletonB\x10\n\x0e_deployment_idB\x10\n\x0e_max_stalenessB\x0e\n\x0c_descriptionB\x08\n\x06_ownerB\x13\n\x11_internal_version"P\n\x15GetFeatureSQLResponse\x12\x37\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32\x1b.chalk.server.v1.FeatureSQLR\x08\x66\x65\x61tures";\n\x14GetFeatureSQLRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId"\x86\x03\n\x0f\x46\x65\x61tureMetadata\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12%\n\x0b\x64\x65scription\x18\x04 \x01(\tH\x00R\x0b\x64\x65scription\x88\x01\x01\x12\x19\n\x05owner\x18\x05 \x01(\tH\x01R\x05owner\x88\x01\x01\x12\x12\n\x04tags\x18\x06 \x03(\tR\x04tags\x12(\n\rmax_staleness\x18\x07 \x01(\tH\x02R\x0cmaxStaleness\x88\x01\x01\x12\x31\n\x15\x65tl_offline_to_online\x18\x08 \x01(\x08R\x12\x65tlOfflineToOnline\x12\x34\n\x08pa_dtype\x18\t \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\x07paDtype\x12\x1a\n\x08nullable\x18\n \x01(\x08R\x08nullableB\x0e\n\x0c_descriptionB\x08\n\x06_ownerB\x10\n\x0e_max_staleness"\xa7\x01\n\x1bGetFeaturesMetadataResponse\x12<\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32 .chalk.server.v1.FeatureMetadataR\x08\x66\x65\x61tures\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId"=\n\x1aGetFeaturesMetadataRequest\x12\x1f\n\x0b\x66qns_filter\x18\x01 \x03(\tR\nfqnsFilter"\xe6\x01\n\x12UpdateGraphRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12/\n\x05graph\x18\x02 \x01(\x0b\x32\x15.chalk.graph.v1.GraphB\x02\x18\x01R\x05graph\x12\'\n\x0f\x63halkpy_version\x18\x03 \x01(\tR\x0e\x63halkpyVersion\x12\x15\n\x03tag\x18\x04 \x01(\tH\x00R\x03tag\x88\x01\x01\x12\x32\n\x06\x65xport\x18\x05 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\x06\x65xportB\x06\n\x04_tag"\x15\n\x13UpdateGraphResponse"6\n\x0fGetGraphRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId"\xe4\x01\n\x10GetGraphResponse\x12/\n\x05graph\x18\x01 \x01(\x0b\x32\x15.chalk.graph.v1.GraphB\x02\x18\x01R\x05graph\x12\'\n\x0f\x63halkpy_version\x18\x02 \x01(\tR\x0e\x63halkpyVersion\x12\x15\n\x03tag\x18\x03 \x01(\tH\x00R\x03tag\x88\x01\x01\x12\x32\n\x06\x65xport\x18\x04 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\x06\x65xport\x12#\n\rdeployment_id\x18\x05 \x01(\tR\x0c\x64\x65ploymentIdB\x06\n\x04_tag"Q\n\rPythonVersion\x12\x14\n\x05major\x18\x01 \x01(\x03R\x05major\x12\x14\n\x05minor\x18\x02 \x01(\x03R\x05minor\x12\x14\n\x05patch\x18\x03 \x01(\x03R\x05patch"\xe7\x01\n"GetCodegenFeaturesFromGraphRequest\x12(\n\rdeployment_id\x18\x01 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1b\n\x06\x62ranch\x18\x02 \x01(\tH\x01R\x06\x62ranch\x88\x01\x01\x12J\n\x0epython_version\x18\x03 \x01(\x0b\x32\x1e.chalk.server.v1.PythonVersionH\x02R\rpythonVersion\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_branchB\x11\n\x0f_python_version"t\n#GetCodegenFeaturesFromGraphResponse\x12\x18\n\x07\x63odegen\x18\x01 \x01(\tR\x07\x63odegen\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"\xe7\x07\n\rGraphMutation\x12]\n\x16\x61\x64\x64_streaming_resolver\x18\x01 \x01(\x0b\x32%.chalk.server.v1.AddStreamingResolverH\x00R\x14\x61\x64\x64StreamingResolver\x12\x66\n\x19update_streaming_resolver\x18\x02 \x01(\x0b\x32(.chalk.server.v1.UpdateStreamingResolverH\x00R\x17updateStreamingResolver\x12\x66\n\x19\x64\x65lete_streaming_resolver\x18\x03 \x01(\x0b\x32(.chalk.server.v1.DeleteStreamingResolverH\x00R\x17\x64\x65leteStreamingResolver\x12>\n\x0b\x61\x64\x64_feature\x18\x04 \x01(\x0b\x32\x1b.chalk.server.v1.AddFeatureH\x00R\naddFeature\x12G\n\x0eupdate_feature\x18\x05 \x01(\x0b\x32\x1e.chalk.server.v1.UpdateFeatureH\x00R\rupdateFeature\x12G\n\x0e\x64\x65lete_feature\x18\x06 \x01(\x0b\x32\x1e.chalk.server.v1.DeleteFeatureH\x00R\rdeleteFeature\x12H\n\x0f\x61\x64\x64_feature_set\x18\x07 \x01(\x0b\x32\x1e.chalk.server.v1.AddFeatureSetH\x00R\raddFeatureSet\x12Q\n\x12update_feature_set\x18\x08 \x01(\x0b\x32!.chalk.server.v1.UpdateFeatureSetH\x00R\x10updateFeatureSet\x12Q\n\x12\x64\x65lete_feature_set\x18\t \x01(\x0b\x32!.chalk.server.v1.DeleteFeatureSetH\x00R\x10\x64\x65leteFeatureSet\x12\x41\n\x0c\x61\x64\x64_resolver\x18\n \x01(\x0b\x32\x1c.chalk.server.v1.AddResolverH\x00R\x0b\x61\x64\x64Resolver\x12J\n\x0fupdate_resolver\x18\x0b \x01(\x0b\x32\x1f.chalk.server.v1.UpdateResolverH\x00R\x0eupdateResolver\x12J\n\x0f\x64\x65lete_resolver\x18\x0c \x01(\x0b\x32\x1f.chalk.server.v1.DeleteResolverH\x00R\x0e\x64\x65leteResolverB\n\n\x08mutation"R\n\x14\x41\x64\x64StreamingResolver\x12:\n\x08resolver\x18\x01 \x01(\x0b\x32\x1e.chalk.graph.v1.StreamResolverR\x08resolver"z\n\x17UpdateStreamingResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName\x12:\n\x08resolver\x18\x02 \x01(\x0b\x32\x1e.chalk.graph.v1.StreamResolverR\x08resolver">\n\x17\x44\x65leteStreamingResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName"C\n\x0b\x41\x64\x64Resolver\x12\x34\n\x08resolver\x18\x01 \x01(\x0b\x32\x18.chalk.graph.v1.ResolverR\x08resolver"k\n\x0eUpdateResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName\x12\x34\n\x08resolver\x18\x02 \x01(\x0b\x32\x18.chalk.graph.v1.ResolverR\x08resolver"5\n\x0e\x44\x65leteResolver\x12#\n\rresolver_name\x18\x01 \x01(\tR\x0cresolverName"m\n\nAddFeature\x12(\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\tR\x0e\x66\x65\x61tureSetName\x12\x35\n\x07\x66\x65\x61ture\x18\x02 \x01(\x0b\x32\x1b.chalk.graph.v1.FeatureTypeR\x07\x66\x65\x61ture"X\n\rUpdateFeature\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn\x12\x35\n\x07\x66\x65\x61ture\x18\x02 \x01(\x0b\x32\x1b.chalk.graph.v1.FeatureTypeR\x07\x66\x65\x61ture"!\n\rDeleteFeature\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn"L\n\rAddFeatureSet\x12;\n\x0b\x66\x65\x61ture_set\x18\x01 \x01(\x0b\x32\x1a.chalk.graph.v1.FeatureSetR\nfeatureSet"y\n\x10UpdateFeatureSet\x12(\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\tR\x0e\x66\x65\x61tureSetName\x12;\n\x0b\x66\x65\x61ture_set\x18\x02 \x01(\x0b\x32\x1a.chalk.graph.v1.FeatureSetR\nfeatureSet"<\n\x10\x44\x65leteFeatureSet\x12(\n\x10\x66\x65\x61ture_set_name\x18\x01 \x01(\tR\x0e\x66\x65\x61tureSetName"}\n\x18\x41pplyGraphUpdatesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\tmutations\x18\x02 \x03(\x0b\x32\x1e.chalk.server.v1.GraphMutationR\tmutations"\x1b\n\x19\x41pplyGraphUpdatesResponse"~\n\x19TestGraphMutationsRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12<\n\tmutations\x18\x02 \x03(\x0b\x32\x1e.chalk.server.v1.GraphMutationR\tmutations"\x85\x01\n\x1aTestGraphMutationsResponse\x12\x32\n\x06\x65xport\x18\x01 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\x06\x65xport\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"&\n\nColumnList\x12\x18\n\x07\x63olumns\x18\x01 \x03(\tR\x07\x63olumns"\xb1\x01\n\x0cTableLineage\x12G\n\x08\x66\x65\x61tures\x18\x01 \x03(\x0b\x32+.chalk.server.v1.TableLineage.FeaturesEntryR\x08\x66\x65\x61tures\x1aX\n\rFeaturesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x31\n\x05value\x18\x02 \x01(\x0b\x32\x1b.chalk.server.v1.ColumnListR\x05value:\x02\x38\x01"\xb5\x01\n\x11\x44\x61taSourceLineage\x12\x46\n\x06tables\x18\x01 \x03(\x0b\x32..chalk.server.v1.DataSourceLineage.TablesEntryR\x06tables\x1aX\n\x0bTablesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x33\n\x05value\x18\x02 \x01(\x0b\x32\x1d.chalk.server.v1.TableLineageR\x05value:\x02\x38\x01"\xd3\x01\n\x13ResolverDataLineage\x12X\n\x0c\x64\x61ta_sources\x18\x01 \x03(\x0b\x32\x35.chalk.server.v1.ResolverDataLineage.DataSourcesEntryR\x0b\x64\x61taSources\x1a\x62\n\x10\x44\x61taSourcesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x38\n\x05value\x18\x02 \x01(\x0b\x32".chalk.server.v1.DataSourceLineageR\x05value:\x02\x38\x01"\x1c\n\x1aGetDataLineageIndexRequest"\x86\x02\n\x1bGetDataLineageIndexResponse\x12y\n\x15resolver_data_lineage\x18\x01 \x03(\x0b\x32\x45.chalk.server.v1.GetDataLineageIndexResponse.ResolverDataLineageEntryR\x13resolverDataLineage\x1al\n\x18ResolverDataLineageEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12:\n\x05value\x18\x02 \x01(\x0b\x32$.chalk.server.v1.ResolverDataLineageR\x05value:\x02\x38\x01"j\n\x0cOfflineTable\x12)\n\x10internal_version\x18\x01 \x01(\x05R\x0finternalVersion\x12\x1d\n\ntable_name\x18\x02 \x01(\tR\ttableName\x12\x10\n\x03\x66qn\x18\x03 \x01(\tR\x03\x66qn"_\n\x1bGetOfflineStoreTableRequest\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn\x12 \n\tbranch_id\x18\x02 \x01(\tH\x00R\x08\x62ranchId\x88\x01\x01\x42\x0c\n\n_branch_id"U\n\x1cGetOfflineStoreTableResponse\x12\x35\n\x06tables\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.OfflineTableR\x06tables"`\n!GetAllOfflineStoreTablesPageToken\x12\x10\n\x03\x66qn\x18\x01 \x01(\tR\x03\x66qn\x12)\n\x10internal_version\x18\x02 \x01(\x05R\x0finternalVersion"\xaf\x01\n\x1fGetAllOfflineStoreTablesRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x19\n\x05limit\x18\x03 \x01(\x05H\x00R\x05limit\x88\x01\x01\x12"\n\npage_token\x18\x04 \x01(\tH\x01R\tpageToken\x88\x01\x01\x42\x08\n\x06_limitB\r\n\x0b_page_tokenJ\x04\x08\x02\x10\x03R\tbranch_id"\x9a\x01\n GetAllOfflineStoreTablesResponse\x12\x35\n\x06tables\x18\x01 \x03(\x0b\x32\x1d.chalk.server.v1.OfflineTableR\x06tables\x12+\n\x0fnext_page_token\x18\x02 \x01(\tH\x00R\rnextPageToken\x88\x01\x01\x42\x12\n\x10_next_page_token"\xd0\x01\n\x16\x44iffDeploymentsRequest\x12\x35\n\x14\x64\x65ployment_id_before\x18\x01 \x01(\tH\x00R\x12\x64\x65ploymentIdBefore\x88\x01\x01\x12.\n\x13\x64\x65ployment_id_after\x18\x02 \x01(\tR\x11\x64\x65ploymentIdAfter\x12\x36\n\tdiff_mode\x18\x03 \x01(\x0e\x32\x19.chalk.server.v1.DiffModeR\x08\x64iffModeB\x17\n\x15_deployment_id_before"\xb9\x01\n\x17\x44iffDeploymentsResponse\x12-\n\x10\x64\x65ploy_id_before\x18\x01 \x01(\tH\x00R\x0e\x64\x65ployIdBefore\x88\x01\x01\x12&\n\x0f\x64\x65ploy_id_after\x18\x02 \x01(\tR\rdeployIdAfter\x12\x32\n\x04\x64iff\x18\x03 \x01(\x0b\x32\x1e.chalk.artifacts.v1.ExportDiffR\x04\x64iffB\x13\n\x11_deploy_id_before"y\n\x1aSmartDiffDeploymentRequest\x12#\n\rdeployment_id\x18\x01 \x01(\tR\x0c\x64\x65ploymentId\x12\x36\n\tdiff_mode\x18\x02 \x01(\x0e\x32\x19.chalk.server.v1.DiffModeR\x08\x64iffMode"\xbd\x01\n\x1bSmartDiffDeploymentResponse\x12-\n\x10\x64\x65ploy_id_before\x18\x01 \x01(\tH\x00R\x0e\x64\x65ployIdBefore\x88\x01\x01\x12&\n\x0f\x64\x65ploy_id_after\x18\x02 \x01(\tR\rdeployIdAfter\x12\x32\n\x04\x64iff\x18\x03 \x01(\x0b\x32\x1e.chalk.artifacts.v1.ExportDiffR\x04\x64iffB\x13\n\x11_deploy_id_before"\x88\x01\n\x14\x44iffCandidateRequest\x12\x38\n\tcandidate\x18\x01 \x01(\x0b\x32\x1a.chalk.artifacts.v1.ExportR\tcandidate\x12\x36\n\tdiff_mode\x18\x02 \x01(\x0e\x32\x19.chalk.server.v1.DiffModeR\x08\x64iffMode"\x8f\x01\n\x15\x44iffCandidateResponse\x12-\n\x10\x64\x65ploy_id_before\x18\x01 \x01(\tH\x00R\x0e\x64\x65ployIdBefore\x88\x01\x01\x12\x32\n\x04\x64iff\x18\x02 \x01(\x0b\x32\x1e.chalk.artifacts.v1.ExportDiffR\x04\x64iffB\x13\n\x11_deploy_id_before*O\n\x08\x44iffMode\x12\x19\n\x15\x44IFF_MODE_UNSPECIFIED\x10\x00\x12\x12\n\x0e\x44IFF_MODE_FULL\x10\x01\x12\x14\n\x10\x44IFF_MODE_SIMPLE\x10\x02\x32\xf4\x0b\n\x0cGraphService\x12\x66\n\rGetFeatureSQL\x12%.chalk.server.v1.GetFeatureSQLRequest\x1a&.chalk.server.v1.GetFeatureSQLResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13GetFeaturesMetadata\x12+.chalk.server.v1.GetFeaturesMetadataRequest\x1a,.chalk.server.v1.GetFeaturesMetadataResponse"\x06\x90\x02\x01\x80}\x0b\x12W\n\x08GetGraph\x12 .chalk.server.v1.GetGraphRequest\x1a!.chalk.server.v1.GetGraphResponse"\x06\x90\x02\x01\x80}\x0b\x12]\n\x0bUpdateGraph\x12#.chalk.server.v1.UpdateGraphRequest\x1a$.chalk.server.v1.UpdateGraphResponse"\x03\x80}\x0c\x12\x90\x01\n\x1bGetCodegenFeaturesFromGraph\x12\x33.chalk.server.v1.GetCodegenFeaturesFromGraphRequest\x1a\x34.chalk.server.v1.GetCodegenFeaturesFromGraphResponse"\x06\x90\x02\x01\x80}\x0b\x12o\n\x11\x41pplyGraphUpdates\x12).chalk.server.v1.ApplyGraphUpdatesRequest\x1a*.chalk.server.v1.ApplyGraphUpdatesResponse"\x03\x80}\x0c\x12u\n\x12TestGraphMutations\x12*.chalk.server.v1.TestGraphMutationsRequest\x1a+.chalk.server.v1.TestGraphMutationsResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13GetDataLineageIndex\x12+.chalk.server.v1.GetDataLineageIndexRequest\x1a,.chalk.server.v1.GetDataLineageIndexResponse"\x06\x90\x02\x01\x80}\x0b\x12{\n\x14GetOfflineStoreTable\x12,.chalk.server.v1.GetOfflineStoreTableRequest\x1a-.chalk.server.v1.GetOfflineStoreTableResponse"\x06\x90\x02\x01\x80}\x0b\x12\x87\x01\n\x18GetAllOfflineStoreTables\x12\x30.chalk.server.v1.GetAllOfflineStoreTablesRequest\x1a\x31.chalk.server.v1.GetAllOfflineStoreTablesResponse"\x06\x90\x02\x01\x80}\x0b\x12l\n\x0f\x44iffDeployments\x12\'.chalk.server.v1.DiffDeploymentsRequest\x1a(.chalk.server.v1.DiffDeploymentsResponse"\x06\x90\x02\x01\x80}\x0b\x12x\n\x13SmartDiffDeployment\x12+.chalk.server.v1.SmartDiffDeploymentRequest\x1a,.chalk.server.v1.SmartDiffDeploymentResponse"\x06\x90\x02\x01\x80}\x0b\x12\x66\n\rDiffCandidate\x12%.chalk.server.v1.DiffCandidateRequest\x1a&.chalk.server.v1.DiffCandidateResponse"\x06\x90\x02\x01\x80}\x0b\x42\x93\x01\n\x13\x63om.chalk.server.v1B\nGraphProtoP\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()
@@ -73,8 +73,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_GRAPHSERVICE"].methods_by_name["SmartDiffDeployment"]._serialized_options = b"\220\002\001\200}\013"
     _globals["_GRAPHSERVICE"].methods_by_name["DiffCandidate"]._options = None
     _globals["_GRAPHSERVICE"].methods_by_name["DiffCandidate"]._serialized_options = b"\220\002\001\200}\013"
-    _globals["_DIFFMODE"]._serialized_start = 7427
-    _globals["_DIFFMODE"]._serialized_end = 7506
+    _globals["_DIFFMODE"]._serialized_start = 7649
+    _globals["_DIFFMODE"]._serialized_end = 7728
     _globals["_FEATURESQL"]._serialized_start = 237
     _globals["_FEATURESQL"]._serialized_end = 837
     _globals["_GETFEATURESQLRESPONSE"]._serialized_start = 839
@@ -161,22 +161,24 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_GETOFFLINESTORETABLEREQUEST"]._serialized_end = 6128
     _globals["_GETOFFLINESTORETABLERESPONSE"]._serialized_start = 6130
     _globals["_GETOFFLINESTORETABLERESPONSE"]._serialized_end = 6215
-    _globals["_GETALLOFFLINESTORETABLESREQUEST"]._serialized_start = 6217
-    _globals["_GETALLOFFLINESTORETABLESREQUEST"]._serialized_end = 6335
-    _globals["_GETALLOFFLINESTORETABLESRESPONSE"]._serialized_start = 6337
-    _globals["_GETALLOFFLINESTORETABLESRESPONSE"]._serialized_end = 6426
-    _globals["_DIFFDEPLOYMENTSREQUEST"]._serialized_start = 6429
-    _globals["_DIFFDEPLOYMENTSREQUEST"]._serialized_end = 6637
-    _globals["_DIFFDEPLOYMENTSRESPONSE"]._serialized_start = 6640
-    _globals["_DIFFDEPLOYMENTSRESPONSE"]._serialized_end = 6825
-    _globals["_SMARTDIFFDEPLOYMENTREQUEST"]._serialized_start = 6827
-    _globals["_SMARTDIFFDEPLOYMENTREQUEST"]._serialized_end = 6948
-    _globals["_SMARTDIFFDEPLOYMENTRESPONSE"]._serialized_start = 6951
-    _globals["_SMARTDIFFDEPLOYMENTRESPONSE"]._serialized_end = 7140
-    _globals["_DIFFCANDIDATEREQUEST"]._serialized_start = 7143
-    _globals["_DIFFCANDIDATEREQUEST"]._serialized_end = 7279
-    _globals["_DIFFCANDIDATERESPONSE"]._serialized_start = 7282
-    _globals["_DIFFCANDIDATERESPONSE"]._serialized_end = 7425
-    _globals["_GRAPHSERVICE"]._serialized_start = 7509
-    _globals["_GRAPHSERVICE"]._serialized_end = 9033
+    _globals["_GETALLOFFLINESTORETABLESPAGETOKEN"]._serialized_start = 6217
+    _globals["_GETALLOFFLINESTORETABLESPAGETOKEN"]._serialized_end = 6313
+    _globals["_GETALLOFFLINESTORETABLESREQUEST"]._serialized_start = 6316
+    _globals["_GETALLOFFLINESTORETABLESREQUEST"]._serialized_end = 6491
+    _globals["_GETALLOFFLINESTORETABLESRESPONSE"]._serialized_start = 6494
+    _globals["_GETALLOFFLINESTORETABLESRESPONSE"]._serialized_end = 6648
+    _globals["_DIFFDEPLOYMENTSREQUEST"]._serialized_start = 6651
+    _globals["_DIFFDEPLOYMENTSREQUEST"]._serialized_end = 6859
+    _globals["_DIFFDEPLOYMENTSRESPONSE"]._serialized_start = 6862
+    _globals["_DIFFDEPLOYMENTSRESPONSE"]._serialized_end = 7047
+    _globals["_SMARTDIFFDEPLOYMENTREQUEST"]._serialized_start = 7049
+    _globals["_SMARTDIFFDEPLOYMENTREQUEST"]._serialized_end = 7170
+    _globals["_SMARTDIFFDEPLOYMENTRESPONSE"]._serialized_start = 7173
+    _globals["_SMARTDIFFDEPLOYMENTRESPONSE"]._serialized_end = 7362
+    _globals["_DIFFCANDIDATEREQUEST"]._serialized_start = 7365
+    _globals["_DIFFCANDIDATEREQUEST"]._serialized_end = 7501
+    _globals["_DIFFCANDIDATERESPONSE"]._serialized_start = 7504
+    _globals["_DIFFCANDIDATERESPONSE"]._serialized_end = 7647
+    _globals["_GRAPHSERVICE"]._serialized_start = 7731
+    _globals["_GRAPHSERVICE"]._serialized_end = 9255
 # @@protoc_insertion_point(module_scope)

{chalkpy-2.154.2 → chalkpy-2.154.4}/chalk/_gen/chalk/server/v1/graph_pb2.pyi

@@ -559,19 +559,35 @@ class GetOfflineStoreTableResponse(_message.Message):
     tables: _containers.RepeatedCompositeFieldContainer[OfflineTable]
     def __init__(self, tables: _Optional[_Iterable[_Union[OfflineTable, _Mapping]]] = ...) -> None: ...
 
+class GetAllOfflineStoreTablesPageToken(_message.Message):
+    __slots__ = ("fqn", "internal_version")
+    FQN_FIELD_NUMBER: _ClassVar[int]
+    INTERNAL_VERSION_FIELD_NUMBER: _ClassVar[int]
+    fqn: str
+    internal_version: int
+    def __init__(self, fqn: _Optional[str] = ..., internal_version: _Optional[int] = ...) -> None: ...
+
 class GetAllOfflineStoreTablesRequest(_message.Message):
-    __slots__ = ("deployment_id", "branch_id")
+    __slots__ = ("deployment_id", "limit", "page_token")
     DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
-    BRANCH_ID_FIELD_NUMBER: _ClassVar[int]
+    LIMIT_FIELD_NUMBER: _ClassVar[int]
+    PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
     deployment_id: str
-    branch_id: str
-    def __init__(self, deployment_id: _Optional[str] = ..., branch_id: _Optional[str] = ...) -> None: ...
+    limit: int
+    page_token: str
+    def __init__(
+        self, deployment_id: _Optional[str] = ..., limit: _Optional[int] = ..., page_token: _Optional[str] = ...
+    ) -> None: ...
 
 class GetAllOfflineStoreTablesResponse(_message.Message):
-    __slots__ = ("tables",)
+    __slots__ = ("tables", "next_page_token")
     TABLES_FIELD_NUMBER: _ClassVar[int]
+    NEXT_PAGE_TOKEN_FIELD_NUMBER: _ClassVar[int]
     tables: _containers.RepeatedCompositeFieldContainer[OfflineTable]
-    def __init__(self, tables: _Optional[_Iterable[_Union[OfflineTable, _Mapping]]] = ...) -> None: ...
+    next_page_token: str
+    def __init__(
+        self, tables: _Optional[_Iterable[_Union[OfflineTable, _Mapping]]] = ..., next_page_token: _Optional[str] = ...
+    ) -> None: ...
 
 class DiffDeploymentsRequest(_message.Message):
     __slots__ = ("deployment_id_before", "deployment_id_after", "diff_mode")

chalkpy-2.154.4/chalk/_version.py

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

chalkpy-2.154.4/chalk/client/_model_remote.py

@@ -0,0 +1,157 @@
+"""Directly invoke a model deployed to a scaling group."""
+
+from __future__ import annotations
+
+import collections.abc
+import logging
+from typing import TYPE_CHECKING, Any, List, Mapping, Optional, Sequence, Tuple
+from urllib.parse import urlsplit
+
+import grpc
+
+if TYPE_CHECKING:
+    import pyarrow as pa
+
+    from chalk.client.client_grpc import ChalkGRPCClient
+
+logger = logging.getLogger(__name__)
+
+# chalk-remote-call-backed scaling groups always route to "handler".
+DEFAULT_HANDLER = "handler"
+
+
+class ModelRemoteError(RuntimeError):
+    pass
+
+
+class ModelNotDeployedError(ModelRemoteError):
+    """Model/scaling group exists but has no reachable ingress."""
+
+
+def _grpc_target_from_url(web_url: str) -> Tuple[str, bool]:
+    """``https://host`` -> ``("host:443", True)``; explicit port and ``http://`` are honored."""
+    parsed = urlsplit(web_url)
+    use_tls = parsed.scheme != "http"
+    host = parsed.hostname or ""
+    if parsed.port:
+        return f"{host}:{parsed.port}", use_tls
+    if use_tls:
+        return f"{host}:443", True
+    return host, False
+
+
+def resolve_scaling_group_web_url(
+    client: "ChalkGRPCClient",
+    model_name: str,
+    *,
+    version: Optional[int] = None,
+) -> str:
+    """Resolve the public ``web_url`` of the scaling group serving a model version.
+
+    Uses ``version`` (or the model's latest) to look up its deployed scaling group.
+    """
+    from chalk._gen.chalk.modeldeployment.v1 import service_pb2 as md_pb
+    from chalk._gen.chalk.models.v1.model_version_pb2 import ModelVersionIdentifier
+    from chalk._gen.chalk.server.v1.model_registry_pb2 import GetModelRequest
+
+    resolved_version = version
+    if resolved_version is None:
+        try:
+            model_resp = client._stub_refresher.call_model_stub(  # pyright: ignore[reportPrivateUsage]
+                lambda x: x.GetModel(GetModelRequest(model_name=model_name))
+            )
+        except grpc.RpcError as e:
+            raise ModelNotDeployedError(f"Model {model_name!r} not found: {e.details()}") from e
+        resolved_version = model_resp.model.latest_model_version.version
+        if not resolved_version:
+            raise ModelNotDeployedError(f"Model {model_name!r} has no published versions")
+
+    selector = md_pb.ModelVersionSelector(
+        model_name=model_name,
+        identifier=ModelVersionIdentifier(version=resolved_version),
+    )
+    list_resp = client._stub_refresher.call_model_deployment_stub(  # pyright: ignore[reportPrivateUsage]
+        lambda x: x.ListModelScalingGroups(md_pb.ListModelScalingGroupsRequest(model_version=selector))
+    )
+    for group in list_resp.scaling_groups:
+        if group.web_url:
+            return group.web_url
+    raise ModelNotDeployedError(f"Model {model_name!r} v{resolved_version} has no scaling group with a web URL")
+
+
+def _encode_inputs(inputs: "Mapping[str, Sequence[Any]] | pa.RecordBatch | pa.Table") -> bytes:
+    """Serialize inputs to Arrow IPC stream bytes (the format the runtime expects)."""
+    import pyarrow as pa
+
+    if isinstance(inputs, collections.abc.Mapping):
+        arrays = {k: pa.array(v) for k, v in inputs.items()}
+        batch = pa.record_batch(list(arrays.values()), names=list(arrays.keys()))
+    elif isinstance(inputs, pa.RecordBatch):
+        batch = inputs
+    else:
+        batches = inputs.combine_chunks().to_batches()
+        batch = batches[0] if batches else pa.RecordBatch.from_pylist([], schema=inputs.schema)
+
+    sink = pa.BufferOutputStream()
+    with pa.ipc.new_stream(sink, batch.schema) as writer:
+        writer.write_batch(batch)
+    return sink.getvalue().to_pybytes()
+
+
+def _decode_output(chunks: Sequence[bytes]) -> "pa.RecordBatch":
+    import pyarrow as pa
+
+    if not chunks:
+        raise ModelRemoteError("Empty response from scaling group")
+    batches = list(pa.ipc.open_stream(chunks[0]))
+    if not batches:
+        raise ModelRemoteError("Response from scaling group contained no record batches")
+    return batches[0]
+
+
+def _transport_call(
+    target: str,
+    use_tls: bool,
+    handler: str,
+    feather_bytes: bytes,
+    metadata: Sequence[Tuple[str, str]],
+) -> List[bytes]:
+    try:
+        from chalkcompute import RemoteCallClient, RemoteCallClientUnavailable  # pyright: ignore[reportMissingImports]
+    except ImportError as e:
+        raise ImportError(
+            "Install `chalkcompute` (`pip install 'chalkpy[compute]'`) to enable direct model calls."
+        ) from e
+
+    try:
+        remote_client = RemoteCallClient(target, metadata=list(metadata), use_tls=use_tls)
+    except RemoteCallClientUnavailable as e:
+        raise ImportError("Reinstall `chalkcompute`: its native extension is missing.") from e
+
+    try:
+        return list(remote_client.call_ipc(handler, feather_bytes))
+    finally:
+        close = getattr(remote_client, "close", None)
+        if callable(close):
+            close()
+
+
+def call_model_scaling_group(
+    client: "ChalkGRPCClient",
+    model_name: str,
+    inputs: "Mapping[str, Sequence[Any]] | pa.RecordBatch | pa.Table",
+    *,
+    version: Optional[int] = None,
+    handler: str = DEFAULT_HANDLER,
+) -> "pa.RecordBatch":
+    """Invoke a deployed model by calling its scaling group ingress directly.
+
+    ``inputs`` is a column mapping or pyarrow batch/table whose column order
+    matches the model's input schema.
+    """
+    web_url = resolve_scaling_group_web_url(client, model_name, version=version)
+    target, use_tls = _grpc_target_from_url(web_url)
+    metadata = client._get_remote_call_metadata()  # pyright: ignore[reportPrivateUsage]
+    feather_bytes = _encode_inputs(inputs)
+    chunks = _transport_call(target, use_tls, handler, feather_bytes, metadata)
+    return _decode_output(chunks)

{chalkpy-2.154.2 → chalkpy-2.154.4}/chalk/client/client.py

@@ -1655,7 +1655,6 @@ class ChalkClient:
 
     def get_all_offline_store_table_names(
         self,
-        branch_id: str | None = None,
         deployment_id: str | None = None,
     ) -> "list[OfflineStoreTable]":
         """Get the offline store table name for every feature and internal version in a deployment.
@@ -1664,8 +1663,6 @@ class ChalkClient:
 
         Parameters
         ----------
-        branch_id
-            If set, return table names for the given branch deployment.
         deployment_id
             The deployment to look up. If ``None``, uses the environment's active deployment.
 
@@ -1684,7 +1681,6 @@ class ChalkClient:
     def get_feature_from_offline_store_table_name(
         self,
         table_name: str,
-        branch_id: str | None = None,
         deployment_id: str | None = None,
     ) -> "OfflineStoreTable | None":
         """Find the feature for a given offline store table name (reverse lookup).
@@ -1693,8 +1689,6 @@ class ChalkClient:
         ----------
         table_name
             The ``feat_<hash>`` offline store table name to look up.
-        branch_id
-            If set, search the given branch deployment's tables.
         deployment_id
             The deployment to search. If ``None``, uses the environment's active deployment.