@kweaver-ai/kweaver-sdk 0.4.0 → 0.4.2

package/dist/resources/agents.d.ts

@@ -1,5 +1,76 @@
 import type { ChatResult, SendChatRequestStreamCallbacks } from "../api/agent-chat.js";
 import type { ClientContext } from "../client.js";
+export interface AgentLlmConfig {
+    id?: string;
+    name: string;
+    model_type?: string;
+    temperature?: number;
+    top_p?: number;
+    top_k?: number;
+    frequency_penalty?: number;
+    presence_penalty?: number;
+    max_tokens: number;
+}
+export interface AgentLlmItem {
+    is_default: boolean;
+    llm_config: AgentLlmConfig;
+}
+export interface AgentInputField {
+    name: string;
+    type?: string;
+    desc?: string;
+}
+export interface AgentInput {
+    fields: AgentInputField[];
+    rewrite?: Record<string, unknown>;
+    augment?: Record<string, unknown>;
+}
+export interface AgentOutput {
+    variables?: Record<string, unknown>;
+    default_format?: string;
+}
+export interface AgentConfig {
+    input: AgentInput;
+    output: AgentOutput;
+    system_prompt?: string;
+    dolphin?: string;
+    is_dolphin_mode?: number;
+    data_source?: Record<string, unknown>;
+    skills?: Record<string, unknown>;
+    llms?: AgentLlmItem[];
+    opening_remark_config?: Record<string, unknown>;
+    preset_questions?: Array<{
+        question: string;
+    }>;
+    memory?: {
+        is_enabled: boolean;
+    };
+    related_question?: {
+        is_enabled: boolean;
+    };
+    plan_mode?: {
+        is_enabled: boolean;
+    };
+    conversation_history_config?: Record<string, unknown>;
+    [key: string]: unknown;
+}
+export interface CreateAgentBody {
+    name: string;
+    profile: string;
+    avatar_type?: number;
+    avatar?: string;
+    product_key?: string;
+    key?: string;
+    config: AgentConfig;
+}
+export interface UpdateAgentBody {
+    name: string;
+    profile: string;
+    avatar_type: number;
+    avatar: string;
+    product_key: string;
+    config: AgentConfig;
+}
 export declare class AgentsResource {
     private readonly ctx;
     constructor(ctx: ClientContext);
@@ -9,26 +80,29 @@ export declare class AgentsResource {
         offset?: number;
         limit?: number;
     }): Promise<unknown[]>;
-    /** Resolve agent key and version for a given agent id. */
+    get(agentId: string): Promise<unknown>;
+    getByKey(key: string): Promise<unknown>;
+    create(body: CreateAgentBody): Promise<{
+        id: string;
+        version: string;
+    }>;
+    update(agentId: string, body: UpdateAgentBody): Promise<void>;
+    delete(agentId: string): Promise<void>;
+    publish(agentId: string, opts?: {
+        business_domain_id?: string;
+    }): Promise<unknown>;
+    unpublish(agentId: string): Promise<void>;
     info(agentId: string, version?: string): Promise<{
         id: string;
         key: string;
         version: string;
     }>;
-    /**
-     * Send a single message and return the full response.
-     * Automatically resolves the agent key/version before sending.
-     */
     chat(agentId: string, message: string, opts?: {
         conversationId?: string;
         version?: string;
         stream?: boolean;
         verbose?: boolean;
     }): Promise<ChatResult>;
-    /**
-     * Send a message with streaming callbacks.
-     * Automatically resolves the agent key/version before sending.
-     */
     stream(agentId: string, message: string, callbacks: SendChatRequestStreamCallbacks, opts?: {
         conversationId?: string;
         version?: string;

package/dist/resources/agents.js

@@ -1,10 +1,12 @@
-import { listAgents } from "../api/agent-list.js";
+import { listAgents, getAgent, getAgentByKey, createAgent, updateAgent, deleteAgent, publishAgent, unpublishAgent, } from "../api/agent-list.js";
 import { fetchAgentInfo, sendChatRequest, sendChatRequestStream, } from "../api/agent-chat.js";
+// ── AgentsResource ───────────────────────────────────────────────────────────
 export class AgentsResource {
     ctx;
     constructor(ctx) {
         this.ctx = ctx;
     }
+    // ── List (published agents) ──────────────────────────────────────────────
     async list(opts = {}) {
         const { keyword, ...rest } = opts;
         const raw = await listAgents({ ...this.ctx.base(), name: keyword, ...rest });
@@ -16,15 +18,52 @@ export class AgentsResource {
                 : [];
         return items;
     }
-    /** Resolve agent key and version for a given agent id. */
+    // ── Get by ID ────────────────────────────────────────────────────────────
+    async get(agentId) {
+        const raw = await getAgent({ ...this.ctx.base(), agentId });
+        return JSON.parse(raw);
+    }
+    // ── Get by key ───────────────────────────────────────────────────────────
+    async getByKey(key) {
+        const raw = await getAgentByKey({ ...this.ctx.base(), key });
+        return JSON.parse(raw);
+    }
+    // ── Create ───────────────────────────────────────────────────────────────
+    async create(body) {
+        // Apply defaults for required fields the user may omit
+        const payload = {
+            avatar_type: body.avatar_type ?? 1,
+            avatar: body.avatar ?? "icon-dip-agent-default",
+            product_key: body.product_key ?? "DIP",
+            ...body,
+        };
+        const raw = await createAgent({ ...this.ctx.base(), body: JSON.stringify(payload) });
+        return JSON.parse(raw);
+    }
+    // ── Update ───────────────────────────────────────────────────────────────
+    async update(agentId, body) {
+        await updateAgent({ ...this.ctx.base(), agentId, body: JSON.stringify(body) });
+    }
+    // ── Delete ───────────────────────────────────────────────────────────────
+    async delete(agentId) {
+        await deleteAgent({ ...this.ctx.base(), agentId });
+    }
+    // ── Publish ──────────────────────────────────────────────────────────────
+    async publish(agentId, opts = {}) {
+        const body = JSON.stringify({ agent_id: agentId, ...opts });
+        const raw = await publishAgent({ ...this.ctx.base(), agentId, body });
+        return JSON.parse(raw);
+    }
+    // ── Unpublish ────────────────────────────────────────────────────────────
+    async unpublish(agentId) {
+        await unpublishAgent({ ...this.ctx.base(), agentId });
+    }
+    // ── Agent info (resolve key/version) ─────────────────────────────────────
     async info(agentId, version = "v0") {
         const info = await fetchAgentInfo({ ...this.ctx.base(), agentId, version });
         return info;
     }
-    /**
-     * Send a single message and return the full response.
-     * Automatically resolves the agent key/version before sending.
-     */
+    // ── Chat (single-shot) ──────────────────────────────────────────────────
     async chat(agentId, message, opts = {}) {
         const { version = "v0", stream = false, conversationId, verbose } = opts;
         const info = await fetchAgentInfo({ ...this.ctx.base(), agentId, version });
@@ -39,10 +78,7 @@ export class AgentsResource {
             verbose,
         });
     }
-    /**
-     * Send a message with streaming callbacks.
-     * Automatically resolves the agent key/version before sending.
-     */
+    // ── Stream ──────────────────────────────────────────────────────────────
     async stream(agentId, message, callbacks, opts = {}) {
         const { version = "v0", conversationId, verbose } = opts;
         const info = await fetchAgentInfo({ ...this.ctx.base(), agentId, version });

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) {
@@ -11,9 +8,8 @@ export class KnowledgeNetworksResource {
     async list(opts = {}) {
         const raw = await listKnowledgeNetworks({ ...this.ctx.base(), ...opts });
         const parsed = JSON.parse(raw);
-        const data = parsed && typeof parsed === "object" && "data" in parsed
-            ? parsed.data
-            : parsed;
+        // API returns { entries: [...] }
+        const data = parsed?.entries ?? parsed?.data ?? parsed;
         return Array.isArray(data) ? data : [];
     }
     async get(knId, opts = {}) {
@@ -77,29 +73,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 +87,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.2",
   "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": [
@@ -53,6 +54,7 @@
     "typescript": "^5.9.3"
   },
   "dependencies": {
+    "@kweaver-ai/bkn": "^0.1.0",
     "ink": "^6.8.0",
     "ink-spinner": "^5.0.0",
     "ink-text-input": "^6.0.0",