@databricks/sdk-features 0.48.0 → 0.52.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/v1/model.js CHANGED
@@ -1,8 +1,108 @@
1
+ import { z } from "zod";
1
2
  import { Temporal } from "@js-temporal/polyfill";
2
3
  import { FieldMask } from "@databricks/sdk-core/wkt";
3
- import { z } from "zod";
4
4
 
5
5
  //#region src/v1/model.ts
6
+ /** Error codes returned by Databricks APIs to indicate specific failure conditions. */
7
+ const ErrorCode = {
8
+ UNKNOWN: "UNKNOWN",
9
+ INTERNAL_ERROR: "INTERNAL_ERROR",
10
+ TEMPORARILY_UNAVAILABLE: "TEMPORARILY_UNAVAILABLE",
11
+ IO_ERROR: "IO_ERROR",
12
+ BAD_REQUEST: "BAD_REQUEST",
13
+ SERVICE_UNDER_MAINTENANCE: "SERVICE_UNDER_MAINTENANCE",
14
+ WORKSPACE_TEMPORARILY_UNAVAILABLE: "WORKSPACE_TEMPORARILY_UNAVAILABLE",
15
+ DEADLINE_EXCEEDED: "DEADLINE_EXCEEDED",
16
+ CANCELLED: "CANCELLED",
17
+ RESOURCE_EXHAUSTED: "RESOURCE_EXHAUSTED",
18
+ ABORTED: "ABORTED",
19
+ NOT_FOUND: "NOT_FOUND",
20
+ ALREADY_EXISTS: "ALREADY_EXISTS",
21
+ UNAUTHENTICATED: "UNAUTHENTICATED",
22
+ UNAVAILABLE: "UNAVAILABLE",
23
+ INVALID_PARAMETER_VALUE: "INVALID_PARAMETER_VALUE",
24
+ ENDPOINT_NOT_FOUND: "ENDPOINT_NOT_FOUND",
25
+ MALFORMED_REQUEST: "MALFORMED_REQUEST",
26
+ INVALID_STATE: "INVALID_STATE",
27
+ PERMISSION_DENIED: "PERMISSION_DENIED",
28
+ FEATURE_DISABLED: "FEATURE_DISABLED",
29
+ CUSTOMER_UNAUTHORIZED: "CUSTOMER_UNAUTHORIZED",
30
+ REQUEST_LIMIT_EXCEEDED: "REQUEST_LIMIT_EXCEEDED",
31
+ RESOURCE_CONFLICT: "RESOURCE_CONFLICT",
32
+ UNPARSEABLE_HTTP_ERROR: "UNPARSEABLE_HTTP_ERROR",
33
+ NOT_IMPLEMENTED: "NOT_IMPLEMENTED",
34
+ DATA_LOSS: "DATA_LOSS",
35
+ INVALID_STATE_TRANSITION: "INVALID_STATE_TRANSITION",
36
+ COULD_NOT_ACQUIRE_LOCK: "COULD_NOT_ACQUIRE_LOCK",
37
+ RESOURCE_ALREADY_EXISTS: "RESOURCE_ALREADY_EXISTS",
38
+ RESOURCE_DOES_NOT_EXIST: "RESOURCE_DOES_NOT_EXIST",
39
+ QUOTA_EXCEEDED: "QUOTA_EXCEEDED",
40
+ MAX_BLOCK_SIZE_EXCEEDED: "MAX_BLOCK_SIZE_EXCEEDED",
41
+ MAX_READ_SIZE_EXCEEDED: "MAX_READ_SIZE_EXCEEDED",
42
+ PARTIAL_DELETE: "PARTIAL_DELETE",
43
+ MAX_LIST_SIZE_EXCEEDED: "MAX_LIST_SIZE_EXCEEDED",
44
+ DRY_RUN_FAILED: "DRY_RUN_FAILED",
45
+ RESOURCE_LIMIT_EXCEEDED: "RESOURCE_LIMIT_EXCEEDED",
46
+ DIRECTORY_NOT_EMPTY: "DIRECTORY_NOT_EMPTY",
47
+ DIRECTORY_PROTECTED: "DIRECTORY_PROTECTED",
48
+ MAX_NOTEBOOK_SIZE_EXCEEDED: "MAX_NOTEBOOK_SIZE_EXCEEDED",
49
+ MAX_CHILD_NODE_SIZE_EXCEEDED: "MAX_CHILD_NODE_SIZE_EXCEEDED",
50
+ SEARCH_QUERY_TOO_LONG: "SEARCH_QUERY_TOO_LONG",
51
+ SEARCH_QUERY_TOO_SHORT: "SEARCH_QUERY_TOO_SHORT",
52
+ MANAGED_RESOURCE_GROUP_DOES_NOT_EXIST: "MANAGED_RESOURCE_GROUP_DOES_NOT_EXIST",
53
+ PERMISSION_NOT_PROPAGATED: "PERMISSION_NOT_PROPAGATED",
54
+ DEPLOYMENT_TIMEOUT: "DEPLOYMENT_TIMEOUT",
55
+ GIT_CONFLICT: "GIT_CONFLICT",
56
+ GIT_UNKNOWN_REF: "GIT_UNKNOWN_REF",
57
+ GIT_SENSITIVE_TOKEN_DETECTED: "GIT_SENSITIVE_TOKEN_DETECTED",
58
+ GIT_URL_NOT_ON_ALLOW_LIST: "GIT_URL_NOT_ON_ALLOW_LIST",
59
+ GIT_REMOTE_ERROR: "GIT_REMOTE_ERROR",
60
+ PROJECTS_OPERATION_TIMEOUT: "PROJECTS_OPERATION_TIMEOUT",
61
+ IPYNB_FILE_IN_REPO: "IPYNB_FILE_IN_REPO",
62
+ INSECURE_PARTNER_RESPONSE: "INSECURE_PARTNER_RESPONSE",
63
+ MALFORMED_PARTNER_RESPONSE: "MALFORMED_PARTNER_RESPONSE",
64
+ METASTORE_DOES_NOT_EXIST: "METASTORE_DOES_NOT_EXIST",
65
+ DAC_DOES_NOT_EXIST: "DAC_DOES_NOT_EXIST",
66
+ CATALOG_DOES_NOT_EXIST: "CATALOG_DOES_NOT_EXIST",
67
+ SCHEMA_DOES_NOT_EXIST: "SCHEMA_DOES_NOT_EXIST",
68
+ TABLE_DOES_NOT_EXIST: "TABLE_DOES_NOT_EXIST",
69
+ SHARE_DOES_NOT_EXIST: "SHARE_DOES_NOT_EXIST",
70
+ RECIPIENT_DOES_NOT_EXIST: "RECIPIENT_DOES_NOT_EXIST",
71
+ STORAGE_CREDENTIAL_DOES_NOT_EXIST: "STORAGE_CREDENTIAL_DOES_NOT_EXIST",
72
+ EXTERNAL_LOCATION_DOES_NOT_EXIST: "EXTERNAL_LOCATION_DOES_NOT_EXIST",
73
+ PRINCIPAL_DOES_NOT_EXIST: "PRINCIPAL_DOES_NOT_EXIST",
74
+ PROVIDER_DOES_NOT_EXIST: "PROVIDER_DOES_NOT_EXIST",
75
+ METASTORE_ALREADY_EXISTS: "METASTORE_ALREADY_EXISTS",
76
+ DAC_ALREADY_EXISTS: "DAC_ALREADY_EXISTS",
77
+ CATALOG_ALREADY_EXISTS: "CATALOG_ALREADY_EXISTS",
78
+ SCHEMA_ALREADY_EXISTS: "SCHEMA_ALREADY_EXISTS",
79
+ TABLE_ALREADY_EXISTS: "TABLE_ALREADY_EXISTS",
80
+ SHARE_ALREADY_EXISTS: "SHARE_ALREADY_EXISTS",
81
+ RECIPIENT_ALREADY_EXISTS: "RECIPIENT_ALREADY_EXISTS",
82
+ STORAGE_CREDENTIAL_ALREADY_EXISTS: "STORAGE_CREDENTIAL_ALREADY_EXISTS",
83
+ EXTERNAL_LOCATION_ALREADY_EXISTS: "EXTERNAL_LOCATION_ALREADY_EXISTS",
84
+ PROVIDER_ALREADY_EXISTS: "PROVIDER_ALREADY_EXISTS",
85
+ CATALOG_NOT_EMPTY: "CATALOG_NOT_EMPTY",
86
+ SCHEMA_NOT_EMPTY: "SCHEMA_NOT_EMPTY",
87
+ METASTORE_NOT_EMPTY: "METASTORE_NOT_EMPTY",
88
+ PROVIDER_SHARE_NOT_ACCESSIBLE: "PROVIDER_SHARE_NOT_ACCESSIBLE"
89
+ };
90
+ const FunctionFunctionType = {
91
+ FUNCTION_TYPE_UNSPECIFIED: "FUNCTION_TYPE_UNSPECIFIED",
92
+ AVG: "AVG",
93
+ COUNT: "COUNT",
94
+ SUM: "SUM",
95
+ MIN: "MIN",
96
+ MAX: "MAX",
97
+ FIRST: "FIRST",
98
+ LAST: "LAST",
99
+ APPROX_COUNT_DISTINCT: "APPROX_COUNT_DISTINCT",
100
+ APPROX_PERCENTILE: "APPROX_PERCENTILE",
101
+ STDDEV_POP: "STDDEV_POP",
102
+ STDDEV_SAMP: "STDDEV_SAMP",
103
+ VAR_POP: "VAR_POP",
104
+ VAR_SAMP: "VAR_SAMP"
105
+ };
6
106
  /**
7
107
  * Scalar data types for request-time field definitions.
8
108
  * Only flat (non-nested) types are supported.
@@ -21,12 +121,43 @@ const ScalarDataType = {
21
121
  BINARY: "BINARY",
22
122
  DECIMAL: "DECIMAL"
23
123
  };
124
+ /** Lifecycle state of a backfill. */
125
+ const BackfillOperationMetadata_State = {
126
+ STATE_UNSPECIFIED: "STATE_UNSPECIFIED",
127
+ PENDING: "PENDING",
128
+ RUNNING: "RUNNING",
129
+ SUCCEEDED: "SUCCEEDED",
130
+ FAILED: "FAILED",
131
+ CANCELLED: "CANCELLED"
132
+ };
133
+ /** The way a materialization schedule is arrived at. */
134
+ const CronSchedule_Mode = {
135
+ MODE_UNSPECIFIED: "MODE_UNSPECIFIED",
136
+ MANUAL: "MANUAL",
137
+ DERIVED: "DERIVED"
138
+ };
24
139
  const MaterializedFeature_PipelineScheduleState = {
25
140
  PIPELINE_SCHEDULE_STATE_UNSPECIFIED: "PIPELINE_SCHEDULE_STATE_UNSPECIFIED",
26
141
  SNAPSHOT: "SNAPSHOT",
27
142
  ACTIVE: "ACTIVE",
28
143
  PAUSED: "PAUSED"
29
144
  };
145
+ /** Lifecycle state of a feature entity purge. */
146
+ const PurgeFeatureEntitiesMetadata_State = {
147
+ STATE_UNSPECIFIED: "STATE_UNSPECIFIED",
148
+ PENDING: "PENDING",
149
+ RUNNING: "RUNNING",
150
+ SUCCEEDED: "SUCCEEDED",
151
+ FAILED: "FAILED",
152
+ CANCELLED: "CANCELLED"
153
+ };
154
+ /** Terminal state of a purge for one store type. */
155
+ const PurgeFeatureEntitiesResult_State = {
156
+ STATE_UNSPECIFIED: "STATE_UNSPECIFIED",
157
+ SUCCEEDED: "SUCCEEDED",
158
+ FAILED: "FAILED",
159
+ NOT_APPLICABLE: "NOT_APPLICABLE"
160
+ };
30
161
  /** Supported serialization formats for a schema registry schema. */
31
162
  const SchemaLocator_Format = {
32
163
  FORMAT_UNSPECIFIED: "FORMAT_UNSPECIFIED",
@@ -113,6 +244,17 @@ const unmarshalAggregationFunctionSchema = z.object({
113
244
  } : void 0,
114
245
  timeWindow: d.time_window
115
246
  }));
247
+ const unmarshalApiErrorSchema = z.object({
248
+ error_code: z.string().optional(),
249
+ message: z.string().optional(),
250
+ stack_trace: z.string().optional(),
251
+ details: z.array(z.record(z.string(), z.unknown())).optional()
252
+ }).transform((d) => ({
253
+ errorCode: d.error_code,
254
+ message: d.message,
255
+ stackTrace: d.stack_trace,
256
+ details: d.details
257
+ }));
116
258
  const unmarshalApproxCountDistinctFunctionSchema = z.object({
117
259
  input: z.string().optional(),
118
260
  relative_sd: z.number().optional()
@@ -144,6 +286,23 @@ const unmarshalAuthConfigSchema = z.object({
144
286
  mtlsConfig: d.mtls_config
145
287
  } : void 0 }));
146
288
  const unmarshalAvgFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
289
+ const unmarshalBackfillFeaturesResponseSchema = z.object({});
290
+ const unmarshalBackfillOperationMetadataSchema = z.object({
291
+ feature_full_names: z.array(z.string()).optional(),
292
+ backfill_ranges: z.array(z.lazy(() => unmarshalBackfillRangeSchema)).optional(),
293
+ state: z.string().optional()
294
+ }).transform((d) => ({
295
+ featureFullNames: d.feature_full_names,
296
+ backfillRanges: d.backfill_ranges,
297
+ state: d.state
298
+ }));
299
+ const unmarshalBackfillRangeSchema = z.object({
300
+ start_time: z.string().transform((s) => Temporal.Instant.from(s)).optional(),
301
+ end_time: z.string().transform((s) => Temporal.Instant.from(s)).optional()
302
+ }).transform((d) => ({
303
+ startTime: d.start_time,
304
+ endTime: d.end_time
305
+ }));
147
306
  const unmarshalBackfillSourceSchema = z.object({
148
307
  delta_table_source: z.lazy(() => unmarshalDeltaTableSourceSchema).optional(),
149
308
  delta_table_name: z.string().optional()
@@ -155,9 +314,23 @@ const unmarshalBackfillSourceSchema = z.object({
155
314
  deltaTableName: d.delta_table_name
156
315
  } : void 0 }));
157
316
  const unmarshalBatchCreateMaterializedFeaturesResponseSchema = z.object({ materialized_features: z.array(z.lazy(() => unmarshalMaterializedFeatureSchema)).optional() }).transform((d) => ({ materializedFeatures: d.materialized_features }));
317
+ const unmarshalColumnIdentifierSchema = z.object({ variant_expr_path: z.string().optional() }).transform((d) => ({ variantExprPath: d.variant_expr_path }));
158
318
  const unmarshalColumnSelectionSchema = z.object({ column: z.string().optional() }).transform((d) => ({ column: d.column }));
319
+ const unmarshalContinuousWindowSchema = z.object({
320
+ window_duration: z.string().transform((s) => Temporal.Duration.from("PT" + s.toUpperCase())).optional(),
321
+ offset: z.string().transform((s) => Temporal.Duration.from("PT" + s.toUpperCase())).optional()
322
+ }).transform((d) => ({
323
+ windowDuration: d.window_duration,
324
+ offset: d.offset
325
+ }));
159
326
  const unmarshalCountFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
160
- const unmarshalCronScheduleSchema = z.object({ cron_expression: z.string().optional() }).transform((d) => ({ cronExpression: d.cron_expression }));
327
+ const unmarshalCronScheduleSchema = z.object({
328
+ cron_expression: z.string().optional(),
329
+ mode: z.string().optional()
330
+ }).transform((d) => ({
331
+ cronExpression: d.cron_expression,
332
+ mode: d.mode
333
+ }));
161
334
  const unmarshalCustomUdfSchema = z.object({
162
335
  function_path: z.string().optional(),
163
336
  input_bindings: z.array(z.lazy(() => unmarshalInputBindingSchema)).optional()
@@ -170,6 +343,7 @@ const unmarshalDataSourceSchema = z.object({
170
343
  kafka_source: z.lazy(() => unmarshalKafkaSourceSchema).optional(),
171
344
  request_source: z.lazy(() => unmarshalRequestSourceSchema).optional(),
172
345
  stream_source: z.lazy(() => unmarshalStreamSourceSchema).optional(),
346
+ feature_view_source: z.lazy(() => unmarshalFeatureViewSourceSchema).optional(),
173
347
  lateness: z.lazy(() => unmarshalSourceLatenessSchema).optional()
174
348
  }).transform((d) => ({
175
349
  dataSource: d.delta_table_source !== void 0 ? {
@@ -184,16 +358,23 @@ const unmarshalDataSourceSchema = z.object({
184
358
  } : d.stream_source !== void 0 ? {
185
359
  $case: "streamSource",
186
360
  streamSource: d.stream_source
361
+ } : d.feature_view_source !== void 0 ? {
362
+ $case: "featureViewSource",
363
+ featureViewSource: d.feature_view_source
187
364
  } : void 0,
188
365
  lateness: d.lateness
189
366
  }));
190
367
  const unmarshalDeltaTableSourceSchema = z.object({
191
368
  full_name: z.string().optional(),
369
+ entity_columns: z.array(z.string()).optional(),
370
+ timeseries_column: z.string().optional(),
192
371
  filter_condition: z.string().optional(),
193
372
  transformation_sql: z.string().optional(),
194
373
  dataframe_schema: z.string().optional()
195
374
  }).transform((d) => ({
196
375
  fullName: d.full_name,
376
+ entityColumns: d.entity_columns,
377
+ timeseriesColumn: d.timeseries_column,
197
378
  filterCondition: d.filter_condition,
198
379
  transformationSql: d.transformation_sql,
199
380
  dataframeSchema: d.dataframe_schema
@@ -216,8 +397,11 @@ const unmarshalEntityColumnSchema = z.object({ name: z.string().optional() }).tr
216
397
  const unmarshalFeatureSchema = z.object({
217
398
  full_name: z.string().optional(),
218
399
  source: z.lazy(() => unmarshalDataSourceSchema).optional(),
400
+ inputs: z.array(z.string()).optional(),
219
401
  function: z.lazy(() => unmarshalFunctionSchema).optional(),
402
+ time_window: z.lazy(() => unmarshalTimeWindowSchema).optional(),
220
403
  description: z.string().optional(),
404
+ filter_condition: z.string().optional(),
221
405
  lineage_context: z.lazy(() => unmarshalLineageContextSchema).optional(),
222
406
  entities: z.array(z.lazy(() => unmarshalEntityColumnSchema)).optional(),
223
407
  timeseries_column: z.lazy(() => unmarshalTimeseriesColumnSchema).optional(),
@@ -229,8 +413,11 @@ const unmarshalFeatureSchema = z.object({
229
413
  }).transform((d) => ({
230
414
  fullName: d.full_name,
231
415
  source: d.source,
416
+ inputs: d.inputs,
232
417
  function: d.function,
418
+ timeWindow: d.time_window,
233
419
  description: d.description,
420
+ filterCondition: d.filter_condition,
234
421
  lineageContext: d.lineage_context,
235
422
  entities: d.entities,
236
423
  timeseriesColumn: d.timeseries_column,
@@ -240,6 +427,8 @@ const unmarshalFeatureSchema = z.object({
240
427
  createdAt: d.created_at,
241
428
  createdBy: d.created_by
242
429
  }));
430
+ const unmarshalFeatureReferenceSchema = z.object({ feature: z.string().optional() }).transform((d) => ({ feature: d.feature }));
431
+ const unmarshalFeatureViewSourceSchema = z.object({ feature_references: z.array(z.lazy(() => unmarshalFeatureReferenceSchema)).optional() }).transform((d) => ({ featureReferences: d.feature_references }));
243
432
  const unmarshalFieldDefinitionSchema = z.object({
244
433
  name: z.string().optional(),
245
434
  data_type: z.string().optional()
@@ -272,19 +461,32 @@ const unmarshalFirstNFunctionSchema = z.object({
272
461
  }));
273
462
  const unmarshalFlatSchemaSchema = z.object({ fields: z.array(z.lazy(() => unmarshalFieldDefinitionSchema)).optional() }).transform((d) => ({ fields: d.fields }));
274
463
  const unmarshalFunctionSchema = z.object({
464
+ function_type: z.string().optional(),
465
+ extra_parameters: z.array(z.lazy(() => unmarshalFunctionExtraParameterSchema)).optional(),
275
466
  aggregation_function: z.lazy(() => unmarshalAggregationFunctionSchema).optional(),
276
467
  column_selection: z.lazy(() => unmarshalColumnSelectionSchema).optional(),
277
468
  custom_udf: z.lazy(() => unmarshalCustomUdfSchema).optional()
278
- }).transform((d) => ({ function: d.aggregation_function !== void 0 ? {
279
- $case: "aggregationFunction",
280
- aggregationFunction: d.aggregation_function
281
- } : d.column_selection !== void 0 ? {
282
- $case: "columnSelection",
283
- columnSelection: d.column_selection
284
- } : d.custom_udf !== void 0 ? {
285
- $case: "customUdf",
286
- customUdf: d.custom_udf
287
- } : void 0 }));
469
+ }).transform((d) => ({
470
+ functionType: d.function_type,
471
+ extraParameters: d.extra_parameters,
472
+ function: d.aggregation_function !== void 0 ? {
473
+ $case: "aggregationFunction",
474
+ aggregationFunction: d.aggregation_function
475
+ } : d.column_selection !== void 0 ? {
476
+ $case: "columnSelection",
477
+ columnSelection: d.column_selection
478
+ } : d.custom_udf !== void 0 ? {
479
+ $case: "customUdf",
480
+ customUdf: d.custom_udf
481
+ } : void 0
482
+ }));
483
+ const unmarshalFunctionExtraParameterSchema = z.object({
484
+ key: z.string().optional(),
485
+ value: z.string().optional()
486
+ }).transform((d) => ({
487
+ key: d.key,
488
+ value: d.value
489
+ }));
288
490
  const unmarshalIngestionConfigSchema = z.object({
289
491
  ingestion_destination: z.lazy(() => unmarshalIngestionDestinationSchema).optional(),
290
492
  backfill_source: z.lazy(() => unmarshalBackfillSourceSchema).optional(),
@@ -299,14 +501,18 @@ const unmarshalIngestionConfigSchema = z.object({
299
501
  z.number(),
300
502
  z.bigint(),
301
503
  z.string()
302
- ]).transform((v) => BigInt(v)).optional()
504
+ ]).transform((v) => BigInt(v)).optional(),
505
+ tags: z.record(z.string(), z.string()).optional(),
506
+ budget_policy_id: z.string().optional()
303
507
  }).transform((d) => ({
304
508
  ingestionDestination: d.ingestion_destination,
305
509
  backfillSource: d.backfill_source,
306
510
  deduplicationColumns: d.deduplication_columns,
307
511
  ingestionPipelineId: d.ingestion_pipeline_id,
308
512
  ingestionJobId: d.ingestion_job_id,
309
- backfillJobId: d.backfill_job_id
513
+ backfillJobId: d.backfill_job_id,
514
+ tags: d.tags,
515
+ budgetPolicyId: d.budget_policy_id
310
516
  }));
311
517
  const unmarshalIngestionDestinationSchema = z.object({ delta_table_name: z.string().optional() }).transform((d) => ({ ingestionDestination: d.delta_table_name !== void 0 ? {
312
518
  $case: "deltaTableName",
@@ -357,9 +563,13 @@ const unmarshalKafkaConfigSchema = z.object({
357
563
  }));
358
564
  const unmarshalKafkaSourceSchema = z.object({
359
565
  name: z.string().optional(),
566
+ entity_column_identifiers: z.array(z.lazy(() => unmarshalColumnIdentifierSchema)).optional(),
567
+ timeseries_column_identifier: z.lazy(() => unmarshalColumnIdentifierSchema).optional(),
360
568
  filter_condition: z.string().optional()
361
569
  }).transform((d) => ({
362
570
  name: d.name,
571
+ entityColumnIdentifiers: d.entity_column_identifiers,
572
+ timeseriesColumnIdentifier: d.timeseries_column_identifier,
363
573
  filterCondition: d.filter_condition
364
574
  }));
365
575
  const unmarshalKafkaStreamConfigSchema = z.object({
@@ -467,10 +677,14 @@ const unmarshalMaterializedFeatureSchema = z.object({
467
677
  table_name: z.string().optional(),
468
678
  pipeline_schedule_state: z.string().optional(),
469
679
  last_materialization_time: z.string().transform((s) => Temporal.Instant.from(s)).optional(),
680
+ cron_schedule: z.string().optional(),
470
681
  is_online: z.boolean().optional(),
471
682
  cron_schedule_trigger: z.lazy(() => unmarshalCronScheduleSchema).optional(),
472
683
  table_trigger: z.lazy(() => unmarshalTableTriggerSchema).optional(),
473
- streaming_mode: z.lazy(() => unmarshalStreamingModeSchema).optional()
684
+ streaming_mode: z.lazy(() => unmarshalStreamingModeSchema).optional(),
685
+ latest_backfill_operation: z.string().optional(),
686
+ tags: z.record(z.string(), z.string()).optional(),
687
+ budget_policy_id: z.string().optional()
474
688
  }).transform((d) => ({
475
689
  materializedFeatureId: d.materialized_feature_id,
476
690
  featureName: d.feature_name,
@@ -484,6 +698,7 @@ const unmarshalMaterializedFeatureSchema = z.object({
484
698
  tableName: d.table_name,
485
699
  pipelineScheduleState: d.pipeline_schedule_state,
486
700
  lastMaterializationTime: d.last_materialization_time,
701
+ cronSchedule: d.cron_schedule,
487
702
  isOnline: d.is_online,
488
703
  trigger: d.cron_schedule_trigger !== void 0 ? {
489
704
  $case: "cronScheduleTrigger",
@@ -494,7 +709,10 @@ const unmarshalMaterializedFeatureSchema = z.object({
494
709
  } : d.streaming_mode !== void 0 ? {
495
710
  $case: "streamingMode",
496
711
  streamingMode: d.streaming_mode
497
- } : void 0
712
+ } : void 0,
713
+ latestBackfillOperation: d.latest_backfill_operation,
714
+ tags: d.tags,
715
+ budgetPolicyId: d.budget_policy_id
498
716
  }));
499
717
  const unmarshalMaxFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
500
718
  const unmarshalMinFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
@@ -533,6 +751,24 @@ const unmarshalOnlineStoreConfigSchema = z.object({
533
751
  tableNamePrefix: d.table_name_prefix,
534
752
  onlineStoreName: d.online_store_name
535
753
  }));
754
+ const unmarshalOperationSchema = z.object({
755
+ name: z.string().optional(),
756
+ metadata: z.record(z.string(), z.unknown()).optional(),
757
+ done: z.boolean().optional(),
758
+ error: z.lazy(() => unmarshalApiErrorSchema).optional(),
759
+ response: z.record(z.string(), z.unknown()).optional()
760
+ }).transform((d) => ({
761
+ name: d.name,
762
+ metadata: d.metadata,
763
+ done: d.done,
764
+ result: d.error !== void 0 ? {
765
+ $case: "error",
766
+ error: d.error
767
+ } : d.response !== void 0 ? {
768
+ $case: "response",
769
+ response: d.response
770
+ } : void 0
771
+ }));
536
772
  const unmarshalProtoSchemaSpecSchema = z.object({
537
773
  schema_text: z.string().optional(),
538
774
  message_name: z.string().optional()
@@ -540,6 +776,47 @@ const unmarshalProtoSchemaSpecSchema = z.object({
540
776
  schemaText: d.schema_text,
541
777
  messageName: d.message_name
542
778
  }));
779
+ const unmarshalPurgeFeatureEntitiesMetadataSchema = z.object({
780
+ features: z.array(z.string()).optional(),
781
+ entities_table: z.string().optional(),
782
+ entities_table_version: z.string().optional(),
783
+ create_time: z.string().transform((s) => Temporal.Instant.from(s)).optional(),
784
+ state: z.string().optional(),
785
+ job_id: z.union([
786
+ z.number(),
787
+ z.bigint(),
788
+ z.string()
789
+ ]).transform((v) => BigInt(v)).optional()
790
+ }).transform((d) => ({
791
+ features: d.features,
792
+ entitiesTable: d.entities_table,
793
+ entitiesTableVersion: d.entities_table_version,
794
+ createTime: d.create_time,
795
+ state: d.state,
796
+ jobId: d.job_id
797
+ }));
798
+ const unmarshalPurgeFeatureEntitiesResponseSchema = z.object({
799
+ metadata: z.lazy(() => unmarshalPurgeFeatureEntitiesMetadataSchema).optional(),
800
+ results: z.array(z.lazy(() => unmarshalPurgeFeatureEntitiesResultSchema)).optional(),
801
+ state: z.string().optional(),
802
+ error: z.lazy(() => unmarshalApiErrorSchema).optional()
803
+ }).transform((d) => ({
804
+ metadata: d.metadata,
805
+ results: d.results,
806
+ state: d.state,
807
+ error: d.error
808
+ }));
809
+ const unmarshalPurgeFeatureEntitiesResultSchema = z.object({
810
+ feature: z.string().optional(),
811
+ offline_state: z.string().optional(),
812
+ online_state: z.string().optional(),
813
+ error: z.lazy(() => unmarshalApiErrorSchema).optional()
814
+ }).transform((d) => ({
815
+ feature: d.feature,
816
+ offlineState: d.offline_state,
817
+ onlineState: d.online_state,
818
+ error: d.error
819
+ }));
543
820
  const unmarshalRequestSourceSchema = z.object({ flat_schema: z.lazy(() => unmarshalFlatSchemaSchema).optional() }).transform((d) => ({ schema: d.flat_schema !== void 0 ? {
544
821
  $case: "flatSchema",
545
822
  flatSchema: d.flat_schema
@@ -622,6 +899,8 @@ const unmarshalStreamSchema = z.object({
622
899
  connection_config: z.lazy(() => unmarshalStreamConnectionConfigSchema).optional(),
623
900
  schema_config: z.lazy(() => unmarshalStreamSchemaConfigSchema).optional(),
624
901
  ingestion_config: z.lazy(() => unmarshalIngestionConfigSchema).optional(),
902
+ record_type_filter: z.string().optional(),
903
+ excluded_columns: z.array(z.string()).optional(),
625
904
  create_time: z.string().transform((s) => Temporal.Instant.from(s)).optional(),
626
905
  created_by: z.string().optional(),
627
906
  update_time: z.string().transform((s) => Temporal.Instant.from(s)).optional(),
@@ -634,6 +913,8 @@ const unmarshalStreamSchema = z.object({
634
913
  connectionConfig: d.connection_config,
635
914
  schemaConfig: d.schema_config,
636
915
  ingestionConfig: d.ingestion_config,
916
+ recordTypeFilter: d.record_type_filter,
917
+ excludedColumns: d.excluded_columns,
637
918
  createTime: d.create_time,
638
919
  createdBy: d.created_by,
639
920
  updateTime: d.update_time,
@@ -707,13 +988,17 @@ const unmarshalSubscriptionModeSchema = z.object({
707
988
  const unmarshalSumFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
708
989
  const unmarshalTableTriggerSchema = z.object({});
709
990
  const unmarshalTimeWindowSchema = z.object({
991
+ continuous: z.lazy(() => unmarshalContinuousWindowSchema).optional(),
710
992
  tumbling: z.lazy(() => unmarshalTumblingWindowSchema).optional(),
711
993
  sliding: z.lazy(() => unmarshalSlidingWindowSchema).optional(),
712
994
  rolling: z.lazy(() => unmarshalRollingWindowSchema).optional(),
713
995
  sawtooth: z.lazy(() => unmarshalSawtoothWindowSchema).optional(),
714
996
  start_time: z.string().transform((s) => Temporal.Instant.from(s)).optional()
715
997
  }).transform((d) => ({
716
- windowType: d.tumbling !== void 0 ? {
998
+ windowType: d.continuous !== void 0 ? {
999
+ $case: "continuous",
1000
+ continuous: d.continuous
1001
+ } : d.tumbling !== void 0 ? {
717
1002
  $case: "tumbling",
718
1003
  tumbling: d.tumbling
719
1004
  } : d.sliding !== void 0 ? {
@@ -859,6 +1144,22 @@ const marshalAuthConfigSchema = z.object({ authConfig: z.discriminatedUnion("$ca
859
1144
  ...d.authConfig?.$case === "mtlsConfig" && { mtls_config: d.authConfig.mtlsConfig }
860
1145
  }));
861
1146
  const marshalAvgFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
1147
+ const marshalBackfillFeaturesRequestSchema = z.object({
1148
+ featureFullNames: z.array(z.string()).optional(),
1149
+ backfillRanges: z.array(z.lazy(() => marshalBackfillRangeSchema)).optional(),
1150
+ requestId: z.string().optional()
1151
+ }).transform((d) => ({
1152
+ feature_full_names: d.featureFullNames,
1153
+ backfill_ranges: d.backfillRanges,
1154
+ request_id: d.requestId
1155
+ }));
1156
+ const marshalBackfillRangeSchema = z.object({
1157
+ startTime: z.any().transform((d) => d.toString()).optional(),
1158
+ endTime: z.any().transform((d) => d.toString()).optional()
1159
+ }).transform((d) => ({
1160
+ start_time: d.startTime,
1161
+ end_time: d.endTime
1162
+ }));
862
1163
  const marshalBackfillSourceSchema = z.object({ backfillSource: z.discriminatedUnion("$case", [z.object({
863
1164
  $case: z.literal("deltaTableSource"),
864
1165
  deltaTableSource: z.lazy(() => marshalDeltaTableSourceSchema)
@@ -870,10 +1171,25 @@ const marshalBackfillSourceSchema = z.object({ backfillSource: z.discriminatedUn
870
1171
  ...d.backfillSource?.$case === "deltaTableName" && { delta_table_name: d.backfillSource.deltaTableName }
871
1172
  }));
872
1173
  const marshalBatchCreateMaterializedFeaturesRequestSchema = z.object({ requests: z.array(z.lazy(() => marshalCreateMaterializedFeatureRequestSchema)).optional() }).transform((d) => ({ requests: d.requests }));
1174
+ const marshalCancelOperationRequestSchema = z.object({ name: z.string().optional() }).transform((d) => ({ name: d.name }));
1175
+ const marshalColumnIdentifierSchema = z.object({ variantExprPath: z.string().optional() }).transform((d) => ({ variant_expr_path: d.variantExprPath }));
873
1176
  const marshalColumnSelectionSchema = z.object({ column: z.string().optional() }).transform((d) => ({ column: d.column }));
1177
+ const marshalContinuousWindowSchema = z.object({
1178
+ windowDuration: z.any().transform((d) => d.toString().slice(2).toLowerCase()).optional(),
1179
+ offset: z.any().transform((d) => d.toString().slice(2).toLowerCase()).optional()
1180
+ }).transform((d) => ({
1181
+ window_duration: d.windowDuration,
1182
+ offset: d.offset
1183
+ }));
874
1184
  const marshalCountFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
875
1185
  const marshalCreateMaterializedFeatureRequestSchema = z.object({ materializedFeature: z.lazy(() => marshalMaterializedFeatureSchema).optional() }).transform((d) => ({ materialized_feature: d.materializedFeature }));
876
- const marshalCronScheduleSchema = z.object({ cronExpression: z.string().optional() }).transform((d) => ({ cron_expression: d.cronExpression }));
1186
+ const marshalCronScheduleSchema = z.object({
1187
+ cronExpression: z.string().optional(),
1188
+ mode: z.string().optional()
1189
+ }).transform((d) => ({
1190
+ cron_expression: d.cronExpression,
1191
+ mode: d.mode
1192
+ }));
877
1193
  const marshalCustomUdfSchema = z.object({
878
1194
  functionPath: z.string().optional(),
879
1195
  inputBindings: z.array(z.lazy(() => marshalInputBindingSchema)).optional()
@@ -898,6 +1214,10 @@ const marshalDataSourceSchema = z.object({
898
1214
  z.object({
899
1215
  $case: z.literal("streamSource"),
900
1216
  streamSource: z.lazy(() => marshalStreamSourceSchema)
1217
+ }),
1218
+ z.object({
1219
+ $case: z.literal("featureViewSource"),
1220
+ featureViewSource: z.lazy(() => marshalFeatureViewSourceSchema)
901
1221
  })
902
1222
  ]).optional(),
903
1223
  lateness: z.lazy(() => marshalSourceLatenessSchema).optional()
@@ -906,15 +1226,20 @@ const marshalDataSourceSchema = z.object({
906
1226
  ...d.dataSource?.$case === "kafkaSource" && { kafka_source: d.dataSource.kafkaSource },
907
1227
  ...d.dataSource?.$case === "requestSource" && { request_source: d.dataSource.requestSource },
908
1228
  ...d.dataSource?.$case === "streamSource" && { stream_source: d.dataSource.streamSource },
1229
+ ...d.dataSource?.$case === "featureViewSource" && { feature_view_source: d.dataSource.featureViewSource },
909
1230
  lateness: d.lateness
910
1231
  }));
911
1232
  const marshalDeltaTableSourceSchema = z.object({
912
1233
  fullName: z.string().optional(),
1234
+ entityColumns: z.array(z.string()).optional(),
1235
+ timeseriesColumn: z.string().optional(),
913
1236
  filterCondition: z.string().optional(),
914
1237
  transformationSql: z.string().optional(),
915
1238
  dataframeSchema: z.string().optional()
916
1239
  }).transform((d) => ({
917
1240
  full_name: d.fullName,
1241
+ entity_columns: d.entityColumns,
1242
+ timeseries_column: d.timeseriesColumn,
918
1243
  filter_condition: d.filterCondition,
919
1244
  transformation_sql: d.transformationSql,
920
1245
  dataframe_schema: d.dataframeSchema
@@ -937,8 +1262,11 @@ const marshalEntityColumnSchema = z.object({ name: z.string().optional() }).tran
937
1262
  const marshalFeatureSchema = z.object({
938
1263
  fullName: z.string().optional(),
939
1264
  source: z.lazy(() => marshalDataSourceSchema).optional(),
1265
+ inputs: z.array(z.string()).optional(),
940
1266
  function: z.lazy(() => marshalFunctionSchema).optional(),
1267
+ timeWindow: z.lazy(() => marshalTimeWindowSchema).optional(),
941
1268
  description: z.string().optional(),
1269
+ filterCondition: z.string().optional(),
942
1270
  lineageContext: z.lazy(() => marshalLineageContextSchema).optional(),
943
1271
  entities: z.array(z.lazy(() => marshalEntityColumnSchema)).optional(),
944
1272
  timeseriesColumn: z.lazy(() => marshalTimeseriesColumnSchema).optional(),
@@ -950,8 +1278,11 @@ const marshalFeatureSchema = z.object({
950
1278
  }).transform((d) => ({
951
1279
  full_name: d.fullName,
952
1280
  source: d.source,
1281
+ inputs: d.inputs,
953
1282
  function: d.function,
1283
+ time_window: d.timeWindow,
954
1284
  description: d.description,
1285
+ filter_condition: d.filterCondition,
955
1286
  lineage_context: d.lineageContext,
956
1287
  entities: d.entities,
957
1288
  timeseries_column: d.timeseriesColumn,
@@ -961,6 +1292,8 @@ const marshalFeatureSchema = z.object({
961
1292
  created_at: d.createdAt,
962
1293
  created_by: d.createdBy
963
1294
  }));
1295
+ const marshalFeatureReferenceSchema = z.object({ feature: z.string().optional() }).transform((d) => ({ feature: d.feature }));
1296
+ const marshalFeatureViewSourceSchema = z.object({ featureReferences: z.array(z.lazy(() => marshalFeatureReferenceSchema)).optional() }).transform((d) => ({ feature_references: d.featureReferences }));
964
1297
  const marshalFieldDefinitionSchema = z.object({
965
1298
  name: z.string().optional(),
966
1299
  dataType: z.string().optional()
@@ -984,38 +1317,55 @@ const marshalFirstNFunctionSchema = z.object({
984
1317
  n: d.n
985
1318
  }));
986
1319
  const marshalFlatSchemaSchema = z.object({ fields: z.array(z.lazy(() => marshalFieldDefinitionSchema)).optional() }).transform((d) => ({ fields: d.fields }));
987
- const marshalFunctionSchema = z.object({ function: z.discriminatedUnion("$case", [
988
- z.object({
989
- $case: z.literal("aggregationFunction"),
990
- aggregationFunction: z.lazy(() => marshalAggregationFunctionSchema)
991
- }),
992
- z.object({
993
- $case: z.literal("columnSelection"),
994
- columnSelection: z.lazy(() => marshalColumnSelectionSchema)
995
- }),
996
- z.object({
997
- $case: z.literal("customUdf"),
998
- customUdf: z.lazy(() => marshalCustomUdfSchema)
999
- })
1000
- ]).optional() }).transform((d) => ({
1320
+ const marshalFunctionSchema = z.object({
1321
+ functionType: z.string().optional(),
1322
+ extraParameters: z.array(z.lazy(() => marshalFunctionExtraParameterSchema)).optional(),
1323
+ function: z.discriminatedUnion("$case", [
1324
+ z.object({
1325
+ $case: z.literal("aggregationFunction"),
1326
+ aggregationFunction: z.lazy(() => marshalAggregationFunctionSchema)
1327
+ }),
1328
+ z.object({
1329
+ $case: z.literal("columnSelection"),
1330
+ columnSelection: z.lazy(() => marshalColumnSelectionSchema)
1331
+ }),
1332
+ z.object({
1333
+ $case: z.literal("customUdf"),
1334
+ customUdf: z.lazy(() => marshalCustomUdfSchema)
1335
+ })
1336
+ ]).optional()
1337
+ }).transform((d) => ({
1338
+ function_type: d.functionType,
1339
+ extra_parameters: d.extraParameters,
1001
1340
  ...d.function?.$case === "aggregationFunction" && { aggregation_function: d.function.aggregationFunction },
1002
1341
  ...d.function?.$case === "columnSelection" && { column_selection: d.function.columnSelection },
1003
1342
  ...d.function?.$case === "customUdf" && { custom_udf: d.function.customUdf }
1004
1343
  }));
1344
+ const marshalFunctionExtraParameterSchema = z.object({
1345
+ key: z.string().optional(),
1346
+ value: z.string().optional()
1347
+ }).transform((d) => ({
1348
+ key: d.key,
1349
+ value: d.value
1350
+ }));
1005
1351
  const marshalIngestionConfigSchema = z.object({
1006
1352
  ingestionDestination: z.lazy(() => marshalIngestionDestinationSchema).optional(),
1007
1353
  backfillSource: z.lazy(() => marshalBackfillSourceSchema).optional(),
1008
1354
  deduplicationColumns: z.array(z.string()).optional(),
1009
1355
  ingestionPipelineId: z.string().optional(),
1010
1356
  ingestionJobId: z.bigint().optional(),
1011
- backfillJobId: z.bigint().optional()
1357
+ backfillJobId: z.bigint().optional(),
1358
+ tags: z.record(z.string(), z.string()).optional(),
1359
+ budgetPolicyId: z.string().optional()
1012
1360
  }).transform((d) => ({
1013
1361
  ingestion_destination: d.ingestionDestination,
1014
1362
  backfill_source: d.backfillSource,
1015
1363
  deduplication_columns: d.deduplicationColumns,
1016
1364
  ingestion_pipeline_id: d.ingestionPipelineId,
1017
1365
  ingestion_job_id: d.ingestionJobId,
1018
- backfill_job_id: d.backfillJobId
1366
+ backfill_job_id: d.backfillJobId,
1367
+ tags: d.tags,
1368
+ budget_policy_id: d.budgetPolicyId
1019
1369
  }));
1020
1370
  const marshalIngestionDestinationSchema = z.object({ ingestionDestination: z.discriminatedUnion("$case", [z.object({
1021
1371
  $case: z.literal("deltaTableName"),
@@ -1058,9 +1408,13 @@ const marshalKafkaConfigSchema = z.object({
1058
1408
  }));
1059
1409
  const marshalKafkaSourceSchema = z.object({
1060
1410
  name: z.string().optional(),
1411
+ entityColumnIdentifiers: z.array(z.lazy(() => marshalColumnIdentifierSchema)).optional(),
1412
+ timeseriesColumnIdentifier: z.lazy(() => marshalColumnIdentifierSchema).optional(),
1061
1413
  filterCondition: z.string().optional()
1062
1414
  }).transform((d) => ({
1063
1415
  name: d.name,
1416
+ entity_column_identifiers: d.entityColumnIdentifiers,
1417
+ timeseries_column_identifier: d.timeseriesColumnIdentifier,
1064
1418
  filter_condition: d.filterCondition
1065
1419
  }));
1066
1420
  const marshalKafkaStreamConfigSchema = z.object({
@@ -1137,6 +1491,7 @@ const marshalMaterializedFeatureSchema = z.object({
1137
1491
  tableName: z.string().optional(),
1138
1492
  pipelineScheduleState: z.string().optional(),
1139
1493
  lastMaterializationTime: z.any().transform((d) => d.toString()).optional(),
1494
+ cronSchedule: z.string().optional(),
1140
1495
  isOnline: z.boolean().optional(),
1141
1496
  trigger: z.discriminatedUnion("$case", [
1142
1497
  z.object({
@@ -1151,7 +1506,10 @@ const marshalMaterializedFeatureSchema = z.object({
1151
1506
  $case: z.literal("streamingMode"),
1152
1507
  streamingMode: z.lazy(() => marshalStreamingModeSchema)
1153
1508
  })
1154
- ]).optional()
1509
+ ]).optional(),
1510
+ latestBackfillOperation: z.string().optional(),
1511
+ tags: z.record(z.string(), z.string()).optional(),
1512
+ budgetPolicyId: z.string().optional()
1155
1513
  }).transform((d) => ({
1156
1514
  materialized_feature_id: d.materializedFeatureId,
1157
1515
  feature_name: d.featureName,
@@ -1160,10 +1518,14 @@ const marshalMaterializedFeatureSchema = z.object({
1160
1518
  table_name: d.tableName,
1161
1519
  pipeline_schedule_state: d.pipelineScheduleState,
1162
1520
  last_materialization_time: d.lastMaterializationTime,
1521
+ cron_schedule: d.cronSchedule,
1163
1522
  is_online: d.isOnline,
1164
1523
  ...d.trigger?.$case === "cronScheduleTrigger" && { cron_schedule_trigger: d.trigger.cronScheduleTrigger },
1165
1524
  ...d.trigger?.$case === "tableTrigger" && { table_trigger: d.trigger.tableTrigger },
1166
- ...d.trigger?.$case === "streamingMode" && { streaming_mode: d.trigger.streamingMode }
1525
+ ...d.trigger?.$case === "streamingMode" && { streaming_mode: d.trigger.streamingMode },
1526
+ latest_backfill_operation: d.latestBackfillOperation,
1527
+ tags: d.tags,
1528
+ budget_policy_id: d.budgetPolicyId
1167
1529
  }));
1168
1530
  const marshalMaxFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
1169
1531
  const marshalMinFunctionSchema = z.object({ input: z.string().optional() }).transform((d) => ({ input: d.input }));
@@ -1209,6 +1571,18 @@ const marshalProtoSchemaSpecSchema = z.object({
1209
1571
  schema_text: d.schemaText,
1210
1572
  message_name: d.messageName
1211
1573
  }));
1574
+ const marshalPurgeFeatureEntitiesRequestSchema = z.object({
1575
+ features: z.array(z.string()).optional(),
1576
+ entities: z.discriminatedUnion("$case", [z.object({
1577
+ $case: z.literal("entitiesTable"),
1578
+ entitiesTable: z.string()
1579
+ })]).optional(),
1580
+ requestId: z.string().optional()
1581
+ }).transform((d) => ({
1582
+ features: d.features,
1583
+ ...d.entities?.$case === "entitiesTable" && { entities_table: d.entities.entitiesTable },
1584
+ request_id: d.requestId
1585
+ }));
1212
1586
  const marshalRequestSourceSchema = z.object({ schema: z.discriminatedUnion("$case", [z.object({
1213
1587
  $case: z.literal("flatSchema"),
1214
1588
  flatSchema: z.lazy(() => marshalFlatSchemaSchema)
@@ -1295,6 +1669,8 @@ const marshalStreamSchema = z.object({
1295
1669
  connectionConfig: z.lazy(() => marshalStreamConnectionConfigSchema).optional(),
1296
1670
  schemaConfig: z.lazy(() => marshalStreamSchemaConfigSchema).optional(),
1297
1671
  ingestionConfig: z.lazy(() => marshalIngestionConfigSchema).optional(),
1672
+ recordTypeFilter: z.string().optional(),
1673
+ excludedColumns: z.array(z.string()).optional(),
1298
1674
  createTime: z.any().transform((d) => d.toString()).optional(),
1299
1675
  createdBy: z.string().optional(),
1300
1676
  updateTime: z.any().transform((d) => d.toString()).optional(),
@@ -1307,6 +1683,8 @@ const marshalStreamSchema = z.object({
1307
1683
  connection_config: d.connectionConfig,
1308
1684
  schema_config: d.schemaConfig,
1309
1685
  ingestion_config: d.ingestionConfig,
1686
+ record_type_filter: d.recordTypeFilter,
1687
+ excluded_columns: d.excludedColumns,
1310
1688
  create_time: d.createTime,
1311
1689
  created_by: d.createdBy,
1312
1690
  update_time: d.updateTime,
@@ -1385,6 +1763,10 @@ const marshalSumFunctionSchema = z.object({ input: z.string().optional() }).tran
1385
1763
  const marshalTableTriggerSchema = z.object({});
1386
1764
  const marshalTimeWindowSchema = z.object({
1387
1765
  windowType: z.discriminatedUnion("$case", [
1766
+ z.object({
1767
+ $case: z.literal("continuous"),
1768
+ continuous: z.lazy(() => marshalContinuousWindowSchema)
1769
+ }),
1388
1770
  z.object({
1389
1771
  $case: z.literal("tumbling"),
1390
1772
  tumbling: z.lazy(() => marshalTumblingWindowSchema)
@@ -1404,6 +1786,7 @@ const marshalTimeWindowSchema = z.object({
1404
1786
  ]).optional(),
1405
1787
  startTime: z.any().transform((d) => d.toString()).optional()
1406
1788
  }).transform((d) => ({
1789
+ ...d.windowType?.$case === "continuous" && { continuous: d.windowType.continuous },
1407
1790
  ...d.windowType?.$case === "tumbling" && { tumbling: d.windowType.tumbling },
1408
1791
  ...d.windowType?.$case === "sliding" && { sliding: d.windowType.sliding },
1409
1792
  ...d.windowType?.$case === "rolling" && { rolling: d.windowType.rolling },
@@ -1520,9 +1903,17 @@ const backfillSourceFieldMaskSchema = {
1520
1903
  children: () => deltaTableSourceFieldMaskSchema
1521
1904
  }
1522
1905
  };
1906
+ const columnIdentifierFieldMaskSchema = { variantExprPath: { wire: "variant_expr_path" } };
1523
1907
  const columnSelectionFieldMaskSchema = { column: { wire: "column" } };
1908
+ const continuousWindowFieldMaskSchema = {
1909
+ offset: { wire: "offset" },
1910
+ windowDuration: { wire: "window_duration" }
1911
+ };
1524
1912
  const countFunctionFieldMaskSchema = { input: { wire: "input" } };
1525
- const cronScheduleFieldMaskSchema = { cronExpression: { wire: "cron_expression" } };
1913
+ const cronScheduleFieldMaskSchema = {
1914
+ cronExpression: { wire: "cron_expression" },
1915
+ mode: { wire: "mode" }
1916
+ };
1526
1917
  const customUdfFieldMaskSchema = {
1527
1918
  functionPath: { wire: "function_path" },
1528
1919
  inputBindings: { wire: "input_bindings" }
@@ -1532,6 +1923,10 @@ const dataSourceFieldMaskSchema = {
1532
1923
  wire: "delta_table_source",
1533
1924
  children: () => deltaTableSourceFieldMaskSchema
1534
1925
  },
1926
+ featureViewSource: {
1927
+ wire: "feature_view_source",
1928
+ children: () => featureViewSourceFieldMaskSchema
1929
+ },
1535
1930
  kafkaSource: {
1536
1931
  wire: "kafka_source",
1537
1932
  children: () => kafkaSourceFieldMaskSchema
@@ -1551,8 +1946,10 @@ const dataSourceFieldMaskSchema = {
1551
1946
  };
1552
1947
  const deltaTableSourceFieldMaskSchema = {
1553
1948
  dataframeSchema: { wire: "dataframe_schema" },
1949
+ entityColumns: { wire: "entity_columns" },
1554
1950
  filterCondition: { wire: "filter_condition" },
1555
1951
  fullName: { wire: "full_name" },
1952
+ timeseriesColumn: { wire: "timeseries_column" },
1556
1953
  transformationSql: { wire: "transformation_sql" }
1557
1954
  };
1558
1955
  const directMtlsConfigFieldMaskSchema = {
@@ -1578,11 +1975,13 @@ const featureFieldMaskSchema = {
1578
1975
  createdBy: { wire: "created_by" },
1579
1976
  description: { wire: "description" },
1580
1977
  entities: { wire: "entities" },
1978
+ filterCondition: { wire: "filter_condition" },
1581
1979
  fullName: { wire: "full_name" },
1582
1980
  function: {
1583
1981
  wire: "function",
1584
1982
  children: () => functionFieldMaskSchema
1585
1983
  },
1984
+ inputs: { wire: "inputs" },
1586
1985
  lineageContext: {
1587
1986
  wire: "lineage_context",
1588
1987
  children: () => lineageContextFieldMaskSchema
@@ -1593,6 +1992,10 @@ const featureFieldMaskSchema = {
1593
1992
  wire: "source",
1594
1993
  children: () => dataSourceFieldMaskSchema
1595
1994
  },
1995
+ timeWindow: {
1996
+ wire: "time_window",
1997
+ children: () => timeWindowFieldMaskSchema
1998
+ },
1596
1999
  timeseriesColumn: {
1597
2000
  wire: "timeseries_column",
1598
2001
  children: () => timeseriesColumnFieldMaskSchema
@@ -1601,6 +2004,7 @@ const featureFieldMaskSchema = {
1601
2004
  function featureFieldMask(...paths) {
1602
2005
  return FieldMask.build(paths, featureFieldMaskSchema);
1603
2006
  }
2007
+ const featureViewSourceFieldMaskSchema = { featureReferences: { wire: "feature_references" } };
1604
2008
  const firstDistinctFunctionFieldMaskSchema = {
1605
2009
  input: { wire: "input" },
1606
2010
  n: { wire: "n" }
@@ -1623,7 +2027,9 @@ const functionFieldMaskSchema = {
1623
2027
  customUdf: {
1624
2028
  wire: "custom_udf",
1625
2029
  children: () => customUdfFieldMaskSchema
1626
- }
2030
+ },
2031
+ extraParameters: { wire: "extra_parameters" },
2032
+ functionType: { wire: "function_type" }
1627
2033
  };
1628
2034
  const ingestionConfigFieldMaskSchema = {
1629
2035
  backfillJobId: { wire: "backfill_job_id" },
@@ -1631,13 +2037,15 @@ const ingestionConfigFieldMaskSchema = {
1631
2037
  wire: "backfill_source",
1632
2038
  children: () => backfillSourceFieldMaskSchema
1633
2039
  },
2040
+ budgetPolicyId: { wire: "budget_policy_id" },
1634
2041
  deduplicationColumns: { wire: "deduplication_columns" },
1635
2042
  ingestionDestination: {
1636
2043
  wire: "ingestion_destination",
1637
2044
  children: () => ingestionDestinationFieldMaskSchema
1638
2045
  },
1639
2046
  ingestionJobId: { wire: "ingestion_job_id" },
1640
- ingestionPipelineId: { wire: "ingestion_pipeline_id" }
2047
+ ingestionPipelineId: { wire: "ingestion_pipeline_id" },
2048
+ tags: { wire: "tags" }
1641
2049
  };
1642
2050
  const ingestionDestinationFieldMaskSchema = { deltaTableName: { wire: "delta_table_name" } };
1643
2051
  const jobContextFieldMaskSchema = {
@@ -1677,8 +2085,13 @@ function kafkaConfigFieldMask(...paths) {
1677
2085
  return FieldMask.build(paths, kafkaConfigFieldMaskSchema);
1678
2086
  }
1679
2087
  const kafkaSourceFieldMaskSchema = {
2088
+ entityColumnIdentifiers: { wire: "entity_column_identifiers" },
1680
2089
  filterCondition: { wire: "filter_condition" },
1681
- name: { wire: "name" }
2090
+ name: { wire: "name" },
2091
+ timeseriesColumnIdentifier: {
2092
+ wire: "timeseries_column_identifier",
2093
+ children: () => columnIdentifierFieldMaskSchema
2094
+ }
1682
2095
  };
1683
2096
  const kafkaStreamConfigFieldMaskSchema = {
1684
2097
  extraOptions: { wire: "extra_options" },
@@ -1720,6 +2133,8 @@ const lineageContextFieldMaskSchema = {
1720
2133
  notebookId: { wire: "notebook_id" }
1721
2134
  };
1722
2135
  const materializedFeatureFieldMaskSchema = {
2136
+ budgetPolicyId: { wire: "budget_policy_id" },
2137
+ cronSchedule: { wire: "cron_schedule" },
1723
2138
  cronScheduleTrigger: {
1724
2139
  wire: "cron_schedule_trigger",
1725
2140
  children: () => cronScheduleFieldMaskSchema
@@ -1727,6 +2142,7 @@ const materializedFeatureFieldMaskSchema = {
1727
2142
  featureName: { wire: "feature_name" },
1728
2143
  isOnline: { wire: "is_online" },
1729
2144
  lastMaterializationTime: { wire: "last_materialization_time" },
2145
+ latestBackfillOperation: { wire: "latest_backfill_operation" },
1730
2146
  materializedFeatureId: { wire: "materialized_feature_id" },
1731
2147
  offlineStoreConfig: {
1732
2148
  wire: "offline_store_config",
@@ -1745,7 +2161,8 @@ const materializedFeatureFieldMaskSchema = {
1745
2161
  tableTrigger: {
1746
2162
  wire: "table_trigger",
1747
2163
  children: () => tableTriggerFieldMaskSchema
1748
- }
2164
+ },
2165
+ tags: { wire: "tags" }
1749
2166
  };
1750
2167
  function materializedFeatureFieldMask(...paths) {
1751
2168
  return FieldMask.build(paths, materializedFeatureFieldMaskSchema);
@@ -1849,11 +2266,13 @@ const streamFieldMaskSchema = {
1849
2266
  createTime: { wire: "create_time" },
1850
2267
  createdBy: { wire: "created_by" },
1851
2268
  description: { wire: "description" },
2269
+ excludedColumns: { wire: "excluded_columns" },
1852
2270
  ingestionConfig: {
1853
2271
  wire: "ingestion_config",
1854
2272
  children: () => ingestionConfigFieldMaskSchema
1855
2273
  },
1856
2274
  name: { wire: "name" },
2275
+ recordTypeFilter: { wire: "record_type_filter" },
1857
2276
  schemaConfig: {
1858
2277
  wire: "schema_config",
1859
2278
  children: () => streamSchemaConfigFieldMaskSchema
@@ -1915,6 +2334,10 @@ const subscriptionModeFieldMaskSchema = {
1915
2334
  const sumFunctionFieldMaskSchema = { input: { wire: "input" } };
1916
2335
  const tableTriggerFieldMaskSchema = {};
1917
2336
  const timeWindowFieldMaskSchema = {
2337
+ continuous: {
2338
+ wire: "continuous",
2339
+ children: () => continuousWindowFieldMaskSchema
2340
+ },
1918
2341
  rolling: {
1919
2342
  wire: "rolling",
1920
2343
  children: () => rollingWindowFieldMaskSchema
@@ -1943,5 +2366,5 @@ const varPopFunctionFieldMaskSchema = { input: { wire: "input" } };
1943
2366
  const varSampFunctionFieldMaskSchema = { input: { wire: "input" } };
1944
2367
 
1945
2368
  //#endregion
1946
- export { MaterializedFeature_PipelineScheduleState, ScalarDataType, SchemaLocator_Format, StreamingMode_StreamingModeType, featureFieldMask, kafkaConfigFieldMask, marshalAggregationFunctionSchema, marshalApproxCountDistinctFunctionSchema, marshalApproxPercentileFunctionSchema, marshalAuthConfigSchema, marshalAvgFunctionSchema, marshalBackfillSourceSchema, marshalBatchCreateMaterializedFeaturesRequestSchema, marshalColumnSelectionSchema, marshalCountFunctionSchema, marshalCreateMaterializedFeatureRequestSchema, marshalCronScheduleSchema, marshalCustomUdfSchema, marshalDataSourceSchema, marshalDeltaTableSourceSchema, marshalDirectMtlsConfigSchema, marshalDirectSchemasSchema, marshalEntityColumnSchema, marshalFeatureSchema, marshalFieldDefinitionSchema, marshalFirstDistinctFunctionSchema, marshalFirstFunctionSchema, marshalFirstNFunctionSchema, marshalFlatSchemaSchema, marshalFunctionSchema, marshalIngestionConfigSchema, marshalIngestionDestinationSchema, marshalInputBindingSchema, marshalJobContextSchema, marshalKafkaConfigSchema, marshalKafkaSourceSchema, marshalKafkaStreamConfigSchema, marshalKafkaSubscriptionModeSchema, marshalKinesisStreamConfigSchema, marshalLastDistinctFunctionSchema, marshalLastFunctionSchema, marshalLastNFunctionSchema, marshalLineageContextSchema, marshalMaterializedFeatureSchema, marshalMaxFunctionSchema, marshalMinFunctionSchema, marshalMtlsConfigSchema, marshalOfflineStoreConfigSchema, marshalOnlineStoreConfigSchema, marshalProtoSchemaSpecSchema, marshalRequestSourceSchema, marshalRollingWindowSchema, marshalSawtoothWindowSchema, marshalSchemaConfigSchema, marshalSchemaLocatorSchema, marshalSchemaLocator_ConfluentSchemaSchema, marshalSchemaRegistryConfigSchema, marshalSecretScopeReferenceSchema, marshalSlidingWindowSchema, marshalSourceLatenessSchema, marshalStddevPopFunctionSchema, marshalStddevSampFunctionSchema, marshalStreamArnListSchema, marshalStreamConnectionConfigSchema, marshalStreamNameListSchema, marshalStreamSchema, marshalStreamSchemaConfigSchema, marshalStreamSourceConfigSchema, marshalStreamSourceSchema, marshalStreamingModeSchema, marshalSubscriptionModeSchema, marshalSumFunctionSchema, marshalTableTriggerSchema, marshalTimeWindowSchema, marshalTimeseriesColumnSchema, marshalTumblingWindowSchema, marshalVarPopFunctionSchema, marshalVarSampFunctionSchema, materializedFeatureFieldMask, streamFieldMask, unmarshalAggregationFunctionSchema, unmarshalApproxCountDistinctFunctionSchema, unmarshalApproxPercentileFunctionSchema, unmarshalAuthConfigSchema, unmarshalAvgFunctionSchema, unmarshalBackfillSourceSchema, unmarshalBatchCreateMaterializedFeaturesResponseSchema, unmarshalColumnSelectionSchema, unmarshalCountFunctionSchema, unmarshalCronScheduleSchema, unmarshalCustomUdfSchema, unmarshalDataSourceSchema, unmarshalDeltaTableSourceSchema, unmarshalDirectMtlsConfigSchema, unmarshalDirectSchemasSchema, unmarshalEntityColumnSchema, unmarshalFeatureSchema, unmarshalFieldDefinitionSchema, unmarshalFirstDistinctFunctionSchema, unmarshalFirstFunctionSchema, unmarshalFirstNFunctionSchema, unmarshalFlatSchemaSchema, unmarshalFunctionSchema, unmarshalIngestionConfigSchema, unmarshalIngestionDestinationSchema, unmarshalInputBindingSchema, unmarshalJobContextSchema, unmarshalKafkaConfigSchema, unmarshalKafkaSourceSchema, unmarshalKafkaStreamConfigSchema, unmarshalKafkaSubscriptionModeSchema, unmarshalKinesisStreamConfigSchema, unmarshalLastDistinctFunctionSchema, unmarshalLastFunctionSchema, unmarshalLastNFunctionSchema, unmarshalLineageContextSchema, unmarshalListFeaturesResponseSchema, unmarshalListKafkaConfigsResponseSchema, unmarshalListMaterializedFeaturesResponseSchema, unmarshalListStreamsResponseSchema, unmarshalMaterializedFeatureSchema, unmarshalMaxFunctionSchema, unmarshalMinFunctionSchema, unmarshalMtlsConfigSchema, unmarshalOfflineStoreConfigSchema, unmarshalOnlineStoreConfigSchema, unmarshalProtoSchemaSpecSchema, unmarshalRequestSourceSchema, unmarshalRollingWindowSchema, unmarshalSawtoothWindowSchema, unmarshalSchemaConfigSchema, unmarshalSchemaLocatorSchema, unmarshalSchemaLocator_ConfluentSchemaSchema, unmarshalSchemaRegistryConfigSchema, unmarshalSecretScopeReferenceSchema, unmarshalSlidingWindowSchema, unmarshalSourceLatenessSchema, unmarshalStddevPopFunctionSchema, unmarshalStddevSampFunctionSchema, unmarshalStreamArnListSchema, unmarshalStreamConnectionConfigSchema, unmarshalStreamNameListSchema, unmarshalStreamSchema, unmarshalStreamSchemaConfigSchema, unmarshalStreamSourceConfigSchema, unmarshalStreamSourceSchema, unmarshalStreamingModeSchema, unmarshalSubscriptionModeSchema, unmarshalSumFunctionSchema, unmarshalTableTriggerSchema, unmarshalTimeWindowSchema, unmarshalTimeseriesColumnSchema, unmarshalTumblingWindowSchema, unmarshalVarPopFunctionSchema, unmarshalVarSampFunctionSchema };
2369
+ export { BackfillOperationMetadata_State, CronSchedule_Mode, ErrorCode, FunctionFunctionType, MaterializedFeature_PipelineScheduleState, PurgeFeatureEntitiesMetadata_State, PurgeFeatureEntitiesResult_State, ScalarDataType, SchemaLocator_Format, StreamingMode_StreamingModeType, featureFieldMask, kafkaConfigFieldMask, marshalAggregationFunctionSchema, marshalApproxCountDistinctFunctionSchema, marshalApproxPercentileFunctionSchema, marshalAuthConfigSchema, marshalAvgFunctionSchema, marshalBackfillFeaturesRequestSchema, marshalBackfillRangeSchema, marshalBackfillSourceSchema, marshalBatchCreateMaterializedFeaturesRequestSchema, marshalCancelOperationRequestSchema, marshalColumnIdentifierSchema, marshalColumnSelectionSchema, marshalContinuousWindowSchema, marshalCountFunctionSchema, marshalCreateMaterializedFeatureRequestSchema, marshalCronScheduleSchema, marshalCustomUdfSchema, marshalDataSourceSchema, marshalDeltaTableSourceSchema, marshalDirectMtlsConfigSchema, marshalDirectSchemasSchema, marshalEntityColumnSchema, marshalFeatureReferenceSchema, marshalFeatureSchema, marshalFeatureViewSourceSchema, marshalFieldDefinitionSchema, marshalFirstDistinctFunctionSchema, marshalFirstFunctionSchema, marshalFirstNFunctionSchema, marshalFlatSchemaSchema, marshalFunctionExtraParameterSchema, marshalFunctionSchema, marshalIngestionConfigSchema, marshalIngestionDestinationSchema, marshalInputBindingSchema, marshalJobContextSchema, marshalKafkaConfigSchema, marshalKafkaSourceSchema, marshalKafkaStreamConfigSchema, marshalKafkaSubscriptionModeSchema, marshalKinesisStreamConfigSchema, marshalLastDistinctFunctionSchema, marshalLastFunctionSchema, marshalLastNFunctionSchema, marshalLineageContextSchema, marshalMaterializedFeatureSchema, marshalMaxFunctionSchema, marshalMinFunctionSchema, marshalMtlsConfigSchema, marshalOfflineStoreConfigSchema, marshalOnlineStoreConfigSchema, marshalProtoSchemaSpecSchema, marshalPurgeFeatureEntitiesRequestSchema, marshalRequestSourceSchema, marshalRollingWindowSchema, marshalSawtoothWindowSchema, marshalSchemaConfigSchema, marshalSchemaLocatorSchema, marshalSchemaLocator_ConfluentSchemaSchema, marshalSchemaRegistryConfigSchema, marshalSecretScopeReferenceSchema, marshalSlidingWindowSchema, marshalSourceLatenessSchema, marshalStddevPopFunctionSchema, marshalStddevSampFunctionSchema, marshalStreamArnListSchema, marshalStreamConnectionConfigSchema, marshalStreamNameListSchema, marshalStreamSchema, marshalStreamSchemaConfigSchema, marshalStreamSourceConfigSchema, marshalStreamSourceSchema, marshalStreamingModeSchema, marshalSubscriptionModeSchema, marshalSumFunctionSchema, marshalTableTriggerSchema, marshalTimeWindowSchema, marshalTimeseriesColumnSchema, marshalTumblingWindowSchema, marshalVarPopFunctionSchema, marshalVarSampFunctionSchema, materializedFeatureFieldMask, streamFieldMask, unmarshalAggregationFunctionSchema, unmarshalApiErrorSchema, unmarshalApproxCountDistinctFunctionSchema, unmarshalApproxPercentileFunctionSchema, unmarshalAuthConfigSchema, unmarshalAvgFunctionSchema, unmarshalBackfillFeaturesResponseSchema, unmarshalBackfillOperationMetadataSchema, unmarshalBackfillRangeSchema, unmarshalBackfillSourceSchema, unmarshalBatchCreateMaterializedFeaturesResponseSchema, unmarshalColumnIdentifierSchema, unmarshalColumnSelectionSchema, unmarshalContinuousWindowSchema, unmarshalCountFunctionSchema, unmarshalCronScheduleSchema, unmarshalCustomUdfSchema, unmarshalDataSourceSchema, unmarshalDeltaTableSourceSchema, unmarshalDirectMtlsConfigSchema, unmarshalDirectSchemasSchema, unmarshalEntityColumnSchema, unmarshalFeatureReferenceSchema, unmarshalFeatureSchema, unmarshalFeatureViewSourceSchema, unmarshalFieldDefinitionSchema, unmarshalFirstDistinctFunctionSchema, unmarshalFirstFunctionSchema, unmarshalFirstNFunctionSchema, unmarshalFlatSchemaSchema, unmarshalFunctionExtraParameterSchema, unmarshalFunctionSchema, unmarshalIngestionConfigSchema, unmarshalIngestionDestinationSchema, unmarshalInputBindingSchema, unmarshalJobContextSchema, unmarshalKafkaConfigSchema, unmarshalKafkaSourceSchema, unmarshalKafkaStreamConfigSchema, unmarshalKafkaSubscriptionModeSchema, unmarshalKinesisStreamConfigSchema, unmarshalLastDistinctFunctionSchema, unmarshalLastFunctionSchema, unmarshalLastNFunctionSchema, unmarshalLineageContextSchema, unmarshalListFeaturesResponseSchema, unmarshalListKafkaConfigsResponseSchema, unmarshalListMaterializedFeaturesResponseSchema, unmarshalListStreamsResponseSchema, unmarshalMaterializedFeatureSchema, unmarshalMaxFunctionSchema, unmarshalMinFunctionSchema, unmarshalMtlsConfigSchema, unmarshalOfflineStoreConfigSchema, unmarshalOnlineStoreConfigSchema, unmarshalOperationSchema, unmarshalProtoSchemaSpecSchema, unmarshalPurgeFeatureEntitiesMetadataSchema, unmarshalPurgeFeatureEntitiesResponseSchema, unmarshalPurgeFeatureEntitiesResultSchema, unmarshalRequestSourceSchema, unmarshalRollingWindowSchema, unmarshalSawtoothWindowSchema, unmarshalSchemaConfigSchema, unmarshalSchemaLocatorSchema, unmarshalSchemaLocator_ConfluentSchemaSchema, unmarshalSchemaRegistryConfigSchema, unmarshalSecretScopeReferenceSchema, unmarshalSlidingWindowSchema, unmarshalSourceLatenessSchema, unmarshalStddevPopFunctionSchema, unmarshalStddevSampFunctionSchema, unmarshalStreamArnListSchema, unmarshalStreamConnectionConfigSchema, unmarshalStreamNameListSchema, unmarshalStreamSchema, unmarshalStreamSchemaConfigSchema, unmarshalStreamSourceConfigSchema, unmarshalStreamSourceSchema, unmarshalStreamingModeSchema, unmarshalSubscriptionModeSchema, unmarshalSumFunctionSchema, unmarshalTableTriggerSchema, unmarshalTimeWindowSchema, unmarshalTimeseriesColumnSchema, unmarshalTumblingWindowSchema, unmarshalVarPopFunctionSchema, unmarshalVarSampFunctionSchema };
1947
2370
  //# sourceMappingURL=model.js.map