@databricks/sdk-jobs 0.6.0 → 0.7.0

package/dist/v2/model.js

@@ -392,6 +392,20 @@ export const CleanRoomTaskRunResultState_CleanRoomTaskRunResultState = {
     /** 12 is reserved for previously used SUCCESS_WITH_SKIPPED_CELLS */
     DISABLED: 'DISABLED',
 };
+/**
+ * Customer-facing AcceleratorType: hardware accelerator type for the
+ * AiRuntime workload. Per-node accelerator count is encoded in the value
+ * name (e.g. `GPU_8xH100` means 8 H100s per node).
+ */
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Enum-style const object.
+export const ComputeSpec_AcceleratorType = {
+    /** Single A10 GPU per node. Good for development and small workloads. */
+    GPU_1X_A10: 'GPU_1xA10',
+    /** Single H100 GPU per node. */
+    GPU_1X_H100: 'GPU_1xH100',
+    /** Eight H100 GPUs per node. Typical for distributed training. */
+    GPU_8X_H100: 'GPU_8xH100',
+};
 /**
  * * `EQUAL_TO`, `NOT_EQUAL` operators perform string comparison of their operands. This means that `“12.0” == “12”` will evaluate to `false`.
  * * `GREATER_THAN`, `GREATER_THAN_OR_EQUAL`, `LESS_THAN`, `LESS_THAN_OR_EQUAL` operators perform numeric comparison of their operands. `“12.0” >= “12”` will evaluate to `true`, `“10.0” >= “12”` will evaluate to `false`.
@@ -679,6 +693,32 @@ export const unmarshalAdlsgen2InfoSchema = z
     .transform(d => ({
     destination: d.destination,
 }));
+export const unmarshalAiRuntimeTaskSchema = z
+    .object({
+    experiment: z.string().optional(),
+    deployments: z
+        .array(z.lazy(() => unmarshalDeploymentSpecSchema))
+        .optional(),
+    code_source_path: z.string().optional(),
+    mlflow_run: z.string().optional(),
+    mlflow_experiment_directory: z.string().optional(),
+})
+    .transform(d => ({
+    experiment: d.experiment,
+    deployments: d.deployments,
+    codeSourcePath: d.code_source_path,
+    mlflowRun: d.mlflow_run,
+    mlflowExperimentDirectory: d.mlflow_experiment_directory,
+}));
+export const unmarshalAiRuntimeTaskOutputSchema = z
+    .object({
+    mlflow_experiment_id: z.string().optional(),
+    mlflow_run_id: z.string().optional(),
+})
+    .transform(d => ({
+    mlflowExperimentId: d.mlflow_experiment_id,
+    mlflowRunId: d.mlflow_run_id,
+}));
 export const unmarshalAlertTaskSchema = z
     .object({
     alert_id: z.string().optional(),
@@ -1096,6 +1136,15 @@ export const unmarshalComputeConfigSchema = z
     gpuNodePoolId: d.gpu_node_pool_id,
     gpuType: d.gpu_type,
 }));
+export const unmarshalComputeSpecSchema = z
+    .object({
+    accelerator_type: z.string().optional(),
+    accelerator_count: z.number().optional(),
+})
+    .transform(d => ({
+    acceleratorType: d.accelerator_type,
+    acceleratorCount: d.accelerator_count,
+}));
 export const unmarshalConditionTaskSchema = z
     .object({
     op: z.string().optional(),
@@ -1292,6 +1341,17 @@ export const unmarshalDbtTask_DbtTaskOutputSchema = z
 }));
 export const unmarshalDeleteJobResponseSchema = z.object({});
 export const unmarshalDeleteRunResponseSchema = z.object({});
+export const unmarshalDeploymentSpecSchema = z
+    .object({
+    command_path: z.string().optional(),
+    compute: z.lazy(() => unmarshalComputeSpecSchema).optional(),
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    commandPath: d.command_path,
+    compute: d.compute,
+    name: d.name,
+}));
 export const unmarshalDockerBasicAuthSchema = z
     .object({
     username: z.string().optional(),
@@ -1503,6 +1563,9 @@ export const unmarshalGetRunOutputResponseSchema = z
         .lazy(() => unmarshalDbtPlatformTaskOutputSchema)
         .optional(),
     alert_output: z.lazy(() => unmarshalAlertTaskOutputSchema).optional(),
+    ai_runtime_task_output: z
+        .lazy(() => unmarshalAiRuntimeTaskOutputSchema)
+        .optional(),
     logs: z.string().optional(),
     logs_truncated: z.boolean().optional(),
     error_trace: z.string().optional(),
@@ -1550,7 +1613,12 @@ export const unmarshalGetRunOutputResponseSchema = z
                                             $case: 'alertOutput',
                                             alertOutput: d.alert_output,
                                         }
-                                        : undefined,
+                                        : d.ai_runtime_task_output !== undefined
+                                            ? {
+                                                $case: 'aiRuntimeTaskOutput',
+                                                aiRuntimeTaskOutput: d.ai_runtime_task_output,
+                                            }
+                                            : undefined,
     logs: d.logs,
     logsTruncated: d.logs_truncated,
     errorTrace: d.error_trace,
@@ -2327,6 +2395,9 @@ export const unmarshalResolvedValuesSchema = z
     pipeline_task: z
         .lazy(() => unmarshalResolvedValues_PipelineTaskResolvedValuesSchema)
         .optional(),
+    ai_runtime_task: z
+        .lazy(() => unmarshalResolvedValues_AiRuntimeTaskResolvedValuesSchema)
+        .optional(),
 })
     .transform(d => ({
     resolved: d.notebook_task !== undefined
@@ -2372,9 +2443,16 @@ export const unmarshalResolvedValuesSchema = z
                                                     $case: 'pipelineTask',
                                                     pipelineTask: d.pipeline_task,
                                                 }
-                                                : undefined,
+                                                : d.ai_runtime_task !== undefined
+                                                    ? {
+                                                        $case: 'aiRuntimeTask',
+                                                        aiRuntimeTask: d.ai_runtime_task,
+                                                    }
+                                                    : undefined,
 }));
 // eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
+export const unmarshalResolvedValues_AiRuntimeTaskResolvedValuesSchema = z.object({});
+// eslint-disable-next-line @typescript-eslint/naming-convention -- Proto-style nested message name.
 export const unmarshalResolvedValues_ConditionTaskResolvedValuesSchema = z
     .object({
     left: z.string().optional(),
@@ -2741,6 +2819,7 @@ export const unmarshalRunTaskSchema = z
     python_operator_task: z
         .lazy(() => unmarshalPythonOperatorTaskSchema)
         .optional(),
+    ai_runtime_task: z.lazy(() => unmarshalAiRuntimeTaskSchema).optional(),
     existing_cluster_id: z.string().optional(),
     new_cluster: z.lazy(() => unmarshalClusterSpec_NewClusterSchema).optional(),
     job_cluster_key: z.string().optional(),
@@ -2885,7 +2964,12 @@ export const unmarshalRunTaskSchema = z
                                                                                     $case: 'pythonOperatorTask',
                                                                                     pythonOperatorTask: d.python_operator_task,
                                                                                 }
-                                                                                : undefined,
+                                                                                : d.ai_runtime_task !== undefined
+                                                                                    ? {
+                                                                                        $case: 'aiRuntimeTask',
+                                                                                        aiRuntimeTask: d.ai_runtime_task,
+                                                                                    }
+                                                                                    : undefined,
     spec: d.existing_cluster_id !== undefined
         ? {
             $case: 'existingClusterId',
@@ -3297,6 +3381,7 @@ export const unmarshalTaskSettingsSchema = z
     python_operator_task: z
         .lazy(() => unmarshalPythonOperatorTaskSchema)
         .optional(),
+    ai_runtime_task: z.lazy(() => unmarshalAiRuntimeTaskSchema).optional(),
     existing_cluster_id: z.string().optional(),
     new_cluster: z.lazy(() => unmarshalClusterSpec_NewClusterSchema).optional(),
     job_cluster_key: z.string().optional(),
@@ -3404,7 +3489,12 @@ export const unmarshalTaskSettingsSchema = z
                                                                                     $case: 'pythonOperatorTask',
                                                                                     pythonOperatorTask: d.python_operator_task,
                                                                                 }
-                                                                                : undefined,
+                                                                                : d.ai_runtime_task !== undefined
+                                                                                    ? {
+                                                                                        $case: 'aiRuntimeTask',
+                                                                                        aiRuntimeTask: d.ai_runtime_task,
+                                                                                    }
+                                                                                    : undefined,
     spec: d.existing_cluster_id !== undefined
         ? {
             $case: 'existingClusterId',
@@ -3584,6 +3674,21 @@ export const marshalAdlsgen2InfoSchema = z
     .transform(d => ({
     destination: d.destination,
 }));
+export const marshalAiRuntimeTaskSchema = z
+    .object({
+    experiment: z.string().optional(),
+    deployments: z.array(z.lazy(() => marshalDeploymentSpecSchema)).optional(),
+    codeSourcePath: z.string().optional(),
+    mlflowRun: z.string().optional(),
+    mlflowExperimentDirectory: z.string().optional(),
+})
+    .transform(d => ({
+    experiment: d.experiment,
+    deployments: d.deployments,
+    code_source_path: d.codeSourcePath,
+    mlflow_run: d.mlflowRun,
+    mlflow_experiment_directory: d.mlflowExperimentDirectory,
+}));
 export const marshalAlertTaskSchema = z
     .object({
     alertId: z.string().optional(),
@@ -3824,6 +3929,15 @@ export const marshalComputeConfigSchema = z
     gpu_node_pool_id: d.gpuNodePoolId,
     gpu_type: d.gpuType,
 }));
+export const marshalComputeSpecSchema = z
+    .object({
+    acceleratorType: z.string().optional(),
+    acceleratorCount: z.number().optional(),
+})
+    .transform(d => ({
+    accelerator_type: d.acceleratorType,
+    accelerator_count: d.acceleratorCount,
+}));
 export const marshalConditionTaskSchema = z
     .object({
     op: z.string().optional(),
@@ -4003,6 +4117,17 @@ export const marshalDeleteRunRequestSchema = z
     .transform(d => ({
     run_id: d.runId,
 }));
+export const marshalDeploymentSpecSchema = z
+    .object({
+    commandPath: z.string().optional(),
+    compute: z.lazy(() => marshalComputeSpecSchema).optional(),
+    name: z.string().optional(),
+})
+    .transform(d => ({
+    command_path: d.commandPath,
+    compute: d.compute,
+    name: d.name,
+}));
 export const marshalDockerBasicAuthSchema = z
     .object({
     username: z.string().optional(),
@@ -4852,6 +4977,10 @@ export const marshalRunTaskSettingsSchema = z
             $case: z.literal('pythonOperatorTask'),
             pythonOperatorTask: z.lazy(() => marshalPythonOperatorTaskSchema),
         }),
+        z.object({
+            $case: z.literal('aiRuntimeTask'),
+            aiRuntimeTask: z.lazy(() => marshalAiRuntimeTaskSchema),
+        }),
     ])
         .optional(),
     spec: z
@@ -4936,6 +5065,9 @@ export const marshalRunTaskSettingsSchema = z
     ...(d.task?.$case === 'pythonOperatorTask' && {
         python_operator_task: d.task.pythonOperatorTask,
     }),
+    ...(d.task?.$case === 'aiRuntimeTask' && {
+        ai_runtime_task: d.task.aiRuntimeTask,
+    }),
     ...(d.spec?.$case === 'existingClusterId' && {
         existing_cluster_id: d.spec.existingClusterId,
     }),
@@ -5313,6 +5445,10 @@ export const marshalTaskSettingsSchema = z
             $case: z.literal('pythonOperatorTask'),
             pythonOperatorTask: z.lazy(() => marshalPythonOperatorTaskSchema),
         }),
+        z.object({
+            $case: z.literal('aiRuntimeTask'),
+            aiRuntimeTask: z.lazy(() => marshalAiRuntimeTaskSchema),
+        }),
     ])
         .optional(),
     spec: z
@@ -5397,6 +5533,9 @@ export const marshalTaskSettingsSchema = z
     ...(d.task?.$case === 'pythonOperatorTask' && {
         python_operator_task: d.task.pythonOperatorTask,
     }),
+    ...(d.task?.$case === 'aiRuntimeTask' && {
+        ai_runtime_task: d.task.aiRuntimeTask,
+    }),
     ...(d.spec?.$case === 'existingClusterId' && {
         existing_cluster_id: d.spec.existingClusterId,
     }),