@webless/agent 0.2.6 → 0.2.8

package/dist/embed.cjs

@@ -22,6 +22,7 @@ var embed_exports = {};
 __export(embed_exports, {
   DEFAULT_AGENT_PLACEMENT: () => DEFAULT_AGENT_PLACEMENT,
   bootstrapAgent: () => bootstrapAgent,
+  createWeblessAgentPublicApi: () => createWeblessAgentPublicApi,
   getMountedAgent: () => getMountedAgent,
   mountAgent: () => mountAgent,
   normalizeAgentPlacement: () => normalizeAgentPlacement,
@@ -31,7 +32,7 @@ __export(embed_exports, {
 module.exports = __toCommonJS(embed_exports);
 
 // src/embed/mount.tsx
-var import_client4 = require("react-dom/client");
+var import_client5 = require("react-dom/client");
 
 // src/react/panel-controller.ts
 var controllers = /* @__PURE__ */ new Map();
@@ -55,22 +56,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
   };
 }
 
@@ -87,7 +176,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") {
@@ -156,13 +246,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;
@@ -171,6 +268,7 @@ var AgentSession = class {
   session;
   activeResponse;
   renderedPrefix = "";
+  capability;
   getActiveSessionId() {
     return this.session?.state.sessionId ?? loadPersistedAgentSession(this.visitorSessionId, this.storeOptions)?.sessionId;
   }
@@ -193,6 +291,7 @@ var AgentSession = class {
   ensureClient() {
     const config = resolveAgentRuntimeConfig({
       indexId: this.indexId,
+      version: this.version,
       runtimeOrigin: this.runtimeOrigin
     });
     if (!config.indexId) {
@@ -202,7 +301,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;
   }
@@ -221,9 +324,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;
@@ -233,7 +340,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;
@@ -291,18 +401,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(
@@ -317,9 +434,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().";
     }
@@ -385,6 +502,7 @@ async function runStatusSequence(signal, onStep) {
 function useAgentChat({
   customerId,
   indexId,
+  version,
   runtimeOrigin,
   visitorSessionId,
   storageKeyPrefix
@@ -392,8 +510,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 }),
@@ -403,13 +521,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;
@@ -425,12 +544,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;
@@ -533,7 +653,7 @@ function useAgentChat({
         }));
       }
     },
-    [customerId, indexId, runtimeOrigin, resolvedStorageKeyPrefix, visitorId]
+    [customerId, indexId, runtimeOrigin, resolvedStorageKeyPrefix, version, visitorId]
   );
   (0, import_react.useEffect)(() => {
     return () => {
@@ -1004,6 +1124,7 @@ var import_jsx_runtime8 = require("react/jsx-runtime");
 function AgentWidget({
   indexId,
   customerId,
+  version,
   runtimeOrigin,
   placement: placementInput,
   defaultCollapsed = true,
@@ -1016,6 +1137,7 @@ function AgentWidget({
   const { state, submit } = useAgentChat({
     customerId,
     indexId,
+    version,
     runtimeOrigin
   });
   const idle = state.phase === "idle" && !hasVisitorMessages(state.messages);
@@ -1096,6 +1218,7 @@ function AgentWidget2({ manifest }) {
     {
       indexId: manifest.indexId,
       customerId: manifest.customerId,
+      version: manifest.version,
       runtimeOrigin: manifest.runtimeOrigin,
       placement: manifest.placement,
       defaultCollapsed: true,
@@ -1114,15 +1237,20 @@ function normalizeAgentTagManifest(manifest) {
   if (!customerId) {
     throw new Error("Agent manifest requires customerId.");
   }
+  const version = manifest.version ?? "published";
+  if (version !== "published" && version !== "unpublished") {
+    throw new Error("Agent manifest version must be published or unpublished.");
+  }
   const runtime = resolveAgentRuntimeConfig({
     indexId,
+    version,
     runtimeOrigin: manifest.runtimeOrigin?.trim() || DEFAULT_RUNTIME_ORIGIN
   });
   return {
     customerId,
     indexId,
     runtimeOrigin: runtime.origin,
-    version: manifest.version?.trim() || "published",
+    version,
     mount: {
       selector: manifest.mount?.selector?.trim(),
       strategy: manifest.mount?.strategy ?? "body"
@@ -1161,7 +1289,7 @@ function mountAgent(input) {
   const mountTarget = resolveMountHost(manifest, input.script ?? null);
   const host = createHost(manifest.customerId);
   mountTarget.append(host);
-  const root = (0, import_client4.createRoot)(host);
+  const root = (0, import_client5.createRoot)(host);
   root.render(/* @__PURE__ */ (0, import_jsx_runtime10.jsx)(AgentWidget2, { manifest }));
   const handle = {
     customerId: manifest.customerId,
@@ -1203,10 +1331,54 @@ function getMountedAgent(customerId) {
 function bootstrapAgent(manifest, script) {
   return mountAgent({ manifest, script });
 }
+
+// src/embed/iife-bootstrap.ts
+function createWeblessAgentPublicApi(manifest) {
+  const script = document.currentScript instanceof HTMLScriptElement ? document.currentScript : null;
+  function mount(options = {}) {
+    return mountAgent({
+      manifest: {
+        ...manifest,
+        ...options,
+        customerId: manifest.customerId
+      },
+      script
+    });
+  }
+  function unmount(customerId) {
+    return unmountAgent(customerId ?? manifest.customerId);
+  }
+  function open(customerId) {
+    getMountedAgent(customerId ?? manifest.customerId)?.open();
+  }
+  function close(customerId) {
+    getMountedAgent(customerId ?? manifest.customerId)?.close();
+  }
+  function get(customerId) {
+    return getMountedAgent(customerId ?? manifest.customerId);
+  }
+  const api = {
+    mount,
+    unmount,
+    open,
+    close,
+    get
+  };
+  window.WeblessAgent = api;
+  void (async () => {
+    try {
+      mount();
+    } catch (error) {
+      console.error("[Webless] Agent mount failed", error);
+    }
+  })();
+  return api;
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   DEFAULT_AGENT_PLACEMENT,
   bootstrapAgent,
+  createWeblessAgentPublicApi,
   getMountedAgent,
   mountAgent,
   normalizeAgentPlacement,