@kweaver-ai/kweaver-sdk 0.4.6 → 0.4.7

package/bin/kweaver.js

@@ -1,9 +1,24 @@
 #!/usr/bin/env node
+
+function exit(code) {
+  if (process.stdout.writableNeedDrain || process.stderr.writableNeedDrain) {
+    const done = () => {
+      if (!process.stdout.writableNeedDrain && !process.stderr.writableNeedDrain) {
+        process.exit(code);
+      }
+    };
+    process.stdout.once("drain", done);
+    process.stderr.once("drain", done);
+  } else {
+    process.exit(code);
+  }
+}
+
 import("../dist/cli.js").then(({ run }) => {
   run(process.argv.slice(2))
-    .then((code) => process.exit(code))
+    .then((code) => exit(code))
     .catch((err) => {
       console.error(err instanceof Error ? err.message : String(err));
-      process.exit(1);
+      exit(1);
     });
 });

package/dist/cli.js

@@ -87,14 +87,28 @@ export async function run(argv) {
     printHelp();
     return 1;
 }
+function safeExit(code) {
+    if (process.stdout.writableNeedDrain || process.stderr.writableNeedDrain) {
+        const done = () => {
+            if (!process.stdout.writableNeedDrain && !process.stderr.writableNeedDrain) {
+                process.exit(code);
+            }
+        };
+        process.stdout.once("drain", done);
+        process.stderr.once("drain", done);
+    }
+    else {
+        process.exit(code);
+    }
+}
 if (import.meta.url === `file://${process.argv[1]}`) {
     run(process.argv.slice(2))
         .then((code) => {
-        process.exit(code);
+        safeExit(code);
     })
         .catch((error) => {
         const message = error instanceof Error ? error.message : String(error);
         console.error(message);
-        process.exit(1);
+        safeExit(1);
     });
 }

package/dist/commands/bkn.js

@@ -456,6 +456,57 @@ function parseJsonObject(text, errorMessage) {
     }
     return parsed;
 }
+const MAX_OUTPUT_BYTES = 100_000;
+/**
+ * If a query response exceeds MAX_OUTPUT_BYTES, trim the datas array
+ * to fit, preserving valid JSON and the search_after cursor for pagination.
+ */
+function truncateQueryResult(raw) {
+    if (raw.length <= MAX_OUTPUT_BYTES) {
+        return raw;
+    }
+    let parsed;
+    try {
+        parsed = JSON.parse(raw);
+    }
+    catch {
+        return raw;
+    }
+    const datas = parsed.datas;
+    if (!Array.isArray(datas) || datas.length === 0) {
+        return raw;
+    }
+    const originalCount = datas.length;
+    while (datas.length > 1) {
+        datas.pop();
+        const candidate = JSON.stringify(parsed);
+        if (candidate.length <= MAX_OUTPUT_BYTES) {
+            const remaining = originalCount - datas.length;
+            const sa = parsed.search_after;
+            parsed._truncated = {
+                returned: datas.length,
+                total_fetched: originalCount,
+                remaining,
+                next_search_after: sa ?? null,
+                hint: sa
+                    ? `Pass --search-after '${JSON.stringify(sa)}' --limit ${datas.length} to fetch the next page.`
+                    : `Reduce --limit to ${datas.length} or less to avoid truncation.`,
+            };
+            console.error(`[warn] Truncated ${originalCount} → ${datas.length} records (output exceeded ${Math.round(MAX_OUTPUT_BYTES / 1024)}KB). ${parsed._truncated.hint}`);
+            return JSON.stringify(parsed);
+        }
+    }
+    const sa = parsed.search_after;
+    parsed._truncated = {
+        returned: 1,
+        total_fetched: originalCount,
+        remaining: originalCount - 1,
+        next_search_after: sa ?? null,
+        hint: `Single record is very large. Use --limit 1 and --search-after to iterate.`,
+    };
+    console.error(`[warn] Truncated ${originalCount} → 1 record. Single record is very large. Use --limit 1 and --search-after to iterate.`);
+    return JSON.stringify(parsed);
+}
 function parseSearchAfterArray(text) {
     let parsed;
     try {
@@ -1017,11 +1068,7 @@ properties JSON format: {"_instance_identities":[{"<primary-key>":"<value>"}],"p
                 body: options.body,
                 businessDomain: options.businessDomain,
             });
-            const OUTPUT_WARN_BYTES = 100_000;
-            if (result.length > OUTPUT_WARN_BYTES) {
-                console.error(`[warn] Response is ${(result.length / 1024).toFixed(0)}KB. Use a smaller --limit or --search-after to paginate.`);
-            }
-            console.log(formatCallOutput(result, options.pretty));
+            console.log(formatCallOutput(truncateQueryResult(result), options.pretty));
             return 0;
         }
         if (action === "properties") {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kweaver-ai/kweaver-sdk",
-  "version": "0.4.6",
+  "version": "0.4.7",
   "description": "KWeaver TypeScript SDK — CLI tool and programmatic API for knowledge networks and Decision Agents.",
   "type": "module",
   "main": "./dist/index.js",