@flytedan/flytebot-design-system 0.71.2 → 0.71.3

package/dist/index.js

@@ -12437,6 +12437,8 @@ function useChatEngine(opts) {
   const stoppedRef = React70.useRef(false);
   const abortRef = React70.useRef(null);
   const serverCount = React70.useRef(0);
+  const serverIds = React70.useRef(/* @__PURE__ */ new Set());
+  const liveTurn = React70.useRef(null);
   const threadRef = React70.useRef(null);
   const mounted = React70.useRef(true);
   React70.useEffect(() => {
@@ -12465,15 +12467,35 @@ function useChatEngine(opts) {
     const data = await apiAdapter.getThread(id);
     const list = [...data && data.messages || []];
     serverCount.current = list.length;
+    serverIds.current = new Set(list.map((m) => m.id));
     commit(list);
     return list;
   }, [apiAdapter]);
   async function mergeThread() {
     const data = await apiAdapter.getThread(threadRef.current);
     const server = data && data.messages || [];
+    const turn = liveTurn.current;
+    const ownIds = /* @__PURE__ */ new Set();
+    let progress;
+    if (turn) {
+      for (const m of server) {
+        if (turn.known.has(m.id)) continue;
+        ownIds.add(m.id);
+        if (m.role === "assistant") progress = m;
+      }
+    }
     const ids = new Set(server.map((m) => m.id));
     serverCount.current = server.length;
-    commit([...server, ...listRef.current.filter((m) => !ids.has(m.id))]);
+    serverIds.current = ids;
+    const next = [...server.filter((m) => !ownIds.has(m.id)), ...listRef.current.filter((m) => !ids.has(m.id))];
+    commit(turn && progress ? next.map((m) => m.id === turn.holderId ? foldProgress(m, progress) : m) : next);
+  }
+  function foldProgress(holder, progress) {
+    return Object.assign({}, holder, {
+      steps: progress.steps ?? holder.steps,
+      text: progress.text || holder.text,
+      packet: progress.packet ?? holder.packet
+    });
   }
   React70.useEffect(() => {
     if (!apiAdapter) {
@@ -12628,6 +12650,7 @@ function useChatEngine(opts) {
       { id: userId, role: "user", text: turn.text || null, attachments: turn.attachments, pending: true, timestamp: (/* @__PURE__ */ new Date()).toISOString() },
       { id: holderId, role: "assistant", text: "", streaming: !!apiAdapter.streamMessage, working: !apiAdapter.streamMessage, timestamp: (/* @__PURE__ */ new Date()).toISOString() }
     );
+    liveTurn.current = { known: new Set(serverIds.current), holderId };
     const ac = typeof AbortController !== "undefined" ? new AbortController() : { abort() {
     }, signal: void 0 };
     abortRef.current = ac;
@@ -12683,6 +12706,7 @@ function useChatEngine(opts) {
         commit(listRef.current.filter((m) => m.id !== holderId));
       }
     } finally {
+      liveTurn.current = null;
       abortRef.current = null;
       busyRef.current = false;
       setBusy(false);