@assistant-ui/react 0.5.33 → 0.5.34

package/dist/index.mjs

@@ -9,7 +9,7 @@ import {
   toLanguageModelMessages,
   toLanguageModelTools,
   toolResultStream
-} from "./chunk-DXLSOE2J.mjs";
+} from "./chunk-QBSJKWMT.mjs";
 import {
   __export
 } from "./chunk-BJPOCE4O.mjs";
@@ -1142,19 +1142,40 @@ var LocalThreadRuntime = class {
     this.abortController?.abort();
     this.abortController = new AbortController();
     const initialContent = message.content;
-    const initialRoundtrips = message.roundtrips;
+    const initialRoundtrips = message.metadata?.roundtrips;
+    const initalCustom = message.metadata?.custom;
     const updateMessage = (m) => {
       message = {
         ...message,
         ...m.content ? { content: [...initialContent, ...m.content ?? []] } : void 0,
         status: m.status ?? message.status,
-        ...m.roundtrips?.length ? { roundtrips: [...initialRoundtrips ?? [], ...m.roundtrips] } : void 0
+        // TODO deprecated, remove in v0.6
+        ...m.metadata?.roundtrips ? {
+          roundtrips: [
+            ...initialRoundtrips ?? [],
+            ...m.metadata.roundtrips
+          ]
+        } : void 0,
+        ...m.metadata ? {
+          metadata: {
+            ...message.metadata,
+            ...m.metadata.roundtrips ? {
+              roundtrips: [
+                ...initialRoundtrips ?? [],
+                ...m.metadata.roundtrips
+              ]
+            } : void 0,
+            ...m.metadata?.custom ? {
+              custom: { ...initalCustom ?? {}, ...m.metadata.custom }
+            } : void 0
+          }
+        } : void 0
       };
       this.repository.addOrUpdateMessage(parentId, message);
       this.notifySubscribers();
     };
     const maxToolRoundtrips = this.options?.maxToolRoundtrips ?? 1;
-    const toolRoundtrips = message.roundtrips?.length ?? 0;
+    const toolRoundtrips = message.metadata?.roundtrips?.length ?? 0;
     if (toolRoundtrips > maxToolRoundtrips) {
       updateMessage({
         status: {