dataiku-sdk 3.1.0 → 3.2.0

package/dist/src/cli.js

@@ -1,22 +1,32 @@
 #!/usr/bin/env node
 import * as fs from "node:fs/promises";
+import { homedir, } from "node:os";
+import { resolve, } from "node:path";
 import { num, unknownJsonInput, } from "./cli/coerce.js";
-import { commands, } from "./cli/commands/index.js";
-import { AGENT_CONTRACT_USAGE, AUTH_ACTIONS, BATCH_EXAMPLE_PAYLOAD, BATCH_HINT, BATCH_PLAN_EXIT_CODES, BATCH_USAGE, buildAgentContract, buildCommandRegistry, buildMutationPlan, CLEANUP_USAGE, commandActionSummary, COMMANDS_USAGE, inferRequiresProject, isAllowedCleanupAction, supportsCleanupLedger, } from "./cli/contract.js";
-import { runDoctor, runFixtures, } from "./cli/doctor.js";
 import { dataikuEnvironmentEnabled, loadEnvFile, } from "./cli/env.js";
-import { FLAG_ALIASES, isNegativeNumberToken, parseArgs, SHORT_FLAGS, VALUE_FLAGS, } from "./cli/flags.js";
+import { executionMode, FLAG_ALIASES, isNegativeNumberToken, parseArgs, SHORT_FLAGS, VALUE_FLAGS, } from "./cli/flags.js";
 import { cleanupLedgerEntry, } from "./cli/helpers/cleanup.js";
 import { commandFailureExitCode, CommandResultFailure, flushCliWarnings, isFailedWaitResult, planResult, projectResultFields, setOutputFieldProjection, writeCommandResult, } from "./cli/output.js";
 import { currentCommandContext, resolveCredentials, resolveLoginCredentials, } from "./cli/runtime.js";
 import { COMMANDS_RUN_HINT, missingActionError, noCommandError, requireArgs, unknownActionError, unknownResourceError, UsageError, } from "./cli/usage.js";
 import { cliVersionResult, } from "./cli/version.js";
-import { DataikuClient, } from "./client.js";
 import { getCredentialsPath, } from "./config.js";
 import { canonicalStatusText, ClientValidationError, DataikuError, dataikuErrorCode, } from "./errors.js";
-import { AGENTS, detectAgents, findWorkspaceRoot, installSkill, planSkillInstalls, } from "./skill.js";
 import { appendCleanupLedgerEntry, findCleanupLedgerBindingViolation, preflightCleanupLedgerPath, readCleanupLedger, reserveCleanupLedgerDssUrl, } from "./utils/cleanup-ledger.js";
 import { canonicalDssUrl, } from "./utils/dss-url.js";
+let commands;
+let contract;
+let commandRuntimeLoad;
+// Dispatch and error reporting initialize these bindings before synchronous validation.
+function loadCommandRuntime() {
+    return commandRuntimeLoad ??= Promise.all([
+        import("./cli/commands/index.js"),
+        import("./cli/contract.js"),
+    ]).then(([commandModule, contractModule,]) => {
+        commands = commandModule.commands;
+        contract = contractModule;
+    });
+}
 // ---------------------------------------------------------------------------
 // Arg parsing
 // ---------------------------------------------------------------------------
@@ -60,7 +70,7 @@ function findUnboundLifecycleCleanup(entries) {
 async function runCleanup(flags) {
     const filePath = flags["file"];
     if (typeof filePath !== "string" || filePath.trim().length === 0) {
-        throw new UsageError(`--file is required. Usage: ${CLEANUP_USAGE}`);
+        throw new UsageError(`--file is required. Usage: ${contract.CLEANUP_USAGE}`);
     }
     let entries;
     try {
@@ -95,6 +105,7 @@ async function runCleanup(flags) {
     }
     const requestTimeoutMs = num(flags["request-timeout"], "--request-timeout");
     const retryMaxAttempts = num(flags["retries"], "--retries");
+    const { DataikuClient, } = await import("./client.js");
     const client = new DataikuClient({
         url,
         apiKey,
@@ -158,7 +169,7 @@ async function runCleanup(flags) {
                 action,
                 projectKey: parsed.flags["project-key"],
             });
-            if (!resource || !action || !isAllowedCleanupAction(resource, action)
+            if (!resource || !action || !contract.isAllowedCleanupAction(resource, action)
                 || !commands[resource]?.[action]) {
                 throw new UsageError(`Invalid cleanup argv: ${entry.cleanup.argv.join(" ")}`);
             }
@@ -231,11 +242,11 @@ async function runCleanup(flags) {
 }
 function parseBatchSteps(payload) {
     if (!Array.isArray(payload)) {
-        throw new UsageError("Batch payload must be a JSON array of command-argument arrays.", "validation_failed", BATCH_HINT, { example: BATCH_EXAMPLE_PAYLOAD, });
+        throw new UsageError("Batch payload must be a JSON array of command-argument arrays.", "validation_failed", contract.BATCH_HINT, { example: contract.BATCH_EXAMPLE_PAYLOAD, });
     }
     return payload.map((step, index) => {
         if (!Array.isArray(step) || !step.every((token) => typeof token === "string")) {
-            throw new UsageError(`Batch step ${index} must be an array of string arguments.`, "validation_failed", BATCH_HINT);
+            throw new UsageError(`Batch step ${index} must be an array of string arguments.`, "validation_failed", contract.BATCH_HINT);
         }
         return step;
     });
@@ -266,12 +277,12 @@ const META_COMMAND_RESOURCES = {
     fixtures: true,
     batch: true,
 };
-let commandRegistryCache;
-function cachedCommandRegistry() {
-    if (commandRegistryCache === undefined) {
-        commandRegistryCache = buildCommandRegistry();
+const commandRegistryCache = new Map();
+function cachedCommandRegistry(resource) {
+    if (!commandRegistryCache.has(resource)) {
+        commandRegistryCache.set(resource, contract.buildCommandRegistry(resource)[resource]);
     }
-    return commandRegistryCache;
+    return commandRegistryCache.get(resource);
 }
 function supportedCommandFlags(entry) {
     const supported = { ...ALWAYS_ALLOWED_COMMAND_FLAGS, };
@@ -309,7 +320,7 @@ function commandRegistrySelectors(flags) {
     return selectors;
 }
 function validateSupportedCommandFlags(resource, action, flags) {
-    const entry = cachedCommandRegistry()[resource]?.[action];
+    const entry = cachedCommandRegistry(resource)?.[action];
     if (!entry)
         return;
     const supported = supportedCommandFlags(entry);
@@ -318,7 +329,7 @@ function validateSupportedCommandFlags(resource, action, flags) {
             throw new UsageError(`Unknown flag --${flagName} for ${resource} ${action}`, "unknown_flag", scopedCommandDiscoveryHint(resource, action));
         }
     }
-    if (flags["dry-run"] === true && !entry.dryRun) {
+    if (executionMode(flags).dryRun && !entry.dryRun) {
         throw new UsageError(`--dry-run is not supported for ${resource} ${action}.`, "unknown_flag", entry.flags.some((flag) => flag.name === "plan")
             ? "Remove --dry-run; use --plan to preview the mutation instead."
             : `Remove --dry-run; ${resource} ${action} has no dry-run mode.`);
@@ -381,7 +392,7 @@ function validateRequiredCommandInputs(resource, action, args, flags, entry) {
     }
 }
 function validateRegistryCommandInputs(resource, action, args, flags) {
-    const entry = cachedCommandRegistry()[resource]?.[action];
+    const entry = cachedCommandRegistry(resource)?.[action];
     if (!entry)
         return;
     validateSupportedCommandFlags(resource, action, flags);
@@ -418,7 +429,7 @@ async function validateDryRunHandlerPreconditions(handler, args, flags) {
         throw error;
     }
 }
-function validateMetaCommandInputs(resource, action, flags) {
+async function validateMetaCommandInputs(resource, action, flags) {
     if (resource === "doctor") {
         if (action !== undefined && action !== "run") {
             throw unknownActionError("doctor", action, ["run",]);
@@ -426,11 +437,11 @@ function validateMetaCommandInputs(resource, action, flags) {
         return "run";
     }
     if (resource === "auth") {
-        const validActions = Object.keys(AUTH_ACTIONS);
+        const validActions = Object.keys(contract.AUTH_ACTIONS);
         if (!action) {
             throw missingActionError("auth", validActions, "dss auth login --url URL --api-key KEY");
         }
-        if (!AUTH_ACTIONS[action]) {
+        if (!contract.AUTH_ACTIONS[action]) {
             throw unknownActionError("auth", action, validActions, "auth only supports 'login'. To check credentials/connectivity, run 'dss doctor'.");
         }
         return action;
@@ -439,6 +450,7 @@ function validateMetaCommandInputs(resource, action, flags) {
         if (action !== undefined && action !== "run") {
             throw unknownActionError("install-skill", action, ["run",]);
         }
+        const { AGENTS, } = await import("./skill.js");
         const agentFilter = typeof flags["agent"] === "string" ? flags["agent"] : undefined;
         if (agentFilter && !AGENTS[agentFilter]) {
             throw new UsageError(`Unknown agent: ${agentFilter}.`, "usage_error", COMMANDS_RUN_HINT, { agent: agentFilter, validAgents: Object.keys(AGENTS), });
@@ -447,14 +459,14 @@ function validateMetaCommandInputs(resource, action, flags) {
     }
     if (resource === "agent") {
         if (!action)
-            throw missingActionError("agent", ["contract",], AGENT_CONTRACT_USAGE);
+            throw missingActionError("agent", ["contract",], contract.AGENT_CONTRACT_USAGE);
         if (action !== "contract")
             throw unknownActionError("agent", action, ["contract",]);
         return action;
     }
     if (resource === "commands") {
         if (!action)
-            throw missingActionError("commands", ["run",], COMMANDS_USAGE);
+            throw missingActionError("commands", ["run",], contract.COMMANDS_USAGE);
         if (action !== "run")
             throw unknownActionError("commands", action, ["run",]);
         validateSupportedCommandFlags("commands", "run", flags);
@@ -471,7 +483,7 @@ function validateMetaCommandInputs(resource, action, flags) {
         if (action !== undefined && action !== "run") {
             throw unknownActionError("cleanup", action, ["run",]);
         }
-        const entry = cachedCommandRegistry().cleanup?.run;
+        const entry = cachedCommandRegistry("cleanup")?.run;
         if (entry)
             validateRequiredCommandInputs("cleanup", "run", [], flags, entry);
         return "run";
@@ -486,7 +498,7 @@ function validateMetaCommandInputs(resource, action, flags) {
         if (action !== undefined && action !== "run") {
             throw unknownActionError("batch", action, ["run",]);
         }
-        const entry = cachedCommandRegistry().batch?.run;
+        const entry = cachedCommandRegistry("batch")?.run;
         if (entry)
             validateRequiredCommandInputs("batch", "run", [], flags, entry);
         return "run";
@@ -499,7 +511,7 @@ async function validateBatchStep(argv) {
     const action = positional[1];
     if (!resource)
         throw noCommandError();
-    const metaAction = validateMetaCommandInputs(resource, action, flags);
+    const metaAction = await validateMetaCommandInputs(resource, action, flags);
     if (metaAction)
         return { positional, flags, resource, action: metaAction, };
     const resourceActions = commands[resource];
@@ -518,7 +530,7 @@ async function validateBatchStep(argv) {
 function batchStepNeedsClient(argv) {
     try {
         const { positional, flags, } = parseArgs(argv);
-        if (flags["plan"] === true)
+        if (executionMode(flags).plan)
             return false;
         const resource = positional[0];
         if (!resource)
@@ -600,7 +612,7 @@ function validateBatchStepMode(argv, index) {
     catch {
         return;
     }
-    if (flags["plan"] !== true && flags["dry-run"] !== true)
+    if (!executionMode(flags).plan && !executionMode(flags).dryRun)
         return;
     const resource = positional[0];
     const action = positional[1];
@@ -614,13 +626,13 @@ function validateBatchStepMode(argv, index) {
     const meta = resourceActions[action ?? ""];
     if (!meta)
         return;
-    const entry = cachedCommandRegistry()[resource]?.[action ?? ""];
+    const entry = cachedCommandRegistry(resource)?.[action ?? ""];
     if (!entry)
         return;
-    if (flags["plan"] === true && !entry.flags.some((flag) => flag.name === "plan")) {
+    if (executionMode(flags).plan && !entry.flags.some((flag) => flag.name === "plan")) {
         throw new UsageError(`Batch step ${index} requests --plan, which is not supported for ${resource} ${action}.`, "unknown_flag", "Remove --plan from the step; only mutating commands support --plan.");
     }
-    if (flags["dry-run"] === true && !entry.dryRun) {
+    if (executionMode(flags).dryRun && !entry.dryRun) {
         throw new UsageError(`Batch step ${index} requests --dry-run, which is not supported for ${resource} ${action}.`, "unknown_flag", "Remove --dry-run from the step.");
     }
 }
@@ -629,7 +641,8 @@ function validateBatchStepModes(steps) {
         validateBatchStepMode(steps[index], index);
     }
 }
-function runInstallSkill(flags) {
+async function runInstallSkill(flags) {
+    const { AGENTS, detectAgents, findWorkspaceRoot, installSkill, planSkillInstalls, } = await import("./skill.js");
     const agentFilter = typeof flags["agent"] === "string" ? flags["agent"] : undefined;
     const isGlobal = flags["global"] === true;
     const targetDir = typeof flags["target"] === "string" ? flags["target"] : undefined;
@@ -655,25 +668,29 @@ function runInstallSkill(flags) {
         throw new UsageError("No coding agents detected.", "usage_error", "Use --agent NAME to choose one of the supported agents.", { validAgents: Object.keys(AGENTS), });
     }
     const scope = isGlobal ? "global" : "project";
-    const cwd = targetDir ?? (isGlobal ? process.cwd() : findWorkspaceRoot(process.cwd()));
+    const cwd = isGlobal
+        ? homedir()
+        : targetDir
+            ? resolve(targetDir)
+            : findWorkspaceRoot(process.cwd());
     const installed = planSkillInstalls(targets, { global: isGlobal, cwd, });
-    if (flags["plan"] === true) {
+    if (executionMode(flags).plan) {
         return planResult("install-skill", "run", {
             identifiers: { scope, target: cwd, },
             payload: { installed, },
             idempotency: "none",
             asyncKind: "none",
             exitCodesOnFailure: { usage: 1, error: 2, transient: 3, },
-            plannedAndDryRun: flags["dry-run"] === true,
+            plannedAndDryRun: executionMode(flags).dryRun,
         });
     }
     return {
         scope,
         target: cwd,
-        installed: flags["dry-run"] === true
+        installed: executionMode(flags).dryRun
             ? installed
             : installSkill(targets, { global: isGlobal, cwd, }),
-        ...(flags["dry-run"] === true ? { dryRun: true, } : {}),
+        ...(executionMode(flags).dryRun ? { dryRun: true, } : {}),
     };
 }
 const META_PLAN_EXIT_CODES = { usage: 1, error: 2, transient: 3, };
@@ -706,7 +723,7 @@ function authLoginPlan(flags) {
 function cleanupPlan(flags) {
     const filePath = flags["file"];
     if (typeof filePath !== "string" || filePath.trim().length === 0) {
-        throw new UsageError(`--file is required. Usage: ${CLEANUP_USAGE}`);
+        throw new UsageError(`--file is required. Usage: ${contract.CLEANUP_USAGE}`);
     }
     return planResult("cleanup", "run", {
         identifiers: { file: filePath, },
@@ -717,13 +734,13 @@ function cleanupPlan(flags) {
         idempotency: "none",
         asyncKind: "none",
         exitCodesOnFailure: META_PLAN_EXIT_CODES,
-        plannedAndDryRun: flags["dry-run"] === true,
+        plannedAndDryRun: executionMode(flags).dryRun,
     });
 }
 function batchPlan(flags) {
     const payload = unknownJsonInput(flags);
     if (payload === undefined) {
-        throw new UsageError(`Provide steps via --data, --data-file, or --stdin. Usage: ${BATCH_USAGE}`, "missing_required_flag", BATCH_HINT);
+        throw new UsageError(`Provide steps via --data, --data-file, or --stdin. Usage: ${contract.BATCH_USAGE}`, "missing_required_flag", contract.BATCH_HINT);
     }
     const steps = parseBatchSteps(payload);
     return planResult("batch", "run", {
@@ -734,34 +751,37 @@ function batchPlan(flags) {
         payload: {
             steps: steps.map((argv) => redactArgv(argv)),
             continueOnError: flags["continue-on-error"] === true,
-            dryRun: flags["dry-run"] === true,
+            dryRun: executionMode(flags).dryRun,
         },
         idempotency: "none",
         asyncKind: "none",
-        exitCodesOnFailure: BATCH_PLAN_EXIT_CODES,
-        plannedAndDryRun: flags["dry-run"] === true,
+        exitCodesOnFailure: contract.BATCH_PLAN_EXIT_CODES,
+        plannedAndDryRun: executionMode(flags).dryRun,
     });
 }
 async function runMetaCommand(resource, action, flags) {
+    if (resource !== "version")
+        await loadCommandRuntime();
     if (resource === "doctor") {
         if (action !== undefined && action !== "run") {
             throw unknownActionError("doctor", action, ["run",]);
         }
         currentCommandContext.action = action ?? "run";
+        const { runDoctor, } = await import("./cli/doctor.js");
         const { result, exitCode, } = await runDoctor(flags);
         return { action: "run", result, exitCode, };
     }
     if (resource === "auth") {
-        const validActions = Object.keys(AUTH_ACTIONS);
+        const validActions = Object.keys(contract.AUTH_ACTIONS);
         if (!action) {
             throw missingActionError("auth", validActions, "dss auth login --url URL --api-key KEY");
         }
         currentCommandContext.action = action;
-        const authMeta = AUTH_ACTIONS[action];
+        const authMeta = contract.AUTH_ACTIONS[action];
         if (!authMeta) {
             throw unknownActionError("auth", action, validActions, "auth only supports 'login'. To check credentials/connectivity, run 'dss doctor'.");
         }
-        if (flags["plan"] === true) {
+        if (executionMode(flags).plan) {
             return { action, result: authLoginPlan(flags), exitCode: 0, };
         }
         return { action, result: await authMeta.handler(flags), exitCode: 0, };
@@ -771,29 +791,29 @@ async function runMetaCommand(resource, action, flags) {
             throw unknownActionError("install-skill", action, ["run",]);
         }
         currentCommandContext.action = action ?? "run";
-        return { action: "run", result: runInstallSkill(flags), exitCode: 0, };
+        return { action: "run", result: await runInstallSkill(flags), exitCode: 0, };
     }
     if (resource === "agent") {
         if (!action)
-            throw missingActionError("agent", ["contract",], AGENT_CONTRACT_USAGE);
+            throw missingActionError("agent", ["contract",], contract.AGENT_CONTRACT_USAGE);
         currentCommandContext.action = action;
         if (action !== "contract")
             throw unknownActionError("agent", action, ["contract",]);
-        return { action, result: buildAgentContract(), exitCode: 0, };
+        return { action, result: contract.buildAgentContract(), exitCode: 0, };
     }
     if (resource === "commands") {
         if (!action)
-            throw missingActionError("commands", ["run",], COMMANDS_USAGE);
+            throw missingActionError("commands", ["run",], contract.COMMANDS_USAGE);
         currentCommandContext.action = action;
         if (action !== "run")
             throw unknownActionError("commands", action, ["run",]);
         validateSupportedCommandFlags("commands", "run", flags);
         const selectors = commandRegistrySelectors(flags);
-        const registry = buildCommandRegistry();
+        const registry = contract.buildCommandRegistry();
         for (const selector of selectors) {
             const selectorParts = selector.split(".");
             if (selectorParts.some((part) => part.length === 0)) {
-                throw new UsageError(`Invalid --fields selector: ${selector}. Expected RESOURCE or RESOURCE.ACTION[.FIELD...].`, "usage_error", COMMANDS_USAGE, { selector, });
+                throw new UsageError(`Invalid --fields selector: ${selector}. Expected RESOURCE or RESOURCE.ACTION[.FIELD...].`, "usage_error", contract.COMMANDS_USAGE, { selector, });
             }
             const [selectedResource, selectedAction,] = selectorParts;
             const resourceActions = registry[selectedResource];
@@ -814,7 +834,7 @@ async function runMetaCommand(resource, action, flags) {
         }
         return {
             action,
-            result: selectors.length === 0 ? commandActionSummary(registry) : selectedRegistry,
+            result: selectors.length === 0 ? contract.commandActionSummary(registry) : selectedRegistry,
             exitCode: 0,
         };
     }
@@ -830,7 +850,7 @@ async function runMetaCommand(resource, action, flags) {
             throw unknownActionError("cleanup", action, ["run",]);
         }
         currentCommandContext.action = action ?? "run";
-        if (flags["plan"] === true) {
+        if (executionMode(flags).plan) {
             return { action: "run", result: cleanupPlan(flags), exitCode: 0, };
         }
         const { result, exitCode, } = await runCleanup(flags);
@@ -841,6 +861,7 @@ async function runMetaCommand(resource, action, flags) {
             throw unknownActionError("fixtures", action, ["run",]);
         }
         currentCommandContext.action = action ?? "run";
+        const { runFixtures, } = await import("./cli/doctor.js");
         return { action: "run", result: await runFixtures(flags), exitCode: 0, };
     }
     if (resource === "batch") {
@@ -848,7 +869,7 @@ async function runMetaCommand(resource, action, flags) {
             throw unknownActionError("batch", action, ["run",]);
         }
         currentCommandContext.action = action ?? "run";
-        if (flags["plan"] === true) {
+        if (executionMode(flags).plan) {
             return { action: "run", result: batchPlan(flags), exitCode: 0, };
         }
         const { result, exitCode, } = await runBatch(flags);
@@ -862,9 +883,9 @@ async function runMetaCommand(resource, action, flags) {
  * share this guard.
  */
 function assertCleanupLedgerSupported(resource, action, flags) {
-    if (typeof flags["record-cleanup"] !== "string" || flags["dry-run"] === true)
+    if (typeof flags["record-cleanup"] !== "string" || executionMode(flags).dryRun)
         return;
-    if (!supportsCleanupLedger(resource, action)) {
+    if (!contract.supportsCleanupLedger(resource, action)) {
         throw new UsageError(`--record-cleanup is not supported for ${resource} ${action}.`);
     }
     if (resource === "workspace" && action === "create" && flags["stdin"] === true) {
@@ -878,7 +899,7 @@ function assertCleanupLedgerSupported(resource, action, flags) {
  * steps both call it, and dry-run/preview runs never touch the ledger.
  */
 async function preflightCleanupLedgerForFlags(client, flags) {
-    if (flags["dry-run"] === true)
+    if (executionMode(flags).dryRun)
         return;
     const ledgerPath = flags["record-cleanup"];
     if (typeof ledgerPath !== "string")
@@ -929,7 +950,7 @@ async function preflightCleanupLedgerForFlags(client, flags) {
  */
 async function recordCleanupLedgerEntry(client, resource, action, args, flags, result, projectKey) {
     const ledgerPath = flags["record-cleanup"];
-    if (typeof ledgerPath !== "string" || flags["dry-run"] === true)
+    if (typeof ledgerPath !== "string" || executionMode(flags).dryRun)
         return;
     const entry = cleanupLedgerEntry(resource, action, args, flags, result, projectKey);
     if (entry) {
@@ -946,10 +967,10 @@ async function recordCleanupLedgerEntry(client, resource, action, args, flags, r
 async function runBatch(flags) {
     const payload = unknownJsonInput(flags);
     if (payload === undefined) {
-        throw new UsageError(`Provide steps via --data, --data-file, or --stdin. Usage: ${BATCH_USAGE}`, "missing_required_flag", BATCH_HINT);
+        throw new UsageError(`Provide steps via --data, --data-file, or --stdin. Usage: ${contract.BATCH_USAGE}`, "missing_required_flag", contract.BATCH_HINT);
     }
     const steps = parseBatchSteps(payload);
-    if (flags["dry-run"] === true) {
+    if (executionMode(flags).dryRun) {
         const batchProjectKey = typeof flags["project-key"] === "string"
             ? flags["project-key"]
             : dataikuEnvironmentEnabled()
@@ -1016,6 +1037,7 @@ async function runBatch(flags) {
                 env: ["DATAIKU_API_KEY",],
             });
         }
+        const { DataikuClient, } = await import("./client.js");
         client = new DataikuClient({
             url,
             apiKey,
@@ -1078,8 +1100,8 @@ async function runBatch(flags) {
                 if (!meta)
                     throw unknownActionError(resource, action, Object.keys(resourceActions));
                 validateSupportedCommandFlags(resource, action, stepFlags);
-                if (stepFlags["plan"] === true) {
-                    result = buildMutationPlan(resource, action, meta, positional.slice(2), stepFlags);
+                if (executionMode(stepFlags).plan) {
+                    result = contract.buildMutationPlan(resource, action, meta, positional.slice(2), stepFlags);
                 }
                 else {
                     if (!client) {
@@ -1176,7 +1198,7 @@ function commandIsProjectScoped(resource, action) {
     if (!resource)
         return false;
     const usage = commands[resource]?.[action ?? ""]?.usage ?? "";
-    return inferRequiresProject(resource, action ?? "", usage);
+    return contract.inferRequiresProject(resource, action ?? "", usage);
 }
 function rawCommandContext() {
     const argv = process.argv.slice(2);
@@ -1262,7 +1284,7 @@ function isAmbiguousMutationFailure(err, context) {
     if (err.status !== 0 && err.status < 500)
         return false;
     const entry = context.resource && context.action
-        ? cachedCommandRegistry()[context.resource]?.[context.action]
+        ? cachedCommandRegistry(context.resource)?.[context.action]
         : undefined;
     return entry?.sideEffect === "write" && entry.idempotency === "none";
 }
@@ -1480,8 +1502,8 @@ async function main() {
         throw unknownActionError(resource, action, Object.keys(resourceActions));
     validateSupportedCommandFlags(resource, action, flags);
     const args = positional.slice(2);
-    if (flags["plan"] === true) {
-        const plan = buildMutationPlan(resource, action, actionMeta, args, flags);
+    if (executionMode(flags).plan) {
+        const plan = contract.buildMutationPlan(resource, action, actionMeta, args, flags);
         writeCommandResult(plan);
         return;
     }
@@ -1504,6 +1526,7 @@ async function main() {
     }
     const requestTimeoutMs = num(flags["request-timeout"], "--request-timeout");
     const retryMaxAttempts = num(flags["retries"], "--retries");
+    const { DataikuClient, } = await import("./client.js");
     const client = new DataikuClient({
         url,
         apiKey,
@@ -1523,7 +1546,8 @@ async function main() {
         throw new CommandResultFailure(result, failureExitCode);
     writeCommandResult(result);
 }
-main().catch((err) => {
+main().catch(async (err) => {
+    await loadCommandRuntime();
     writeErrorReport(err);
     process.exitCode = errorExitCode(err);
 });

package/dist/src/client.d.ts

@@ -67,7 +67,7 @@ export interface DataikuClientConfig {
     apiKey: string;
     /** Default project key — used when a resource method omits projectKey */
     projectKey?: string;
-    /** Per-request timeout in milliseconds (default 30 000) */
+    /** Request/header timeout and buffered-body budget; idle timeout per raw stream read (default 30 000 ms). */
     requestTimeoutMs?: number;
     /** Max retry attempts for idempotent requests (default 4, capped at 10) */
     retryMaxAttempts?: number;
@@ -247,6 +247,8 @@ export declare class DataikuClient {
     upload(path: string, filePath: string, fileName?: string): Promise<void>;
     uploadJson<T>(path: string, filePath: string, fileName?: string): Promise<T>;
     stream(path: string): Promise<Response>;
+    /** Bound each pending read, not total transfer time; preserve large downloads and backpressure. */
+    private withBodyDeadline;
     private getHeaders;
     private getAnyHeaders;
     private responseMeta;