@kweaver-ai/kweaver-sdk 0.4.2 → 0.4.5

package/dist/commands/context-loader.js

@@ -1,4 +1,4 @@
-import { ensureValidToken, formatHttpError } from "../auth/oauth.js";
+import { ensureValidToken, formatHttpError, with401RefreshRetry } from "../auth/oauth.js";
 import { knSearch, knSchemaSearch, queryObjectInstance, queryInstanceSubgraph, getLogicPropertiesValues, getActionInfo, listTools, listResources, readResource, listResourceTemplates, listPrompts, getPrompt, } from "../api/context-loader.js";
 import { addContextLoaderEntry, getCurrentContextLoaderKn, getCurrentPlatform, loadContextLoaderConfig, removeContextLoaderEntry, setCurrentContextLoader, } from "../config/store.js";
 const MCP_NOT_CONFIGURED = "Context-loader MCP is not configured. Run: kweaver context-loader config set --kn-id <kn-id>";
@@ -54,59 +54,56 @@ Examples:
     if (subcommand === "config") {
         return runConfigCommand(rest);
     }
-    const token = await ensureValidToken();
-    const base = ensureContextLoaderConfig();
-    const options = { ...base, accessToken: token.accessToken };
     let pretty = true;
     const prettyIdx = rest.indexOf("--pretty");
     if (prettyIdx !== -1) {
         pretty = true;
         rest.splice(prettyIdx, 1);
     }
+    const dispatch = async () => {
+        const token = await ensureValidToken();
+        const base = ensureContextLoaderConfig();
+        const options = { ...base, accessToken: token.accessToken };
+        if (subcommand === "tools")
+            return runListTools(options, rest, pretty);
+        if (subcommand === "resources")
+            return runListResources(options, rest, pretty);
+        if (subcommand === "resource")
+            return runReadResource(options, rest, pretty);
+        if (subcommand === "templates")
+            return runListTemplates(options, rest, pretty);
+        if (subcommand === "prompts")
+            return runListPrompts(options, rest, pretty);
+        if (subcommand === "prompt")
+            return runGetPrompt(options, rest, pretty);
+        if (subcommand === "kn-search")
+            return runKnSearch(options, rest, pretty);
+        if (subcommand === "kn-schema-search")
+            return runKnSchemaSearch(options, rest, pretty);
+        if (subcommand === "query-object-instance")
+            return runQueryObjectInstance(options, rest, pretty);
+        if (subcommand === "query-instance-subgraph")
+            return runQueryInstanceSubgraph(options, rest, pretty);
+        if (subcommand === "get-logic-properties")
+            return runGetLogicProperties(options, rest, pretty);
+        if (subcommand === "get-action-info")
+            return runGetActionInfo(options, rest, pretty);
+        return -1;
+    };
     try {
-        if (subcommand === "tools") {
-            return await runListTools(options, rest, pretty);
-        }
-        if (subcommand === "resources") {
-            return await runListResources(options, rest, pretty);
-        }
-        if (subcommand === "resource") {
-            return await runReadResource(options, rest, pretty);
-        }
-        if (subcommand === "templates") {
-            return await runListTemplates(options, rest, pretty);
-        }
-        if (subcommand === "prompts") {
-            return await runListPrompts(options, rest, pretty);
-        }
-        if (subcommand === "prompt") {
-            return await runGetPrompt(options, rest, pretty);
-        }
-        if (subcommand === "kn-search") {
-            return await runKnSearch(options, rest, pretty);
-        }
-        if (subcommand === "kn-schema-search") {
-            return await runKnSchemaSearch(options, rest, pretty);
-        }
-        if (subcommand === "query-object-instance") {
-            return await runQueryObjectInstance(options, rest, pretty);
-        }
-        if (subcommand === "query-instance-subgraph") {
-            return await runQueryInstanceSubgraph(options, rest, pretty);
-        }
-        if (subcommand === "get-logic-properties") {
-            return await runGetLogicProperties(options, rest, pretty);
-        }
-        if (subcommand === "get-action-info") {
-            return await runGetActionInfo(options, rest, pretty);
-        }
+        return await with401RefreshRetry(async () => {
+            const code = await dispatch();
+            if (code === -1) {
+                console.error(`Unknown context-loader subcommand: ${subcommand}`);
+                return 1;
+            }
+            return code;
+        });
     }
     catch (error) {
         console.error(formatHttpError(error));
         return 1;
     }
-    console.error(`Unknown context-loader subcommand: ${subcommand}`);
-    return 1;
 }
 async function runConfigCommand(args) {
     const [action, ...rest] = args;

package/dist/commands/ds.js

@@ -1,7 +1,8 @@
 import { createInterface } from "node:readline";
-import { ensureValidToken, formatHttpError } from "../auth/oauth.js";
+import { ensureValidToken, formatHttpError, with401RefreshRetry } from "../auth/oauth.js";
 import { testDatasource, createDatasource, listDatasources, getDatasource, deleteDatasource, listTablesWithColumns, } from "../api/datasources.js";
 import { formatCallOutput } from "./call.js";
+import { resolveBusinessDomain } from "../config/store.js";
 function confirmYes(prompt) {
     return new Promise((resolve) => {
         const rl = createInterface({ input: process.stdin, output: process.stdout });
@@ -34,24 +35,28 @@ Subcommands:
     Test connectivity, register datasource, and discover tables.`);
         return 0;
     }
-    try {
-        if (subcommand === "list") {
+    const dispatch = () => {
+        if (subcommand === "list")
             return runDsListCommand(rest);
-        }
-        if (subcommand === "get") {
+        if (subcommand === "get")
             return runDsGetCommand(rest);
-        }
-        if (subcommand === "delete") {
+        if (subcommand === "delete")
             return runDsDeleteCommand(rest);
-        }
-        if (subcommand === "tables") {
+        if (subcommand === "tables")
             return runDsTablesCommand(rest);
-        }
-        if (subcommand === "connect") {
+        if (subcommand === "connect")
             return runDsConnectCommand(rest);
-        }
-        console.error(`Unknown ds subcommand: ${subcommand}`);
-        return 1;
+        return Promise.resolve(-1);
+    };
+    try {
+        return await with401RefreshRetry(async () => {
+            const code = await dispatch();
+            if (code === -1) {
+                console.error(`Unknown ds subcommand: ${subcommand}`);
+                return 1;
+            }
+            return code;
+        });
     }
     catch (error) {
         console.error(formatHttpError(error));
@@ -61,7 +66,7 @@ Subcommands:
 export function parseDsListArgs(args) {
     let keyword;
     let type;
-    let businessDomain = "bd_public";
+    let businessDomain = "";
     let pretty = true;
     for (let i = 0; i < args.length; i += 1) {
         const arg = args[i];
@@ -84,6 +89,8 @@ export function parseDsListArgs(args) {
             continue;
         }
     }
+    if (!businessDomain)
+        businessDomain = resolveBusinessDomain();
     return { keyword, type, businessDomain, pretty };
 }
 async function runDsListCommand(args) {

package/dist/commands/vega.js

@@ -1,6 +1,7 @@
-import { ensureValidToken, formatHttpError } from "../auth/oauth.js";
+import { ensureValidToken, formatHttpError, with401RefreshRetry } from "../auth/oauth.js";
 import { vegaHealth, listVegaCatalogs, getVegaCatalog, vegaCatalogHealthStatus, testVegaCatalogConnection, discoverVegaCatalog, listVegaCatalogResources, listVegaResources, getVegaResource, queryVegaResourceData, previewVegaResource, listVegaConnectorTypes, getVegaConnectorType, listVegaDiscoverTasks, } from "../api/vega.js";
 import { formatCallOutput } from "./call.js";
+import { resolveBusinessDomain } from "../config/store.js";
 // ---------------------------------------------------------------------------
 // Help
 // ---------------------------------------------------------------------------
@@ -32,7 +33,7 @@ Common flags:
 // Common flag parser
 // ---------------------------------------------------------------------------
 function parseCommonFlags(args) {
-    let businessDomain = "bd_public";
+    let businessDomain = "";
     let pretty = true;
     const remaining = [];
     for (let i = 0; i < args.length; i += 1) {
@@ -47,6 +48,8 @@ function parseCommonFlags(args) {
         }
         remaining.push(arg);
     }
+    if (!businessDomain)
+        businessDomain = resolveBusinessDomain();
     return { remaining, businessDomain, pretty };
 }
 // ---------------------------------------------------------------------------
@@ -58,21 +61,30 @@ export async function runVegaCommand(args) {
         printVegaHelp();
         return 0;
     }
-    try {
+    const dispatch = () => {
         if (subcommand === "health")
-            return await runVegaHealthCommand(rest);
+            return runVegaHealthCommand(rest);
         if (subcommand === "stats")
-            return await runVegaStatsCommand(rest);
+            return runVegaStatsCommand(rest);
         if (subcommand === "inspect")
-            return await runVegaInspectCommand(rest);
+            return runVegaInspectCommand(rest);
         if (subcommand === "catalog")
-            return await runVegaCatalogCommand(rest);
+            return runVegaCatalogCommand(rest);
         if (subcommand === "resource")
-            return await runVegaResourceCommand(rest);
+            return runVegaResourceCommand(rest);
         if (subcommand === "connector-type")
-            return await runVegaConnectorTypeCommand(rest);
-        console.error(`Unknown vega subcommand: ${subcommand}`);
-        return 1;
+            return runVegaConnectorTypeCommand(rest);
+        return Promise.resolve(-1);
+    };
+    try {
+        return await with401RefreshRetry(async () => {
+            const code = await dispatch();
+            if (code === -1) {
+                console.error(`Unknown vega subcommand: ${subcommand}`);
+                return 1;
+            }
+            return code;
+        });
     }
     catch (error) {
         console.error(formatHttpError(error));

package/dist/config/store.d.ts

@@ -1,14 +1,3 @@
-export interface ClientConfig {
-    baseUrl: string;
-    clientId: string;
-    clientSecret: string;
-    redirectUri: string;
-    logoutRedirectUri: string;
-    scope: string;
-    lang?: string;
-    product?: string;
-    xForwardedPrefix?: string;
-}
 export interface TokenConfig {
     baseUrl: string;
     accessToken: string;
@@ -20,13 +9,17 @@ export interface TokenConfig {
     idToken?: string;
     obtainedAt: string;
 }
-export interface CallbackSession {
+/** OAuth2 client registration (per platform), used for refresh_token grant. */
+export interface ClientConfig {
     baseUrl: string;
-    redirectUri: string;
-    code: string;
-    state: string;
+    clientId: string;
+    clientSecret: string;
+    redirectUri?: string;
+    logoutRedirectUri?: string;
     scope?: string;
-    receivedAt: string;
+    lang?: string;
+    product?: string;
+    xForwardedPrefix?: string;
 }
 /** Single context-loader entry (named kn_id). */
 export interface ContextLoaderEntry {
@@ -51,12 +44,10 @@ export declare function setPlatformAlias(baseUrl: string, alias: string): void;
 export declare function deletePlatformAlias(baseUrl: string): void;
 export declare function getPlatformAlias(baseUrl: string): string | null;
 export declare function resolvePlatformIdentifier(value: string): string | null;
-export declare function loadClientConfig(baseUrl?: string): ClientConfig | null;
-export declare function saveClientConfig(config: ClientConfig): void;
 export declare function loadTokenConfig(baseUrl?: string): TokenConfig | null;
 export declare function saveTokenConfig(config: TokenConfig): void;
-export declare function loadCallbackSession(baseUrl?: string): CallbackSession | null;
-export declare function saveCallbackSession(session: CallbackSession): void;
+export declare function loadClientConfig(baseUrl?: string): ClientConfig | null;
+export declare function saveClientConfig(baseUrl: string, config: ClientConfig): void;
 export declare function loadContextLoaderConfig(baseUrl?: string): ContextLoaderConfig | null;
 export declare function saveContextLoaderConfig(baseUrl: string, config: ContextLoaderConfig): void;
 export interface CurrentContextLoaderKn {
@@ -69,9 +60,19 @@ export declare function setCurrentContextLoader(baseUrl: string, name: string):
 export declare function removeContextLoaderEntry(baseUrl: string, name: string): void;
 export declare function hasPlatform(baseUrl: string): boolean;
 /**
- * Remove token and callback for a platform so the next auth will do a full login.
- * Keeps client config so the same app registration can be reused.
+ * Remove token for a platform so the next auth will do a full login.
  */
 export declare function clearPlatformSession(baseUrl: string): void;
 export declare function deletePlatform(baseUrl: string): void;
 export declare function listPlatforms(): PlatformSummary[];
+/** Per-platform config (not auth — general settings). */
+export interface PlatformConfig {
+    businessDomain?: string;
+}
+export declare function loadPlatformBusinessDomain(baseUrl: string): string | null;
+export declare function savePlatformBusinessDomain(baseUrl: string, businessDomain: string): void;
+/**
+ * Resolve businessDomain: env var > per-platform config > "bd_public".
+ * If baseUrl is omitted, uses the current platform.
+ */
+export declare function resolveBusinessDomain(baseUrl?: string): string;

package/dist/config/store.js

@@ -175,19 +175,6 @@ export function resolvePlatformIdentifier(value) {
     }
     return normalized;
 }
-export function loadClientConfig(baseUrl) {
-    ensureStoreReady();
-    const targetBaseUrl = baseUrl ?? getCurrentPlatform();
-    if (!targetBaseUrl) {
-        return null;
-    }
-    return readJsonFile(getPlatformFile(targetBaseUrl, "client.json"));
-}
-export function saveClientConfig(config) {
-    ensureStoreReady();
-    ensurePlatformDir(config.baseUrl);
-    writeJsonFile(getPlatformFile(config.baseUrl, "client.json"), config);
-}
 export function loadTokenConfig(baseUrl) {
     ensureStoreReady();
     const targetBaseUrl = baseUrl ?? getCurrentPlatform();
@@ -201,18 +188,18 @@ export function saveTokenConfig(config) {
     ensurePlatformDir(config.baseUrl);
     writeJsonFile(getPlatformFile(config.baseUrl, "token.json"), config);
 }
-export function loadCallbackSession(baseUrl) {
+export function loadClientConfig(baseUrl) {
     ensureStoreReady();
     const targetBaseUrl = baseUrl ?? getCurrentPlatform();
     if (!targetBaseUrl) {
         return null;
     }
-    return readJsonFile(getPlatformFile(targetBaseUrl, "callback.json"));
+    return readJsonFile(getPlatformFile(targetBaseUrl, "client.json"));
 }
-export function saveCallbackSession(session) {
+export function saveClientConfig(baseUrl, config) {
     ensureStoreReady();
-    ensurePlatformDir(session.baseUrl);
-    writeJsonFile(getPlatformFile(session.baseUrl, "callback.json"), session);
+    ensurePlatformDir(baseUrl);
+    writeJsonFile(getPlatformFile(baseUrl, "client.json"), { ...config, baseUrl });
 }
 function migrateLegacyContextLoader(raw) {
     const leg = raw;
@@ -258,8 +245,8 @@ export function getCurrentContextLoaderKn(baseUrl) {
     const targetBaseUrl = baseUrl ?? getCurrentPlatform();
     if (!targetBaseUrl)
         return null;
-    const client = loadClientConfig(targetBaseUrl);
-    if (!client?.baseUrl)
+    const token = loadTokenConfig(targetBaseUrl);
+    if (!token?.baseUrl)
         return null;
     const config = loadContextLoaderConfig(targetBaseUrl);
     if (!config)
@@ -268,7 +255,7 @@ export function getCurrentContextLoaderKn(baseUrl) {
     if (!entry)
         return null;
     return {
-        mcpUrl: buildMcpUrl(client.baseUrl),
+        mcpUrl: buildMcpUrl(token.baseUrl),
         knId: entry.knId,
     };
 }
@@ -320,22 +307,17 @@ export function removeContextLoaderEntry(baseUrl, name) {
 }
 export function hasPlatform(baseUrl) {
     ensureStoreReady();
-    return existsSync(getPlatformFile(baseUrl, "client.json"));
+    return existsSync(getPlatformFile(baseUrl, "token.json"));
 }
 /**
- * Remove token and callback for a platform so the next auth will do a full login.
- * Keeps client config so the same app registration can be reused.
+ * Remove token for a platform so the next auth will do a full login.
  */
 export function clearPlatformSession(baseUrl) {
     ensureStoreReady();
     const tokenFile = getPlatformFile(baseUrl, "token.json");
-    const callbackFile = getPlatformFile(baseUrl, "callback.json");
     if (existsSync(tokenFile)) {
         rmSync(tokenFile, { force: true });
     }
-    if (existsSync(callbackFile)) {
-        rmSync(callbackFile, { force: true });
-    }
 }
 export function deletePlatform(baseUrl) {
     ensureStoreReady();
@@ -364,17 +346,49 @@ export function listPlatforms() {
         if (!statSync(dirPath).isDirectory()) {
             continue;
         }
-        const client = readJsonFile(join(dirPath, "client.json"));
-        if (!client?.baseUrl) {
+        const token = readJsonFile(join(dirPath, "token.json"));
+        if (!token?.baseUrl) {
             continue;
         }
         items.push({
-            baseUrl: client.baseUrl,
-            hasToken: existsSync(join(dirPath, "token.json")),
-            isCurrent: client.baseUrl === currentPlatform,
-            alias: getPlatformAlias(client.baseUrl) ?? undefined,
+            baseUrl: token.baseUrl,
+            hasToken: true,
+            isCurrent: token.baseUrl === currentPlatform,
+            alias: getPlatformAlias(token.baseUrl) ?? undefined,
         });
     }
     items.sort((a, b) => a.baseUrl.localeCompare(b.baseUrl));
     return items;
 }
+function loadPlatformConfig(baseUrl) {
+    ensureStoreReady();
+    return readJsonFile(getPlatformFile(baseUrl, "config.json"));
+}
+function savePlatformConfig(baseUrl, config) {
+    ensureStoreReady();
+    ensurePlatformDir(baseUrl);
+    writeJsonFile(getPlatformFile(baseUrl, "config.json"), config);
+}
+export function loadPlatformBusinessDomain(baseUrl) {
+    return loadPlatformConfig(baseUrl)?.businessDomain ?? null;
+}
+export function savePlatformBusinessDomain(baseUrl, businessDomain) {
+    const existing = loadPlatformConfig(baseUrl) ?? {};
+    savePlatformConfig(baseUrl, { ...existing, businessDomain });
+}
+/**
+ * Resolve businessDomain: env var > per-platform config > "bd_public".
+ * If baseUrl is omitted, uses the current platform.
+ */
+export function resolveBusinessDomain(baseUrl) {
+    const fromEnv = process.env.KWEAVER_BUSINESS_DOMAIN;
+    if (fromEnv)
+        return fromEnv;
+    const targetUrl = baseUrl ?? getCurrentPlatform();
+    if (targetUrl) {
+        const fromConfig = loadPlatformBusinessDomain(targetUrl);
+        if (fromConfig)
+            return fromConfig;
+    }
+    return "bd_public";
+}

package/dist/index.d.ts

@@ -50,5 +50,5 @@ export { BknResource } from "./resources/bkn.js";
 export { ConversationsResource } from "./resources/conversations.js";
 export { ContextLoaderResource } from "./resources/context-loader.js";
 export { HttpError, NetworkRequestError, fetchTextOrThrow } from "./utils/http.js";
-export type { ClientConfig, TokenConfig, ContextLoaderEntry, ContextLoaderConfig, PlatformSummary, } from "./config/store.js";
+export type { TokenConfig, ContextLoaderEntry, ContextLoaderConfig, } from "./config/store.js";
 export { getConfigDir, getCurrentPlatform } from "./config/store.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kweaver-ai/kweaver-sdk",
-  "version": "0.4.2",
+  "version": "0.4.5",
   "description": "KWeaver TypeScript SDK — CLI tool and programmatic API for knowledge networks and Decision Agents.",
   "type": "module",
   "main": "./dist/index.js",
@@ -53,6 +53,14 @@
     "tsx": "^4.20.5",
     "typescript": "^5.9.3"
   },
+  "peerDependencies": {
+    "playwright": ">=1.40.0"
+  },
+  "peerDependenciesMeta": {
+    "playwright": {
+      "optional": true
+    }
+  },
   "dependencies": {
     "@kweaver-ai/bkn": "^0.1.0",
     "ink": "^6.8.0",