@truefoundry/assistant-ui-runtime 0.1.14 → 0.1.16

package/dist/index.d.ts

@@ -25,7 +25,6 @@ type TrueFoundryAgentRuntimeBaseOptions = ExternalStoreSharedOptions & {
     threadId?: string | undefined;
     onThreadIdChange?: ((threadId: string | undefined) => void) | undefined;
     onError?: ((error: unknown) => void) | undefined;
-    listEventsConcurrency?: number | undefined;
     /**
      * Optional filter forwarded to `listSessions({ agentId })`.
      * Omit for all chats; hosts that key agents by name pass that name as the id.

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;
@@ -2082,56 +2140,20 @@ async function buildSnapshotFromSession(server, sessionId, concurrency = DEFAULT
     pendingUser: void 0
   });
 }
-async function buildSnapshotBeforeTurn(server, sessionId, beforeTurnId, concurrency = DEFAULT_LIST_EVENTS_CONCURRENCY) {
-  const turns = await listSessionTurnsOrdered(server, sessionId);
-  const beforeIndex = turns.findIndex((turn) => turn.id === beforeTurnId);
-  if (beforeIndex === -1) {
-    throw new Error(`Turn ${beforeTurnId} not found in session`);
-  }
-  return buildSnapshotBeforeTurnIndex(
-    server,
-    sessionId,
-    beforeIndex,
-    concurrency,
-    turns
-  );
-}
-async function buildSnapshotBeforeTurnIndex(server, sessionId, turnIndex, _concurrency = DEFAULT_LIST_EVENTS_CONCURRENCY, orderedTurns) {
-  if (turnIndex <= 0) {
+async function buildSnapshotThroughTurn(server, sessionId, anchorTurnId) {
+  if (anchorTurnId == null) {
     return createEmptySessionSnapshot();
   }
-  const turns = orderedTurns ?? await listSessionTurnsOrdered(server, sessionId);
-  const turnsToInclude = turns.slice(0, turnIndex);
-  const lastTurnId = turnsToInclude.at(-1)?.id;
-  if (lastTurnId == null) {
-    return createEmptySessionSnapshot();
-  }
-  const items = await fetchAllSessionEvents(server, sessionId, { lastTurnId });
+  const items = await fetchAllSessionEvents(server, sessionId, {
+    lastTurnId: anchorTurnId
+  });
   const snapshot = createEmptySessionSnapshot();
   ingestSessionEventsIntoSnapshot(snapshot, items);
   return snapshot;
 }
-async function resolveGatewayBranchPreviousTurnId(server, sessionId, turnIndex, orderedTurns) {
-  if (turnIndex <= 0) {
-    return "none";
-  }
-  const turns = orderedTurns ?? await listSessionTurnsOrdered(server, sessionId);
-  return turns[turnIndex - 1]?.id ?? "none";
-}
 async function resolveGatewayBranchPreviousTurnIdForTurn(server, sessionId, turnId) {
-  const turns = await listSessionTurnsOrdered(server, sessionId);
-  const turnIndex = turns.findIndex((turn) => turn.id === turnId);
-  return resolveGatewayBranchPreviousTurnId(server, sessionId, turnIndex, turns);
-}
-async function listSessionTurnsOrdered(server, sessionId) {
-  const turns = await drainListPages(
-    (pageToken) => server.listTurns({
-      sessionId,
-      ...pageToken != null ? { pageToken } : {}
-    })
-  );
-  turns.reverse();
-  return turns;
+  const turn = await server.getTurn({ sessionId, turnId });
+  return turn.previousTurnId ?? "none";
 }
 async function buildTurnAssistantContent(server, sessionId, turn, foldState) {
   const state = foldState ?? new PeerThreadFoldState();
@@ -3127,7 +3149,6 @@ function useTrueFoundryAgentMessages({
   sessionId,
   isMain,
   isInitialSession,
-  listEventsConcurrency,
   onError,
   initializeSession,
   resolveConversationSessionId,
@@ -3658,11 +3679,10 @@ function useTrueFoundryAgentMessages({
           conversationSessionId,
           turnId
         );
-        rewound = await buildSnapshotBeforeTurn(
+        rewound = await buildSnapshotThroughTurn(
           server,
           conversationSessionId,
-          turnId,
-          listEventsConcurrency
+          previousTurnId === "none" ? null : previousTurnId
         );
         createdAtByMessageIdRef.current = /* @__PURE__ */ new Map();
         snapshotRef.current = rewound;
@@ -3678,13 +3698,7 @@ function useTrueFoundryAgentMessages({
         branchRollbackSnapshot: committed
       });
     },
-    [
-      cancel,
-      server,
-      listEventsConcurrency,
-      sendTurn,
-      sessionId
-    ]
+    [cancel, server, sendTurn, sessionId]
   );
   const resetFromTurn = useCallback2(
     async (turnId) => {
@@ -3794,7 +3808,6 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     agent,
     adapters,
     onError,
-    listEventsConcurrency,
     ...sharedOptions
   } = options;
   const draftBridgeRef = useRef3(
@@ -3855,7 +3868,6 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     sessionId,
     isMain,
     isInitialSession,
-    listEventsConcurrency,
     onError,
     initializeSession,
     getTurnHeaders: agent.mode === "draft" ? getTurnHeaders : void 0