@truefoundry/assistant-ui-runtime 0.1.14 → 0.1.15

package/dist/index.js

@@ -1534,7 +1534,6 @@ function attachRunningTurn(snapshot, runningTurn) {
   if (runningTurn == null) {
     return snapshot;
   }
-  applyUserToolResponsesToFold(snapshot.fold, runningTurn.input ?? []);
   const pendingUserText = extractTurnUserText(runningTurn.input);
   return replaceSessionSnapshot(snapshot, {
     runningTurn,
@@ -1549,10 +1548,63 @@ function attachRunningTurn(snapshot, runningTurn) {
     } : {}
   });
 }
-async function buildSnapshotFromSessionEvents(server, sessionId, onProgress) {
+function findOpenTurnCreated(itemsAsc) {
+  let open;
+  for (const item of itemsAsc) {
+    if (item.event.type === "turn.created") {
+      open = { turnId: item.turnId, event: item.event };
+    } else if (item.event.type === "turn.done") {
+      open = void 0;
+    }
+  }
+  return open;
+}
+function turnFromCreatedEvent(options) {
+  const { sessionId, turnId, event } = options;
+  return {
+    id: turnId,
+    sessionId,
+    state: { status: "running" },
+    createdAt: event.createdAt,
+    ...event.input != null ? { input: event.input } : {},
+    ...event.previousTurnId === void 0 ? {} : { previousTurnId: event.previousTurnId }
+  };
+}
+async function resolveSessionTip(options) {
+  const { server, sessionId, itemsAsc } = options;
+  const open = findOpenTurnCreated(itemsAsc);
+  if (open != null) {
+    const continuationInput = open.event.input ?? [];
+    if (typeof server.getTurn === "function") {
+      try {
+        const turn = await server.getTurn({
+          sessionId,
+          turnId: open.turnId
+        });
+        if (turn.state.status === "running") {
+          return {
+            runningTurn: turn,
+            continuationInput: turn.input ?? continuationInput
+          };
+        }
+        return { continuationInput: turn.input ?? continuationInput };
+      } catch {
+      }
+    }
+    return {
+      runningTurn: turnFromCreatedEvent({
+        sessionId,
+        turnId: open.turnId,
+        event: open.event
+      }),
+      continuationInput
+    };
+  }
   const turnsPage = await server.listTurns({ sessionId, limit: 1 });
-  const newestTurn = turnsPage.data[0];
-  const runningTurn = newestTurn?.state?.status === "running" ? newestTurn : void 0;
+  const tip = turnsPage.data[0];
+  return tip?.state?.status === "running" ? { runningTurn: tip, continuationInput: tip.input ?? [] } : { continuationInput: [] };
+}
+async function buildSnapshotFromSessionEvents(server, sessionId, onProgress) {
   const window = await fetchSessionEventsWindow(server, sessionId);
   const historyPagination = {
     hasOlder: window.hasOlder,
@@ -1564,7 +1616,13 @@ async function buildSnapshotFromSessionEvents(server, sessionId, onProgress) {
     historyEvents: window.itemsAsc,
     historyPagination
   });
-  return attachRunningTurn(withHistory, runningTurn);
+  const tip = await resolveSessionTip({
+    server,
+    sessionId,
+    itemsAsc: window.itemsAsc
+  });
+  applyUserToolResponsesToFold(withHistory.fold, tip.continuationInput);
+  return attachRunningTurn(withHistory, tip.runningTurn);
 }
 async function prependOlderSessionHistory(server, sessionId, snapshot) {
   const pagination = snapshot.historyPagination;