@thinkingai/ae-cli 6.0.41 → 6.0.44

package/dist/{te-dataops-KXCEB4CS.js → te-dataops-EJP56W3K.js}

@@ -1,25 +1,33 @@
 import {
   printError
-} from "./chunk-WZRX4KOH.js";
+} from "./chunk-GT46FPXN.js";
+import {
+  internalCallSourceHeaders
+} from "./chunk-6EIJSNBD.js";
 import {
   clearCliToken,
   getCliToken
-} from "./chunk-S5NTSDBS.js";
-import {
-  resolveHost
-} from "./chunk-ZQKDZXDO.js";
+} from "./chunk-4SGZG4XY.js";
 import {
+  resolveHost,
   safeJsonParse
-} from "./chunk-3FY3RJ26.js";
+} from "./chunk-ZZUOD757.js";
 import {
   CliValidationError,
   PermissionError
-} from "./chunk-AXDXJTPC.js";
+} from "./chunk-UW5UN47B.js";
 import {
   logger
-} from "./chunk-JHENBQ5B.js";
+} from "./chunk-LYVNONC4.js";
+import "./chunk-QGM4M3NI.js";
 
 // src/commands/te-dataops/shared.ts
+import { randomBytes } from "crypto";
+import { createWriteStream } from "fs";
+import { mkdir, rename, unlink } from "fs/promises";
+import { basename, dirname, join, resolve } from "path";
+import { Readable } from "stream";
+import { pipeline } from "stream/promises";
 var TOOL_API_MAP = {
   repo_list_spaces: { method: "GET", path: "/api/cli/dataops/v1/gaia/repo/spaces" },
   datatable_get_table_detail: { method: "GET", path: "/api/cli/dataops/v1/gaia/datatable/table-detail" },
@@ -110,16 +118,16 @@ function buildDataopsApiDryRun(ctx, toolName59, args = {}) {
     body: params
   };
 }
-async function downloadDataopsApi(ctx, path2, params, retry = true) {
+async function downloadDataopsApi(ctx, path, params, targetPath, retry = true) {
   const host = resolveHost(ctx.host());
   const token = await getCliToken(host);
-  const url = buildUrl(host, path2, compactArgs(params));
+  const url = buildUrl(host, path, compactArgs(params));
   const resp = await fetch(url, {
     method: "GET",
     headers: {
       "cli-token": token,
       "Accept": "*/*",
-      "X-Source": "ae-cli"
+      ...internalCallSourceHeaders()
     }
   });
   if (resp.status === 403) {
@@ -128,23 +136,44 @@ async function downloadDataopsApi(ctx, path2, params, retry = true) {
   }
   if (resp.status === 401 && retry) {
     clearCliToken(host);
-    return downloadDataopsApi(ctx, path2, params, false);
+    return downloadDataopsApi(ctx, path, params, targetPath, false);
   }
   if (!resp.ok) {
     const data = parseResponseBody(await resp.text(), url, resp.status, true);
     throw new Error(formatHttpError(resp.status, resp.statusText, data));
   }
-  return Buffer.from(await resp.arrayBuffer());
-}
-async function dataopsRequest(ctx, method, path2, params = {}, body, retry = true) {
+  if (!resp.body) {
+    throw new Error("DataOps download response has no body.");
+  }
+  const outputPath = resolve(targetPath);
+  await mkdir(dirname(outputPath), { recursive: true });
+  const tempPath = join(
+    dirname(outputPath),
+    `.${basename(outputPath)}.part-${process.pid}-${randomBytes(6).toString("hex")}`
+  );
+  let tempPresent = false;
+  try {
+    tempPresent = true;
+    await pipeline(
+      Readable.fromWeb(resp.body),
+      createWriteStream(tempPath, { flags: "wx" })
+    );
+    await rename(tempPath, outputPath);
+    tempPresent = false;
+    return outputPath;
+  } finally {
+    if (tempPresent) await unlink(tempPath).catch(() => void 0);
+  }
+}
+async function dataopsRequest(ctx, method, path, params = {}, body, retry = true) {
   const host = resolveHost(ctx.host());
   const token = await getCliToken(host);
-  const url = buildUrl(host, path2, params);
+  const url = buildUrl(host, path, params);
   const headers = {
     "cli-token": token,
     "Accept": "application/json",
     "Content-Type": "application/json",
-    "X-Source": "ae-cli"
+    ...internalCallSourceHeaders()
   };
   const options = { method, headers };
   if (method !== "GET") {
@@ -160,16 +189,16 @@ async function dataopsRequest(ctx, method, path2, params = {}, body, retry = tru
   if (resp.status === 401 && retry) {
     logger.warn(`DataOps request failed (HTTP 401) for ${host}, refreshing CLI token`);
     clearCliToken(host);
-    return dataopsRequest(ctx, method, path2, params, body, false);
+    return dataopsRequest(ctx, method, path, params, body, false);
   }
   if (!resp.ok) {
     throw new Error(formatHttpError(resp.status, resp.statusText, data));
   }
   return unwrapDataResponse(data);
 }
-function buildUrl(host, path2, params = {}) {
+function buildUrl(host, path, params = {}) {
   const base = host.replace(/\/+$/, "");
-  const normalizedPath = path2.startsWith("/") ? path2 : `/${path2}`;
+  const normalizedPath = path.startsWith("/") ? path : `/${path}`;
   const url = new URL(`${base}${normalizedPath}`);
   for (const [key, value] of Object.entries(params)) {
     if (value !== void 0 && value !== null) {
@@ -219,8 +248,8 @@ function permissionMessage(data) {
 }
 function formatHttpError(status, statusText, data) {
   const detail = typeof data === "string" ? data : data?.message || data?.error || data?.return_message || data?.returnMessage;
-  const path2 = data?.path ? ` path: ${data.path}` : "";
-  return `AE API error: HTTP ${status} ${statusText}${detail ? ` - ${detail}` : ""}${path2}`;
+  const path = data?.path ? ` path: ${data.path}` : "";
+  return `AE API error: HTTP ${status} ${statusText}${detail ? ` - ${detail}` : ""}${path}`;
 }
 
 // src/commands/te-dataops/datatable/get-table-detail.ts
@@ -1845,8 +1874,6 @@ var cancelSqlQuery = {
 };
 
 // src/commands/te-dataops/ide/get-sql-query-status.ts
-import { mkdir, writeFile } from "fs/promises";
-import path from "path";
 var toolName51 = "ide_get_sql_query_status";
 var downloadPath = "/api/cli/dataops/v1/gaia/ide/sql-query-download";
 function optionalString5(ctx, name) {
@@ -1865,24 +1892,20 @@ function isDownloadReady(data) {
 }
 async function downloadResult(ctx, data, targetPath) {
   const params = data?.downloadParams ?? {};
-  const bytes = await downloadDataopsApi(ctx, downloadPath, {
+  return downloadDataopsApi(ctx, downloadPath, {
     spaceCode: params.spaceCode,
     taskId: params.taskId
-  });
-  const absPath = path.resolve(targetPath);
-  await mkdir(path.dirname(absPath), { recursive: true });
-  await writeFile(absPath, bytes);
-  return absPath;
+  }, targetPath);
 }
 var getSqlQueryStatus = {
   service: "dataops_ide",
   command: "+get_sql_query_status",
-  description: "Poll a Gaia download-center task created by +submit_sql_query. Requires spaceCode and downloadTaskId. requestId is trace-only. Returns status and download metadata; --downloadTo saves the result zip locally after SUCCESS and adds localFile.",
+  description: "Poll a Gaia download-center task created by +submit_sql_query. Requires spaceCode and downloadTaskId. requestId is trace-only. Returns status and download metadata; --downloadTo streams the result zip to a local file after SUCCESS and adds localFile.",
   flags: [
     { name: "spaceCode", type: "string", required: true, desc: "Space code returned by +submit_sql_query" },
     { name: "downloadTaskId", type: "number", required: true, desc: "Download task ID returned by +submit_sql_query" },
     { name: "requestId", type: "string", required: false, desc: "Optional request ID returned by +submit_sql_query, used only for trace display" },
-    { name: "downloadTo", type: "string", required: false, desc: "Optional local file path. When downloadStatus=SUCCESS, save the result zip to this path; use a .zip suffix, for example ./result.zip." }
+    { name: "downloadTo", type: "string", required: false, desc: "Optional local file path. After SUCCESS, stream the result zip and replace this path only when the download completes; use a .zip suffix, for example ./result.zip." }
   ],
   risk: "read",
   dryRun: (ctx) => buildDataopsApiDryRun(ctx, toolName51, buildArgs47(ctx)),
@@ -2144,10 +2167,10 @@ function buildArgs54(ctx) {
 var submitSqlQuery = {
   service: "dataops_ide",
   command: "+submit_sql_query",
-  description: "Submit an IDE SQL query and create a Gaia download-center task. Requires spaceCode and sql. Optional repoCode/engineType default to te_etl/TASK_ENGINE_TRINO. Returns download task metadata; result rows are never returned by MCP/CLI.",
+  description: "Submit exactly one read-only IDE SQL query and create a Gaia download-center task. Optional repoCode/engineType default to te_etl/TASK_ENGINE_TRINO. Results are platform-bounded; Gaia returns downloadRowLimit when supported, and rows are never returned by MCP/CLI.",
   flags: [
     { name: "spaceCode", type: "string", required: true, desc: "Space code" },
-    { name: "sql", type: "string", required: true, desc: "SQL statement to execute" },
+    { name: "sql", type: "string", required: true, desc: "Exactly one read-only SQL query" },
     { name: "repoCode", type: "string", required: false, desc: "Repository code. Defaults to te_etl." },
     { name: "engineType", type: "string", required: false, desc: "SQL execution engine. Defaults to TASK_ENGINE_TRINO. Optional: TASK_ENGINE_STARROCKS." }
   ],

package/dist/{te-dataops-KQPYNAE3.js → te-dataops-HDRUXY4K.js}

@@ -1,25 +1,33 @@
 import {
   printError
-} from "./chunk-WZRX4KOH.js";
+} from "./chunk-GT46FPXN.js";
+import {
+  internalCallSourceHeaders
+} from "./chunk-C4MGVGJW.js";
 import {
   clearCliToken,
   getCliToken
-} from "./chunk-S5NTSDBS.js";
-import {
-  resolveHost
-} from "./chunk-ZQKDZXDO.js";
+} from "./chunk-4SGZG4XY.js";
 import {
+  resolveHost,
   safeJsonParse
-} from "./chunk-3FY3RJ26.js";
+} from "./chunk-ZZUOD757.js";
 import {
   CliValidationError,
   PermissionError
-} from "./chunk-AXDXJTPC.js";
+} from "./chunk-UW5UN47B.js";
 import {
   logger
-} from "./chunk-JHENBQ5B.js";
+} from "./chunk-LYVNONC4.js";
+import "./chunk-QGM4M3NI.js";
 
 // src/commands/te-dataops/shared.ts
+import { randomBytes } from "crypto";
+import { createWriteStream } from "fs";
+import { mkdir, rename, unlink } from "fs/promises";
+import { basename, dirname, join, resolve } from "path";
+import { Readable } from "stream";
+import { pipeline } from "stream/promises";
 var TOOL_API_MAP = {
   repo_list_spaces: { method: "GET", path: "/api/cli/dataops/v1/gaia/repo/spaces" },
   datatable_get_table_detail: { method: "GET", path: "/api/cli/dataops/v1/gaia/datatable/table-detail" },
@@ -110,15 +118,16 @@ function buildDataopsApiDryRun(ctx, toolName59, args = {}) {
     body: params
   };
 }
-async function downloadDataopsApi(ctx, path2, params, retry = true) {
+async function downloadDataopsApi(ctx, path, params, targetPath, retry = true) {
   const host = resolveHost(ctx.host());
   const token = await getCliToken(host);
-  const url = buildUrl(host, path2, compactArgs(params));
+  const url = buildUrl(host, path, compactArgs(params));
   const resp = await fetch(url, {
     method: "GET",
     headers: {
       "cli-token": token,
-      "Accept": "*/*"
+      "Accept": "*/*",
+      ...internalCallSourceHeaders()
     }
   });
   if (resp.status === 403) {
@@ -127,22 +136,44 @@ async function downloadDataopsApi(ctx, path2, params, retry = true) {
   }
   if (resp.status === 401 && retry) {
     clearCliToken(host);
-    return downloadDataopsApi(ctx, path2, params, false);
+    return downloadDataopsApi(ctx, path, params, targetPath, false);
   }
   if (!resp.ok) {
     const data = parseResponseBody(await resp.text(), url, resp.status, true);
     throw new Error(formatHttpError(resp.status, resp.statusText, data));
   }
-  return Buffer.from(await resp.arrayBuffer());
-}
-async function dataopsRequest(ctx, method, path2, params = {}, body, retry = true) {
+  if (!resp.body) {
+    throw new Error("DataOps download response has no body.");
+  }
+  const outputPath = resolve(targetPath);
+  await mkdir(dirname(outputPath), { recursive: true });
+  const tempPath = join(
+    dirname(outputPath),
+    `.${basename(outputPath)}.part-${process.pid}-${randomBytes(6).toString("hex")}`
+  );
+  let tempPresent = false;
+  try {
+    tempPresent = true;
+    await pipeline(
+      Readable.fromWeb(resp.body),
+      createWriteStream(tempPath, { flags: "wx" })
+    );
+    await rename(tempPath, outputPath);
+    tempPresent = false;
+    return outputPath;
+  } finally {
+    if (tempPresent) await unlink(tempPath).catch(() => void 0);
+  }
+}
+async function dataopsRequest(ctx, method, path, params = {}, body, retry = true) {
   const host = resolveHost(ctx.host());
   const token = await getCliToken(host);
-  const url = buildUrl(host, path2, params);
+  const url = buildUrl(host, path, params);
   const headers = {
     "cli-token": token,
     "Accept": "application/json",
-    "Content-Type": "application/json"
+    "Content-Type": "application/json",
+    ...internalCallSourceHeaders()
   };
   const options = { method, headers };
   if (method !== "GET") {
@@ -158,16 +189,16 @@ async function dataopsRequest(ctx, method, path2, params = {}, body, retry = tru
   if (resp.status === 401 && retry) {
     logger.warn(`DataOps request failed (HTTP 401) for ${host}, refreshing CLI token`);
     clearCliToken(host);
-    return dataopsRequest(ctx, method, path2, params, body, false);
+    return dataopsRequest(ctx, method, path, params, body, false);
   }
   if (!resp.ok) {
     throw new Error(formatHttpError(resp.status, resp.statusText, data));
   }
   return unwrapDataResponse(data);
 }
-function buildUrl(host, path2, params = {}) {
+function buildUrl(host, path, params = {}) {
   const base = host.replace(/\/+$/, "");
-  const normalizedPath = path2.startsWith("/") ? path2 : `/${path2}`;
+  const normalizedPath = path.startsWith("/") ? path : `/${path}`;
   const url = new URL(`${base}${normalizedPath}`);
   for (const [key, value] of Object.entries(params)) {
     if (value !== void 0 && value !== null) {
@@ -217,8 +248,8 @@ function permissionMessage(data) {
 }
 function formatHttpError(status, statusText, data) {
   const detail = typeof data === "string" ? data : data?.message || data?.error || data?.return_message || data?.returnMessage;
-  const path2 = data?.path ? ` path: ${data.path}` : "";
-  return `AE API error: HTTP ${status} ${statusText}${detail ? ` - ${detail}` : ""}${path2}`;
+  const path = data?.path ? ` path: ${data.path}` : "";
+  return `AE API error: HTTP ${status} ${statusText}${detail ? ` - ${detail}` : ""}${path}`;
 }
 
 // src/commands/te-dataops/datatable/get-table-detail.ts
@@ -1843,8 +1874,6 @@ var cancelSqlQuery = {
 };
 
 // src/commands/te-dataops/ide/get-sql-query-status.ts
-import { mkdir, writeFile } from "fs/promises";
-import path from "path";
 var toolName51 = "ide_get_sql_query_status";
 var downloadPath = "/api/cli/dataops/v1/gaia/ide/sql-query-download";
 function optionalString5(ctx, name) {
@@ -1863,24 +1892,20 @@ function isDownloadReady(data) {
 }
 async function downloadResult(ctx, data, targetPath) {
   const params = data?.downloadParams ?? {};
-  const bytes = await downloadDataopsApi(ctx, downloadPath, {
+  return downloadDataopsApi(ctx, downloadPath, {
     spaceCode: params.spaceCode,
     taskId: params.taskId
-  });
-  const absPath = path.resolve(targetPath);
-  await mkdir(path.dirname(absPath), { recursive: true });
-  await writeFile(absPath, bytes);
-  return absPath;
+  }, targetPath);
 }
 var getSqlQueryStatus = {
   service: "dataops_ide",
   command: "+get_sql_query_status",
-  description: "Poll a Gaia download-center task created by +submit_sql_query. Requires spaceCode and downloadTaskId. requestId is trace-only. Returns status and download metadata; --downloadTo saves the result zip locally after SUCCESS and adds localFile.",
+  description: "Poll a Gaia download-center task created by +submit_sql_query. Requires spaceCode and downloadTaskId. requestId is trace-only. Returns status and download metadata; --downloadTo streams the result zip to a local file after SUCCESS and adds localFile.",
   flags: [
     { name: "spaceCode", type: "string", required: true, desc: "Space code returned by +submit_sql_query" },
     { name: "downloadTaskId", type: "number", required: true, desc: "Download task ID returned by +submit_sql_query" },
     { name: "requestId", type: "string", required: false, desc: "Optional request ID returned by +submit_sql_query, used only for trace display" },
-    { name: "downloadTo", type: "string", required: false, desc: "Optional local file path. When downloadStatus=SUCCESS, save the result zip to this path; use a .zip suffix, for example ./result.zip." }
+    { name: "downloadTo", type: "string", required: false, desc: "Optional local file path. After SUCCESS, stream the result zip and replace this path only when the download completes; use a .zip suffix, for example ./result.zip." }
   ],
   risk: "read",
   dryRun: (ctx) => buildDataopsApiDryRun(ctx, toolName51, buildArgs47(ctx)),
@@ -2142,10 +2167,10 @@ function buildArgs54(ctx) {
 var submitSqlQuery = {
   service: "dataops_ide",
   command: "+submit_sql_query",
-  description: "Submit an IDE SQL query and create a Gaia download-center task. Requires spaceCode and sql. Optional repoCode/engineType default to te_etl/TASK_ENGINE_TRINO. Returns download task metadata; result rows are never returned by MCP/CLI.",
+  description: "Submit exactly one read-only IDE SQL query and create a Gaia download-center task. Optional repoCode/engineType default to te_etl/TASK_ENGINE_TRINO. Results are platform-bounded; Gaia returns downloadRowLimit when supported, and rows are never returned by MCP/CLI.",
   flags: [
     { name: "spaceCode", type: "string", required: true, desc: "Space code" },
-    { name: "sql", type: "string", required: true, desc: "SQL statement to execute" },
+    { name: "sql", type: "string", required: true, desc: "Exactly one read-only SQL query" },
     { name: "repoCode", type: "string", required: false, desc: "Repository code. Defaults to te_etl." },
     { name: "engineType", type: "string", required: false, desc: "SQL execution engine. Defaults to TASK_ENGINE_TRINO. Optional: TASK_ENGINE_STARROCKS." }
   ],