@kweaver-ai/kweaver-sdk 0.4.0 → 0.4.1

package/dist/commands/context-loader.js

@@ -340,7 +340,7 @@ function parseJsonArg(args) {
 async function runQueryObjectInstance(options, args, pretty) {
     const body = parseJsonArg(args);
     if (!body.ot_id || !body.condition) {
-        console.error("JSON must include ot_id and condition. See ref/contextloader/examples.md");
+        console.error("JSON must include ot_id and condition. See references/json-formats.md#context-loader");
         return 1;
     }
     const result = await queryObjectInstance(options, {
@@ -354,7 +354,7 @@ async function runQueryObjectInstance(options, args, pretty) {
 async function runQueryInstanceSubgraph(options, args, pretty) {
     const body = parseJsonArg(args);
     if (!Array.isArray(body.relation_type_paths)) {
-        console.error("JSON must include relation_type_paths array. See ref/contextloader/examples.md");
+        console.error("JSON must include relation_type_paths array. See references/json-formats.md#context-loader");
         return 1;
     }
     const result = await queryInstanceSubgraph(options, body);
@@ -364,7 +364,7 @@ async function runQueryInstanceSubgraph(options, args, pretty) {
 async function runGetLogicProperties(options, args, pretty) {
     const body = parseJsonArg(args);
     if (!body.ot_id || !body.query || !body._instance_identities || !body.properties) {
-        console.error("JSON must include ot_id, query, _instance_identities, properties. See ref/contextloader/examples.md");
+        console.error("JSON must include ot_id, query, _instance_identities, properties. See references/json-formats.md#context-loader");
         return 1;
     }
     const result = await getLogicPropertiesValues(options, body);
@@ -374,7 +374,7 @@ async function runGetLogicProperties(options, args, pretty) {
 async function runGetActionInfo(options, args, pretty) {
     const body = parseJsonArg(args);
     if (!body.at_id || !body._instance_identity) {
-        console.error("JSON must include at_id and _instance_identity. See ref/contextloader/examples.md");
+        console.error("JSON must include at_id and _instance_identity. See references/json-formats.md#context-loader");
         return 1;
     }
     const result = await getActionInfo(options, body);

package/dist/commands/ds.d.ts

@@ -0,0 +1,7 @@
+export declare function runDsCommand(args: string[]): Promise<number>;
+export declare function parseDsListArgs(args: string[]): {
+    keyword?: string;
+    type?: string;
+    businessDomain: string;
+    pretty: boolean;
+};

package/dist/commands/ds.js

@@ -0,0 +1,283 @@
+import { createInterface } from "node:readline";
+import { ensureValidToken, formatHttpError } from "../auth/oauth.js";
+import { testDatasource, createDatasource, listDatasources, getDatasource, deleteDatasource, listTablesWithColumns, } from "../api/datasources.js";
+import { formatCallOutput } from "./call.js";
+function confirmYes(prompt) {
+    return new Promise((resolve) => {
+        const rl = createInterface({ input: process.stdin, output: process.stdout });
+        rl.question(`${prompt} [y/N] `, (answer) => {
+            rl.close();
+            const trimmed = answer.trim().toLowerCase();
+            resolve(trimmed === "y" || trimmed === "yes");
+        });
+    });
+}
+function extractDatasourceId(body) {
+    const parsed = JSON.parse(body);
+    const item = Array.isArray(parsed) ? parsed[0] : parsed;
+    if (!item || typeof item !== "object")
+        return "";
+    const id = item.id ?? item.ds_id;
+    return id != null ? String(id) : "";
+}
+export async function runDsCommand(args) {
+    const [subcommand, ...rest] = args;
+    if (!subcommand || subcommand === "--help" || subcommand === "-h") {
+        console.log(`kweaver ds
+
+Subcommands:
+  list [--keyword X] [--type Y]     List datasources
+  get <id>                          Get datasource details
+  delete <id> [-y]                  Delete a datasource
+  tables <id> [--keyword X]         List tables with columns
+  connect <db_type> <host> <port> <database> --account X --password Y [--schema Z] [--name N]
+    Test connectivity, register datasource, and discover tables.`);
+        return 0;
+    }
+    try {
+        if (subcommand === "list") {
+            return runDsListCommand(rest);
+        }
+        if (subcommand === "get") {
+            return runDsGetCommand(rest);
+        }
+        if (subcommand === "delete") {
+            return runDsDeleteCommand(rest);
+        }
+        if (subcommand === "tables") {
+            return runDsTablesCommand(rest);
+        }
+        if (subcommand === "connect") {
+            return runDsConnectCommand(rest);
+        }
+        console.error(`Unknown ds subcommand: ${subcommand}`);
+        return 1;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
+}
+export function parseDsListArgs(args) {
+    let keyword;
+    let type;
+    let businessDomain = "bd_public";
+    let pretty = true;
+    for (let i = 0; i < args.length; i += 1) {
+        const arg = args[i];
+        if (arg === "--help" || arg === "-h")
+            throw new Error("help");
+        if (arg === "--keyword" && args[i + 1]) {
+            keyword = args[++i];
+            continue;
+        }
+        if (arg === "--type" && args[i + 1]) {
+            type = args[++i];
+            continue;
+        }
+        if ((arg === "-bd" || arg === "--biz-domain") && args[i + 1]) {
+            businessDomain = args[++i];
+            continue;
+        }
+        if (arg === "--pretty") {
+            pretty = true;
+            continue;
+        }
+    }
+    return { keyword, type, businessDomain, pretty };
+}
+async function runDsListCommand(args) {
+    try {
+        const opts = parseDsListArgs(args);
+        const token = await ensureValidToken();
+        const body = await listDatasources({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            keyword: opts.keyword,
+            type: opts.type,
+            businessDomain: opts.businessDomain,
+        });
+        console.log(formatCallOutput(body, opts.pretty));
+        return 0;
+    }
+    catch (error) {
+        if (error instanceof Error && error.message === "help") {
+            console.log(`kweaver ds list [options]
+
+Options:
+  --keyword <s>   Filter by keyword
+  --type <s>      Filter by database type
+  -bd, --biz-domain  Business domain (default: bd_public)
+  --pretty         Pretty-print JSON (default)`);
+            return 0;
+        }
+        throw error;
+    }
+}
+async function runDsGetCommand(args) {
+    const id = args.find((a) => !a.startsWith("-"));
+    if (!id) {
+        console.error("Usage: kweaver ds get <id>");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await getDatasource({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+    });
+    console.log(formatCallOutput(body, true));
+    return 0;
+}
+async function runDsDeleteCommand(args) {
+    let id = "";
+    let yes = false;
+    for (let i = 0; i < args.length; i += 1) {
+        const arg = args[i];
+        if (arg === "--yes" || arg === "-y")
+            yes = true;
+        else if (!arg.startsWith("-"))
+            id = arg;
+    }
+    if (!id) {
+        console.error("Usage: kweaver ds delete <id> [-y]");
+        return 1;
+    }
+    if (!yes) {
+        const confirmed = await confirmYes("Are you sure you want to delete this datasource?");
+        if (!confirmed) {
+            console.error("Aborted.");
+            return 1;
+        }
+    }
+    const token = await ensureValidToken();
+    await deleteDatasource({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+    });
+    console.error(`Deleted ${id}`);
+    return 0;
+}
+async function runDsTablesCommand(args) {
+    let id = "";
+    let keyword;
+    let pretty = true;
+    for (let i = 0; i < args.length; i += 1) {
+        const arg = args[i];
+        if (arg === "--keyword" && args[i + 1]) {
+            keyword = args[++i];
+            continue;
+        }
+        if (arg === "--pretty") {
+            pretty = true;
+            continue;
+        }
+        if (!arg.startsWith("-"))
+            id = arg;
+    }
+    if (!id) {
+        console.error("Usage: kweaver ds tables <id> [--keyword X]");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const body = await listTablesWithColumns({
+        baseUrl: token.baseUrl,
+        accessToken: token.accessToken,
+        id,
+        keyword,
+    });
+    console.log(formatCallOutput(body, pretty));
+    return 0;
+}
+async function runDsConnectCommand(args) {
+    let dbType = "";
+    let host = "";
+    let port = 0;
+    let database = "";
+    let account = "";
+    let password = "";
+    let schema;
+    let name;
+    for (let i = 0; i < args.length; i += 1) {
+        const arg = args[i];
+        if (arg === "--account" && args[i + 1]) {
+            account = args[++i];
+            continue;
+        }
+        if (arg === "--password" && args[i + 1]) {
+            password = args[++i];
+            continue;
+        }
+        if (arg === "--schema" && args[i + 1]) {
+            schema = args[++i];
+            continue;
+        }
+        if (arg === "--name" && args[i + 1]) {
+            name = args[++i];
+            continue;
+        }
+        if (!arg.startsWith("-")) {
+            if (!dbType)
+                dbType = arg;
+            else if (!host)
+                host = arg;
+            else if (port === 0)
+                port = parseInt(arg, 10);
+            else if (!database)
+                database = arg;
+        }
+    }
+    if (!dbType || !host || !database || !account || !password) {
+        console.error("Usage: kweaver ds connect <db_type> <host> <port> <database> --account X --password Y [--schema Z] [--name N]");
+        return 1;
+    }
+    if (Number.isNaN(port) || port < 1) {
+        console.error("Invalid port");
+        return 1;
+    }
+    const token = await ensureValidToken();
+    const base = { baseUrl: token.baseUrl, accessToken: token.accessToken };
+    console.error("Testing connectivity ...");
+    await testDatasource({
+        ...base,
+        type: dbType,
+        host,
+        port,
+        database,
+        account,
+        password,
+        schema,
+    });
+    const dsName = name ?? database;
+    const createBody = await createDatasource({
+        ...base,
+        name: dsName,
+        type: dbType,
+        host,
+        port,
+        database,
+        account,
+        password,
+        schema,
+    });
+    const dsId = extractDatasourceId(createBody);
+    if (!dsId) {
+        console.error("Failed to get datasource ID from create response");
+        return 1;
+    }
+    const tablesBody = await listTablesWithColumns({
+        ...base,
+        id: dsId,
+    });
+    const tables = JSON.parse(tablesBody);
+    const output = {
+        datasource_id: dsId,
+        tables: tables.map((t) => ({
+            name: t.name,
+            columns: t.columns.map((c) => ({ name: c.name, type: c.type, comment: c.comment })),
+        })),
+    };
+    console.log(JSON.stringify(output, null, 2));
+    return 0;
+}

package/dist/resources/bkn.d.ts

@@ -42,4 +42,16 @@ export declare class BknResource {
     }): Promise<unknown[]>;
     getActionLog(knId: string, logId: string): Promise<unknown>;
     cancelActionLog(knId: string, logId: string): Promise<unknown>;
+    /**
+     * Search KN schema — finds matching object types, relation types, and action types.
+     * Uses MCP protocol via the context-loader (public endpoint).
+     */
+    knSearch(knId: string, query: string, opts?: {
+        onlySchema?: boolean;
+    }): Promise<{
+        object_types?: unknown[];
+        relation_types?: unknown[];
+        action_types?: unknown[];
+        nodes?: unknown[];
+    }>;
 }

package/dist/resources/bkn.js

@@ -83,4 +83,16 @@ export class BknResource {
         const raw = await actionLogCancel({ ...this.ctx.base(), knId, logId });
         return JSON.parse(raw);
     }
+    /**
+     * Search KN schema — finds matching object types, relation types, and action types.
+     * Uses MCP protocol via the context-loader (public endpoint).
+     */
+    async knSearch(knId, query, opts = {}) {
+        const { ContextLoaderResource } = await import("./context-loader.js");
+        const { baseUrl } = this.ctx.base();
+        const mcpUrl = `${baseUrl}/api/agent-retrieval/v1/mcp`;
+        const cl = new ContextLoaderResource(this.ctx, mcpUrl, knId);
+        const result = await cl.search({ query, only_schema: opts.onlySchema ?? false });
+        return result;
+    }
 }

package/dist/resources/knowledge-networks.js

@@ -1,8 +1,5 @@
 import { listKnowledgeNetworks, getKnowledgeNetwork, createKnowledgeNetwork, updateKnowledgeNetwork, deleteKnowledgeNetwork, listObjectTypes, listRelationTypes, listActionTypes, } from "../api/knowledge-networks.js";
 import { fetchTextOrThrow } from "../utils/http.js";
-function is404(err) {
-    return !!(err && typeof err === "object" && "status" in err && err.status === 404);
-}
 export class KnowledgeNetworksResource {
     ctx;
     constructor(ctx) {
@@ -77,29 +74,11 @@ export class KnowledgeNetworksResource {
             token: accessToken,
             "x-business-domain": businessDomain,
         };
-        try {
-            await fetchTextOrThrow(`${baseUrl}/api/agent-retrieval/in/v1/kn/full_build_ontology`, { method: "POST", headers, body: JSON.stringify({ kn_id: bknId }) });
-        }
-        catch (err) {
-            if (!is404(err))
-                throw err;
-            // Fallback: call ontology-manager jobs endpoint directly
-            try {
-                await fetchTextOrThrow(`${baseUrl}/api/ontology-manager/in/v1/knowledge-networks/${encodeURIComponent(bknId)}/jobs`, {
-                    method: "POST",
-                    headers,
-                    body: JSON.stringify({ name: `sdk_build_${bknId.slice(0, 8)}`, job_type: "full" }),
-                });
-            }
-            catch (err2) {
-                if (is404(err2)) {
-                    throw new Error(`No build endpoint available for BKN ${bknId}. ` +
-                        `Both agent-retrieval and ontology-manager returned 404. ` +
-                        `This deployment may not support index rebuilds.`);
-                }
-                throw err2;
-            }
-        }
+        await fetchTextOrThrow(`${baseUrl}/api/ontology-manager/v1/knowledge-networks/${encodeURIComponent(bknId)}/jobs`, {
+            method: "POST",
+            headers,
+            body: JSON.stringify({ name: `sdk_build_${bknId.slice(0, 8)}`, job_type: "full" }),
+        });
     }
     /** Poll build status for a BKN. */
     async buildStatus(bknId) {
@@ -109,36 +88,19 @@ export class KnowledgeNetworksResource {
             token: accessToken,
             "x-business-domain": businessDomain,
         };
-        try {
-            const { body } = await fetchTextOrThrow(`${baseUrl}/api/agent-retrieval/in/v1/kn/full_ontology_building_status?kn_id=${encodeURIComponent(bknId)}`, { headers });
-            const data = JSON.parse(body);
-            return { state: data.state ?? "running", state_detail: data.state_detail };
-        }
-        catch (err) {
-            if (!is404(err))
-                throw err;
-            // Fallback: check ontology-manager jobs for latest status
-            try {
-                const { body } = await fetchTextOrThrow(`${baseUrl}/api/ontology-manager/in/v1/knowledge-networks/${encodeURIComponent(bknId)}/jobs?limit=1&direction=desc`, { headers });
-                const data = JSON.parse(body);
-                const jobs = Array.isArray(data)
-                    ? data
-                    : data && typeof data === "object" && "entries" in data
-                        ? (data.entries ?? [])
-                        : data && typeof data === "object" && "data" in data
-                            ? (data.data ?? [])
-                            : [];
-                if (jobs.length > 0) {
-                    return { state: jobs[0].state ?? "running" };
-                }
-                return { state: "completed" };
-            }
-            catch (err2) {
-                if (is404(err2))
-                    return { state: "completed" };
-                throw err2;
-            }
+        const { body } = await fetchTextOrThrow(`${baseUrl}/api/ontology-manager/v1/knowledge-networks/${encodeURIComponent(bknId)}/jobs?limit=1&direction=desc`, { headers });
+        const data = JSON.parse(body);
+        const jobs = Array.isArray(data)
+            ? data
+            : data && typeof data === "object" && "entries" in data
+                ? (data.entries ?? [])
+                : data && typeof data === "object" && "data" in data
+                    ? (data.data ?? [])
+                    : [];
+        if (jobs.length > 0) {
+            return { state: jobs[0].state ?? "running", state_detail: jobs[0].state_detail };
         }
+        return { state: "completed" };
     }
     /**
      * Trigger a full BKN build and wait for it to complete.

package/dist/utils/crypto.d.ts

@@ -0,0 +1,10 @@
+/**
+ * RSA password encryption for the KWeaver data-connection API.
+ * The KWeaver backend requires datasource passwords to be RSA-encrypted
+ * (PKCS1v15) using a platform-wide public key before transmission.
+ */
+/**
+ * Encrypt a password with the KWeaver platform RSA public key.
+ * Returns a base64-encoded ciphertext string.
+ */
+export declare function encryptPassword(plaintext: string): string;

package/dist/utils/crypto.js

@@ -0,0 +1,31 @@
+/**
+ * RSA password encryption for the KWeaver data-connection API.
+ * The KWeaver backend requires datasource passwords to be RSA-encrypted
+ * (PKCS1v15) using a platform-wide public key before transmission.
+ */
+import { publicEncrypt, createPublicKey, constants } from "node:crypto";
+const KWEAVER_PUBLIC_KEY_PEM = `-----BEGIN PUBLIC KEY-----
+MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA22GOSQ1jeDhpdzxhJddS
+f+U10F4Ivut7giYhchFAIJgRonMamDT86MSqQUc8DdTFdPGLm7M3GUKcsG1qbC3S
+qk4XJ9NjmQXbs7IMWyWEWQrN7Iv7S2QjDYJI+ppvIN03I0Km3WKsmnrle2bLzT/V
+G8e72YX69dfXAeiX6uDhht1va/JxZVFMIV3pHa6AQQ9gn5SAUTX2akEhRfe1bPJj
+fVyoM+dfNtvgdfaraqV1rOhVDEqd0NlOWt2RHwETQwU8gIJib2baj2MtyIAY+fQw
+KlKWxUs1GcFbECnhVPiVN6BEhXD7OhRt9QE/cuYl5v4a6ypugGaMBK6VKOqFHDvf
+mwIDAQAB
+-----END PUBLIC KEY-----`;
+let cachedKey = null;
+function getPublicKey() {
+    if (!cachedKey) {
+        cachedKey = createPublicKey(KWEAVER_PUBLIC_KEY_PEM);
+    }
+    return cachedKey;
+}
+/**
+ * Encrypt a password with the KWeaver platform RSA public key.
+ * Returns a base64-encoded ciphertext string.
+ */
+export function encryptPassword(plaintext) {
+    const key = getPublicKey();
+    const ciphertext = publicEncrypt({ key, padding: constants.RSA_PKCS1_PADDING }, Buffer.from(plaintext, "utf8"));
+    return ciphertext.toString("base64");
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kweaver-ai/kweaver-sdk",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "KWeaver TypeScript SDK — CLI tool and programmatic API for knowledge networks and Decision Agents.",
   "type": "module",
   "main": "./dist/index.js",
@@ -27,7 +27,8 @@
     "build": "tsc -p tsconfig.json",
     "start": "node ./dist/cli.js",
     "lint": "tsc --noEmit -p tsconfig.json",
-    "test": "node --import tsx --test test/**/*.test.ts",
+    "test": "node --import tsx --test test/*.test.ts",
+    "test:e2e": "node --import tsx --test test/e2e/**/*.test.ts",
     "prepublishOnly": "npm run build"
   },
   "keywords": [