dataiku-sdk 3.2.0 → 3.4.0

package/dist/src/cli/commands/ml-task.js

@@ -1,4 +1,4 @@
-import { requiredJsonInput, requiredStringFlag, } from "../coerce.js";
+import { num, requiredJsonInput, requiredStringFlag, } from "../coerce.js";
 import { executionMode, } from "../flags.js";
 import { requireArgs, UsageError, } from "../usage.js";
 function optionalStringFlag(flags, name) {
@@ -78,21 +78,33 @@ export const mlTaskCommands = {
         ],
     },
     train: {
-        handler: (c, a, f) => {
-            requireArgs(a, 2, "dss ml-task train <analysisId> <mlTaskId> [--session-name NAME] [--wait] [--project-key KEY]");
+        handler: async (c, a, f) => {
+            const usage = "dss ml-task train <analysisId> <mlTaskId> [--session-name NAME] [--wait] [--timeout MS] [--poll-interval MS] [--dry-run] [--project-key KEY]";
+            requireArgs(a, 2, usage);
             const options = {
                 analysisId: a[0],
                 mlTaskId: a[1],
                 sessionName: optionalStringFlag(f, "session-name"),
                 wait: f["wait"] === true,
+                timeoutMs: num(f["timeout"], "--timeout"),
+                pollIntervalMs: num(f["poll-interval"], "--poll-interval"),
                 projectKey: f["project-key"],
             };
+            if (executionMode(f).dryRun) {
+                return {
+                    dryRun: true,
+                    action: "train",
+                    resource: "ml-task",
+                    ...options,
+                };
+            }
             return c.mlTasks.train(options);
         },
-        usage: "dss ml-task train <analysisId> <mlTaskId> [--session-name NAME] [--wait] [--project-key KEY]",
-        description: "Start ML task training, optionally waiting for trained model IDs.",
+        usage: "dss ml-task train <analysisId> <mlTaskId> [--session-name NAME] [--wait] [--timeout MS] [--poll-interval MS] [--dry-run] [--project-key KEY]",
+        description: "Start ML task training, optionally waiting for trained model IDs. --timeout bounds only the wait phase after DSS accepts training; an invalid timeout fails before any request.",
         examples: [
-            "dss ml-task train ANALYSIS_ID TASK_ID --session-name baseline --wait --project-key PROJECT",
+            "dss ml-task train ANALYSIS_ID TASK_ID --session-name baseline --wait --timeout 60000 --project-key PROJECT",
+            "dss ml-task train ANALYSIS_ID TASK_ID --wait --timeout 0 --poll-interval 1000 --project-key PROJECT",
         ],
     },
     "list-models": {

package/dist/src/cli/commands/notebook.js

@@ -233,14 +233,18 @@ export const notebookCommands = {
                 expectHash: f["expect-hash"],
             });
             return {
-                saved: a[0],
+                // `saved` is the persisted notebook id: on create DSS allocates it
+                // server-side, so every later command must use this id. `requested`
+                // echoes the handle, which becomes the notebook's display name.
+                saved: result.id,
+                requested: a[0],
                 resource: "sql-notebook",
                 created: result.created,
                 hash: result.hash,
             };
         },
         usage: "dss notebook save-sql <id> (--data '{...}' | --data-file PATH | --stdin) [--expect-hash SHA256] [--dry-run] [--project-key KEY]",
-        description: "Save content to a SQL notebook, creating it if missing. Reports created versus updated and the persisted content hash; --expect-hash rejects a stale read before writing.",
+        description: "Save content to a SQL notebook, creating it if missing. On create DSS allocates the notebook id: `saved` is the persisted id to use for every later read/update/delete, and `requested` echoes the handle, which becomes the display name. Reports created versus updated and the persisted content hash; --expect-hash rejects a stale read before writing and refuses a missing notebook.",
         examples: ["dss notebook save-sql my_sql_notebook --data-file content.json --dry-run",],
     },
     "clear-sql-history": {

package/dist/src/cli/commands/plugin.d.ts

@@ -0,0 +1,2 @@
+import type { CommandMeta } from "../types.js";
+export declare const pluginCommands: Record<string, CommandMeta>;