@webless/agent 0.2.7 → 0.2.9

package/dist/embed.cjs

@@ -32,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();
@@ -56,7 +56,89 @@ 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";
@@ -170,6 +252,11 @@ var AgentSession = class {
     this.runtimeOrigin = runtimeOrigin;
     this.visitorSessionId = visitorSessionId;
     this.storeOptions = storeOptions;
+    this.capability = createAgentRuntimeCapability({
+      indexId,
+      runtimeOrigin,
+      visitorSessionId
+    });
   }
   indexId;
   version;
@@ -180,7 +267,7 @@ var AgentSession = class {
   clientHost;
   session;
   activeResponse;
-  renderedPrefix = "";
+  capability;
   getActiveSessionId() {
     return this.session?.state.sessionId ?? loadPersistedAgentSession(this.visitorSessionId, this.storeOptions)?.sessionId;
   }
@@ -189,7 +276,6 @@ var AgentSession = class {
     });
     this.activeResponse = void 0;
     this.session = void 0;
-    this.renderedPrefix = "";
     clearPersistedAgentSession(this.visitorSessionId, this.storeOptions);
   }
   persistSessionCursor(session) {
@@ -213,14 +299,17 @@ 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;
   }
   attachPersistedSession(client) {
     const persisted = loadPersistedAgentSession(this.visitorSessionId, this.storeOptions);
     if (!persisted?.sessionId) return void 0;
-    this.renderedPrefix = "";
     return client.sessions.attach(persisted.sessionId, {
       streamIndex: persisted.streamIndex
     });
@@ -232,9 +321,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;
@@ -244,15 +337,17 @@ 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;
-      this.renderedPrefix = "";
       this.persistSessionCursor(session);
     }
     this.activeResponse = response;
-    let rendered = this.renderedPrefix;
+    let rendered = "";
     try {
       for await (const event of response) {
         if (signal.aborted) break;
@@ -278,7 +373,6 @@ var AgentSession = class {
       }
     } finally {
       this.activeResponse = void 0;
-      this.renderedPrefix = rendered;
       if (session) {
         this.persistSessionCursor(session);
       }
@@ -302,8 +396,12 @@ 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,
@@ -331,9 +429,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().";
     }
@@ -372,11 +470,6 @@ var STATUS_SEQUENCE = [
   { id: "s1", label: "Starting Eve session", ms: 400 },
   { id: "s2", label: "Connecting to runtime", ms: 500 }
 ];
-var DEFAULT_FOLLOW_UPS = [
-  { id: "fu-1", label: "What can you help me with?" },
-  { id: "fu-2", label: "Summarize your capabilities" },
-  { id: "fu-3", label: "What should I ask next?" }
-];
 function delay(ms, signal) {
   return new Promise((resolve, reject) => {
     const timer = window.setTimeout(resolve, ms);
@@ -533,7 +626,7 @@ function useAgentChat({
           messages: [...prev.messages, agentMessage],
           toolSteps: [],
           streamingText: "",
-          followUps: DEFAULT_FOLLOW_UPS,
+          followUps: [],
           journey: null
         }));
       } catch (error) {
@@ -837,12 +930,17 @@ function AgentRail({
   const transcriptRef = (0, import_react4.useRef)(null);
   const railStyle = theme?.railMaxWidth ? { ["--rail-width"]: theme.railMaxWidth, ["--as-rail-max-width"]: theme.railMaxWidth } : void 0;
   const isBusy = state.phase === "thinking" || state.phase === "running-tools" || state.phase === "streaming";
-  const showActivity = state.toolSteps.length > 0 && (state.phase === "thinking" || state.phase === "running-tools" || state.phase === "streaming");
-  const showStreamingInActivity = state.phase === "streaming" && Boolean(state.streamingText);
+  const showActivity = state.toolSteps.length > 0 && (state.phase === "thinking" || state.phase === "running-tools");
   const hasVisitorMessages2 = state.messages.some((message) => message.role === "visitor");
   const showIdleFollowUps = !hasVisitorMessages2 && state.followUps.length > 0;
-  const showCompleteFollowUps = state.phase === "complete" && state.followUps.length > 0 && hasVisitorMessages2;
-  const showDockFollowUps = expanded && (showIdleFollowUps || showCompleteFollowUps);
+  const showDockFollowUps = expanded && showIdleFollowUps;
+  const streamingMessage = state.phase === "streaming" && state.streamingText ? {
+    createdAt: 0,
+    id: "streaming-response",
+    role: "agent",
+    streaming: true,
+    text: state.streamingText
+  } : null;
   (0, import_react4.useEffect)(() => {
     const node = transcriptRef.current;
     if (!node) return;
@@ -880,14 +978,8 @@ function AgentRail({
         ] }) }),
         /* @__PURE__ */ (0, import_jsx_runtime6.jsxs)("div", { ref: transcriptRef, className: "agent-rail__transcript", children: [
           state.messages.map((message) => /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(MessageBubble, { message }, message.id)),
-          showActivity ? /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
-            AgentActivityBubble,
-            {
-              steps: state.toolSteps,
-              streamingText: state.streamingText,
-              showStreaming: showStreamingInActivity
-            }
-          ) : null,
+          streamingMessage ? /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(MessageBubble, { message: streamingMessage }) : null,
+          showActivity ? /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(AgentActivityBubble, { steps: state.toolSteps }) : null,
           !expanded && showIdleFollowUps ? /* @__PURE__ */ (0, import_jsx_runtime6.jsx)("div", { className: "agent-rail__followups-slot", children: /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
             FollowUpChips,
             {
@@ -897,14 +989,6 @@ function AgentRail({
               onSelect: (suggestion) => onFollowUpSelect?.(suggestion.label)
             }
           ) }) : null,
-          !expanded && showCompleteFollowUps ? /* @__PURE__ */ (0, import_jsx_runtime6.jsx)("div", { className: "agent-rail__followups-slot", children: /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
-            FollowUpChips,
-            {
-              suggestions: state.followUps,
-              disabled: isBusy,
-              onSelect: (suggestion) => onFollowUpSelect?.(suggestion.label)
-            }
-          ) }) : null,
           state.error ? /* @__PURE__ */ (0, import_jsx_runtime6.jsx)("p", { role: "alert", style: { fontSize: 13, color: "var(--as-danger)", margin: 0 }, children: state.error }) : null
         ] }),
         expanded ? /* @__PURE__ */ (0, import_jsx_runtime6.jsxs)("div", { className: "agent-rail__dock-wrap", children: [
@@ -1186,7 +1270,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,