@kweaver-ai/kweaver-sdk 0.4.2 → 0.4.5

package/dist/api/agent-chat.js

@@ -254,7 +254,8 @@ export async function sendChatRequest(options) {
     };
     if (verbose) {
         console.error(`POST ${url}`);
-        console.error(`Headers: ${JSON.stringify(headers)}`);
+        const safeHeaders = Object.fromEntries(Object.entries(headers).map(([k, v]) => k.toLowerCase() === "authorization" ? [k, "Bearer ***"] : [k, v]));
+        console.error(`Headers: ${JSON.stringify(safeHeaders)}`);
         console.error(`Body: ${JSON.stringify(body)}`);
     }
     let response;
@@ -312,7 +313,8 @@ export async function sendChatRequestStream(options, callbacks) {
     };
     if (verbose) {
         console.error(`POST ${url}`);
-        console.error(`Headers: ${JSON.stringify(headers)}`);
+        const safeHeaders = Object.fromEntries(Object.entries(headers).map(([k, v]) => k.toLowerCase() === "authorization" ? [k, "Bearer ***"] : [k, v]));
+        console.error(`Headers: ${JSON.stringify(safeHeaders)}`);
         console.error(`Body: ${JSON.stringify(body)}`);
     }
     let response;

package/dist/api/context-loader.js

@@ -1,5 +1,6 @@
 import { fetchTextOrThrow } from "../utils/http.js";
 const MCP_PROTOCOL_VERSION = "2024-11-05";
+const SESSION_TTL_MS = 300_000; // 5 minutes
 const sessionCache = new Map();
 function sessionKey(options) {
     return `${options.mcpUrl}:${options.knId}`;
@@ -20,8 +21,11 @@ function buildHeaders(options, sessionId) {
 async function ensureSession(options) {
     const key = sessionKey(options);
     const cached = sessionCache.get(key);
+    if (cached && Date.now() - cached.createdAt < SESSION_TTL_MS)
+        return cached.id;
+    // Remove stale entry if expired
     if (cached)
-        return cached;
+        sessionCache.delete(key);
     const initBody = JSON.stringify({
         jsonrpc: "2.0",
         id: 1,
@@ -50,7 +54,7 @@ async function ensureSession(options) {
         headers: buildHeaders(options, sessionId),
         body: initNotifBody,
     });
-    sessionCache.set(key, sessionId);
+    sessionCache.set(key, { id: sessionId, createdAt: Date.now() });
     return sessionId;
 }
 function isMissingInputParams(obj) {

package/dist/api/conversations.js

@@ -30,7 +30,7 @@ export async function listConversations(opts) {
     }
     const body = await response.text();
     if (!response.ok) {
-        return "[]";
+        throw new Error(`listConversations failed: HTTP ${response.status} ${response.statusText} — ${body.slice(0, 200)}`);
     }
     return body || "[]";
 }
@@ -58,7 +58,7 @@ export async function listMessages(opts) {
     }
     const body = await response.text();
     if (!response.ok) {
-        return "[]";
+        throw new Error(`listMessages failed: HTTP ${response.status} ${response.statusText} — ${body.slice(0, 200)}`);
     }
     return body || "[]";
 }

package/dist/api/datasources.js

@@ -204,7 +204,8 @@ export async function scanMetadata(options) {
     const scanResult = await scanResponse.json();
     const taskId = scanResult.id ?? "";
     for (let i = 0; i < 30; i += 1) {
-        await new Promise((r) => setTimeout(r, 2000));
+        const delay = Math.min(2000 * Math.pow(1.5, i), 15000);
+        await new Promise((r) => setTimeout(r, delay));
         const statusResponse = await fetch(statusUrl(taskId), {
             method: "GET",
             headers: buildHeaders(accessToken, businessDomain),

package/dist/api/vega.js

@@ -13,8 +13,11 @@ function buildHeaders(accessToken, businessDomain) {
 export async function vegaHealth(options) {
     const { baseUrl, accessToken, businessDomain = "bd_public", } = options;
     const base = baseUrl.replace(/\/+$/, "");
-    const url = `${base}/health`;
-    const response = await fetch(url, {
+    // Vega backend has no dedicated /health endpoint.
+    // Probe the catalogs list as a lightweight reachability check.
+    const url = new URL(`${base}${VEGA_BASE}/catalogs`);
+    url.searchParams.set("limit", "1");
+    const response = await fetch(url.toString(), {
         method: "GET",
         headers: buildHeaders(accessToken, businessDomain),
     });
@@ -22,7 +25,7 @@ export async function vegaHealth(options) {
     if (!response.ok) {
         throw new HttpError(response.status, response.statusText, body);
     }
-    return body;
+    return JSON.stringify({ status: "healthy", probe: "catalogs", statusCode: response.status });
 }
 export async function listVegaCatalogs(options) {
     const { baseUrl, accessToken, status, limit, offset, businessDomain = "bd_public", } = options;

package/dist/auth/oauth.d.ts

@@ -1,77 +1,24 @@
-import { type CallbackSession, type ClientConfig, type TokenConfig } from "../config/store.js";
-interface RegisterClientOptions {
-    baseUrl: string;
-    clientName: string;
-    redirectUri: string;
-    logoutRedirectUri: string;
-    lang?: string;
-    product?: string;
-    xForwardedPrefix?: string;
-}
+import { type TokenConfig } from "../config/store.js";
 export declare function normalizeBaseUrl(value: string): string;
-export declare function getAuthorizationSuccessMessage(): string;
-export declare function registerClient(options: RegisterClientOptions): Promise<ClientConfig>;
-export interface EnsureClientOptions {
-    baseUrl: string;
-    port: number;
-    clientName: string;
-    forceRegister: boolean;
-    host?: string;
-    redirectUriOverride?: string;
-    lang?: string;
-    product?: string;
-    xForwardedPrefix?: string;
-}
-export interface EnsuredClientConfig {
-    client: ClientConfig;
-    created: boolean;
-}
-export interface AuthRedirectConfig {
-    redirectUri: string;
-    logoutRedirectUri: string;
-    listenHost: string;
-    listenPort: number;
-    callbackPath: string;
-}
-export declare function buildAuthRedirectConfig(options: {
-    port: number;
-    host?: string;
-    redirectUriOverride?: string;
-}): AuthRedirectConfig;
-export declare function ensureClientConfig(options: EnsureClientOptions): Promise<EnsuredClientConfig>;
-export declare function buildAuthorizationUrl(client: ClientConfig, state?: string): string;
+export declare function playwrightLogin(baseUrl: string, options?: {
+    username?: string;
+    password?: string;
+}): Promise<TokenConfig>;
 /**
- * Call the platform's end-session endpoint so the server invalidates the session.
- * Best-effort: failures are ignored so local logout still proceeds.
+ * Exchange refresh_token for a new access token (OAuth2 password grant style, same as Python ConfigAuth).
+ * Persists the new token to ~/.kweaver/ and returns it.
  */
-export declare function callLogoutEndpoint(client: ClientConfig, token: TokenConfig | null): Promise<void>;
-export declare function refreshAccessToken(client: ClientConfig, refreshToken: string): Promise<TokenConfig>;
+export declare function refreshAccessToken(token: TokenConfig): Promise<TokenConfig>;
 export declare function ensureValidToken(opts?: {
     forceRefresh?: boolean;
 }): Promise<TokenConfig>;
-export interface AuthLoginOptions {
-    baseUrl: string;
-    port: number;
-    clientName: string;
-    open: boolean;
-    forceRegister: boolean;
-    host?: string;
-    redirectUriOverride?: string;
-    lang?: string;
-    product?: string;
-    xForwardedPrefix?: string;
-}
-export declare function login(options: AuthLoginOptions): Promise<{
-    client: ClientConfig;
-    token: TokenConfig;
-    authorizationUrl: string;
-    callback: CallbackSession;
-    created: boolean;
-}>;
-export declare function getStoredAuthSummary(baseUrl?: string): {
-    client: ClientConfig | null;
-    token: TokenConfig | null;
-    callback: CallbackSession | null;
-};
+/**
+ * Run an operation; on HTTP 401, refresh the access token once and retry.
+ * Does not call `ensureValidToken` first — use for CLI routers so `--help` works without login.
+ */
+export declare function with401RefreshRetry<T>(fn: () => Promise<T>): Promise<T>;
+/**
+ * Load a valid token, run `fn(token)`, and on 401 refresh once and retry with the new token.
+ */
+export declare function withTokenRetry<T>(fn: (token: TokenConfig) => Promise<T>): Promise<T>;
 export declare function formatHttpError(error: unknown): string;
-export {};