@webless/agent 0.2.12 → 0.2.14

package/dist/embed.cjs

@@ -72,7 +72,10 @@ function parseBootstrapResponse(value, indexId, now) {
   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));
+  const refreshSkew = Math.min(
+    MAX_REFRESH_SKEW_MS,
+    Math.floor((expiresAt - now) / 10)
+  );
   return {
     accessToken: value.accessToken,
     refreshAt: expiresAt - refreshSkew
@@ -91,17 +94,31 @@ function createAgentRuntimeCapability(options) {
   const fetchImplementation = options.fetchImplementation ?? fetch;
   const now = options.now ?? Date.now;
   let capability;
+  let generation = 0;
   let pendingBootstrap;
   const bootstrap = async () => {
-    const response = await fetchImplementation(`${options.runtimeOrigin}/webless/v1/bootstrap`, {
-      body: JSON.stringify({
-        clientSessionId: options.visitorSessionId,
-        indexId: options.indexId,
-        version: options.version
-      }),
-      headers: { "content-type": "application/json" },
-      method: "POST"
-    });
+    let previewGrant;
+    if (options.version === "unpublished") {
+      previewGrant = (await options.getUnpublishedPreviewGrant?.())?.trim();
+      if (!previewGrant) {
+        throw new Error(
+          "An unpublished preview grant is required to use the Agent Runtime preview."
+        );
+      }
+    }
+    const response = await fetchImplementation(
+      `${options.runtimeOrigin}/webless/v1/bootstrap`,
+      {
+        body: JSON.stringify({
+          clientSessionId: options.visitorSessionId,
+          indexId: options.indexId,
+          ...previewGrant ? { previewGrant } : {},
+          version: options.version
+        }),
+        headers: { "content-type": "application/json" },
+        method: "POST"
+      }
+    );
     if (!response.ok) {
       throw new Error(await readBootstrapError(response));
     }
@@ -113,19 +130,32 @@ function createAgentRuntimeCapability(options) {
     }
     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;
+  const getAccessToken = async () => {
+    if (capability && capability.refreshAt > now()) {
       return capability.accessToken;
-    },
+    }
+    const requestedGeneration = generation;
+    if (!pendingBootstrap) {
+      const trackedBootstrap = bootstrap().finally(() => {
+        if (pendingBootstrap === trackedBootstrap) {
+          pendingBootstrap = void 0;
+        }
+      });
+      pendingBootstrap = trackedBootstrap;
+    }
+    const nextCapability = await pendingBootstrap;
+    if (requestedGeneration !== generation) {
+      return getAccessToken();
+    }
+    capability = nextCapability;
+    return capability.accessToken;
+  };
+  return {
+    getAccessToken,
     invalidate: () => {
       capability = void 0;
+      generation += 1;
+      pendingBootstrap = void 0;
     }
   };
 }
@@ -295,13 +325,14 @@ function mapStepLabel(event) {
   };
 }
 var AgentSession = class {
-  constructor(indexId, version, runtimeOrigin, visitorSessionId, storeOptions) {
+  constructor(indexId, version, runtimeOrigin, visitorSessionId, storeOptions, getUnpublishedPreviewGrant) {
     this.indexId = indexId;
     this.version = version;
     this.runtimeOrigin = runtimeOrigin;
     this.visitorSessionId = visitorSessionId;
     this.storeOptions = storeOptions;
     this.capability = createAgentRuntimeCapability({
+      getUnpublishedPreviewGrant,
       indexId,
       runtimeOrigin,
       version,
@@ -331,6 +362,7 @@ var AgentSession = class {
     }
     this.activeResponse = void 0;
     this.session = void 0;
+    this.capability.invalidate();
     clearPersistedAgentSession(this.visitorSessionId, this.storeOptions);
   }
   persistSessionCursor(session) {
@@ -364,7 +396,10 @@ var AgentSession = class {
     return this.client;
   }
   attachPersistedSession(client) {
-    const persisted = loadPersistedAgentSession(this.visitorSessionId, this.storeOptions);
+    const persisted = loadPersistedAgentSession(
+      this.visitorSessionId,
+      this.storeOptions
+    );
     if (!persisted?.sessionId) return void 0;
     return client.sessions.attach(persisted.sessionId, {
       streamIndex: persisted.streamIndex
@@ -431,7 +466,10 @@ var AgentSession = class {
     return rendered.trim();
   }
   async resumeTurn(message, signal, handlers, initialText = "") {
-    const persisted = loadPersistedAgentSession(this.visitorSessionId, this.storeOptions);
+    const persisted = loadPersistedAgentSession(
+      this.visitorSessionId,
+      this.storeOptions
+    );
     if (!persisted) return null;
     const client = this.ensureClient();
     const attached = client.sessions.attach(persisted.sessionId, {
@@ -470,7 +508,9 @@ var AgentSession = class {
     }
     if (snapshotBoundary) {
       if (snapshotBoundary.type === "session.failed") {
-        throw new Error(snapshotBoundary.data.message || snapshotBoundary.data.code);
+        throw new Error(
+          snapshotBoundary.data.message || snapshotBoundary.data.code
+        );
       }
       handlers.onComplete?.();
       if (!rendered.trim()) throw new Error("Empty response from runtime");
@@ -527,7 +567,14 @@ function createAgentClient(options) {
     })
   };
   const visitorSessionId = options.visitorSessionId?.trim() || getOrCreateVisitorSessionId(storeOptions);
-  const session = new AgentSession(indexId, version, runtimeOrigin, visitorSessionId, storeOptions);
+  const session = new AgentSession(
+    indexId,
+    version,
+    runtimeOrigin,
+    visitorSessionId,
+    storeOptions,
+    options.getUnpublishedPreviewGrant
+  );
   return {
     indexId,
     version,
@@ -668,24 +715,50 @@ function delay(ms, signal) {
 }
 async function runStatusSequence(signal, onStep) {
   for (const item of STATUS_SEQUENCE) {
-    onStep({ id: item.id, label: item.label, detail: item.detail, state: "active" });
+    onStep({
+      id: item.id,
+      label: item.label,
+      detail: item.detail,
+      state: "active"
+    });
     await delay(item.ms, signal);
   }
 }
 function useAgentChat({
   customerId,
+  getUnpublishedPreviewGrant,
   indexId,
   version,
   runtimeOrigin,
   visitorSessionId,
   storageKeyPrefix
 }) {
+  const previewGrantProviderRef = (0, import_react.useRef)(getUnpublishedPreviewGrant);
+  previewGrantProviderRef.current = getUnpublishedPreviewGrant;
+  const resolveUnpublishedPreviewGrant = () => {
+    const provider = previewGrantProviderRef.current;
+    if (!provider) {
+      return Promise.reject(
+        new Error(
+          "An unpublished preview grant is required to use the Agent Runtime preview."
+        )
+      );
+    }
+    return provider();
+  };
   const resolvedStorageKeyPrefix = (0, import_react.useMemo)(
-    () => storageKeyPrefix?.trim() || buildAgentStorageKeyPrefix({ customerId, indexId, version, runtimeOrigin }),
+    () => storageKeyPrefix?.trim() || buildAgentStorageKeyPrefix({
+      customerId,
+      indexId,
+      version,
+      runtimeOrigin
+    }),
     [customerId, indexId, runtimeOrigin, storageKeyPrefix, version]
   );
   const visitorId = (0, import_react.useMemo)(
-    () => visitorSessionId?.trim() || getOrCreateVisitorSessionId({ storageKeyPrefix: resolvedStorageKeyPrefix }),
+    () => visitorSessionId?.trim() || getOrCreateVisitorSessionId({
+      storageKeyPrefix: resolvedStorageKeyPrefix
+    }),
     [resolvedStorageKeyPrefix, visitorSessionId]
   );
   const [state, setState] = (0, import_react.useState)(
@@ -697,6 +770,7 @@ function useAgentChat({
   const clientRef = (0, import_react.useRef)(
     createAgentClient({
       customerId,
+      getUnpublishedPreviewGrant: resolveUnpublishedPreviewGrant,
       indexId,
       version,
       runtimeOrigin,
@@ -715,6 +789,7 @@ function useAgentChat({
     runRef.current = null;
     clientRef.current = createAgentClient({
       customerId,
+      getUnpublishedPreviewGrant: resolveUnpublishedPreviewGrant,
       indexId,
       version,
       runtimeOrigin,
@@ -726,7 +801,15 @@ function useAgentChat({
         loadPersistedAgentConversation(resolvedStorageKeyPrefix, visitorId)
       )
     );
-  }, [customerId, identityKey, indexId, runtimeOrigin, resolvedStorageKeyPrefix, version, visitorId]);
+  }, [
+    customerId,
+    identityKey,
+    indexId,
+    runtimeOrigin,
+    resolvedStorageKeyPrefix,
+    version,
+    visitorId
+  ]);
   (0, import_react.useEffect)(() => {
     if (!hasVisitorMessages(state.messages)) return;
     savePersistedAgentConversation(resolvedStorageKeyPrefix, visitorId, {
@@ -734,7 +817,13 @@ function useAgentChat({
       pending: isAgentBusy(state.phase),
       streamingText: state.streamingText
     });
-  }, [resolvedStorageKeyPrefix, state.messages, state.phase, state.streamingText, visitorId]);
+  }, [
+    resolvedStorageKeyPrefix,
+    state.messages,
+    state.phase,
+    state.streamingText,
+    visitorId
+  ]);
   const reset = (0, import_react.useCallback)(() => {
     runRef.current?.abort();
     runRef.current = null;
@@ -752,7 +841,11 @@ function useAgentChat({
         let streamed = initialText;
         const planningPromise = resume ? Promise.resolve() : runStatusSequence(signal, (step) => {
           if (streamStarted || !isActiveRun()) return;
-          setState((prev) => ({ ...prev, phase: "running-tools", toolSteps: [step] }));
+          setState((prev) => ({
+            ...prev,
+            phase: "running-tools",
+            toolSteps: [step]
+          }));
         });
         const handlers = {
           onStep: (label, detail) => {
@@ -760,7 +853,9 @@ function useAgentChat({
             setState((prev) => ({
               ...prev,
               phase: "running-tools",
-              toolSteps: [{ id: `step-${label}`, label, detail, state: "active" }]
+              toolSteps: [
+                { id: `step-${label}`, label, detail, state: "active" }
+              ]
             }));
           },
           onDelta: (delta) => {
@@ -777,7 +872,11 @@ function useAgentChat({
               }));
             }
             streamed += delta;
-            setState((prev) => ({ ...prev, phase: "streaming", streamingText: streamed }));
+            setState((prev) => ({
+              ...prev,
+              phase: "streaming",
+              streamingText: streamed
+            }));
           },
           onComplete: () => {
             if (!isActiveRun()) return;
@@ -791,7 +890,10 @@ function useAgentChat({
           signal
         }) : await clientRef.current.sendTurn(visitorText, { handlers, signal });
         if (resume && finalText === null) {
-          finalText = await clientRef.current.sendTurn(visitorText, { handlers, signal });
+          finalText = await clientRef.current.sendTurn(visitorText, {
+            handlers,
+            signal
+          });
         }
         await planningPromise.catch(() => {
         });
@@ -813,7 +915,8 @@ function useAgentChat({
         }));
         runRef.current = null;
       } catch (error) {
-        if (error instanceof DOMException && error.name === "AbortError") return;
+        if (error instanceof DOMException && error.name === "AbortError")
+          return;
         if (!isActiveRun()) return;
         const message = formatAgentError(error);
         if (!message) return;
@@ -847,7 +950,9 @@ function useAgentChat({
         ...prev,
         phase: "thinking",
         messages: [...prev.messages, visitorMessage],
-        toolSteps: [{ id: "s1", label: "Starting Eve session", state: "active" }],
+        toolSteps: [
+          { id: "s1", label: "Starting Eve session", state: "active" }
+        ],
         journey: null,
         followUps: [],
         streamingText: "",
@@ -1355,6 +1460,7 @@ var import_jsx_runtime8 = require("react/jsx-runtime");
 function AgentWidget({
   indexId,
   customerId,
+  getUnpublishedPreviewGrant,
   version,
   runtimeOrigin,
   placement: placementInput,
@@ -1367,6 +1473,7 @@ function AgentWidget({
   const [railExpanded, setRailExpanded] = (0, import_react5.useState)(false);
   const { state, submit } = useAgentChat({
     customerId,
+    getUnpublishedPreviewGrant,
     indexId,
     version,
     runtimeOrigin