@kweaver-ai/kweaver-sdk 0.4.1 → 0.4.2

package/dist/cli.js

@@ -5,6 +5,7 @@ import { runCallCommand } from "./commands/call.js";
 import { runContextLoaderCommand } from "./commands/context-loader.js";
 import { runDsCommand } from "./commands/ds.js";
 import { runTokenCommand } from "./commands/token.js";
+import { runVegaCommand } from "./commands/vega.js";
 function printHelp() {
     console.log(`kweaver
 
@@ -31,6 +32,7 @@ Usage:
   kweaver bkn create [options]
   kweaver bkn update <kn-id> [options]
   kweaver bkn delete <kn-id>
+  kweaver vega [health|stats|inspect|catalog|resource|connector-type]
   kweaver context-loader [config|kn-search|...]
   kweaver --help
 
@@ -41,6 +43,7 @@ Commands:
   ds             Manage datasources (list, get, delete, tables, connect)
   agent          Chat with a KWeaver agent (agent chat <id>), list published agents (agent list)
   bkn           Business knowledge network (list/get/create/update/delete/export/stats; object-type, subgraph, action-type, action-log)
+  vega           Vega observability platform (catalogs, resources, connector-types, health)
   context-loader Call context-loader MCP (tools, resources, prompts; kn-search, query-*, etc.)
   help           Show this message`);
 }
@@ -68,6 +71,9 @@ export async function run(argv) {
     if (command === "bkn") {
         return runKnCommand(rest);
     }
+    if (command === "vega") {
+        return runVegaCommand(rest);
+    }
     if (command === "context-loader" || command === "context") {
         return runContextLoaderCommand(rest);
     }

package/dist/client.d.ts

@@ -31,6 +31,12 @@ export interface KWeaverClientOptions {
      * Override with KWEAVER_BUSINESS_DOMAIN env var or pass explicitly.
      */
     businessDomain?: string;
+    /**
+     * When true, read credentials exclusively from ~/.kweaver/ (saved by
+     * `kweaver auth login`), ignoring KWEAVER_BASE_URL / KWEAVER_TOKEN env vars.
+     * Useful when env vars hold stale tokens or are intended for other tooling.
+     */
+    config?: boolean;
 }
 /**
  * Main entry point for the KWeaver TypeScript SDK.
@@ -73,6 +79,19 @@ export declare class KWeaverClient implements ClientContext {
     /** Conversation and message history. */
     readonly conversations: ConversationsResource;
     constructor(opts?: KWeaverClientOptions);
+    /**
+     * Async factory that auto-refreshes expired or revoked tokens.
+     *
+     * Reads credentials from `~/.kweaver/` and refreshes the access token
+     * if it has expired or been revoked (using the saved refresh token).
+     * If the initial token fails with 401, forces a refresh and retries.
+     *
+     * @example
+     * ```typescript
+     * const client = await KWeaverClient.connect();
+     * ```
+     */
+    static connect(opts?: KWeaverClientOptions): Promise<KWeaverClient>;
     /** @internal — used by resource classes to build API call options. */
     base(): {
         baseUrl: string;

package/dist/client.js

@@ -1,4 +1,5 @@
 import { getCurrentPlatform, loadTokenConfig, } from "./config/store.js";
+import { ensureValidToken } from "./auth/oauth.js";
 import { AgentsResource } from "./resources/agents.js";
 import { ConversationsResource } from "./resources/conversations.js";
 import { ContextLoaderResource } from "./resources/context-loader.js";
@@ -46,20 +47,37 @@ export class KWeaverClient {
     /** Conversation and message history. */
     conversations;
     constructor(opts = {}) {
-        const envUrl = process.env.KWEAVER_BASE_URL;
-        const envToken = process.env.KWEAVER_TOKEN;
         const envDomain = process.env.KWEAVER_BUSINESS_DOMAIN;
-        // Resolve baseUrl: explicit > env > saved config
-        let baseUrl = opts.baseUrl ?? envUrl;
-        let accessToken = opts.accessToken ?? envToken;
-        if (!baseUrl || !accessToken) {
+        let baseUrl;
+        let accessToken;
+        if (opts.config) {
+            // config: true — read exclusively from ~/.kweaver/, ignore env vars
             const platform = getCurrentPlatform();
-            if (platform) {
-                const stored = loadTokenConfig(platform);
-                if (!baseUrl)
-                    baseUrl = platform;
-                if (!accessToken && stored)
-                    accessToken = stored.accessToken;
+            if (!platform) {
+                throw new Error("No active platform. Run `kweaver auth login` first.");
+            }
+            const stored = loadTokenConfig(platform);
+            if (!stored?.accessToken) {
+                throw new Error(`No token for ${platform}. Run \`kweaver auth login\` first.`);
+            }
+            baseUrl = opts.baseUrl ?? platform;
+            accessToken = opts.accessToken ?? stored.accessToken;
+        }
+        else {
+            // Default: explicit > env > saved config
+            const envUrl = process.env.KWEAVER_BASE_URL;
+            const envToken = process.env.KWEAVER_TOKEN;
+            baseUrl = opts.baseUrl ?? envUrl;
+            accessToken = opts.accessToken ?? envToken;
+            if (!baseUrl || !accessToken) {
+                const platform = getCurrentPlatform();
+                if (platform) {
+                    const stored = loadTokenConfig(platform);
+                    if (!baseUrl)
+                        baseUrl = platform;
+                    if (!accessToken && stored)
+                        accessToken = stored.accessToken;
+                }
             }
         }
         if (!baseUrl) {
@@ -71,13 +89,58 @@ export class KWeaverClient {
                 "Pass it explicitly, set KWEAVER_TOKEN, or run `kweaver auth login`.");
         }
         this._baseUrl = baseUrl.replace(/\/+$/, "");
-        this._accessToken = accessToken;
+        // Strip "Bearer " prefix if present — callers (env vars, config files) may
+        // include it, but API helpers always add their own "Bearer " prefix.
+        this._accessToken = accessToken.replace(/^Bearer\s+/i, "");
         this._businessDomain = opts.businessDomain ?? envDomain ?? "bd_public";
         this.knowledgeNetworks = new KnowledgeNetworksResource(this);
         this.agents = new AgentsResource(this);
         this.bkn = new BknResource(this);
         this.conversations = new ConversationsResource(this);
     }
+    /**
+     * Async factory that auto-refreshes expired or revoked tokens.
+     *
+     * Reads credentials from `~/.kweaver/` and refreshes the access token
+     * if it has expired or been revoked (using the saved refresh token).
+     * If the initial token fails with 401, forces a refresh and retries.
+     *
+     * @example
+     * ```typescript
+     * const client = await KWeaverClient.connect();
+     * ```
+     */
+    static async connect(opts = {}) {
+        // Try with current token first
+        let token = await ensureValidToken();
+        const client = new KWeaverClient({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            ...opts,
+        });
+        // Quick probe — if the token was revoked server-side, force refresh
+        try {
+            const probe = await fetch(`${token.baseUrl.replace(/\/+$/, "")}/api/ontology-manager/v1/knowledge-networks?limit=1`, { headers: { authorization: `Bearer ${token.accessToken}`, token: token.accessToken } });
+            if (probe.status === 401) {
+                try {
+                    token = await ensureValidToken({ forceRefresh: true });
+                }
+                catch {
+                    throw new Error("Access token revoked and refresh token also expired. " +
+                        "Run `kweaver auth login` to re-authenticate.");
+                }
+                return new KWeaverClient({
+                    baseUrl: token.baseUrl,
+                    accessToken: token.accessToken,
+                    ...opts,
+                });
+            }
+        }
+        catch {
+            // Network error — return client as-is, let the caller deal with it
+        }
+        return client;
+    }
     /** @internal — used by resource classes to build API call options. */
     base() {
         return {

package/dist/commands/agent.js

@@ -1,6 +1,6 @@
 import { ensureValidToken, formatHttpError } from "../auth/oauth.js";
 import { runAgentChatCommand } from "./agent-chat.js";
-import { listAgents, getAgent } from "../api/agent-list.js";
+import { listAgents, getAgent, getAgentByKey, createAgent, updateAgent, deleteAgent, publishAgent, unpublishAgent, } from "../api/agent-list.js";
 import { listConversations, listMessages } from "../api/conversations.js";
 import { formatCallOutput } from "./call.js";
 function readStringField(value, ...keys) {
@@ -211,22 +211,20 @@ export function runAgentCommand(args) {
         console.log(`kweaver agent
 
 Subcommands:
+  list [options]                     List published agents
+  get <agent_id> [--verbose]         Get agent details
+  get-by-key <key>                   Get agent by key
+  create --name <n> --profile <p>    Create a new agent
+       [--key <key>] [--product-key <pk>] [--system-prompt <sp>]
+       [--llm-id <id>] [--llm-max-tokens <n>]
+  update <agent_id> --name <n> ...   Update an existing agent
+  delete <agent_id> [-y]             Delete an agent
+  publish <agent_id>                 Publish an agent
+  unpublish <agent_id>               Unpublish an agent
   chat <agent_id>                    Start interactive chat with an agent
   chat <agent_id> -m "message"       Send a single message (non-interactive)
-       [--conversation-id id]         Continue an existing conversation
-       [-cid id]                      Short alias for --conversation-id
-       [--session-id id]             Alias for --conversation-id
-       [-conversation_id id]         Compatibility alias for reference examples
-       [--version value]             Resolve agent key from a specific version (default: v0)
-       [--stream] [--no-stream]      Enable or disable streaming (default: stream in interactive, no-stream in -m mode)
-       [--verbose]                   Print request details to stderr
-       [-bd|--biz-domain value]      Override x-business-domain (default: bd_public)
-  list [options]                    List published agents
-  get <agent_id> [--verbose]         Get agent details
   sessions <agent_id>                List all conversations for an agent
-       [--limit n] [-bd domain] [--pretty]
-  history <conversation_id>          Show message history for a conversation
-       [--limit n] [-bd domain] [--pretty]`);
+  history <conversation_id>          Show message history for a conversation`);
         return Promise.resolve(0);
     }
     if (subcommand === "chat") {
@@ -318,6 +316,24 @@ Options:
         }
         return runAgentHistoryCommand(rest);
     }
+    if (subcommand === "get-by-key") {
+        return runAgentGetByKeyCommand(rest);
+    }
+    if (subcommand === "create") {
+        return runAgentCreateCommand(rest);
+    }
+    if (subcommand === "update") {
+        return runAgentUpdateCommand(rest);
+    }
+    if (subcommand === "delete") {
+        return runAgentDeleteCommand(rest);
+    }
+    if (subcommand === "publish") {
+        return runAgentPublishCommand(rest);
+    }
+    if (subcommand === "unpublish") {
+        return runAgentUnpublishCommand(rest);
+    }
     console.error(`Unknown agent subcommand: ${subcommand}`);
     return Promise.resolve(1);
 }
@@ -530,3 +546,265 @@ Options:
         return 1;
     }
 }
+// ── Get by key ───────────────────────────────────────────────────────────────
+async function runAgentGetByKeyCommand(args) {
+    const key = args[0];
+    if (!key || key.startsWith("-")) {
+        console.error("Usage: kweaver agent get-by-key <key>");
+        return 1;
+    }
+    try {
+        const token = await ensureValidToken();
+        const body = await getAgentByKey({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            key,
+        });
+        console.log(formatCallOutput(body, true));
+        return 0;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
+}
+// ── Create ───────────────────────────────────────────────────────────────────
+async function runAgentCreateCommand(args) {
+    let name = "";
+    let profile = "";
+    let key = "";
+    let productKey = "DIP";
+    let systemPrompt = "";
+    let llmId = "";
+    let llmMaxTokens = 4096;
+    let businessDomain = "bd_public";
+    for (let i = 0; i < args.length; i += 1) {
+        const arg = args[i];
+        if (arg === "--help" || arg === "-h") {
+            console.log(`kweaver agent create --name <name> --profile <profile> [options]
+
+Create a new agent.
+
+Required:
+  --name <text>            Agent name (max 50)
+  --profile <text>         Agent description (max 500)
+
+Optional:
+  --key <text>             Agent unique key (auto-generated if omitted)
+  --product-key <text>     Product key: DIP, AnyShare, ChatBI (default: DIP)
+  --system-prompt <text>   System prompt
+  --llm-id <id>            LLM model ID (required for public API)
+  --llm-max-tokens <n>     LLM max tokens (default: 4096)
+  -bd, --biz-domain <val>  Business domain (default: bd_public)`);
+            return 0;
+        }
+        if (arg === "--name") {
+            name = args[++i] ?? "";
+            continue;
+        }
+        if (arg === "--profile") {
+            profile = args[++i] ?? "";
+            continue;
+        }
+        if (arg === "--key") {
+            key = args[++i] ?? "";
+            continue;
+        }
+        if (arg === "--product-key") {
+            productKey = args[++i] ?? "DIP";
+            continue;
+        }
+        if (arg === "--system-prompt") {
+            systemPrompt = args[++i] ?? "";
+            continue;
+        }
+        if (arg === "--llm-id") {
+            llmId = args[++i] ?? "";
+            continue;
+        }
+        if (arg === "--llm-max-tokens") {
+            llmMaxTokens = parseInt(args[++i] ?? "4096", 10);
+            continue;
+        }
+        if (arg === "-bd" || arg === "--biz-domain") {
+            businessDomain = args[++i] ?? "bd_public";
+            continue;
+        }
+    }
+    if (!name) {
+        console.error("--name is required");
+        return 1;
+    }
+    if (!profile) {
+        console.error("--profile is required");
+        return 1;
+    }
+    const config = {
+        input: { fields: [{ name: "user_input", type: "string", desc: "" }] },
+        output: { default_format: "markdown" },
+        system_prompt: systemPrompt,
+    };
+    if (llmId) {
+        config.llms = [{ is_default: true, llm_config: { id: llmId, name: llmId, max_tokens: llmMaxTokens } }];
+    }
+    const payload = {
+        name,
+        profile,
+        avatar_type: 1,
+        avatar: "icon-dip-agent-default",
+        product_key: productKey,
+        config,
+    };
+    if (key)
+        payload.key = key;
+    try {
+        const token = await ensureValidToken();
+        const body = await createAgent({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            businessDomain,
+            body: JSON.stringify(payload),
+        });
+        console.log(body);
+        return 0;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
+}
+// ── Update ───────────────────────────────────────────────────────────────────
+async function runAgentUpdateCommand(args) {
+    const agentId = args[0];
+    if (!agentId || agentId.startsWith("-")) {
+        console.error("Usage: kweaver agent update <agent_id> [--name <n>] [--profile <p>] [--system-prompt <sp>]");
+        return 1;
+    }
+    try {
+        const token = await ensureValidToken();
+        const currentRaw = await getAgent({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            agentId,
+        });
+        const current = JSON.parse(currentRaw);
+        for (let i = 1; i < args.length; i += 1) {
+            const arg = args[i];
+            if (arg === "--name") {
+                current.name = args[++i] ?? current.name;
+                continue;
+            }
+            if (arg === "--profile") {
+                current.profile = args[++i] ?? current.profile;
+                continue;
+            }
+            if (arg === "--system-prompt") {
+                const config = (current.config ?? {});
+                config.system_prompt = args[++i] ?? "";
+                current.config = config;
+                continue;
+            }
+        }
+        const body = await updateAgent({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            agentId,
+            body: JSON.stringify({
+                name: current.name,
+                profile: current.profile,
+                avatar_type: current.avatar_type,
+                avatar: current.avatar,
+                product_key: current.product_key,
+                config: current.config,
+            }),
+        });
+        if (body)
+            console.log(body);
+        else
+            console.log("Updated.");
+        return 0;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
+}
+// ── Delete ───────────────────────────────────────────────────────────────────
+async function runAgentDeleteCommand(args) {
+    const agentId = args[0];
+    if (!agentId || agentId.startsWith("-")) {
+        console.error("Usage: kweaver agent delete <agent_id> [-y]");
+        return 1;
+    }
+    const autoConfirm = args.includes("-y") || args.includes("--yes");
+    if (!autoConfirm) {
+        process.stdout.write(`Delete agent ${agentId}? [y/N] `);
+        const answer = await new Promise((resolve) => {
+            process.stdin.setEncoding("utf8");
+            process.stdin.once("data", (data) => resolve(String(data).trim().toLowerCase()));
+        });
+        if (answer !== "y" && answer !== "yes") {
+            console.log("Cancelled.");
+            return 0;
+        }
+    }
+    try {
+        const token = await ensureValidToken();
+        await deleteAgent({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            agentId,
+        });
+        console.log(`Deleted agent ${agentId}.`);
+        return 0;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
+}
+// ── Publish ──────────────────────────────────────────────────────────────────
+async function runAgentPublishCommand(args) {
+    const agentId = args[0];
+    if (!agentId || agentId.startsWith("-")) {
+        console.error("Usage: kweaver agent publish <agent_id>");
+        return 1;
+    }
+    try {
+        const token = await ensureValidToken();
+        const body = await publishAgent({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            agentId,
+            body: JSON.stringify({ agent_id: agentId }),
+        });
+        console.log(body);
+        return 0;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
+}
+// ── Unpublish ────────────────────────────────────────────────────────────────
+async function runAgentUnpublishCommand(args) {
+    const agentId = args[0];
+    if (!agentId || agentId.startsWith("-")) {
+        console.error("Usage: kweaver agent unpublish <agent_id>");
+        return 1;
+    }
+    try {
+        const token = await ensureValidToken();
+        await unpublishAgent({
+            baseUrl: token.baseUrl,
+            accessToken: token.accessToken,
+            agentId,
+        });
+        console.log(`Unpublished agent ${agentId}.`);
+        return 0;
+    }
+    catch (error) {
+        console.error(formatHttpError(error));
+        return 1;
+    }
+}

package/dist/commands/bkn.d.ts

@@ -41,6 +41,20 @@ export interface KnDeleteOptions {
     yes: boolean;
 }
 export declare function parseKnDeleteArgs(args: string[]): KnDeleteOptions;
+export interface KnPushOptions {
+    directory: string;
+    branch: string;
+    businessDomain: string;
+    pretty: boolean;
+}
+export declare function parseKnPushArgs(args: string[]): KnPushOptions;
+export interface KnPullOptions {
+    knId: string;
+    directory: string;
+    branch: string;
+    businessDomain: string;
+}
+export declare function parseKnPullArgs(args: string[]): KnPullOptions;
 export interface KnObjectTypeQueryOptions {
     knId: string;
     otId: string;
@@ -66,6 +80,8 @@ export declare function parseKnBuildArgs(args: string[]): {
     timeout: number;
     businessDomain: string;
 };
+export declare function packDirectoryToTar(dirPath: string): Buffer;
+export declare function extractTarToDirectory(tarBuffer: Buffer, dirPath: string): void;
 export declare function parseKnSearchArgs(args: string[]): {
     knId: string;
     query: string;