dataiku-sdk 0.12.5 → 0.12.6

package/README.md

@@ -29,6 +29,7 @@ Bun is the primary development runtime and package manager. Examples below assum
 - `--fields a,b,c` projects those fields from an object or array-of-objects result; dotted paths (`a.b.c`) drill into nested objects, and missing fields become `null`; string and scalar results pass through unchanged.
 - No prompts, help screens, tables, banners, or prose output are part of the contract.
 - Exit codes: `0` success, `1` usage/configuration error, `2` DSS/internal error, `3` transient DSS error, `4` a failed long-running result or synchronous assertion.
+- `--retries N` controls idempotent GET retries only. `dss sql query --start-retries N` explicitly retries transient failures while starting a query; use it only when executing the SQL more than once is safe.
 - The exit code is the success signal. For portable multi-step mutations, prefer `dss batch`; shell chaining and pipeline exit semantics differ across POSIX shells, Windows PowerShell, PowerShell 7, and Command Prompt.
 - `--raw` is only for recipe payload commands. Without `--output`, stdout is raw bytes; with `--output PATH`, stdout is the JSON string equal to `PATH` and the file contains the exact raw bytes.
 

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

@@ -3,7 +3,7 @@ import { mkdir, writeFile, } from "node:fs/promises";
 import { dirname, resolve, } from "node:path";
 import { readStdinText, stripUtf8Bom, } from "../coerce.js";
 import { UsageError, } from "../usage.js";
-const SQL_QUERY_USAGE = "dss sql query (SQL | --sql QUERY | --sql-file PATH | --sql - | --stdin) (--connection CONN | --dataset FULL_NAME) [--database DB] [--output PATH|--output-file PATH] [--preview N] [--request-timeout MS] [--project-key KEY]";
+const SQL_QUERY_USAGE = "dss sql query (SQL | --sql QUERY | --sql-file PATH | --sql - | --stdin) (--connection CONN | --dataset FULL_NAME) [--database DB] [--output PATH|--output-file PATH] [--preview N] [--start-retries N] [--request-timeout MS] [--project-key KEY]";
 const DEFAULT_SQL_PREVIEW_ROWS = 5;
 /**
  * Parse `--preview N` into a non-negative row count. Rejects non-integers,
@@ -21,6 +21,19 @@ function parseSqlPreviewCount(value) {
     }
     return parsed;
 }
+function parseSqlStartRetries(value) {
+    if (value === undefined)
+        return undefined;
+    if (typeof value !== "string") {
+        throw new UsageError(`--start-retries requires a positive integer. Usage: ${SQL_QUERY_USAGE}`, "validation_failed");
+    }
+    const trimmed = value.trim();
+    const parsed = Number(trimmed);
+    if (trimmed.length === 0 || !Number.isInteger(parsed) || parsed < 1) {
+        throw new UsageError(`--start-retries must be a positive integer (got "${value}"). Usage: ${SQL_QUERY_USAGE}`, "validation_failed");
+    }
+    return parsed;
+}
 function resolveSqlInput(args, flags) {
     const sources = [];
     if (typeof flags["sql"] === "string") {
@@ -80,6 +93,7 @@ export const sqlCommands = {
                 datasetFullName,
                 database: f["database"],
                 projectKey: f["project-key"],
+                retryMaxAttempts: parseSqlStartRetries(f["start-retries"]),
             });
             if (!outputFile)
                 return result;
@@ -97,13 +111,14 @@ export const sqlCommands = {
             };
         },
         usage: SQL_QUERY_USAGE,
-        description: "Run a SQL query against a DSS connection or dataset.",
+        description: "Run a SQL query against a DSS connection or dataset. --start-retries may execute the SQL more than once; use it only when repetition is safe.",
         examples: [
             "dss sql query 'SELECT * FROM orders LIMIT 10' --connection my_pg",
             "dss sql query --sql-file query.sql --connection my_pg",
             "echo 'SELECT 1' | dss sql query --stdin --dataset MYPROJ.orders",
             "dss sql query --sql-file query.sql --connection my_pg --output results.json --request-timeout 120000",
             "dss sql query --sql-file query.sql --connection my_pg --output results.json --preview 10",
+            "dss sql query 'SELECT 1' --connection my_pg --start-retries 4",
         ],
     },
 };

package/dist/src/cli/flags.js

@@ -160,6 +160,7 @@ export const VALUE_FLAGS = new Set([
     "sql",
     "schema",
     "sql-file",
+    "start-retries",
     "standard",
     "state",
     "streaming-endpoint",

package/dist/src/client.d.ts

@@ -165,7 +165,13 @@ export declare class DataikuClient {
         text: string;
         truncated: boolean;
     }>;
-    post<T = unknown>(path: string, body?: unknown): Promise<T>;
+    /**
+     * POST with optional transient retry. Only opt in when repeating the request
+     * is safe: the server may have accepted an attempt whose response was lost.
+     */
+    post<T = unknown>(path: string, body?: unknown, options?: {
+        retryMaxAttempts?: number;
+    }): Promise<T>;
     postText(path: string, body?: unknown): Promise<string>;
     put<T = unknown>(path: string, body: unknown): Promise<T>;
     del(path: string): Promise<void>;

package/dist/src/client.js

@@ -406,12 +406,21 @@ export class DataikuClient {
         const usable = truncated ? trimToUtf8Boundary(collected) : collected;
         return { text: new TextDecoder().decode(usable), truncated, };
     }
-    async post(path, body) {
+    /**
+     * POST with optional transient retry. Only opt in when repeating the request
+     * is safe: the server may have accepted an attempt whose response was lost.
+     */
+    async post(path, body, options) {
+        const retryMaxAttempts = options?.retryMaxAttempts;
+        if (retryMaxAttempts !== undefined
+            && (!Number.isInteger(retryMaxAttempts) || retryMaxAttempts < 1)) {
+            throw new ClientValidationError("retryMaxAttempts must be a positive integer.");
+        }
         const res = await this.fetchWithRetry(`${this.baseUrl}${path}`, {
             method: "POST",
             headers: this.getHeaders(),
             body: body !== undefined ? JSON.stringify(body) : undefined,
-        });
+        }, retryMaxAttempts);
         return this.parseJsonResponse(res);
     }
     async postText(path, body) {
@@ -538,10 +547,14 @@ export class DataikuClient {
         }
     }
     /* ---- private: retry loop ---- */
-    async fetchWithRetry(url, init) {
+    async fetchWithRetry(url, init, retryMaxAttempts) {
         const method = (init.method ?? "GET").toUpperCase();
-        const retryEnabled = shouldRetryMethod(method);
-        const maxAttempts = retryEnabled ? this.retryMaxAttempts : 1;
+        const retryEnabled = shouldRetryMethod(method) || retryMaxAttempts !== undefined;
+        const maxAttempts = retryMaxAttempts === undefined
+            ? retryEnabled
+                ? this.retryMaxAttempts
+                : 1
+            : Math.min(retryMaxAttempts, MAX_RETRY_ATTEMPTS_CAP);
         const delaysMs = [];
         for (let attempt = 1; attempt <= maxAttempts; attempt++) {
             let timedOut = false;

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

@@ -9,6 +9,11 @@ type SqlQueryOptions = {
     postQueries?: string[];
     type?: string;
     projectKey?: string;
+    /**
+     * Opt in to transient retries while starting the query. The SQL may execute
+     * more than once when DSS accepts a request but loses its response.
+     */
+    retryMaxAttempts?: number;
 };
 export declare class SqlResource extends BaseResource {
     private resolveOptionalProjectKey;

package/dist/src/resources/sql.js

@@ -126,11 +126,12 @@ export class SqlResource extends BaseResource {
      * or `datasetFullName` (run against a dataset's connection).
      */
     async startQuery(opts) {
+        const { retryMaxAttempts, ...queryOpts } = opts;
         return this.client.post("/public/api/sql/queries/", {
-            ...opts,
-            projectKey: opts.projectKey ?? this.resolveOptionalProjectKey(opts.projectKey),
-            type: opts.type ?? "sql",
-        });
+            ...queryOpts,
+            projectKey: queryOpts.projectKey ?? this.resolveOptionalProjectKey(queryOpts.projectKey),
+            type: queryOpts.type ?? "sql",
+        }, retryMaxAttempts === undefined ? undefined : { retryMaxAttempts, });
     }
     /**
      * Stream results of a started query as parsed JSON (array of arrays).

package/dist/src/skill.js

@@ -45,7 +45,7 @@ For disposable agent tests, set \`DSS_CONFIG_DIR\` to a temporary directory so s
 - Use \`--dry-run\` only when that action's registry entry has \`dryRun:true\`. \`--plan\` explains the operation; \`--dry-run\` exercises the action-specific simulation path. Never add an unsupported flag.
 - When a create/upload action advertises \`--record-cleanup\`, pass \`--record-cleanup cleanup.jsonl\`. \`dss cleanup --file cleanup.jsonl\` previews the recorded cleanup steps in reverse order and does not mutate DSS; add \`--apply\` only after checking the preview.
 - For JSON payload actions, follow \`inputContract\`, \`requiredFlags\`, and \`requiredOneOf\`. Prefer \`--data-file PATH\` or \`--stdin\` when advertised instead of inline \`--data\`; this preserves exact JSON across shells and keeps large or sensitive payloads out of process arguments.
-- Authenticated actions advertise \`--request-timeout MS\` and \`--retries N\`. Long-running actions separately advertise controls such as \`--timeout MS\`, \`--poll-interval MS\`, and log limits; use only the flags in that action's registry entry.
+- Authenticated actions advertise \`--request-timeout MS\` and \`--retries N\`; \`--retries\` applies to idempotent GET requests. Long-running actions separately advertise controls such as \`--timeout MS\`, \`--poll-interval MS\`, and log limits; use only the flags in that action's registry entry.
 - Before live mutation tests, use \`dss fixtures --json\` to discover compatible test resources instead of guessing project objects.
 
 ## Authentication
@@ -179,6 +179,7 @@ Mutations print a small JSON ack to stdout and exit 0 on success (e.g. \`{"updat
 - \`dss api-service list-packages SERVICE\` first verifies the parent service through its settings. A \`not_found\` error means the service is missing (verify the service ID and project key); an empty array means the service exists but has no deployable packages. Do not reinterpret \`not_found\` as an empty package list or retry the lower-level packages route.
 - \`dss notebook clear-jupyter-outputs NAME --dry-run\` returns the full \`current\` and \`next\` notebook states. Applying it clears \`outputs\` and resets \`execution_count\` to \`0\` only on code cells, preserving markdown and raw cells unchanged.
 - \`dss sql query ... --dataset PROJECT.NAME\` first queries through the dataset. If DSS rejects that dataset connection as neither SQL nor HDFS, the CLI reads dataset metadata and retries with \`params.connection\` (using its schema or catalog as the database when available). A readable dataset with no usable connection exits with \`validation_failed\` and advises \`--connection\`; dataset metadata \`404\` and \`403\` errors propagate as \`not_found\` and \`permission_denied\`.
+- \`dss sql query ... --start-retries N\` opts the query-start POST into transient retries with exponential backoff. A lost response can cause DSS to execute the SQL more than once, so use it only for SQL whose repetition is safe; ordinary \`--retries N\` remains GET-only.
 
 ## Error envelope
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "dataiku-sdk",
-	"version": "0.12.5",
+	"version": "0.12.6",
 	"description": "Dataiku DSS SDK and CLI for programmatic access to DSS REST APIs",
 	"type": "module",
 	"workspaces": [
@@ -43,7 +43,9 @@
 		"dprint": "^0.55.2",
 		"js-tiktoken": "1.0.21",
 		"oxlint": "^1.78.0",
-		"typescript": "^7.0.2"
+		"typescript": "^7.0.2",
+		"typescript-language-server": "5.3.0",
+		"typescript-language-service": "npm:typescript@5.9.3"
 	},
 	"dependencies": {
 		"@sinclair/typebox": "^0.34.52"