@databricks/sdk-features 0.1.0-dev.3 → 0.1.0-dev.4

package/dist/v1/model.js

@@ -6,69 +6,66 @@ import { z } from 'zod';
  * Scalar data types for request-time field definitions.
  * Only flat (non-nested) types are supported.
  */
-export var ScalarDataType;
-(function (ScalarDataType) {
-    ScalarDataType["SCALAR_DATA_TYPE_UNSPECIFIED"] = "SCALAR_DATA_TYPE_UNSPECIFIED";
-    ScalarDataType["INTEGER"] = "INTEGER";
-    ScalarDataType["FLOAT"] = "FLOAT";
-    ScalarDataType["BOOLEAN"] = "BOOLEAN";
-    ScalarDataType["STRING"] = "STRING";
-    ScalarDataType["DOUBLE"] = "DOUBLE";
-    ScalarDataType["LONG"] = "LONG";
-    ScalarDataType["TIMESTAMP"] = "TIMESTAMP";
-    ScalarDataType["DATE"] = "DATE";
-    ScalarDataType["SHORT"] = "SHORT";
-    ScalarDataType["BINARY"] = "BINARY";
-    ScalarDataType["DECIMAL"] = "DECIMAL";
-})(ScalarDataType || (ScalarDataType = {}));
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
+export const ScalarDataType = {
+    SCALAR_DATA_TYPE_UNSPECIFIED: 'SCALAR_DATA_TYPE_UNSPECIFIED',
+    INTEGER: 'INTEGER',
+    FLOAT: 'FLOAT',
+    BOOLEAN: 'BOOLEAN',
+    STRING: 'STRING',
+    DOUBLE: 'DOUBLE',
+    LONG: 'LONG',
+    TIMESTAMP: 'TIMESTAMP',
+    DATE: 'DATE',
+    SHORT: 'SHORT',
+    BINARY: 'BINARY',
+    DECIMAL: 'DECIMAL',
+};
 /** Deprecated: Use the function-specific messages in AggregationFunction.function_type oneof instead. Kept for backwards compatibility. */
-// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested enum name.
-export var Function_FunctionType;
-(function (Function_FunctionType) {
-    Function_FunctionType["FUNCTION_TYPE_UNSPECIFIED"] = "FUNCTION_TYPE_UNSPECIFIED";
-    Function_FunctionType["AVG"] = "AVG";
-    Function_FunctionType["COUNT"] = "COUNT";
-    Function_FunctionType["SUM"] = "SUM";
-    Function_FunctionType["MIN"] = "MIN";
-    Function_FunctionType["MAX"] = "MAX";
-    Function_FunctionType["FIRST"] = "FIRST";
-    Function_FunctionType["LAST"] = "LAST";
-    Function_FunctionType["APPROX_COUNT_DISTINCT"] = "APPROX_COUNT_DISTINCT";
-    Function_FunctionType["APPROX_PERCENTILE"] = "APPROX_PERCENTILE";
-    Function_FunctionType["STDDEV_POP"] = "STDDEV_POP";
-    Function_FunctionType["STDDEV_SAMP"] = "STDDEV_SAMP";
-    Function_FunctionType["VAR_POP"] = "VAR_POP";
-    Function_FunctionType["VAR_SAMP"] = "VAR_SAMP";
-})(Function_FunctionType || (Function_FunctionType = {}));
-// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested enum name.
-export var MaterializedFeature_PipelineScheduleState;
-(function (MaterializedFeature_PipelineScheduleState) {
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
+export const Function_FunctionType = {
+    FUNCTION_TYPE_UNSPECIFIED: 'FUNCTION_TYPE_UNSPECIFIED',
+    AVG: 'AVG',
+    COUNT: 'COUNT',
+    SUM: 'SUM',
+    MIN: 'MIN',
+    MAX: 'MAX',
+    FIRST: 'FIRST',
+    LAST: 'LAST',
+    APPROX_COUNT_DISTINCT: 'APPROX_COUNT_DISTINCT',
+    APPROX_PERCENTILE: 'APPROX_PERCENTILE',
+    STDDEV_POP: 'STDDEV_POP',
+    STDDEV_SAMP: 'STDDEV_SAMP',
+    VAR_POP: 'VAR_POP',
+    VAR_SAMP: 'VAR_SAMP',
+};
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
+export const MaterializedFeature_PipelineScheduleState = {
     /** Default value, not used. */
-    MaterializedFeature_PipelineScheduleState["PIPELINE_SCHEDULE_STATE_UNSPECIFIED"] = "PIPELINE_SCHEDULE_STATE_UNSPECIFIED";
+    PIPELINE_SCHEDULE_STATE_UNSPECIFIED: 'PIPELINE_SCHEDULE_STATE_UNSPECIFIED',
     /** Pipeline was configured to run once then stop. */
-    MaterializedFeature_PipelineScheduleState["SNAPSHOT"] = "SNAPSHOT";
+    SNAPSHOT: 'SNAPSHOT',
     /** Pipeline is actively running and computing features. */
-    MaterializedFeature_PipelineScheduleState["ACTIVE"] = "ACTIVE";
+    ACTIVE: 'ACTIVE',
     /** Pipeline is paused and not computing features. */
-    MaterializedFeature_PipelineScheduleState["PAUSED"] = "PAUSED";
-})(MaterializedFeature_PipelineScheduleState || (MaterializedFeature_PipelineScheduleState = {}));
-// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested enum name.
-export var StreamingMode_StreamingModeType;
-(function (StreamingMode_StreamingModeType) {
+    PAUSED: 'PAUSED',
+};
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
+export const StreamingMode_StreamingModeType = {
     /** Default value, not used. */
-    StreamingMode_StreamingModeType["STREAMING_MODE_TYPE_UNSPECIFIED"] = "STREAMING_MODE_TYPE_UNSPECIFIED";
+    STREAMING_MODE_TYPE_UNSPECIFIED: 'STREAMING_MODE_TYPE_UNSPECIFIED',
     /**
      * Real-time mode. Ultra-low-latency trigger intended for operational workloads
      * that need responses in milliseconds or sub-second latency.
      */
-    StreamingMode_StreamingModeType["STREAMING_MODE_TYPE_RTM"] = "STREAMING_MODE_TYPE_RTM";
+    STREAMING_MODE_TYPE_RTM: 'STREAMING_MODE_TYPE_RTM',
     /**
      * Micro-batch mode in Structured Streaming. Better suited for ETL and analytics
      * workloads where latency is measured in seconds or minutes and cost efficiency
      * matters more.
      */
-    StreamingMode_StreamingModeType["STREAMING_MODE_TYPE_MBM"] = "STREAMING_MODE_TYPE_MBM";
-})(StreamingMode_StreamingModeType || (StreamingMode_StreamingModeType = {}));
+    STREAMING_MODE_TYPE_MBM: 'STREAMING_MODE_TYPE_MBM',
+};
 export const unmarshalAggregationFunctionSchema = z
     .object({
     avg: z.lazy(() => unmarshalAvgFunctionSchema).optional(),
@@ -259,6 +256,7 @@ export const unmarshalDataSourceSchema = z
         .optional(),
     kafka_source: z.lazy(() => unmarshalKafkaSourceSchema).optional(),
     request_source: z.lazy(() => unmarshalRequestSourceSchema).optional(),
+    stream_source: z.lazy(() => unmarshalStreamSourceSchema).optional(),
 })
     .transform(d => ({
     dataSource: d.delta_table_source !== undefined
@@ -270,7 +268,9 @@ export const unmarshalDataSourceSchema = z
             ? { $case: 'kafkaSource', kafkaSource: d.kafka_source }
             : d.request_source !== undefined
                 ? { $case: 'requestSource', requestSource: d.request_source }
-                : undefined,
+                : d.stream_source !== undefined
+                    ? { $case: 'streamSource', streamSource: d.stream_source }
+                    : undefined,
 }));
 export const unmarshalDeltaTableSourceSchema = z
     .object({
@@ -355,7 +355,7 @@ export const unmarshalFeatureSchema = z
 export const unmarshalFieldDefinitionSchema = z
     .object({
     name: z.string().optional(),
-    data_type: z.enum(ScalarDataType).optional(),
+    data_type: z.string().optional(),
 })
     .transform(d => ({
     name: d.name,
@@ -377,7 +377,7 @@ export const unmarshalFlatSchemaSchema = z
 }));
 export const unmarshalFunctionSchema = z
     .object({
-    function_type: z.enum(Function_FunctionType).optional(),
+    function_type: z.string().optional(),
     extra_parameters: z
         .array(z.lazy(() => unmarshalFunction_ExtraParameterSchema))
         .optional(),
@@ -473,6 +473,7 @@ export const unmarshalKafkaConfigSchema = z
     value_schema: z.lazy(() => unmarshalSchemaConfigSchema).optional(),
     extra_options: z.record(z.string(), z.string()).optional(),
     backfill_source: z.lazy(() => unmarshalBackfillSourceSchema).optional(),
+    ingestion_config: z.lazy(() => unmarshalIngestionConfigSchema).optional(),
 })
     .transform(d => ({
     name: d.name,
@@ -483,6 +484,7 @@ export const unmarshalKafkaConfigSchema = z
     valueSchema: d.value_schema,
     extraOptions: d.extra_options,
     backfillSource: d.backfill_source,
+    ingestionConfig: d.ingestion_config,
 }));
 export const unmarshalKafkaSourceSchema = z
     .object({
@@ -600,9 +602,7 @@ export const unmarshalMaterializedFeatureSchema = z
         .lazy(() => unmarshalOnlineStoreConfigSchema)
         .optional(),
     table_name: z.string().optional(),
-    pipeline_schedule_state: z
-        .enum(MaterializedFeature_PipelineScheduleState)
-        .optional(),
+    pipeline_schedule_state: z.string().optional(),
     last_materialization_time: z
         .string()
         .transform(s => Temporal.Instant.from(s))
@@ -844,6 +844,13 @@ export const unmarshalStreamSchemaConfigSchema = z
         ? { $case: 'directSchemas', directSchemas: d.direct_schemas }
         : undefined,
 }));
+export const unmarshalStreamSourceSchema = z
+    .object({
+    full_name: z.string().optional(),
+})
+    .transform(d => ({
+    fullName: d.full_name,
+}));
 export const unmarshalStreamSourceConfigSchema = z
     .object({
     kafka_stream_config: z
@@ -860,7 +867,7 @@ export const unmarshalStreamSourceConfigSchema = z
 }));
 export const unmarshalStreamingModeSchema = z
     .object({
-    mode: z.enum(StreamingMode_StreamingModeType).optional(),
+    mode: z.string().optional(),
 })
     .transform(d => ({
     mode: d.mode,
@@ -1176,6 +1183,10 @@ export const marshalDataSourceSchema = z
             $case: z.literal('requestSource'),
             requestSource: z.lazy(() => marshalRequestSourceSchema),
         }),
+        z.object({
+            $case: z.literal('streamSource'),
+            streamSource: z.lazy(() => marshalStreamSourceSchema),
+        }),
     ])
         .optional(),
 })
@@ -1189,6 +1200,9 @@ export const marshalDataSourceSchema = z
     ...(d.dataSource?.$case === 'requestSource' && {
         request_source: d.dataSource.requestSource,
     }),
+    ...(d.dataSource?.$case === 'streamSource' && {
+        stream_source: d.dataSource.streamSource,
+    }),
 }));
 export const marshalDeltaTableSourceSchema = z
     .object({
@@ -1273,7 +1287,7 @@ export const marshalFeatureSchema = z
 export const marshalFieldDefinitionSchema = z
     .object({
     name: z.string().optional(),
-    dataType: z.enum(ScalarDataType).optional(),
+    dataType: z.string().optional(),
 })
     .transform(d => ({
     name: d.name,
@@ -1295,7 +1309,7 @@ export const marshalFlatSchemaSchema = z
 }));
 export const marshalFunctionSchema = z
     .object({
-    functionType: z.enum(Function_FunctionType).optional(),
+    functionType: z.string().optional(),
     extraParameters: z
         .array(z.lazy(() => marshalFunction_ExtraParameterSchema))
         .optional(),
@@ -1386,6 +1400,7 @@ export const marshalKafkaConfigSchema = z
     valueSchema: z.lazy(() => marshalSchemaConfigSchema).optional(),
     extraOptions: z.record(z.string(), z.string()).optional(),
     backfillSource: z.lazy(() => marshalBackfillSourceSchema).optional(),
+    ingestionConfig: z.lazy(() => marshalIngestionConfigSchema).optional(),
 })
     .transform(d => ({
     name: d.name,
@@ -1396,6 +1411,7 @@ export const marshalKafkaConfigSchema = z
     value_schema: d.valueSchema,
     extra_options: d.extraOptions,
     backfill_source: d.backfillSource,
+    ingestion_config: d.ingestionConfig,
 }));
 export const marshalKafkaSourceSchema = z
     .object({
@@ -1482,9 +1498,7 @@ export const marshalMaterializedFeatureSchema = z
     ])
         .optional(),
     tableName: z.string().optional(),
-    pipelineScheduleState: z
-        .enum(MaterializedFeature_PipelineScheduleState)
-        .optional(),
+    pipelineScheduleState: z.string().optional(),
     lastMaterializationTime: z
         .any()
         .transform((d) => d.toString())
@@ -1743,6 +1757,13 @@ export const marshalStreamSchemaConfigSchema = z
         direct_schemas: d.schemaConfig.directSchemas,
     }),
 }));
+export const marshalStreamSourceSchema = z
+    .object({
+    fullName: z.string().optional(),
+})
+    .transform(d => ({
+    full_name: d.fullName,
+}));
 export const marshalStreamSourceConfigSchema = z
     .object({
     sourceConfig: z
@@ -1761,7 +1782,7 @@ export const marshalStreamSourceConfigSchema = z
 }));
 export const marshalStreamingModeSchema = z
     .object({
-    mode: z.enum(StreamingMode_StreamingModeType).optional(),
+    mode: z.string().optional(),
 })
     .transform(d => ({
     mode: d.mode,
@@ -1945,6 +1966,10 @@ const dataSourceFieldMaskSchema = {
         wire: 'request_source',
         children: () => requestSourceFieldMaskSchema,
     },
+    streamSource: {
+        wire: 'stream_source',
+        children: () => streamSourceFieldMaskSchema,
+    },
 };
 const deltaTableSourceFieldMaskSchema = {
     dataframeSchema: { wire: 'dataframe_schema' },
@@ -2038,6 +2063,10 @@ const kafkaConfigFieldMaskSchema = {
     },
     bootstrapServers: { wire: 'bootstrap_servers' },
     extraOptions: { wire: 'extra_options' },
+    ingestionConfig: {
+        wire: 'ingestion_config',
+        children: () => ingestionConfigFieldMaskSchema,
+    },
     keySchema: { wire: 'key_schema', children: () => schemaConfigFieldMaskSchema },
     name: { wire: 'name' },
     subscriptionMode: {
@@ -2211,6 +2240,9 @@ const streamSchemaConfigFieldMaskSchema = {
         children: () => directSchemasFieldMaskSchema,
     },
 };
+const streamSourceFieldMaskSchema = {
+    fullName: { wire: 'full_name' },
+};
 const streamSourceConfigFieldMaskSchema = {
     kafkaStreamConfig: {
         wire: 'kafka_stream_config',