@cargo-ai/cli 1.0.13 → 1.0.15

package/build/commands/orchestration/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/commands/orchestration/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAcxC,wBAAgB,6BAA6B,CAC3C,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CAiBN"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/commands/orchestration/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAexC,wBAAgB,6BAA6B,CAC3C,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CAkBN"}

package/build/commands/orchestration/index.js

@@ -4,6 +4,7 @@ import { registerDraftReleaseCommands } from "./draftRelease.js";
 import { registerLogCommands } from "./log.js";
 import { registerNodeCommands } from "./node.js";
 import { registerPlayCommands } from "./play.js";
+import { registerQueryCommands } from "./query.js";
 import { registerRecordCommands } from "./record.js";
 import { registerReleaseCommands } from "./release.js";
 import { registerRunCommands } from "./run.js";
@@ -26,4 +27,5 @@ export function registerOrchestrationCommands(parent, getApi) {
     registerRecordCommands(orchestration, getApi);
     registerTemplateCommands(orchestration, getApi);
     registerNodeCommands(orchestration, getApi);
+    registerQueryCommands(orchestration, getApi);
 }

package/build/commands/orchestration/query.d.ts

@@ -0,0 +1,4 @@
+import type { Command } from "commander";
+import type { Api } from "../../api.js";
+export declare function registerQueryCommands(parent: Command, getApi: () => Api): void;
+//# sourceMappingURL=query.d.ts.map

package/build/commands/orchestration/query.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../../../src/commands/orchestration/query.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CA2BN"}

package/build/commands/orchestration/query.js

@@ -0,0 +1,23 @@
+import { handleApiCall, outputJson } from "../runHandler.js";
+export function registerQueryCommands(parent, getApi) {
+    const query = parent
+        .command("query")
+        .description("Execute SQL queries against orchestration tables");
+    query
+        .command("execute")
+        .description("Execute a read-only SQL query")
+        .argument("<sql>", "SQL query string")
+        .addHelpText("after", `
+Available tables: logs, runs, batches, records
+
+Examples:
+  $ cargo-ai orchestration query execute "SELECT count() FROM runs"
+  $ cargo-ai orchestration query execute "SELECT status, count() FROM batches GROUP BY status"
+  $ cargo-ai orchestration query execute "SELECT * FROM logs ORDER BY execution_started_at DESC LIMIT 10"
+  $ cargo-ai orchestration query execute "WITH recent AS (SELECT * FROM runs WHERE created_at > now() - INTERVAL 1 DAY) SELECT status, count() FROM recent GROUP BY status"`)
+        .action(async (sql) => {
+        const api = getApi();
+        const result = await handleApiCall(() => api.orchestration.query.execute({ query: sql }));
+        outputJson(result);
+    });
+}

package/build/commands/orchestration/run.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/commands/orchestration/run.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAQxC,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG,IAAI,CAsiB5E"}
+{"version":3,"file":"run.d.ts","sourceRoot":"","sources":["../../../src/commands/orchestration/run.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAQxC,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG,IAAI,CA+iB5E"}

package/build/commands/orchestration/run.js

@@ -222,11 +222,14 @@ Examples:
         .option("--executions-filter <json>", "Filter by node executions (JSON object, same format as 'run list')")
         .option("--created-after <date>", "Filter runs created after this date (ISO 8601 string)")
         .option("--created-before <date>", "Filter runs created before this date (ISO 8601 string)")
+        .option("--limit <n>", "Max runs to include in the export (integer)")
+        .option("--offset <n>", "Number of runs to skip before exporting (integer)")
         .addHelpText("after", `
 Examples:
   $ cargo-ai orchestration run download-outputs --workflow-uuid 550e8400-... --output-node-slug my-output
   $ cargo-ai orchestration run download-outputs --workflow-uuid 550e8400-... --output-node-slug my-output --format json
-  $ cargo-ai orchestration run download-outputs --workflow-uuid 550e8400-... --output-node-slug my-output --statuses finished --is-finished`)
+  $ cargo-ai orchestration run download-outputs --workflow-uuid 550e8400-... --output-node-slug my-output --statuses finished --is-finished
+  $ cargo-ai orchestration run download-outputs --workflow-uuid 550e8400-... --output-node-slug my-output --limit 1000 --offset 2000`)
         .action(async (opts) => {
         const api = getApi();
         const result = await handleApiCall(() => api.orchestration.run.downloadOutputs({
@@ -251,6 +254,8 @@ Examples:
                 : undefined,
             createdAfter: opts.createdAfter,
             createdBefore: opts.createdBefore,
+            limit: opts.limit !== undefined ? parseInt(opts.limit, 10) : undefined,
+            offset: opts.offset !== undefined ? parseInt(opts.offset, 10) : undefined,
         }));
         outputJson(result);
     });

package/build/commands/runHandler.d.ts

@@ -1,7 +1,38 @@
+export declare const ExitCodes: {
+    readonly Success: 0;
+    readonly GenericError: 1;
+    readonly InvalidUsage: 2;
+    readonly NotAuthenticated: 3;
+    readonly PermissionDenied: 4;
+    readonly NotFound: 5;
+};
+export type ExitCode = (typeof ExitCodes)[keyof typeof ExitCodes];
+export declare const colors: {
+    red: (s: string) => string;
+    green: (s: string) => string;
+    yellow: (s: string) => string;
+    cyan: (s: string) => string;
+    dim: (s: string) => string;
+    bold: (s: string) => string;
+};
 export declare function outputJson(value: unknown): void;
+export declare function info(message: string): void;
+export declare function success(message: string): void;
+type FailWithOpts = {
+    code?: ExitCode;
+    extra?: Record<string, unknown>;
+};
+export declare function failWith(message: string, opts?: FailWithOpts): never;
+export declare function startSpinner(message: string): {
+    stop: () => void;
+};
 export declare function parseJson(value: string, optionName: string): any;
-export declare function handleApiCall<T>(fn: () => Promise<T>): Promise<T | never>;
+type HandleApiCallOpts = {
+    spinner?: string | false;
+};
+export declare function handleApiCall<T>(fn: () => Promise<T>, opts?: HandleApiCallOpts): Promise<T>;
 export declare function pollRunUntilFinished<T>(getResult: () => Promise<T>, getStatus: (result: T) => string, intervalMs: number): Promise<T>;
 export declare function pollBatchUntilFinished<T>(getResult: () => Promise<T>, getStatus: (result: T) => string, intervalMs: number): Promise<T>;
 export declare function pollMessageUntilFinished<T>(getResult: () => Promise<T>, getFinishedAt: (result: T) => string | null | undefined, intervalMs: number): Promise<T>;
+export {};
 //# sourceMappingURL=runHandler.d.ts.map

package/build/commands/runHandler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runHandler.d.ts","sourceRoot":"","sources":["../../src/commands/runHandler.ts"],"names":[],"mappings":"AAEA,wBAAgB,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAE/C;AAGD,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,GAAG,CAWhE;AAED,wBAAsB,aAAa,CAAC,CAAC,EACnC,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GACnB,OAAO,CAAC,CAAC,GAAG,KAAK,CAAC,CAiBpB;AAoBD,wBAAsB,oBAAoB,CAAC,CAAC,EAC1C,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,SAAS,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,MAAM,EAChC,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,CAAC,CAAC,CASZ;AAED,wBAAsB,sBAAsB,CAAC,CAAC,EAC5C,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,SAAS,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,MAAM,EAChC,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,CAAC,CAAC,CASZ;AAED,wBAAsB,wBAAwB,CAAC,CAAC,EAC9C,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,aAAa,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,MAAM,GAAG,IAAI,GAAG,SAAS,EACvD,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,CAAC,CAAC,CAWZ"}
+{"version":3,"file":"runHandler.d.ts","sourceRoot":"","sources":["../../src/commands/runHandler.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,SAAS;;;;;;;CAOZ,CAAC;AAEX,MAAM,MAAM,QAAQ,GAAG,CAAC,OAAO,SAAS,CAAC,CAAC,MAAM,OAAO,SAAS,CAAC,CAAC;AAclE,eAAO,MAAM,MAAM;aACR,MAAM,KAAG,MAAM;eACb,MAAM,KAAG,MAAM;gBACd,MAAM,KAAG,MAAM;cACjB,MAAM,KAAG,MAAM;aAChB,MAAM,KAAG,MAAM;cACd,MAAM,KAAG,MAAM;CAC1B,CAAC;AAEF,wBAAgB,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI,CAM/C;AAED,wBAAgB,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAE1C;AAED,wBAAgB,OAAO,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAE7C;AAED,KAAK,YAAY,GAAG;IAClB,IAAI,CAAC,EAAE,QAAQ,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACjC,CAAC;AAEF,wBAAgB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,YAAY,GAAG,KAAK,CAyBpE;AAKD,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG;IAAE,IAAI,EAAE,MAAM,IAAI,CAAA;CAAE,CA4BlE;AAGD,wBAAgB,SAAS,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,GAAG,CAQhE;AAKD,KAAK,iBAAiB,GAAG;IACvB,OAAO,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;CAC1B,CAAC;AAEF,wBAAsB,aAAa,CAAC,CAAC,EACnC,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EACpB,IAAI,CAAC,EAAE,iBAAiB,GACvB,OAAO,CAAC,CAAC,CAAC,CAiDZ;AAmFD,wBAAsB,oBAAoB,CAAC,CAAC,EAC1C,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,SAAS,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,MAAM,EAChC,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,CAAC,CAAC,CASZ;AAED,wBAAsB,sBAAsB,CAAC,CAAC,EAC5C,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,SAAS,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,MAAM,EAChC,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,CAAC,CAAC,CASZ;AAED,wBAAsB,wBAAwB,CAAC,CAAC,EAC9C,SAAS,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAC3B,aAAa,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,MAAM,GAAG,IAAI,GAAG,SAAS,EACvD,UAAU,EAAE,MAAM,GACjB,OAAO,CAAC,CAAC,CAAC,CAWZ"}

package/build/commands/runHandler.js

@@ -1,6 +1,92 @@
 import { determineIfIsFetcherError } from "@cargo-ai/api";
+export const ExitCodes = {
+    Success: 0,
+    GenericError: 1,
+    InvalidUsage: 2,
+    NotAuthenticated: 3,
+    PermissionDenied: 4,
+    NotFound: 5,
+};
+const useColor = process.stderr.isTTY === true &&
+    process.env["NO_COLOR"] === undefined &&
+    process.env["CI"] !== "true";
+const colorize = (text, code) => {
+    if (useColor !== true) {
+        return text;
+    }
+    return `\x1B[${code}m${text}\x1B[0m`;
+};
+export const colors = {
+    red: (s) => colorize(s, "31"),
+    green: (s) => colorize(s, "32"),
+    yellow: (s) => colorize(s, "33"),
+    cyan: (s) => colorize(s, "36"),
+    dim: (s) => colorize(s, "2"),
+    bold: (s) => colorize(s, "1"),
+};
 export function outputJson(value) {
-    console.log(JSON.stringify(value, null, 2));
+    if (process.stdout.isTTY === true) {
+        console.log(JSON.stringify(value, null, 2));
+        return;
+    }
+    console.log(JSON.stringify(value));
+}
+export function info(message) {
+    console.error(message);
+}
+export function success(message) {
+    console.error(`${colors.green("✓")} ${message}`);
+}
+export function failWith(message, opts) {
+    const code = opts !== undefined && opts.code !== undefined
+        ? opts.code
+        : ExitCodes.GenericError;
+    const extra = opts !== undefined ? opts.extra : undefined;
+    if (process.stderr.isTTY === true) {
+        process.stderr.write(`${colors.red("✗")} ${message}\n`);
+        if (extra !== undefined) {
+            for (const [key, value] of Object.entries(extra)) {
+                const valueStr = typeof value === "string" ? value : JSON.stringify(value);
+                process.stderr.write(`  ${colors.dim(`${key}:`)} ${valueStr}\n`);
+            }
+        }
+    }
+    else {
+        const payload = { error: message };
+        if (extra !== undefined) {
+            Object.assign(payload, extra);
+        }
+        console.error(JSON.stringify(payload));
+    }
+    process.exit(code);
+}
+const SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];
+const SPINNER_INTERVAL_MS = 80;
+export function startSpinner(message) {
+    if (process.stderr.isTTY !== true) {
+        process.stderr.write(`${message}\n`);
+        return { stop: () => undefined };
+    }
+    process.stderr.write("\x1B[?25l");
+    let frame = 0;
+    const id = setInterval(() => {
+        const ch = SPINNER_FRAMES[frame % SPINNER_FRAMES.length];
+        if (ch !== undefined) {
+            process.stderr.write(`\r${colors.cyan(ch)} ${colors.dim(message)}`);
+        }
+        frame += 1;
+    }, SPINNER_INTERVAL_MS);
+    let stopped = false;
+    return {
+        stop: () => {
+            if (stopped) {
+                return;
+            }
+            stopped = true;
+            clearInterval(id);
+            process.stderr.write("\r\x1B[K\x1B[?25h");
+        },
+    };
 }
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 export function parseJson(value, optionName) {
@@ -8,28 +94,109 @@ export function parseJson(value, optionName) {
         return JSON.parse(value);
     }
     catch {
-        console.error(JSON.stringify({
-            error: `Invalid JSON for ${optionName}: ${value}`,
-        }));
-        process.exit(1);
+        failWith(`Invalid JSON for ${optionName}: ${value}`, {
+            code: ExitCodes.InvalidUsage,
+        });
     }
 }
-export async function handleApiCall(fn) {
+const SPINNER_DELAY_MS = 200;
+const DEFAULT_SPINNER_MESSAGE = "Loading...";
+export async function handleApiCall(fn, opts) {
+    const spinnerMessage = opts !== undefined && opts.spinner === false
+        ? undefined
+        : opts !== undefined && typeof opts.spinner === "string"
+            ? opts.spinner
+            : DEFAULT_SPINNER_MESSAGE;
+    let spinner;
+    let startTimer;
+    if (spinnerMessage !== undefined) {
+        startTimer = setTimeout(() => {
+            spinner = startSpinner(spinnerMessage);
+        }, SPINNER_DELAY_MS);
+    }
+    const stopAll = () => {
+        if (startTimer !== undefined) {
+            clearTimeout(startTimer);
+        }
+        if (spinner !== undefined) {
+            spinner.stop();
+        }
+    };
     try {
-        return await fn();
-    }
-    catch (err) {
-        if (determineIfIsFetcherError(err)) {
-            const e = err;
-            console.error(JSON.stringify({
-                error: "API error",
-                status: e.status !== undefined ? e.status : "unknown",
-                body: e.body !== undefined ? e.body : String(err),
-            }));
-            process.exit(1);
+        const result = await fn();
+        stopAll();
+        return result;
+    }
+    catch (error) {
+        stopAll();
+        if (determineIfIsFetcherError(error)) {
+            const parsedError = error;
+            const status = parsedError.status;
+            const body = parsedError.body !== undefined ? parsedError.body : String(error);
+            const { message, code } = describeApiError(status, body);
+            failWith(message, {
+                code,
+                extra: { status: status !== undefined ? status : "unknown", body },
+            });
+        }
+        throw error;
+    }
+}
+function describeApiError(status, body) {
+    const detail = extractApiErrorDetail(body);
+    const detailSuffix = detail !== undefined ? `: ${detail}` : "";
+    if (status === 401) {
+        return {
+            message: `Not authenticated${detailSuffix}. Run \`cargo-ai login\` to refresh credentials.`,
+            code: ExitCodes.NotAuthenticated,
+        };
+    }
+    if (status === 403) {
+        return {
+            message: `Permission denied${detailSuffix}. Check your workspace permissions or selected workspace UUID.`,
+            code: ExitCodes.PermissionDenied,
+        };
+    }
+    if (status === 404) {
+        return {
+            message: `Not found${detailSuffix}. Verify the UUID(s) you passed.`,
+            code: ExitCodes.NotFound,
+        };
+    }
+    if (status === 429) {
+        return {
+            message: `Rate limited${detailSuffix}. Retry in a moment.`,
+            code: ExitCodes.GenericError,
+        };
+    }
+    if (status !== undefined && status >= 500) {
+        return {
+            message: `Server error (${String(status)})${detailSuffix}. Retry shortly or check status.cargo.io.`,
+            code: ExitCodes.GenericError,
+        };
+    }
+    return {
+        message: `API error (${status !== undefined ? String(status) : "unknown"})${detailSuffix}`,
+        code: ExitCodes.GenericError,
+    };
+}
+function extractApiErrorDetail(body) {
+    if (typeof body === "string" && body.length > 0) {
+        return body;
+    }
+    if (body !== null && typeof body === "object") {
+        const record = body;
+        if (typeof record["message"] === "string") {
+            return record["message"];
+        }
+        if (typeof record["reason"] === "string") {
+            return record["reason"];
+        }
+        if (typeof record["error"] === "string") {
+            return record["error"];
         }
-        throw err;
     }
+    return undefined;
 }
 function sleep(ms) {
     return new Promise((resolve) => setTimeout(resolve, ms));
@@ -47,27 +214,27 @@ const TERMINAL_BATCH_STATUSES = new Set([
     "skipped",
 ]);
 export async function pollRunUntilFinished(getResult, getStatus, intervalMs) {
-    let result = await handleApiCall(getResult);
+    let result = await handleApiCall(getResult, { spinner: false });
     while (!TERMINAL_RUN_STATUSES.has(getStatus(result))) {
         await sleep(intervalMs);
-        result = await handleApiCall(getResult);
+        result = await handleApiCall(getResult, { spinner: false });
     }
     return result;
 }
 export async function pollBatchUntilFinished(getResult, getStatus, intervalMs) {
-    let result = await handleApiCall(getResult);
+    let result = await handleApiCall(getResult, { spinner: false });
     while (!TERMINAL_BATCH_STATUSES.has(getStatus(result))) {
         await sleep(intervalMs);
-        result = await handleApiCall(getResult);
+        result = await handleApiCall(getResult, { spinner: false });
     }
     return result;
 }
 export async function pollMessageUntilFinished(getResult, getFinishedAt, intervalMs) {
-    let result = await handleApiCall(getResult);
+    let result = await handleApiCall(getResult, { spinner: false });
     const finishedAt = getFinishedAt(result);
     while (finishedAt === null || finishedAt === undefined) {
         await sleep(intervalMs);
-        result = await handleApiCall(getResult);
+        result = await handleApiCall(getResult, { spinner: false });
     }
     return result;
 }

package/build/commands/storage/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/commands/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAQxC,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CAWN"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/commands/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AASxC,wBAAgB,uBAAuB,CACrC,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CAYN"}

package/build/commands/storage/index.js

@@ -1,6 +1,7 @@
 import { registerColumnCommands } from "./column.js";
 import { registerDatasetCommands } from "./dataset.js";
 import { registerModelCommands } from "./model.js";
+import { registerQueryCommands } from "./query.js";
 import { registerRecordCommands } from "./record.js";
 import { registerRelationshipCommands } from "./relationship.js";
 import { registerRunCommands } from "./run.js";
@@ -14,4 +15,5 @@ export function registerStorageCommands(parent, getApi) {
     registerRunCommands(storage, getApi);
     registerRecordCommands(storage, getApi);
     registerColumnCommands(storage, getApi);
+    registerQueryCommands(storage, getApi);
 }

package/build/commands/storage/query.d.ts

@@ -0,0 +1,4 @@
+import type { Command } from "commander";
+import type { Api } from "../../api.js";
+export declare function registerQueryCommands(parent: Command, getApi: () => Api): void;
+//# sourceMappingURL=query.d.ts.map

package/build/commands/storage/query.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"query.d.ts","sourceRoot":"","sources":["../../../src/commands/storage/query.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CA4DN"}

package/build/commands/storage/query.js

@@ -0,0 +1,46 @@
+import { handleApiCall, outputJson } from "../runHandler.js";
+export function registerQueryCommands(parent, getApi) {
+    const query = parent
+        .command("query")
+        .description("Execute SQL queries against storage models");
+    query
+        .command("execute")
+        .description("Execute a read-only SQL query against storage models")
+        .argument("<sql>", "SQL query string")
+        .addHelpText("after", `
+Tables must be referenced as [datasetSlug].[modelSlug] and are rewritten
+to the underlying warehouse table under the hood.
+
+Examples:
+  $ cargo-ai storage query execute "SELECT count(*) FROM my_dataset.my_model"
+  $ cargo-ai storage query execute "SELECT a._id FROM sales.accounts a JOIN sales.contacts c ON c.account_id = a._id"
+  $ cargo-ai storage query execute "WITH recent AS (SELECT * FROM sales.accounts) SELECT count(*) FROM recent"`)
+        .action(async (sql) => {
+        const api = getApi();
+        const result = await handleApiCall(() => api.storage.query.execute({ query: sql }));
+        outputJson(result);
+    });
+    query
+        .command("download")
+        .description("Download the result of a SQL query against storage models as files")
+        .requiredOption("--query <sql>", "SQL query string")
+        .option("--format <format>", "Download format (csv or parquet)", "csv")
+        .addHelpText("after", `
+Tables must be referenced as [datasetSlug].[modelSlug] and are rewritten
+to the underlying warehouse table under the hood. Returns signed URLs to
+download the resulting files.
+
+Examples:
+  $ cargo-ai storage query download --query "SELECT * FROM my_dataset.my_model"
+  $ cargo-ai storage query download --query "SELECT * FROM sales.accounts" --format parquet`)
+        .action(async (opts) => {
+        const api = getApi();
+        const format = opts.format === "parquet"
+            ? "parquet"
+            : opts.format === "csv"
+                ? "csv"
+                : undefined;
+        const result = await handleApiCall(() => api.storage.query.download({ query: opts.query, format }));
+        outputJson(result);
+    });
+}

package/build/commands/workspaceManagement/file.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"file.d.ts","sourceRoot":"","sources":["../../../src/commands/workspaceManagement/file.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG,IAAI,CA+C7E"}

package/build/commands/workspaceManagement/folder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"folder.d.ts","sourceRoot":"","sources":["../../../src/commands/workspaceManagement/folder.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CA6FN"}

package/build/commands/workspaceManagement/index.d.ts

@@ -0,0 +1,4 @@
+import type { Command } from "commander";
+import type { Api } from "../../api.js";
+export declare function registerWorkspaceManagementCommands(parent: Command, getApi: () => Api): void;
+//# sourceMappingURL=index.d.ts.map

package/build/commands/workspaceManagement/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/commands/workspaceManagement/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AASxC,wBAAgB,mCAAmC,CACjD,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CAYN"}

package/build/commands/workspaceManagement/index.js

@@ -0,0 +1,19 @@
+import { registerFileCommands } from "./file.js";
+import { registerFolderCommands } from "./folder.js";
+import { registerReportCommands } from "./report.js";
+import { registerRoleCommands } from "./role.js";
+import { registerTokenCommands } from "./token.js";
+import { registerUserCommands } from "./user.js";
+import { registerWorkspacesCommands } from "./workspaces.js";
+export function registerWorkspaceManagementCommands(parent, getApi) {
+    const workspaceManagement = parent
+        .command("workspaceManagement")
+        .description("Workspaces, users, tokens, roles, folders, reports");
+    registerWorkspacesCommands(workspaceManagement, getApi);
+    registerUserCommands(workspaceManagement, getApi);
+    registerTokenCommands(workspaceManagement, getApi);
+    registerRoleCommands(workspaceManagement, getApi);
+    registerFolderCommands(workspaceManagement, getApi);
+    registerFileCommands(workspaceManagement, getApi);
+    registerReportCommands(workspaceManagement, getApi);
+}

package/build/commands/workspaceManagement/report.d.ts

@@ -0,0 +1,4 @@
+import type { Command } from "commander";
+import type { Api } from "../../api.js";
+export declare function registerReportCommands(parent: Command, getApi: () => Api): void;
+//# sourceMappingURL=report.d.ts.map

package/build/commands/workspaceManagement/report.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"report.d.ts","sourceRoot":"","sources":["../../../src/commands/workspaceManagement/report.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CAkBN"}

package/build/commands/workspaceManagement/report.js

@@ -0,0 +1,17 @@
+import { handleApiCall, outputJson } from "../runHandler.js";
+export function registerReportCommands(parent, getApi) {
+    const report = parent.command("report").description("Report operations");
+    report
+        .command("create")
+        .description("Submit a report for the current workspace")
+        .requiredOption("--title <title>", "Report title")
+        .requiredOption("--description <description>", "Report description")
+        .action(async (opts) => {
+        const api = getApi();
+        const result = await handleApiCall(() => api.workspaceManagement.report.create({
+            title: opts.title,
+            description: opts.description,
+        }));
+        outputJson(result);
+    });
+}

package/build/commands/workspaceManagement/role.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"role.d.ts","sourceRoot":"","sources":["../../../src/commands/workspaceManagement/role.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG,IAAI,CAa7E"}

package/build/commands/workspaceManagement/token.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"token.d.ts","sourceRoot":"","sources":["../../../src/commands/workspaceManagement/token.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,GAAG,GAChB,IAAI,CAqCN"}

package/build/commands/{workspace → workspaceManagement}/token.js

@@ -12,11 +12,12 @@ export function registerTokenCommands(parent, getApi) {
     token
         .command("create")
         .description("Create a token")
-        .option("--from-user", "Create token from current user")
+        .requiredOption("--name <name>", "Name of the token")
         .action(async (opts) => {
         const api = getApi();
         const result = await handleApiCall(() => api.workspaceManagement.token.create({
-            fromUser: opts.fromUser !== undefined ? opts.fromUser : true,
+            name: opts.name,
+            permissions: null,
         }));
         outputJson(result);
     });

package/build/commands/workspaceManagement/user.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"user.d.ts","sourceRoot":"","sources":["../../../src/commands/workspaceManagement/user.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,KAAK,EAAE,GAAG,EAAE,MAAM,cAAc,CAAC;AAGxC,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG,IAAI,CA8M7E"}