@kweaver-ai/kweaver-sdk 0.4.1 → 0.4.4

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,47 @@ 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) {
+                throw new Error("Access token revoked. Run `kweaver auth login` to re-authenticate.");
+            }
+        }
+        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-chat.js

@@ -1,5 +1,6 @@
 import { ensureValidToken, formatHttpError } from "../auth/oauth.js";
 import { fetchAgentInfo, sendChatRequest } from "../api/agent-chat.js";
+import { resolveBusinessDomain } from "../config/store.js";
 function formatCliArg(value) {
     if (/^[A-Za-z0-9_./:=+-]+$/.test(value)) {
         return value;
@@ -42,7 +43,7 @@ export function parseChatArgs(args) {
     let conversationId;
     let stream;
     let verbose = false;
-    let businessDomain = "bd_public";
+    let businessDomain = "";
     for (let i = 0; i < args.length; i += 1) {
         const arg = args[i];
         if (arg === "-m" || arg === "--message") {
@@ -104,6 +105,8 @@ export function parseChatArgs(args) {
     if (!agentId) {
         throw new Error("Missing agent_id. Usage: kweaver agent chat <agent_id> [-m \"message\"]");
     }
+    if (!businessDomain)
+        businessDomain = resolveBusinessDomain();
     return { agentId, version, message, conversationId, stream, verbose, businessDomain };
 }
 export async function runAgentChatCommand(args) {

package/dist/commands/agent.js

@@ -1,8 +1,10 @@
 import { ensureValidToken, formatHttpError } from "../auth/oauth.js";
+import { HttpError } from "../utils/http.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";
+import { resolveBusinessDomain } from "../config/store.js";
 function readStringField(value, ...keys) {
     for (const key of keys) {
         const candidate = value[key];
@@ -48,7 +50,7 @@ export function parseAgentListArgs(args) {
     let category_id = "";
     let custom_space_id = "";
     let is_to_square = 1;
-    let businessDomain = "bd_public";
+    let businessDomain = "";
     let pretty = true;
     let verbose = false;
     for (let i = 0; i < args.length; i += 1) {
@@ -113,6 +115,8 @@ export function parseAgentListArgs(args) {
         }
         throw new Error(`Unsupported agent list argument: ${arg}`);
     }
+    if (!businessDomain)
+        businessDomain = resolveBusinessDomain();
     return {
         name,
         offset,
@@ -130,7 +134,7 @@ export function parseAgentSessionsArgs(args) {
     if (!agentId || agentId.startsWith("-")) {
         throw new Error("Missing agent_id");
     }
-    let businessDomain = "bd_public";
+    let businessDomain = "";
     let limit;
     let pretty = true;
     for (let i = 1; i < args.length; i += 1) {
@@ -163,6 +167,8 @@ export function parseAgentSessionsArgs(args) {
         }
         throw new Error(`Unsupported agent sessions argument: ${arg}`);
     }
+    if (!businessDomain)
+        businessDomain = resolveBusinessDomain();
     return { agentId, businessDomain, limit, pretty };
 }
 export function parseAgentHistoryArgs(args) {
@@ -170,7 +176,7 @@ export function parseAgentHistoryArgs(args) {
     if (!conversationId || conversationId.startsWith("-")) {
         throw new Error("Missing conversation_id");
     }
-    let businessDomain = "bd_public";
+    let businessDomain = "";
     let limit;
     let pretty = true;
     for (let i = 1; i < args.length; i += 1) {
@@ -203,32 +209,58 @@ export function parseAgentHistoryArgs(args) {
         }
         throw new Error(`Unsupported agent history argument: ${arg}`);
     }
+    if (!businessDomain)
+        businessDomain = resolveBusinessDomain();
     return { conversationId, businessDomain, limit, pretty };
 }
-export function runAgentCommand(args) {
+export async function runAgentCommand(args) {
     const [subcommand, ...rest] = args;
     if (!subcommand || subcommand === "--help" || subcommand === "-h") {
         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);
     }
+    const dispatch = async () => {
+        if (subcommand === "chat")
+            return runAgentChatCommand(rest);
+        if (subcommand === "get")
+            return runAgentGetCommand(rest);
+        if (subcommand === "list")
+            return runAgentListCommand(rest);
+        if (subcommand === "sessions")
+            return runAgentSessionsCommand(rest);
+        if (subcommand === "history")
+            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);
+        return -1;
+    };
+    // Show subcommand-specific help inline (no retry needed)
     if (subcommand === "chat") {
         if (rest.length === 1 && (rest[0] === "--help" || rest[0] === "-h")) {
             console.log(`kweaver agent chat <agent_id> [-m "message"] [options]
@@ -266,9 +298,8 @@ Options:
   --verbose, -v             Show full JSON response
   -bd, --biz-domain <value>  Business domain (default: bd_public)
   --pretty                   Pretty-print JSON output (default)`);
-            return Promise.resolve(0);
+            return 0;
         }
-        return runAgentGetCommand(rest);
     }
     if (subcommand === "list") {
         if (rest.length === 1 && (rest[0] === "--help" || rest[0] === "-h")) {
@@ -286,9 +317,8 @@ Options:
   --verbose, -v             Show full JSON response
   -bd, --biz-domain <value>  Business domain (default: bd_public)
   --pretty                  Pretty-print JSON output (applies to both modes)`);
-            return Promise.resolve(0);
+            return 0;
         }
-        return runAgentListCommand(rest);
     }
     if (subcommand === "sessions") {
         if (rest.length === 1 && (rest[0] === "--help" || rest[0] === "-h")) {
@@ -300,9 +330,8 @@ Options:
   --limit <n>              Max conversations to return
   -bd, --biz-domain <value> Business domain (default: bd_public)
   --pretty                  Pretty-print JSON output (default)`);
-            return Promise.resolve(0);
+            return 0;
         }
-        return runAgentSessionsCommand(rest);
     }
     if (subcommand === "history") {
         if (rest.length === 1 && (rest[0] === "--help" || rest[0] === "-h")) {
@@ -314,19 +343,38 @@ Options:
   --limit <n>              Max messages to return
   -bd, --biz-domain <value> Business domain (default: bd_public)
   --pretty                  Pretty-print JSON output (default)`);
-            return Promise.resolve(0);
+            return 0;
+        }
+    }
+    try {
+        const code = await dispatch();
+        if (code === -1) {
+            console.error(`Unknown agent subcommand: ${subcommand}`);
+            return 1;
+        }
+        return code;
+    }
+    catch (error) {
+        if (error instanceof HttpError && error.status === 401) {
+            try {
+                await ensureValidToken({ forceRefresh: true });
+                return await dispatch();
+            }
+            catch (retryError) {
+                console.error(formatHttpError(retryError));
+                return 1;
+            }
         }
-        return runAgentHistoryCommand(rest);
+        console.error(formatHttpError(error));
+        return 1;
     }
-    console.error(`Unknown agent subcommand: ${subcommand}`);
-    return Promise.resolve(1);
 }
 export function parseAgentGetArgs(args) {
     const agentId = args[0];
     if (!agentId || agentId.startsWith("-")) {
         throw new Error("Missing agent_id. Usage: kweaver agent get <agent_id> [options]");
     }
-    let businessDomain = "bd_public";
+    let businessDomain = "";
     let pretty = true;
     let verbose = false;
     for (let i = 1; i < args.length; i += 1) {
@@ -352,6 +400,8 @@ export function parseAgentGetArgs(args) {
         }
         throw new Error(`Unsupported agent get argument: ${arg}`);
     }
+    if (!businessDomain)
+        businessDomain = resolveBusinessDomain();
     return { agentId, businessDomain, pretty, verbose };
 }
 function formatSimpleAgentGet(text, pretty) {
@@ -530,3 +580,267 @@ 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 = "";
+    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 (!businessDomain)
+        businessDomain = resolveBusinessDomain();
+    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/auth.d.ts

@@ -1,9 +1 @@
-import type { CallbackSession, ClientConfig, TokenConfig } from "../config/store.js";
-export declare function getClientProvisioningMessage(created: boolean): string;
-export declare function formatAuthStatusSummary(input: {
-    client: ClientConfig;
-    token: TokenConfig | null;
-    callback: CallbackSession | null;
-    isCurrent?: boolean;
-}): string[];
 export declare function runAuthCommand(args: string[]): Promise<number>;