dataiku-sdk 3.2.0 → 3.3.0

package/dist/src/resources/llms.js

@@ -0,0 +1,62 @@
+import { ClientValidationError, } from "../errors.js";
+import { BaseResource, requireNonEmpty, } from "./base.js";
+export class LlmsResource extends BaseResource {
+    /**
+     * List the LLMs available in a project, including Retrieval-Augmented
+     * Generation ones. Requires READ_CONF.
+     */
+    async list(options = {}) {
+        const params = new URLSearchParams();
+        if (options.purpose !== undefined) {
+            params.set("purpose", requireNonEmpty(options.purpose, "purpose"));
+        }
+        const query = params.size > 0 ? `?${params.toString()}` : "";
+        return this.client.get(`/public/api/projects/${this.enc(options.projectKey)}/llms/${query}`);
+    }
+    /**
+     * Perform completions on an LLM. This call is cost-bearing: it invokes an
+     * external LLM provider and may incur charges.
+     */
+    async completions(body, projectKey) {
+        const payload = validateLlmInferenceBody(body, "completions");
+        return this.client.post(`/public/api/projects/${this.enc(projectKey)}/llms/completions`, payload);
+    }
+    /**
+     * Perform embeddings on an LLM. This call is cost-bearing: it invokes an
+     * external LLM provider and may incur charges.
+     */
+    async embeddings(body, projectKey) {
+        const payload = validateLlmInferenceBody(body, "embeddings");
+        return this.client.post(`/public/api/projects/${this.enc(projectKey)}/llms/embeddings`, payload);
+    }
+}
+/**
+ * Validate a completions/embeddings request body on the client boundary:
+ * `llmId` must be a non-empty string and `queries` a non-empty array of
+ * objects. Unknown documented fields pass through untouched.
+ */
+function validateLlmInferenceBody(body, action) {
+    if (body === null || typeof body !== "object" || Array.isArray(body)) {
+        throw new ClientValidationError(`llms.${action} body must be an object with llmId and queries.`);
+    }
+    const llmId = requireNonEmpty(body.llmId, "llmId");
+    if (!Array.isArray(body.queries) || body.queries.length === 0) {
+        throw new ClientValidationError(`llms.${action} queries must be a non-empty array of query objects.`, "validation_failed", 'Pass at least one query, e.g. queries: [{ messages: [...] }] for completions or [{ text: "..." }] for embeddings.');
+    }
+    for (const [index, query,] of body.queries.entries()) {
+        if (query === null || typeof query !== "object" || Array.isArray(query)) {
+            throw new ClientValidationError(`llms.${action} queries[${index}] must be an object.`);
+        }
+    }
+    const payload = {
+        llmId,
+        queries: body.queries,
+    };
+    if (body.settings !== undefined) {
+        if (body.settings === null || typeof body.settings !== "object" || Array.isArray(body.settings)) {
+            throw new ClientValidationError("llms.completions settings must be an object.");
+        }
+        payload["settings"] = body.settings;
+    }
+    return payload;
+}

package/dist/src/resources/macros.d.ts

@@ -0,0 +1,121 @@
+import type { MacroDefinition, MacroRunStart, MacroState, MacroSummary, MacroWaitResult } from "../schemas.js";
+import { BaseResource } from "./base.js";
+export interface MacroRunOptions {
+    /**
+     * Server-side wait: when true the POST call blocks until the run finishes
+     * (docs: `POST /runnables/{runnableType}?wait=true`). Default false.
+     */
+    wait?: boolean;
+    projectKey?: string;
+    /**
+     * Macro parameter values keyed by the param names declared in the macro
+     * definition (`params` array). Sent as the POST body. Values must be
+     * JSON-serializable primitives or structures matching the declared types.
+     */
+    params?: Record<string, unknown>;
+    /**
+     * Admin-only parameter values keyed by the adminParams names declared in
+     * the macro definition. Sent as the POST body's adminParams entry.
+     */
+    adminParams?: Record<string, unknown>;
+}
+export interface MacroWaitOptions {
+    pollIntervalMs?: number;
+    timeoutMs?: number;
+    projectKey?: string;
+    /** Macro parameter values (see {@link MacroRunOptions.params}). */
+    params?: Record<string, unknown>;
+    /** Admin-only parameter values (see {@link MacroRunOptions.adminParams}). */
+    adminParams?: Record<string, unknown>;
+}
+export interface MacroResultOptions {
+    projectKey?: string;
+    /**
+     * Cap on the number of bytes buffered from the result body. Macro results
+     * can be arbitrary documents (HTML, text, JSON); the default 8 MiB guards
+     * against oversized bodies while remaining generous for typical reports.
+     */
+    maxBytes?: number;
+}
+/**
+ * Normalize a byte cap coming from user input (CLI `--max-bytes`) into the
+ * inclusive range [1, 2 GiB], rejecting non-integers before any DSS request.
+ */
+export declare function macroResultMaxBytes(maxBytes: number | undefined): number;
+declare function isRecord(value: unknown): value is Record<string, unknown>;
+/**
+ * Run result of {@link MacrosResource.run} when the caller opted into
+ * `wait: false` (the default): a run id for abort/state/result tracking.
+ */
+export type MacroRunHandle = MacroRunStart;
+/**
+ * When `wait: true`, DSS blocks server-side until the run finishes and then
+ * returns the run id plus the terminal poll state (resultType, errors, …).
+ */
+export type MacroRunWaitedResult = MacroRunStart & {
+    state?: MacroState;
+};
+/**
+ * Error payload surfaced by DSS on a failed macro run, from either
+ * `resultError` or `storedError` in the poll state.
+ */
+export interface MacroRunFailure {
+    source: "resultError" | "storedError";
+    details: unknown;
+}
+declare function macroRunFailure(state: Record<string, unknown>): MacroRunFailure | undefined;
+/**
+ * DSS Macros — the plugin "runnable" surface under
+ * `/public/api/projects/{pk}/runnables`.
+ *
+ * A macro run executes arbitrary plugin code on the DSS backend with
+ * side effects that DSS cannot predict; every method here that starts or
+ * stops a run is a write. Result bodies may be plain text or HTML rather
+ * than JSON, so {@link MacrosResource.result} returns the raw body.
+ */
+export declare class MacrosResource extends BaseResource {
+    /**
+     * List the macros (runnables) available in a project. The official client
+     * lists at `/runnables/` (trailing slash): the slashless path 404s on live
+     * DSS 15, while get/run/state/result stay slashless.
+     */
+    list(projectKey?: string): Promise<MacroSummary[]>;
+    /** Get the definition of a macro: params, result type, owning plugin. */
+    definition(runnableType: string, opts?: {
+        projectKey?: string;
+    }): Promise<MacroDefinition>;
+    /**
+     * Start a macro run. Executes arbitrary plugin code: classify as
+     * destructive. With `wait: true` the server call blocks until completion
+     * and the resolved state is attached to the returned run handle.
+     */
+    run(runnableType: string, opts?: MacroRunOptions): Promise<MacroRunHandle | MacroRunWaitedResult>;
+    /** Request abort of a running macro run (204 response). */
+    abort(runnableType: string, runId: string, projectKey?: string): Promise<void>;
+    /** Get the poll state of a macro run (running flag, progress, errors). */
+    state(runnableType: string, runId: string, opts?: {
+        projectKey?: string;
+        /**
+         * TOTAL duration budget in ms for this state GET, measured from
+         * call start (bounds fetch, retries, backoff, and body read).
+         * Polling loops pass their remaining wait budget so a stalled
+         * state endpoint cannot outlive the wait deadline.
+         */
+        timeoutMs?: number;
+    }): Promise<MacroState>;
+    /**
+     * Download the result of a macro run. Macro results are documents of the
+     * macro's declared `resultType` (HTML, text, JSON, …) and are frequently
+     * NOT JSON: the raw body is returned verbatim. JSON bodies are returned
+     * as parsed values so programmatic callers can read fields directly.
+     */
+    result(runnableType: string, runId: string, opts?: MacroResultOptions): Promise<unknown>;
+    /**
+     * Run a macro and poll its state until it finishes or times out, using a
+     * bounded client-side loop on the state endpoint. Returns
+     * `{ success: false, timedOut: true }` on timeout rather than throwing.
+     */
+    runAndWait(runnableType: string, opts?: MacroWaitOptions): Promise<MacroWaitResult>;
+    private base;
+}
+export { isRecord as isMacroStateRecord, macroRunFailure, };

package/dist/src/resources/macros.js

@@ -0,0 +1,231 @@
+import { ClientValidationError, DataikuError, } from "../errors.js";
+import { MacroDefinitionSchema, MacroRunStartSchema, MacroStateSchema, MacroSummaryArraySchema, MacroWaitResultSchema, } from "../schemas.js";
+import { computeNextPollDelayMs, isRequestDeadlineError, } from "../utils/polling.js";
+import { BaseResource, requireNonEmpty, } from "./base.js";
+const DEFAULT_POLL_INTERVAL_MS = 2_000;
+const DEFAULT_TIMEOUT_MS = 120_000;
+/** Hard floor for the result download cap; keeps a caller typo from hanging the read. */
+const MIN_RESULT_MAX_BYTES = 1;
+const MAX_RESULT_MAX_BYTES = 2 * 1024 * 1024 * 1024;
+/**
+ * Normalize a byte cap coming from user input (CLI `--max-bytes`) into the
+ * inclusive range [1, 2 GiB], rejecting non-integers before any DSS request.
+ */
+export function macroResultMaxBytes(maxBytes) {
+    const raw = maxBytes ?? 8 * 1024 * 1024;
+    if (!Number.isInteger(raw) || raw < MIN_RESULT_MAX_BYTES) {
+        throw new ClientValidationError(`maxBytes must be a positive integer >= ${String(MIN_RESULT_MAX_BYTES)} (got ${String(raw)},)`);
+    }
+    return Math.min(raw, MAX_RESULT_MAX_BYTES);
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+/**
+ * Build the run POST body from params/adminParams. DSS macro runs accept
+ * `params` keyed by the macro's declared param names and `adminParams` keyed
+ * by its declared adminParams names; both are optional.
+ */
+function macroRunBody(params, adminParams) {
+    const body = {};
+    if (params !== undefined)
+        body["params"] = params;
+    if (adminParams !== undefined)
+        body["adminParams"] = adminParams;
+    return body;
+}
+function macroRunFailure(state) {
+    if (state["resultError"] !== undefined && state["resultError"] !== null) {
+        return { source: "resultError", details: state["resultError"], };
+    }
+    if (state["storedError"] !== undefined && state["storedError"] !== null) {
+        return { source: "storedError", details: state["storedError"], };
+    }
+    return undefined;
+}
+/**
+ * DSS Macros — the plugin "runnable" surface under
+ * `/public/api/projects/{pk}/runnables`.
+ *
+ * A macro run executes arbitrary plugin code on the DSS backend with
+ * side effects that DSS cannot predict; every method here that starts or
+ * stops a run is a write. Result bodies may be plain text or HTML rather
+ * than JSON, so {@link MacrosResource.result} returns the raw body.
+ */
+export class MacrosResource extends BaseResource {
+    /**
+     * List the macros (runnables) available in a project. The official client
+     * lists at `/runnables/` (trailing slash): the slashless path 404s on live
+     * DSS 15, while get/run/state/result stay slashless.
+     */
+    async list(projectKey) {
+        const raw = await this.client.get(`/public/api/projects/${this.enc(projectKey)}/runnables/`);
+        return this.client.safeParse(MacroSummaryArraySchema, raw, "macros.list");
+    }
+    /** Get the definition of a macro: params, result type, owning plugin. */
+    async definition(runnableType, opts) {
+        requireNonEmpty(runnableType, "runnableType");
+        const raw = await this.client.get(`${this.base(runnableType, opts?.projectKey)}`);
+        return this.client.safeParse(MacroDefinitionSchema, raw, "macros.definition");
+    }
+    /**
+     * Start a macro run. Executes arbitrary plugin code: classify as
+     * destructive. With `wait: true` the server call blocks until completion
+     * and the resolved state is attached to the returned run handle.
+     */
+    async run(runnableType, opts) {
+        requireNonEmpty(runnableType, "runnableType");
+        const wait = opts?.wait === true;
+        const query = wait ? "?wait=true" : "?wait=false";
+        const raw = await this.client.post(`${this.base(runnableType, opts?.projectKey)}${query}`, macroRunBody(opts?.params, opts?.adminParams));
+        const run = this.client.safeParse(MacroRunStartSchema, raw, "macros.run");
+        if (typeof run?.runId !== "string" || run.runId.trim().length === 0) {
+            throw new DataikuError(200, "Unexpected Response", "macros.run: DSS did not return a run identifier.");
+        }
+        if (!wait)
+            return run;
+        try {
+            const state = await this.state(runnableType, run.runId, {
+                projectKey: opts?.projectKey,
+            });
+            return { ...run, state, };
+        }
+        catch {
+            // The run started; a failed post-wait state probe must not lose the runId.
+            return run;
+        }
+    }
+    /** Request abort of a running macro run (204 response). */
+    async abort(runnableType, runId, projectKey) {
+        requireNonEmpty(runnableType, "runnableType");
+        requireNonEmpty(runId, "runId");
+        await this.client.post(`${this.base(runnableType, projectKey)}/abort/${encodeURIComponent(runId)}`, {});
+    }
+    /** Get the poll state of a macro run (running flag, progress, errors). */
+    async state(runnableType, runId, opts) {
+        requireNonEmpty(runnableType, "runnableType");
+        requireNonEmpty(runId, "runId");
+        const remainingMs = opts?.timeoutMs;
+        const raw = await this.client.get(`${this.base(runnableType, opts?.projectKey)}/state/${encodeURIComponent(runId)}`, ...(remainingMs !== undefined ? [{ timeoutMs: remainingMs, },] : []));
+        return this.client.safeParse(MacroStateSchema, raw, "macros.state");
+    }
+    /**
+     * Download the result of a macro run. Macro results are documents of the
+     * macro's declared `resultType` (HTML, text, JSON, …) and are frequently
+     * NOT JSON: the raw body is returned verbatim. JSON bodies are returned
+     * as parsed values so programmatic callers can read fields directly.
+     */
+    async result(runnableType, runId, opts) {
+        requireNonEmpty(runnableType, "runnableType");
+        requireNonEmpty(runId, "runId");
+        const maxBytes = macroResultMaxBytes(opts?.maxBytes);
+        const raw = await this.client.getTextLimited(`${this.base(runnableType, opts?.projectKey)}/result/${encodeURIComponent(runId)}`, maxBytes);
+        if (raw.truncated) {
+            throw new Error(`Macro result exceeded ${String(maxBytes)} bytes; retry with a larger maxBytes.`);
+        }
+        const text = raw.text;
+        const trimmed = text.trim();
+        if (trimmed.startsWith("{") && trimmed.endsWith("}")
+            || trimmed.startsWith("[") && trimmed.endsWith("]")) {
+            try {
+                return JSON.parse(text);
+            }
+            catch {
+                // Not JSON after all — fall through and return the text.
+            }
+        }
+        return text;
+    }
+    /**
+     * Run a macro and poll its state until it finishes or times out, using a
+     * bounded client-side loop on the state endpoint. Returns
+     * `{ success: false, timedOut: true }` on timeout rather than throwing.
+     */
+    async runAndWait(runnableType, opts) {
+        requireNonEmpty(runnableType, "runnableType");
+        // Validate the wait knobs BEFORE the run POST: an invalid budget must be
+        // a caller error, not a mutation whose wait behavior is undefined.
+        if (opts?.pollIntervalMs !== undefined) {
+            if (!Number.isFinite(opts.pollIntervalMs) || opts.pollIntervalMs < 1) {
+                throw new ClientValidationError("pollIntervalMs must be a finite number >= 1.");
+            }
+        }
+        if (opts?.timeoutMs !== undefined) {
+            if (!Number.isFinite(opts.timeoutMs) || opts.timeoutMs < 0) {
+                throw new ClientValidationError("timeoutMs must be a finite number >= 0.");
+            }
+        }
+        const baseIntervalMs = Math.max(1, opts?.pollIntervalMs ?? DEFAULT_POLL_INTERVAL_MS);
+        const adaptivePolling = opts?.pollIntervalMs === undefined;
+        // One state observation always happens regardless of budget (poll
+        // precedes deadline check), but never round a budget up to a full poll.
+        const timeoutMs = Math.max(0, opts?.timeoutMs ?? DEFAULT_TIMEOUT_MS);
+        const startedAt = Date.now();
+        const run = await this.run(runnableType, {
+            wait: false,
+            projectKey: opts?.projectKey,
+            params: opts?.params,
+            adminParams: opts?.adminParams,
+        });
+        const runId = run.runId;
+        let pollCount = 0;
+        let lastState;
+        const timedOutResult = (elapsedMs, state) => this.client.safeParse(MacroWaitResultSchema, {
+            runnableType,
+            runId,
+            running: true,
+            success: false,
+            timedOut: true,
+            elapsedMs,
+            pollCount,
+            ...(state ? { state, } : {}),
+        }, "macros.runAndWait");
+        while (true) {
+            pollCount += 1;
+            try {
+                lastState = await this.state(runnableType, runId, {
+                    projectKey: opts?.projectKey,
+                    timeoutMs: Math.max(1, timeoutMs - (Date.now() - startedAt)),
+                });
+            }
+            catch (error) {
+                if (!isRequestDeadlineError(error, startedAt + timeoutMs))
+                    throw error;
+                return timedOutResult(Date.now() - startedAt, lastState);
+            }
+            const running = lastState.running === true;
+            if (running !== true) {
+                return this.client.safeParse(MacroWaitResultSchema, {
+                    runnableType,
+                    runId,
+                    running: false,
+                    success: macroRunFailure(lastState) === undefined
+                        && lastState.exists !== false,
+                    elapsedMs: Date.now() - startedAt,
+                    pollCount,
+                    ...(macroRunFailure(lastState)
+                        ? { failure: macroRunFailure(lastState), }
+                        : {}),
+                    state: lastState,
+                }, "macros.runAndWait");
+            }
+            const elapsedMs = Date.now() - startedAt;
+            if (elapsedMs >= timeoutMs) {
+                return timedOutResult(elapsedMs, lastState);
+            }
+            const nextDelayMs = computeNextPollDelayMs({
+                pollCount,
+                baseIntervalMs,
+                adaptiveEnabled: adaptivePolling,
+            });
+            const { promise, resolve, } = Promise.withResolvers();
+            setTimeout(resolve, Math.min(nextDelayMs, Math.max(1, timeoutMs - (Date.now() - startedAt))));
+            await promise;
+        }
+    }
+    base(runnableType, projectKey) {
+        return `/public/api/projects/${this.enc(projectKey)}/runnables/${encodeURIComponent(runnableType)}`;
+    }
+}
+// Re-exported for CLI/plan use: failure extraction from a poll state record.
+export { isRecord as isMacroStateRecord, macroRunFailure, };

package/dist/src/resources/ml-tasks.d.ts

@@ -39,6 +39,19 @@ export interface MlTaskTrainOptions {
     sessionDescription?: string;
     runQueue?: boolean;
     wait?: boolean;
+    /**
+     * Deadline in milliseconds for the whole `wait` phase after DSS has
+     * accepted the training request. Validated before any request is sent:
+     * an invalid deadline must not start training work on the server.
+     * `timeoutMs: 0` disables waiting (single status check). Requires `wait`.
+     */
+    timeoutMs?: number;
+    /**
+     * Fixed poll interval in milliseconds for the training status loop.
+     * When omitted the loop backs off adaptively (doubles every 3 polls, capped).
+     * Requires `wait`.
+     */
+    pollIntervalMs?: number;
     projectKey?: string;
 }
 export interface MlTaskTrainingSession extends Record<string, unknown> {
@@ -48,7 +61,17 @@ export interface MlTaskTrainCompletedResult extends Record<string, unknown> {
     sessionId: string;
     trainedModelIds: string[];
 }
-export type MlTaskTrainResult = MlTaskTrainingSession | MlTaskTrainCompletedResult;
+/** Result returned when the training wait budget was exhausted. */
+export interface MlTaskTrainTimedOutResult extends Record<string, unknown> {
+    sessionId: string;
+    trainedModelIds: string[];
+    timedOut: true;
+    success: false;
+    state: "RUNNING";
+    elapsedMs: number;
+    pollCount: number;
+}
+export type MlTaskTrainResult = MlTaskTrainingSession | MlTaskTrainCompletedResult | MlTaskTrainTimedOutResult;
 export interface MlTrainedModelDetails extends Record<string, unknown> {
     id?: string;
     algorithm?: string;
@@ -77,7 +100,19 @@ export declare class MlTasksResource extends BaseResource {
     getSettings(analysisId: string, mlTaskId: string, projectKey?: string): Promise<MlTaskSettings>;
     /** Save the full settings object of an ML task. */
     saveSettings(analysisId: string, mlTaskId: string, settings: MlTaskSettings, projectKey?: string): Promise<MlTaskActionResult | undefined>;
-    /** Start training, optionally waiting for the task to finish. */
+    /**
+     * Start training, optionally waiting for the task to finish.
+     *
+     * `timeoutMs` is validated BEFORE the training POST: an invalid deadline
+     * never reaches the server, so bad input cannot start training work. Once
+     * DSS has accepted the training request, the deadline scopes only the
+     * wait phase — each status poll is issued with the remaining budget as a
+     * total request deadline, so a hung status request cannot outlast it.
+     *
+     * When the budget is exhausted the returned result keeps the documented
+     * success shape plus `timedOut: true` and `success: false` instead of
+     * throwing, so a caller can distinguish "still training" from "finished".
+     */
     train(opts: MlTaskTrainOptions): Promise<MlTaskTrainResult>;
     /** List identifiers for every trained model currently present on the task. */
     listTrainedModels(analysisId: string, mlTaskId: string, projectKey?: string): Promise<string[]>;

package/dist/src/resources/ml-tasks.js

@@ -1,12 +1,8 @@
-import { UsageError, } from "../cli/usage.js";
-import { BaseResource, } from "./base.js";
+import { ClientValidationError, } from "../errors.js";
+import { computeNextPollDelayMs, isRequestDeadlineError, } from "../utils/polling.js";
+import { BaseResource, requireNonEmpty, } from "./base.js";
 const TRAIN_POLL_INTERVAL_MS = 5_000;
-function requireNonEmpty(value, name) {
-    if (typeof value !== "string" || value.trim().length === 0) {
-        throw new UsageError(`${name} must be a non-empty string.`, "validation_failed");
-    }
-    return value;
-}
+const TRAIN_DEFAULT_TIMEOUT_MS = 120_000;
 function delay(ms) {
     const { promise, resolve, } = Promise.withResolvers();
     setTimeout(resolve, ms);
@@ -26,11 +22,11 @@ export class MlTasksResource extends BaseResource {
     /** Create a prediction or clustering task in an existing visual analysis. */
     async create(opts) {
         if (opts.taskType !== "PREDICTION" && opts.taskType !== "CLUSTERING") {
-            throw new UsageError("taskType must be PREDICTION or CLUSTERING.", "invalid_enum");
+            throw new ClientValidationError("taskType must be PREDICTION or CLUSTERING.", "invalid_enum");
         }
         if (opts.taskType === "PREDICTION"
             && (typeof opts.targetVariable !== "string" || opts.targetVariable.trim().length === 0)) {
-            throw new UsageError("targetVariable is required for PREDICTION ML tasks.", "missing_required_arg");
+            throw new ClientValidationError("targetVariable is required for PREDICTION ML tasks.", "missing_required_arg");
         }
         const analysisId = encodeURIComponent(requireNonEmpty(opts.analysisId, "analysisId"));
         const backendType = opts.backendType ?? "PY_MEMORY";
@@ -65,17 +61,49 @@ export class MlTasksResource extends BaseResource {
     /** Save the full settings object of an ML task. */
     async saveSettings(analysisId, mlTaskId, settings, projectKey) {
         if (settings === null || typeof settings !== "object" || Array.isArray(settings)) {
-            throw new UsageError("settings must be an object.", "validation_failed");
+            throw new ClientValidationError("settings must be an object.", "validation_failed");
         }
         return this.client.post(`${this.taskPath(analysisId, mlTaskId, projectKey)}/settings`, settings);
     }
-    /** Start training, optionally waiting for the task to finish. */
+    /**
+     * Start training, optionally waiting for the task to finish.
+     *
+     * `timeoutMs` is validated BEFORE the training POST: an invalid deadline
+     * never reaches the server, so bad input cannot start training work. Once
+     * DSS has accepted the training request, the deadline scopes only the
+     * wait phase — each status poll is issued with the remaining budget as a
+     * total request deadline, so a hung status request cannot outlast it.
+     *
+     * When the budget is exhausted the returned result keeps the documented
+     * success shape plus `timedOut: true` and `success: false` instead of
+     * throwing, so a caller can distinguish "still training" from "finished".
+     */
     async train(opts) {
         if (opts.sessionName !== undefined)
             requireNonEmpty(opts.sessionName, "sessionName");
         if (opts.sessionDescription !== undefined) {
             requireNonEmpty(opts.sessionDescription, "sessionDescription");
         }
+        // Deadline validation happens before any network request: a malformed
+        // budget must not start training work on the server.
+        const explicitIntervalMs = opts.pollIntervalMs;
+        if (explicitIntervalMs !== undefined && opts.wait !== true) {
+            throw new ClientValidationError("pollIntervalMs requires wait: true.", "validation_failed");
+        }
+        if (explicitIntervalMs !== undefined
+            && (!Number.isFinite(explicitIntervalMs) || explicitIntervalMs <= 0)) {
+            throw new ClientValidationError("pollIntervalMs must be a finite positive number.", "validation_failed");
+        }
+        let timeoutMs;
+        if (opts.timeoutMs !== undefined) {
+            if (opts.wait !== true) {
+                throw new ClientValidationError("timeoutMs requires wait: true.", "validation_failed");
+            }
+            if (!Number.isFinite(opts.timeoutMs) || opts.timeoutMs < 0) {
+                throw new ClientValidationError("timeoutMs must be a finite non-negative number.", "validation_failed");
+            }
+            timeoutMs = opts.timeoutMs;
+        }
         const taskPath = this.taskPath(opts.analysisId, opts.mlTaskId, opts.projectKey);
         const session = await this.client.post(`${taskPath}/train`, {
             sessionName: opts.sessionName,
@@ -84,15 +112,67 @@ export class MlTasksResource extends BaseResource {
         });
         if (opts.wait !== true)
             return session;
-        let status = await this.status(opts.analysisId, opts.mlTaskId, opts.projectKey);
-        while (status.training !== false) {
-            await delay(TRAIN_POLL_INTERVAL_MS);
-            status = await this.status(opts.analysisId, opts.mlTaskId, opts.projectKey);
+        const deadlineMs = timeoutMs ?? TRAIN_DEFAULT_TIMEOUT_MS;
+        const baseIntervalMs = Math.max(1, explicitIntervalMs ?? TRAIN_POLL_INTERVAL_MS);
+        const adaptiveEnabled = explicitIntervalMs === undefined;
+        const startedAt = Date.now();
+        let pollCount = 0;
+        // One status observation always happens; the poll precedes the deadline
+        // check, so `timeoutMs: 0` still reports the freshly-started state once.
+        while (true) {
+            pollCount += 1;
+            // One status observation always happens, even when the budget is
+            // already spent (the poll precedes the deadline check); only the
+            // requests issued while budget remains are bounded by the remaining
+            // time, so a stalled status endpoint can never defeat the deadline.
+            const remainingMs = deadlineMs - (Date.now() - startedAt);
+            let status;
+            try {
+                status = await this.client.get(`${this.taskPath(opts.analysisId, opts.mlTaskId, opts.projectKey)}/status`, remainingMs > 0
+                    ? { timeoutMs: remainingMs, }
+                    : undefined);
+            }
+            catch (error) {
+                // The poll budget ran out: report the structured timeout instead
+                // of letting the transport deadline error escape the loop.
+                if (!isRequestDeadlineError(error, startedAt + deadlineMs))
+                    throw error;
+                return {
+                    sessionId: session.sessionId,
+                    trainedModelIds: [],
+                    timedOut: true,
+                    success: false,
+                    state: "RUNNING",
+                    elapsedMs: Date.now() - startedAt,
+                    pollCount,
+                };
+            }
+            const elapsedMs = Date.now() - startedAt;
+            if (status.training !== false) {
+                if (elapsedMs >= deadlineMs) {
+                    return {
+                        sessionId: session.sessionId,
+                        trainedModelIds: modelIdsFromStatus(status),
+                        timedOut: true,
+                        success: false,
+                        state: "RUNNING",
+                        elapsedMs,
+                        pollCount,
+                    };
+                }
+                const nextDelayMs = computeNextPollDelayMs({
+                    pollCount,
+                    baseIntervalMs,
+                    adaptiveEnabled,
+                });
+                await delay(Math.min(nextDelayMs, deadlineMs - elapsedMs));
+                continue;
+            }
+            return {
+                sessionId: session.sessionId,
+                trainedModelIds: modelIdsFromStatus(status),
+            };
         }
-        return {
-            sessionId: session.sessionId,
-            trainedModelIds: modelIdsFromStatus(status),
-        };
     }
     /** List identifiers for every trained model currently present on the task. */
     async listTrainedModels(analysisId, mlTaskId, projectKey) {

package/dist/src/resources/model-evaluation-stores.js

@@ -1,11 +1,4 @@
-import { UsageError, } from "../cli/usage.js";
-import { BaseResource, } from "./base.js";
-function requireNonEmpty(value, name) {
-    if (typeof value !== "string" || value.trim().length === 0) {
-        throw new UsageError(`${name} must be a non-empty string.`, "validation_failed");
-    }
-    return value;
-}
+import { BaseResource, requireNonEmpty, } from "./base.js";
 export class ModelEvaluationStoresResource extends BaseResource {
     /** List model evaluation stores in a project. */
     async list(projectKey) {