@opentrace/opentrace 0.4.0-rc.649 → 0.4.0-rc.656

package/dist/lib/{SidePanel-CuGhFGJi.cjs → SidePanel-C8fYuv1B.cjs}

@@ -1,6 +1,7 @@
 "use strict";
 const jsxRuntime = require("react/jsx-runtime");
 const React = require("react");
+const reactDom = require("react-dom");
 const _commonjsHelpers = require("./_commonjsHelpers-cpYYNffD.cjs");
 const v4$2 = require("./v4-DT1ViLGh.cjs");
 const ChatTemplates = require("./ChatTemplates-A_lcy38z.cjs");
@@ -70785,22 +70786,22 @@ ${att.textContent}
       setStreaming(false);
       progress.setListener(null);
       const hasUsage = usageAccum.inputTokens > 0 || usageAccum.outputTokens > 0;
-      let updatedMessages = messagesRef.current;
-      if (hasUsage) {
-        const updated = [...updatedMessages];
-        const last = updated[updated.length - 1];
-        if (last?.role === "assistant") {
+      reactDom.flushSync(() => {
+        setMessages((prev) => {
+          if (!hasUsage) return prev;
+          const updated = [...prev];
+          const last = updated[updated.length - 1];
+          if (last?.role !== "assistant") return prev;
           updated[updated.length - 1] = {
             ...last,
             usage: usageAccum
           };
-        }
-        updatedMessages = updated;
-        setMessages(updated);
-      }
+          return updated;
+        });
+      });
       if (!controller.signal.aborted) {
         persistMessages(
-          updatedMessages,
+          messagesRef.current,
           providerId,
           modelId,
           Array.from(chatFoundNodesRef.current)
@@ -79238,4 +79239,4 @@ exports.useGraph = useGraph;
 exports.useGraphData = useGraphData;
 exports.useGraphInteraction = useGraphInteraction;
 exports.useGraphViewer = useGraphViewer;
-//# sourceMappingURL=SidePanel-CuGhFGJi.cjs.map
+//# sourceMappingURL=SidePanel-C8fYuv1B.cjs.map