@dudousxd/nestjs-agent-react 0.35.1 → 0.35.3

package/dist/index.cjs

@@ -1380,6 +1380,39 @@ function useAmbientRenderUi() {
 // src/genui/generative-ui.tsx
 var import_react4 = require("react");
 
+// src/genui/share-structure.ts
+function isPlainObject(value) {
+  if (typeof value !== "object" || value === null || Array.isArray(value)) return false;
+  const prototype = Object.getPrototypeOf(value);
+  return prototype === Object.prototype || prototype === null;
+}
+function shareStructure(previous, next) {
+  if (Object.is(previous, next)) return next;
+  if (Array.isArray(next)) {
+    if (!Array.isArray(previous)) return next;
+    let same = previous.length === next.length;
+    const shared3 = next.map((value, index) => {
+      const kept = shareStructure(previous[index], value);
+      if (kept !== previous[index]) same = false;
+      return kept;
+    });
+    return same ? previous : shared3;
+  }
+  if (isPlainObject(next)) {
+    if (!isPlainObject(previous)) return next;
+    const keys = Object.keys(next);
+    let same = keys.length === Object.keys(previous).length;
+    const shared3 = {};
+    for (const key of keys) {
+      const kept = shareStructure(previous[key], next[key]);
+      if (kept !== previous[key] || !Object.hasOwn(previous, key)) same = false;
+      shared3[key] = kept;
+    }
+    return same ? previous : shared3;
+  }
+  return next;
+}
+
 // src/genui/use-generative-ui.ts
 var import_react3 = require("react");
 
@@ -1551,10 +1584,10 @@ var TreeScopeContext = (0, import_react4.createContext)(null);
 var GenuiNodeContext = sharedContext(
   "@dudousxd/nestjs-agent-react:genui-node"
 );
-function renderPlaceholder(scope, node) {
-  const { placeholder } = scope;
+function renderPlaceholder(look, node) {
+  const { placeholder } = look;
   if (typeof placeholder === "function") return placeholder(node);
-  return placeholder !== void 0 ? placeholder : scope.loading;
+  return placeholder !== void 0 ? placeholder : look.loading;
 }
 function renderFallback(fallback2, problem) {
   if (typeof fallback2 === "function") return fallback2(problem);
@@ -1588,9 +1621,33 @@ var ItemBoundary = class extends import_react4.Component {
 function nodeItem(node, id) {
   return { id, component: node.type, props: node.props ?? {}, version: null, toolCallId: null };
 }
-function TreeNode({ node, id }) {
+function failsWhole(scope) {
+  return !scope.hasFallback && scope.fallbackText !== void 0;
+}
+var TreeNode = (0, import_react4.memo)(
+  function TreeNode2({ node, id, bounded }) {
+    const scope = (0, import_react4.useContext)(TreeScopeContext);
+    const item = (0, import_react4.useMemo)(() => nodeItem(node, id), [node, id]);
+    if (scope === null) return null;
+    if (!bounded || failsWhole(scope)) return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeNodeBody, { node, id, item });
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+      ItemBoundary,
+      {
+        item,
+        fallback: scope.look.current.fallback,
+        onError: scope.look.current.onError,
+        children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeNodeBody, { node, id, item })
+      }
+    );
+  },
+  (previous, next) => previous.node === next.node && previous.id === next.id && previous.bounded === next.bounded
+);
+function TreeNodeBody({
+  node,
+  id,
+  item
+}) {
   const scope = (0, import_react4.useContext)(TreeScopeContext);
-  const item = (0, import_react4.useMemo)(() => nodeItem(node, id), [node, id]);
   const incomplete = node.incomplete === true;
   const held = node.held === true;
   const state = (0, import_react4.useMemo)(
@@ -1612,45 +1669,37 @@ function TreeNode({ node, id }) {
     item.props
   );
   if (scope === null) return null;
+  const look = scope.look.current;
   if (held) {
-    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(GenuiNodeContext.Provider, { value: state, children: renderPlaceholder(scope, state) });
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(GenuiNodeContext.Provider, { value: state, children: renderPlaceholder(look, state) });
   }
-  if (resolution.status === "loading" || validation.status === "pending") return scope.loading;
+  if (resolution.status === "loading" || validation.status === "pending") return look.loading;
   if (resolution.status === "unknown") {
-    if (scope.fallback === void 0 && scope.fallbackText !== void 0)
-      throw new Error("Unknown tree component");
-    return renderFallback(scope.fallback, { reason: "unknown", item });
+    if (failsWhole(scope)) throw new Error("Unknown tree component");
+    return renderFallback(look.fallback, { reason: "unknown", item });
   }
   if (resolution.status === "error") {
-    if (scope.fallback === void 0 && scope.fallbackText !== void 0) throw resolution.error;
-    return renderFallback(scope.fallback, { reason: "error", item, error: resolution.error });
+    if (failsWhole(scope)) throw resolution.error;
+    return renderFallback(look.fallback, { reason: "error", item, error: resolution.error });
   }
   if (validation.status === "invalid") {
-    if (scope.fallback === void 0 && scope.fallbackText !== void 0)
-      throw new Error("Invalid tree component");
-    return renderFallback(scope.fallback, { reason: "invalid", item, issues: validation.issues });
+    if (failsWhole(scope)) throw new Error("Invalid tree component");
+    return renderFallback(look.fallback, { reason: "invalid", item, issues: validation.issues });
   }
   const Renderer = resolution.Component;
   const children = Array.isArray(node.children) ? node.children : [];
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(GenuiNodeContext.Provider, { value: state, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(Renderer, { ...validation.props, children: children.length > 0 ? children.map((child, index) => {
     const childId = `${id}.${index}`;
-    if (scope.fallback === void 0 && scope.fallbackText !== void 0)
-      return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeNode, { node: child, id: childId }, childId);
-    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
-      ItemBoundary,
-      {
-        item: nodeItem(child, childId),
-        fallback: scope.fallback,
-        onError: scope.onError,
-        children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeNode, { node: child, id: childId })
-      },
-      childId
-    );
+    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeNode, { node: child, id: childId, bounded: true }, childId);
   }) : void 0 }) });
 }
 function GenuiTree({ root }) {
-  if (root === void 0 || root === null || typeof root.type !== "string") return null;
-  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeNode, { node: root, id: "root" });
+  const previous = (0, import_react4.useRef)(void 0);
+  const valid = root !== void 0 && root !== null && typeof root.type === "string";
+  const shared3 = valid ? shareStructure(previous.current, root) : void 0;
+  previous.current = shared3;
+  if (shared3 === void 0) return null;
+  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeNode, { node: shared3, id: "root", bounded: false });
 }
 function GenerativeUIScope({
   registry,
@@ -1664,6 +1713,12 @@ function GenerativeUIScope({
   onError,
   children
 }) {
+  const versionsRef = (0, import_react4.useRef)(void 0);
+  const versions = shareStructure(versionsRef.current, componentVersions);
+  versionsRef.current = versions;
+  const look = (0, import_react4.useRef)({ fallback: fallback2, loading, placeholder, onError });
+  look.current = { fallback: fallback2, loading, placeholder, onError };
+  const hasFallback = fallback2 !== void 0;
   const scope = (0, import_react4.useMemo)(
     () => ({
       options: {
@@ -1671,24 +1726,12 @@ function GenerativeUIScope({
         ...catalog !== void 0 ? { catalog } : {},
         ...resolveComponent !== void 0 ? { resolveComponent } : {}
       },
-      fallback: fallback2,
-      ...componentVersions !== void 0 ? { componentVersions } : {},
+      hasFallback,
+      ...versions !== void 0 ? { componentVersions: versions } : {},
       ...fallbackText !== void 0 ? { fallbackText } : {},
-      loading,
-      placeholder,
-      onError
+      look
     }),
-    [
-      registry,
-      catalog,
-      resolveComponent,
-      fallback2,
-      fallbackText,
-      componentVersions,
-      loading,
-      placeholder,
-      onError
-    ]
+    [registry, catalog, resolveComponent, hasFallback, fallbackText, versions]
   );
   return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(TreeScopeContext.Provider, { value: scope, children });
 }
@@ -4245,6 +4288,7 @@ function useTranscriptItems({
   );
   const callbackCache = (0, import_react17.useRef)(/* @__PURE__ */ new Map());
   const reasoningToggleCache = (0, import_react17.useRef)(/* @__PURE__ */ new Map());
+  const uiBlocks = (0, import_react17.useRef)(/* @__PURE__ */ new Map());
   const toggleReasoning = (0, import_react17.useCallback)((key, open) => {
     setOpenReasoning((current) => {
       const next = new Map(current);
@@ -4439,6 +4483,14 @@ function useTranscriptItems({
   }, []);
   const lastAssistantId = findLastAssistant(messages)?.id ?? null;
   const items = [];
+  const previousUi = uiBlocks.current;
+  const keptUi = /* @__PURE__ */ new Map();
+  const keepUiIdentity = (blocks) => blocks.map((block) => {
+    if (block.kind !== "ui") return block;
+    const kept = shareStructure(previousUi.get(block.key), block);
+    keptUi.set(block.key, kept);
+    return kept;
+  });
   for (let index = visibleStart; index < messages.length; index++) {
     const message = messages[index];
     if (!message) {
@@ -4461,37 +4513,39 @@ function useTranscriptItems({
       isAssistant,
       isLastAssistant,
       isStreaming: message.id === streamingMessageId,
-      blocks: buildTranscriptBlocks(message, {
-        isReasoningOpen: (key, isStreamingRun) => openReasoning.get(key) ?? isStreamingRun,
-        toggleReasoning: (key, open) => stableToggle(key)(open),
-        ...options.sources !== void 0 ? { sources: options.sources } : {},
-        ...options.toolCatalog !== void 0 ? { toolCatalog: options.toolCatalog } : {},
-        // Read-only still lifts question sets into their block, so the outcome shows.
-        ...handlers.onAnswer !== void 0 || options.readOnly === true && options.onAnswer !== null ? {
-          elicitation: {
-            picked: (toolCallId, questionId) => picks.get(pickKey(toolCallId, questionId)),
-            pick,
-            canAnswer: handlers.onAnswer !== void 0,
-            canSkip: handlers.onSkip !== void 0,
-            answer,
-            skip,
-            submitting: (toolCallId) => settling.get(toolCallId) ?? null,
-            errorOf: (toolCallId) => settleErrors.get(toolCallId)?.message ?? null,
-            errorCodeOf: (toolCallId) => settleErrors.get(toolCallId)?.code ?? null
-          }
-        } : {},
-        ...handlers.onApprove !== void 0 || handlers.onReject !== void 0 ? {
-          approval: {
-            canApprove: handlers.onApprove !== void 0,
-            canReject: handlers.onReject !== void 0,
-            approve,
-            reject,
-            submitting: (toolCallId) => settling.get(toolCallId) ?? null,
-            errorOf: (toolCallId) => settleErrors.get(toolCallId)?.message ?? null,
-            errorCodeOf: (toolCallId) => settleErrors.get(toolCallId)?.code ?? null
-          }
-        } : {}
-      }),
+      blocks: keepUiIdentity(
+        buildTranscriptBlocks(message, {
+          isReasoningOpen: (key, isStreamingRun) => openReasoning.get(key) ?? isStreamingRun,
+          toggleReasoning: (key, open) => stableToggle(key)(open),
+          ...options.sources !== void 0 ? { sources: options.sources } : {},
+          ...options.toolCatalog !== void 0 ? { toolCatalog: options.toolCatalog } : {},
+          // Read-only still lifts question sets into their block, so the outcome shows.
+          ...handlers.onAnswer !== void 0 || options.readOnly === true && options.onAnswer !== null ? {
+            elicitation: {
+              picked: (toolCallId, questionId) => picks.get(pickKey(toolCallId, questionId)),
+              pick,
+              canAnswer: handlers.onAnswer !== void 0,
+              canSkip: handlers.onSkip !== void 0,
+              answer,
+              skip,
+              submitting: (toolCallId) => settling.get(toolCallId) ?? null,
+              errorOf: (toolCallId) => settleErrors.get(toolCallId)?.message ?? null,
+              errorCodeOf: (toolCallId) => settleErrors.get(toolCallId)?.code ?? null
+            }
+          } : {},
+          ...handlers.onApprove !== void 0 || handlers.onReject !== void 0 ? {
+            approval: {
+              canApprove: handlers.onApprove !== void 0,
+              canReject: handlers.onReject !== void 0,
+              approve,
+              reject,
+              submitting: (toolCallId) => settling.get(toolCallId) ?? null,
+              errorOf: (toolCallId) => settleErrors.get(toolCallId)?.message ?? null,
+              errorCodeOf: (toolCallId) => settleErrors.get(toolCallId)?.code ?? null
+            }
+          } : {}
+        })
+      ),
       text,
       usage: usage ? describeUsage(usage) : null,
       timestamp: describeTimestamp((options.getCreatedAt ?? createdAtFromMetadata)(message)),
@@ -4526,6 +4580,7 @@ function useTranscriptItems({
       }
     });
   }
+  uiBlocks.current = keptUi;
   return items;
 }
 function settleHandlers(options, backend) {
@@ -5719,9 +5774,24 @@ function outcomeId(value) {
 }
 
 // src/approvals/use-action-proposals.ts
+var UNSUPPORTED_STATUSES = /* @__PURE__ */ new Set([404, 405, 501]);
+var UNKNOWN_THREAD = /* @__PURE__ */ new Set(["Thread not found", "Unknown thread"]);
+var MAX_BACKOFF_MS = 3e4;
+var unsupportedBackends = /* @__PURE__ */ new WeakSet();
+function statusOf(cause) {
+  const status = cause?.status;
+  return typeof status === "number" ? status : void 0;
+}
+function isProposalsUnsupported(cause) {
+  const status = statusOf(cause);
+  if (status === void 0 || !UNSUPPORTED_STATUSES.has(status)) return false;
+  return !(status === 404 && cause instanceof Error && UNKNOWN_THREAD.has(cause.message));
+}
 function useActionProposals(backend, threadId, setMessages, refreshKey, pollMs = 1e3) {
   const [items, setItems] = (0, import_react22.useState)([]);
   const [error, setError] = (0, import_react22.useState)(null);
+  const [unsupported, setUnsupported] = (0, import_react22.useState)(() => unsupportedBackends.has(backend));
+  const [failures, setFailures] = (0, import_react22.useState)(0);
   const latest = (0, import_react22.useRef)({ threadId, setMessages });
   latest.current = { threadId, setMessages };
   const mounted = (0, import_react22.useRef)(true);
@@ -5734,6 +5804,7 @@ function useActionProposals(backend, threadId, setMessages, refreshKey, pollMs =
   }, []);
   const refresh = (0, import_react22.useCallback)(async () => {
     if (threadId === void 0 || backend.listActionProposals === void 0) return;
+    if (unsupportedBackends.has(backend)) return;
     const pending = requests.current.get(threadId);
     if (pending) return pending;
     const request = (async () => {
@@ -5750,12 +5821,23 @@ function useActionProposals(backend, threadId, setMessages, refreshKey, pollMs =
         if (!mounted.current || latest.current.threadId !== threadId) return;
         setItems(proposals);
         setError(null);
+        setFailures(0);
         latest.current.setMessages(
           (current) => reconcileProposalMessages(current, proposals, facts)
         );
       } catch (cause) {
-        if (mounted.current && latest.current.threadId === threadId)
+        if (isProposalsUnsupported(cause)) {
+          unsupportedBackends.add(backend);
+          if (mounted.current) {
+            setUnsupported(true);
+            setError(null);
+          }
+          return;
+        }
+        if (mounted.current && latest.current.threadId === threadId) {
           setError(cause instanceof Error ? cause : new Error(String(cause)));
+          setFailures((count) => count + 1);
+        }
       }
     })();
     requests.current.set(threadId, request);
@@ -5768,18 +5850,24 @@ function useActionProposals(backend, threadId, setMessages, refreshKey, pollMs =
   (0, import_react22.useEffect)(() => {
     setItems([]);
     setError(null);
+    setFailures(0);
   }, [threadId]);
+  (0, import_react22.useEffect)(() => {
+    setUnsupported(unsupportedBackends.has(backend));
+  }, [backend]);
   (0, import_react22.useEffect)(() => {
     void refresh();
   }, [refresh, refreshKey]);
   (0, import_react22.useEffect)(() => {
-    if (!items.some(proposalNeedsPolling) || pollMs <= 0) return;
-    const timer = setInterval(() => {
+    if (unsupported || !items.some(proposalNeedsPolling) || pollMs <= 0) return;
+    const wait = failures === 0 ? pollMs : Math.min(pollMs * 2 ** failures, MAX_BACKOFF_MS);
+    const timer = setTimeout(() => {
       void refresh();
-    }, pollMs);
-    return () => clearInterval(timer);
-  }, [items, pollMs, refresh]);
+    }, wait);
+    return () => clearTimeout(timer);
+  }, [items, failures, unsupported, pollMs, refresh]);
   (0, import_react22.useEffect)(() => {
+    if (unsupported) return;
     const refetch = () => {
       void refresh();
     };
@@ -5794,8 +5882,8 @@ function useActionProposals(backend, threadId, setMessages, refreshKey, pollMs =
       window.removeEventListener("online", refetch);
       document.removeEventListener("visibilitychange", visible);
     };
-  }, [refresh]);
-  return { items, error, refresh };
+  }, [refresh, unsupported]);
+  return { items, error, unsupported, refresh };
 }
 
 // src/use-agent-chat.ts