@webless/agent 0.2.6 → 0.2.8

package/dist/react.cjs

@@ -40,22 +40,110 @@ var import_react5 = require("react");
 var import_react = require("react");
 
 // src/runtime/client.ts
+var import_client2 = require("eve/client");
+
+// src/runtime/capability.ts
 var import_client = require("eve/client");
+var MAX_REFRESH_SKEW_MS = 3e4;
+function isRecord(value) {
+  return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function parseBootstrapResponse(value, indexId, now) {
+  if (!isRecord(value) || value.apiVersion !== "webless.ai/agent-runtime-bootstrap/v1" || typeof value.accessToken !== "string" || !value.accessToken || typeof value.expiresAt !== "string" || !isRecord(value.identity) || value.identity.indexId !== indexId || typeof value.identity.revision !== "string" || !value.identity.revision || typeof value.identity.tenantId !== "string" || !value.identity.tenantId || typeof value.origin !== "string" || !value.origin || value.tokenType !== "Bearer" || typeof value.visitorSubject !== "string" || !value.visitorSubject) {
+    throw new Error("Agent Runtime returned an invalid access response.");
+  }
+  const expiresAt = Date.parse(value.expiresAt);
+  if (!Number.isFinite(expiresAt) || expiresAt <= now) {
+    throw new Error("Agent Runtime returned an expired access response.");
+  }
+  const refreshSkew = Math.min(MAX_REFRESH_SKEW_MS, Math.floor((expiresAt - now) / 10));
+  return {
+    accessToken: value.accessToken,
+    refreshAt: expiresAt - refreshSkew
+  };
+}
+async function readBootstrapError(response) {
+  const fallback = `Agent Runtime is unavailable (${response.status}).`;
+  try {
+    const value = await response.json();
+    return isRecord(value) && typeof value.error === "string" && value.error ? value.error : fallback;
+  } catch {
+    return fallback;
+  }
+}
+function createAgentRuntimeCapability(options) {
+  const fetchImplementation = options.fetchImplementation ?? fetch;
+  const now = options.now ?? Date.now;
+  let capability;
+  let pendingBootstrap;
+  const bootstrap = async () => {
+    const response = await fetchImplementation(`${options.runtimeOrigin}/webless/v1/bootstrap`, {
+      body: JSON.stringify({
+        clientSessionId: options.visitorSessionId,
+        indexId: options.indexId
+      }),
+      headers: { "content-type": "application/json" },
+      method: "POST"
+    });
+    if (!response.ok) {
+      throw new Error(await readBootstrapError(response));
+    }
+    let value;
+    try {
+      value = await response.json();
+    } catch {
+      throw new Error("Agent Runtime returned an invalid access response.");
+    }
+    return parseBootstrapResponse(value, options.indexId, now());
+  };
+  return {
+    getAccessToken: async () => {
+      if (capability && capability.refreshAt > now()) {
+        return capability.accessToken;
+      }
+      pendingBootstrap ??= bootstrap().finally(() => {
+        pendingBootstrap = void 0;
+      });
+      capability = await pendingBootstrap;
+      return capability.accessToken;
+    },
+    invalidate: () => {
+      capability = void 0;
+    }
+  };
+}
+async function withCapabilityRefresh(capability, request) {
+  try {
+    return await request();
+  } catch (error) {
+    if (!(error instanceof import_client.ClientError) || error.status !== 401) {
+      throw error;
+    }
+    capability.invalidate();
+    return await request();
+  }
+}
 
 // src/runtime/config.ts
 var DEFAULT_RUNTIME_ORIGIN = "https://runtime.staging.webless.ai";
 var trimTrailingSlash = (value) => value.replace(/\/+$/, "");
-function buildEveHost(origin, indexId) {
+function buildEveHost(origin, indexId, version = "published") {
   const base = trimTrailingSlash(origin);
-  return `${base}?indexId=${encodeURIComponent(indexId.trim())}`;
+  const searchParams = new URLSearchParams({ indexId: indexId.trim() });
+  if (version === "unpublished") {
+    searchParams.set("version", version);
+  }
+  return `${base}?${searchParams.toString()}`;
 }
 function resolveAgentRuntimeConfig(input) {
   const origin = trimTrailingSlash(input.runtimeOrigin?.trim() || DEFAULT_RUNTIME_ORIGIN);
   const resolvedIndexId = input.indexId.trim();
+  const version = input.version ?? "published";
   return {
     origin,
     indexId: resolvedIndexId,
-    host: resolvedIndexId ? buildEveHost(origin, resolvedIndexId) : origin
+    version,
+    host: resolvedIndexId ? buildEveHost(origin, resolvedIndexId, version) : origin
   };
 }
 
@@ -72,7 +160,8 @@ function buildAgentStorageKeyPrefix(input) {
   if (!indexId) {
     throw new Error("indexId is required to build agent storage keys.");
   }
-  return `${DEFAULT_STORAGE_KEY_PREFIX}:${customerId}:${indexId}:${normalizeRuntimeOrigin(input.runtimeOrigin)}`;
+  const versionSuffix = input.version === "unpublished" ? ":unpublished" : "";
+  return `${DEFAULT_STORAGE_KEY_PREFIX}:${customerId}:${indexId}${versionSuffix}:${normalizeRuntimeOrigin(input.runtimeOrigin)}`;
 }
 function createSessionId() {
   if (typeof crypto !== "undefined" && typeof crypto.randomUUID === "function") {
@@ -141,13 +230,20 @@ function mapStepLabel(event) {
   };
 }
 var AgentSession = class {
-  constructor(indexId, runtimeOrigin, visitorSessionId, storeOptions) {
+  constructor(indexId, version, runtimeOrigin, visitorSessionId, storeOptions) {
     this.indexId = indexId;
+    this.version = version;
     this.runtimeOrigin = runtimeOrigin;
     this.visitorSessionId = visitorSessionId;
     this.storeOptions = storeOptions;
+    this.capability = createAgentRuntimeCapability({
+      indexId,
+      runtimeOrigin,
+      visitorSessionId
+    });
   }
   indexId;
+  version;
   runtimeOrigin;
   visitorSessionId;
   storeOptions;
@@ -156,6 +252,7 @@ var AgentSession = class {
   session;
   activeResponse;
   renderedPrefix = "";
+  capability;
   getActiveSessionId() {
     return this.session?.state.sessionId ?? loadPersistedAgentSession(this.visitorSessionId, this.storeOptions)?.sessionId;
   }
@@ -178,6 +275,7 @@ var AgentSession = class {
   ensureClient() {
     const config = resolveAgentRuntimeConfig({
       indexId: this.indexId,
+      version: this.version,
       runtimeOrigin: this.runtimeOrigin
     });
     if (!config.indexId) {
@@ -187,7 +285,11 @@ var AgentSession = class {
       return this.client;
     }
     this.reset();
-    this.client = new import_client.Client({ host: config.host });
+    this.client = new import_client2.Client({
+      auth: { bearer: () => this.capability.getAccessToken() },
+      host: config.host,
+      redirect: "error"
+    });
     this.clientHost = config.host;
     return this.client;
   }
@@ -206,9 +308,13 @@ var AgentSession = class {
     if (session) {
       this.session = session;
       try {
-        response = await session.send(message, { signal });
+        const activeSession = session;
+        response = await withCapabilityRefresh(
+          this.capability,
+          () => activeSession.send(message, { signal })
+        );
       } catch (error) {
-        if (error instanceof import_client.ClientError && error.status === 409 && error.code === "session_not_active") {
+        if (error instanceof import_client2.ClientError && error.status === 409 && error.code === "session_not_active") {
           clearPersistedAgentSession(this.visitorSessionId, this.storeOptions);
           this.session = void 0;
           session = void 0;
@@ -218,7 +324,10 @@ var AgentSession = class {
       }
     }
     if (!response) {
-      const created = await client.sessions.create({ message, signal });
+      const created = await withCapabilityRefresh(
+        this.capability,
+        () => client.sessions.create({ message, signal })
+      );
       response = created.response;
       session = created.session;
       this.session = session;
@@ -276,18 +385,25 @@ function createAgentClient(options) {
   if (!indexId) {
     throw new Error("indexId is required.");
   }
-  const runtimeOrigin = options.runtimeOrigin?.trim() || resolveAgentRuntimeConfig({ indexId }).origin;
+  const version = options.version ?? "published";
+  const runtimeOrigin = resolveAgentRuntimeConfig({
+    indexId,
+    runtimeOrigin: options.runtimeOrigin,
+    version
+  }).origin;
   const storeOptions = {
     storageKeyPrefix: options.storageKeyPrefix?.trim() || buildAgentStorageKeyPrefix({
       customerId: options.customerId,
       indexId,
+      version,
       runtimeOrigin
     })
   };
   const visitorSessionId = options.visitorSessionId?.trim() || getOrCreateVisitorSessionId(storeOptions);
-  const session = new AgentSession(indexId, runtimeOrigin, visitorSessionId, storeOptions);
+  const session = new AgentSession(indexId, version, runtimeOrigin, visitorSessionId, storeOptions);
   return {
     indexId,
+    version,
     runtimeOrigin,
     visitorSessionId,
     sendTurn: (message, sendOptions) => session.sendTurn(
@@ -302,9 +418,9 @@ function createAgentClient(options) {
 }
 
 // src/runtime/errors.ts
-var import_client2 = require("eve/client");
+var import_client3 = require("eve/client");
 function formatAgentError(error) {
-  if (error instanceof import_client2.ClientError) {
+  if (error instanceof import_client3.ClientError) {
     if (error.status === 401 && error.code === "index_required") {
       return "Missing indexId \u2014 pass a published index id to createAgentClient().";
     }
@@ -370,6 +486,7 @@ async function runStatusSequence(signal, onStep) {
 function useAgentChat({
   customerId,
   indexId,
+  version,
   runtimeOrigin,
   visitorSessionId,
   storageKeyPrefix
@@ -377,8 +494,8 @@ function useAgentChat({
   const [state, setState] = (0, import_react.useState)(INITIAL_STATE);
   const runRef = (0, import_react.useRef)(null);
   const resolvedStorageKeyPrefix = (0, import_react.useMemo)(
-    () => storageKeyPrefix?.trim() || buildAgentStorageKeyPrefix({ customerId, indexId, runtimeOrigin }),
-    [customerId, indexId, runtimeOrigin, storageKeyPrefix]
+    () => storageKeyPrefix?.trim() || buildAgentStorageKeyPrefix({ customerId, indexId, version, runtimeOrigin }),
+    [customerId, indexId, runtimeOrigin, storageKeyPrefix, version]
   );
   const visitorId = (0, import_react.useMemo)(
     () => visitorSessionId?.trim() || getOrCreateVisitorSessionId({ storageKeyPrefix: resolvedStorageKeyPrefix }),
@@ -388,13 +505,14 @@ function useAgentChat({
     createAgentClient({
       customerId,
       indexId,
+      version,
       runtimeOrigin,
       visitorSessionId: visitorId,
       storageKeyPrefix: resolvedStorageKeyPrefix
     })
   );
   const identityRef = (0, import_react.useRef)(null);
-  const identityKey = `${customerId ?? ""}|${indexId}|${runtimeOrigin ?? ""}|${resolvedStorageKeyPrefix}|${visitorId}`;
+  const identityKey = `${customerId ?? ""}|${indexId}|${version ?? "published"}|${runtimeOrigin ?? ""}|${resolvedStorageKeyPrefix}|${visitorId}`;
   (0, import_react.useEffect)(() => {
     if (identityRef.current === null) {
       identityRef.current = identityKey;
@@ -410,12 +528,13 @@ function useAgentChat({
     clientRef.current = createAgentClient({
       customerId,
       indexId,
+      version,
       runtimeOrigin,
       visitorSessionId: visitorId,
       storageKeyPrefix: resolvedStorageKeyPrefix
     });
     setState(INITIAL_STATE);
-  }, [customerId, identityKey, indexId, runtimeOrigin, resolvedStorageKeyPrefix, visitorId]);
+  }, [customerId, identityKey, indexId, runtimeOrigin, resolvedStorageKeyPrefix, version, visitorId]);
   const reset = (0, import_react.useCallback)(() => {
     runRef.current?.abort();
     runRef.current = null;
@@ -518,7 +637,7 @@ function useAgentChat({
         }));
       }
     },
-    [customerId, indexId, runtimeOrigin, resolvedStorageKeyPrefix, visitorId]
+    [customerId, indexId, runtimeOrigin, resolvedStorageKeyPrefix, version, visitorId]
   );
   (0, import_react.useEffect)(() => {
     return () => {
@@ -1001,6 +1120,7 @@ var import_jsx_runtime8 = require("react/jsx-runtime");
 function AgentWidget({
   indexId,
   customerId,
+  version,
   runtimeOrigin,
   placement: placementInput,
   defaultCollapsed = true,
@@ -1013,6 +1133,7 @@ function AgentWidget({
   const { state, submit } = useAgentChat({
     customerId,
     indexId,
+    version,
     runtimeOrigin
   });
   const idle = state.phase === "idle" && !hasVisitorMessages(state.messages);