@truefoundry/assistant-ui-runtime 0.1.15 → 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

@@ -2140,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();
@@ -3185,7 +3149,6 @@ function useTrueFoundryAgentMessages({
   sessionId,
   isMain,
   isInitialSession,
-  listEventsConcurrency,
   onError,
   initializeSession,
   resolveConversationSessionId,
@@ -3716,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;
@@ -3736,13 +3698,7 @@ function useTrueFoundryAgentMessages({
         branchRollbackSnapshot: committed
       });
     },
-    [
-      cancel,
-      server,
-      listEventsConcurrency,
-      sendTurn,
-      sessionId
-    ]
+    [cancel, server, sendTurn, sessionId]
   );
   const resetFromTurn = useCallback2(
     async (turnId) => {
@@ -3852,7 +3808,6 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     agent,
     adapters,
     onError,
-    listEventsConcurrency,
     ...sharedOptions
   } = options;
   const draftBridgeRef = useRef3(
@@ -3913,7 +3868,6 @@ function useTrueFoundryAgentRuntimeImpl(options, pendingAgentSpecRef) {
     sessionId,
     isMain,
     isInitialSession,
-    listEventsConcurrency,
     onError,
     initializeSession,
     getTurnHeaders: agent.mode === "draft" ? getTurnHeaders : void 0