@kweaver-ai/kweaver-sdk 0.4.11 → 0.4.13

package/dist/commands/auth.js

@@ -1,11 +1,12 @@
-import { clearPlatformSession, deletePlatform, getConfigDir, getCurrentPlatform, getPlatformAlias, hasPlatform, listPlatforms, loadTokenConfig, resolvePlatformIdentifier, setCurrentPlatform, setPlatformAlias, } from "../config/store.js";
-import { formatHttpError, normalizeBaseUrl, oauth2Login, playwrightLogin, } from "../auth/oauth.js";
+import { clearPlatformSession, deletePlatform, getConfigDir, getCurrentPlatform, getPlatformAlias, hasPlatform, listPlatforms, loadClientConfig, loadTokenConfig, resolvePlatformIdentifier, setCurrentPlatform, setPlatformAlias, } from "../config/store.js";
+import { buildCopyCommand, formatHttpError, normalizeBaseUrl, oauth2Login, playwrightLogin, refreshTokenLogin, } from "../auth/oauth.js";
 export async function runAuthCommand(args) {
     const target = args[0];
     const rest = args.slice(1);
     if (!target || target === "--help" || target === "-h") {
         console.log(`kweaver auth login <url> [options]   Login to a platform (browser OAuth2 by default)
 kweaver auth <url>                   Login (shorthand; same options as login)
+kweaver auth export [url|alias] [--json]   Export credentials; run printed command on a headless host
 kweaver auth status [url|alias]      Show current auth status
 kweaver auth list                    List saved platforms
 kweaver auth use <url|alias>         Switch active platform
@@ -18,6 +19,9 @@ Login options:
                          Use the platform's web app client ID to get the same permissions
                          as the browser. Find it in DevTools: /oauth2/auth?client_id=<id>
   --client-secret <s>    Client secret (omit for public/PKCE clients)
+  --refresh-token <t>    Use on a machine without a browser: exchange refresh token for access token.
+                         Requires --client-id and --client-secret.
+                         Get these from the callback page after browser login or \`auth export\`.
   -u, --username         Username (with -p triggers Playwright headless login)
   -p, --password         Password
   --playwright           Force Playwright browser login even without -u/-p
@@ -26,7 +30,7 @@ Login options:
     }
     if (target === "login") {
         if (rest[0] === "--help" || rest[0] === "-h") {
-            console.log(`kweaver auth login <platform-url> [--alias <name>] [-u user] [-p pass] [--playwright]`);
+            console.log(`kweaver auth login <platform-url> [--alias <name>] [-u user] [-p pass] [--playwright] [--refresh-token T --client-id ID --client-secret S]`);
             return 0;
         }
         const url = rest[0];
@@ -36,7 +40,11 @@ Login options:
         }
         return runAuthCommand([url, ...rest.slice(1)]);
     }
-    if (target && target !== "status" && target !== "list" && target !== "use" && target !== "delete" && target !== "logout") {
+    if (target === "export") {
+        return runAuthExportCommand(rest);
+    }
+    const LOGIN_SUBCOMMANDS = new Set(["status", "list", "use", "delete", "logout", "export"]);
+    if (target && !LOGIN_SUBCOMMANDS.has(target)) {
         try {
             const normalizedTarget = normalizeBaseUrl(target);
             const alias = readOption(args, "--alias");
@@ -45,9 +53,21 @@ Login options:
             const usePlaywright = args.includes("--playwright");
             const clientId = readOption(args, "--client-id");
             const clientSecret = readOption(args, "--client-secret");
+            const refreshToken = readOption(args, "--refresh-token");
             const tlsInsecure = args.includes("--insecure") || args.includes("-k");
             let token;
-            if (username && password) {
+            if (refreshToken) {
+                if (!clientId || !clientSecret) {
+                    console.error("--refresh-token requires --client-id and --client-secret.\n");
+                    console.error("Get these values from the callback page after a browser login or `kweaver auth export`.");
+                    return 1;
+                }
+                console.log("Logging in with refresh token (no browser)...");
+                token = await refreshTokenLogin(normalizedTarget, {
+                    clientId, clientSecret, refreshToken, tlsInsecure,
+                });
+            }
+            else if (username && password) {
                 // Headless Playwright login with credentials
                 console.log("Logging in (headless)...");
                 token = await playwrightLogin(normalizedTarget, { username, password, tlsInsecure });
@@ -213,7 +233,7 @@ Login options:
         return 0;
     }
     console.error("Usage: kweaver auth login <platform-url> [--alias <name>] [-u user] [-p pass] [--playwright]");
-    console.error("       kweaver auth <platform-url> [--alias <name>] [-u user] [-p pass] [--playwright]");
+    console.error("       kweaver auth export [platform-url|alias] [--json]");
     console.error("       kweaver auth status [platform-url|alias]");
     console.error("       kweaver auth list");
     console.error("       kweaver auth use <platform-url|alias>");
@@ -221,6 +241,60 @@ Login options:
     console.error("       kweaver auth delete <platform-url|alias>");
     return 1;
 }
+async function runAuthExportCommand(args) {
+    if (args[0] === "--help" || args[0] === "-h") {
+        console.log(`kweaver auth export [platform-url|alias] [--json]
+
+Export OAuth2 credentials for copying to a headless host (no browser there).
+Prints clientId, clientSecret, refreshToken, and a command to run on that machine.
+
+Options:
+  --json   Output as JSON (machine-readable)`);
+        return 0;
+    }
+    const jsonOutput = args.includes("--json");
+    const positional = args.find((a) => !a.startsWith("-"));
+    const resolved = positional ? resolvePlatformIdentifier(positional) : null;
+    const platform = resolved && /^https?:\/\//.test(resolved) ? normalizeBaseUrl(resolved) : resolved ?? getCurrentPlatform();
+    if (!platform) {
+        console.error("No active platform. Run `kweaver auth login <platform-url>` first.");
+        return 1;
+    }
+    const client = loadClientConfig(platform);
+    const token = loadTokenConfig(platform);
+    const clientId = client?.clientId ?? "";
+    const clientSecret = client?.clientSecret ?? "";
+    const refreshToken = token?.refreshToken ?? "";
+    const tlsInsecure = token?.tlsInsecure;
+    if (!clientId || !refreshToken) {
+        console.error(`Incomplete credentials for ${platform}.\n` +
+            (!clientId ? "  Missing: client registration (client.json)\n" : "") +
+            (!refreshToken ? "  Missing: refresh token (token.json)\n" : "") +
+            `Run \`kweaver auth login ${platform}\` first.`);
+        return 1;
+    }
+    if (jsonOutput) {
+        console.log(JSON.stringify({
+            baseUrl: platform,
+            clientId,
+            clientSecret,
+            refreshToken,
+            ...(tlsInsecure ? { tlsInsecure: true } : {}),
+        }));
+        return 0;
+    }
+    const cmd = buildCopyCommand(platform, clientId, clientSecret, refreshToken, tlsInsecure);
+    console.log(`Platform:       ${platform}`);
+    console.log(`Client ID:      ${clientId}`);
+    console.log(`Client Secret:  ${clientSecret || "(none)"}`);
+    console.log(`Refresh Token:  ${refreshToken}`);
+    console.log("");
+    console.log("On a machine without a browser, run:\n");
+    console.log(`  ${cmd}`);
+    console.log("");
+    console.log("Keep these credentials secure.");
+    return 0;
+}
 function readOption(args, name) {
     const index = args.findIndex((arg) => arg === name);
     if (index === -1) {

package/dist/commands/bkn.d.ts

@@ -1,4 +1,15 @@
 import { type BknEncodingImportOptions } from "../utils/bkn-encoding.js";
+export interface PollOptions<T> {
+    fn: () => Promise<{
+        done: boolean;
+        value: T;
+    }>;
+    interval: number;
+    timeout: number;
+    maxInterval?: number;
+    _sleep?: (ms: number) => Promise<void>;
+}
+export declare function pollWithBackoff<T>(opts: PollOptions<T>): Promise<T>;
 export interface KnListOptions {
     offset: number;
     limit: number;

package/dist/commands/bkn.js

@@ -9,11 +9,24 @@ import { listKnowledgeNetworks, getKnowledgeNetwork, createKnowledgeNetwork, upd
 import { objectTypeQuery, objectTypeProperties, subgraph, actionTypeQuery, actionTypeExecute, actionExecutionGet, actionLogsList, actionLogGet, actionLogCancel, } from "../api/ontology-query.js";
 import { semanticSearch } from "../api/semantic-search.js";
 import { listTablesWithColumns, scanMetadata, getDatasource } from "../api/datasources.js";
-import { createDataView } from "../api/dataviews.js"; // used by runKnCreateFromDsCommand
+import { createDataView, findDataView } from "../api/dataviews.js"; // used by runKnCreateFromDsCommand
 import { downloadBkn, uploadBkn } from "../api/bkn-backend.js";
 import { formatCallOutput } from "./call.js";
 import { resolveBusinessDomain } from "../config/store.js";
 import { runDsImportCsv } from "./ds.js";
+export async function pollWithBackoff(opts) {
+    const { fn, timeout, maxInterval = 15000, _sleep = (ms) => new Promise(r => setTimeout(r, ms)) } = opts;
+    let currentInterval = opts.interval;
+    const deadline = Date.now() + timeout;
+    while (Date.now() < deadline) {
+        const result = await fn();
+        if (result.done)
+            return result.value;
+        await _sleep(currentInterval);
+        currentInterval = Math.min(currentInterval * 2, maxInterval);
+    }
+    throw new Error(`Polling timed out after ${timeout}ms`);
+}
 export function formatSimpleKnList(text, pretty, includeDetail = false) {
     const parsed = JSON.parse(text);
     const entries = Array.isArray(parsed.entries) ? parsed.entries : [];
@@ -27,7 +40,7 @@ export function formatSimpleKnList(text, pretty, includeDetail = false) {
 }
 export function parseKnListArgs(args) {
     let offset = 0;
-    let limit = 50;
+    let limit = 30;
     let sort = "update_time";
     let direction = "desc";
     let businessDomain = "";
@@ -49,9 +62,9 @@ export function parseKnListArgs(args) {
             continue;
         }
         if (arg === "--limit") {
-            limit = parseInt(args[i + 1] ?? "50", 10);
+            limit = parseInt(args[i + 1] ?? "30", 10);
             if (Number.isNaN(limit) || limit < 1)
-                limit = 50;
+                limit = 30;
             i += 1;
             continue;
         }
@@ -582,7 +595,7 @@ export function parseKnObjectTypeQueryArgs(args) {
         body.search_after = searchAfter;
     }
     if (typeof body.limit !== "number" || !Number.isFinite(body.limit) || body.limit < 1) {
-        body.limit = 30;
+        body.limit = 50;
     }
     if (!businessDomain)
         businessDomain = resolveBusinessDomain();
@@ -1126,7 +1139,7 @@ kweaver bkn object-type properties <kn-id> <ot-id> '<json>' [--pretty] [-bd valu
 list: List object types (schema) from ontology-manager.
 get: Get single object type details.
 create/update/delete: Schema CRUD (create requires dataview-id). update: merge flags (--add-property / --update-property / --remove-property, etc.) GET-merge-PUT; or full JSON as third arg.
-query: Query via ontology-query API. Default limit is 30 if not specified. Use --search-after for pagination.
+query: Query via ontology-query API. Default limit is 50 if not specified. Use --search-after for pagination.
 properties: Query instance properties by primary key.
 
 properties JSON format: {"_instance_identities":[{"<primary-key>":"<value>"}],"properties":["prop1","prop2"]}`);
@@ -1719,26 +1732,35 @@ query/execute: Query or execute actions. execute has side effects - only use whe
                 console.log(formatCallOutput(result, options.pretty));
                 return 0;
             }
-            const deadline = Date.now() + options.timeout * 1000;
             let lastBody = result;
-            while (Date.now() < deadline) {
-                const status = extractStatus(lastBody);
-                if (TERMINAL_STATUSES.includes(status.toUpperCase())) {
-                    console.log(formatCallOutput(lastBody, options.pretty));
-                    return status.toUpperCase() === "SUCCESS" ? 0 : 1;
-                }
-                await new Promise((r) => setTimeout(r, 2000));
-                lastBody = await actionExecutionGet({
-                    baseUrl: token.baseUrl,
-                    accessToken: token.accessToken,
-                    knId: options.knId,
-                    executionId,
-                    businessDomain: options.businessDomain,
+            try {
+                lastBody = await pollWithBackoff({
+                    fn: async () => {
+                        const status = extractStatus(lastBody);
+                        if (TERMINAL_STATUSES.includes(status.toUpperCase())) {
+                            return { done: true, value: lastBody };
+                        }
+                        lastBody = await actionExecutionGet({
+                            baseUrl: token.baseUrl,
+                            accessToken: token.accessToken,
+                            knId: options.knId,
+                            executionId,
+                            businessDomain: options.businessDomain,
+                        });
+                        return { done: false, value: lastBody };
+                    },
+                    interval: 2000,
+                    timeout: options.timeout * 1000,
                 });
             }
-            console.error(`Action execution did not complete within ${options.timeout}s`);
+            catch {
+                console.error(`Action execution did not complete within ${options.timeout}s`);
+                console.log(formatCallOutput(lastBody, options.pretty));
+                return 1;
+            }
+            const finalStatus = extractStatus(lastBody);
             console.log(formatCallOutput(lastBody, options.pretty));
-            return 1;
+            return finalStatus.toUpperCase() === "SUCCESS" ? 0 : 1;
         }
         catch (error) {
             console.error(formatHttpError(error));
@@ -1802,7 +1824,7 @@ Options for list: --limit, --need-total, --action-type-id, --status, --trigger-t
     }
     let pretty = true;
     let businessDomain = "";
-    let limit;
+    let limit = 30;
     let needTotal;
     let actionTypeId;
     let status;
@@ -1955,7 +1977,7 @@ List business knowledge networks from the ontology-manager API.
 
 Options:
   --offset <n>       Offset (default: 0)
-  --limit <n>        Limit (default: 50)
+  --limit <n>        Limit (default: 30)
   --sort <key>       Sort field (default: update_time)
   --direction <asc|desc>  Sort direction (default: desc)
   --name-pattern <s> Filter by name pattern
@@ -2195,13 +2217,21 @@ async function runKnCreateFromDsCommand(args, sampleRows) {
         console.error(`Creating data views for ${targetTables.length} table(s) ...`);
         const viewMap = {};
         for (const t of targetTables) {
-            const dvId = await createDataView({
+            const found = await findDataView({
                 ...base,
                 name: t.name,
                 datasourceId: options.dsId,
-                table: t.name,
-                fields: t.columns.map((c) => ({ name: c.name, type: c.type })),
+                exact: true,
+                wait: true,
             });
+            const dvId = found[0]?.id ??
+                (await createDataView({
+                    ...base,
+                    name: t.name,
+                    datasourceId: options.dsId,
+                    table: t.name,
+                    fields: t.columns.map((c) => ({ name: c.name, type: c.type })),
+                }));
             viewMap[t.name] = dvId;
         }
         // Phase 2: Create the KN record
@@ -2266,18 +2296,24 @@ async function runKnCreateFromDsCommand(args, sampleRows) {
         if (options.build) {
             console.error("Building ...");
             await buildKnowledgeNetwork({ ...base, knId });
-            const deadline = Date.now() + options.timeout * 1000;
             const TERMINAL = ["completed", "failed", "success"];
-            while (Date.now() < deadline) {
-                await new Promise((r) => setTimeout(r, 2000));
-                const statusBody = await getBuildStatus({ ...base, knId });
-                const statusParsed = JSON.parse(statusBody);
-                const jobs = Array.isArray(statusParsed) ? statusParsed : (statusParsed.entries ?? []);
-                const state = (jobs[0]?.state ?? "running").toLowerCase();
-                if (TERMINAL.includes(state)) {
-                    statusStr = state;
-                    break;
-                }
+            try {
+                statusStr = await pollWithBackoff({
+                    fn: async () => {
+                        const statusBody = await getBuildStatus({ ...base, knId });
+                        const statusParsed = JSON.parse(statusBody);
+                        const jobs = Array.isArray(statusParsed) ? statusParsed : (statusParsed.entries ?? []);
+                        const state = (jobs[0]?.state ?? "running").toLowerCase();
+                        if (TERMINAL.includes(state))
+                            return { done: true, value: state };
+                        return { done: false, value: "running" };
+                    },
+                    interval: 2000,
+                    timeout: options.timeout * 1000,
+                });
+            }
+            catch {
+                // timeout — statusStr remains "skipped"
             }
         }
         const output = {
@@ -2448,30 +2484,37 @@ async function runKnBuildCommand(args) {
             return 0;
         }
         console.error("Waiting for build to complete ...");
-        const deadline = Date.now() + options.timeout * 1000;
-        while (Date.now() < deadline) {
-            await new Promise((r) => setTimeout(r, 2000));
-            const body = await getBuildStatus({
-                baseUrl: token.baseUrl,
-                accessToken: token.accessToken,
-                knId: options.knId,
-                businessDomain: options.businessDomain,
+        try {
+            const { state, detail } = await pollWithBackoff({
+                fn: async () => {
+                    const body = await getBuildStatus({
+                        baseUrl: token.baseUrl,
+                        accessToken: token.accessToken,
+                        knId: options.knId,
+                        businessDomain: options.businessDomain,
+                    });
+                    const parsed = JSON.parse(body);
+                    const jobs = Array.isArray(parsed) ? parsed : (parsed.entries ?? parsed.data ?? []);
+                    const job = jobs[0];
+                    const st = (job?.state ?? "running").toLowerCase();
+                    const dt = job?.state_detail;
+                    if (TERMINAL_STATES.includes(st))
+                        return { done: true, value: { state: st, detail: dt } };
+                    return { done: false, value: { state: st } };
+                },
+                interval: 2000,
+                timeout: options.timeout * 1000,
             });
-            const parsed = JSON.parse(body);
-            const jobs = Array.isArray(parsed) ? parsed : (parsed.entries ?? parsed.data ?? []);
-            const job = jobs[0];
-            const state = (job?.state ?? "running").toLowerCase();
-            const detail = job?.state_detail;
-            if (TERMINAL_STATES.includes(state)) {
-                console.log(state);
-                if (detail) {
-                    console.log(`Detail: ${detail}`);
-                }
-                return state === "failed" ? 1 : 0;
+            console.log(state);
+            if (detail) {
+                console.log(`Detail: ${detail}`);
             }
+            return state === "failed" ? 1 : 0;
+        }
+        catch {
+            console.error(`Build did not complete within ${options.timeout}s`);
+            return 1;
         }
-        console.error(`Build did not complete within ${options.timeout}s`);
-        return 1;
     }
     catch (error) {
         console.error(formatHttpError(error));

package/dist/commands/dataview.d.ts

@@ -0,0 +1 @@
+export declare function runDataviewCommand(args: string[]): Promise<number>;