@athenaintel/react 0.57.1 → 0.59.0

package/dist/threads/statewire-thread-list.js

@@ -1,7 +1,7 @@
 import { jsx } from "react/jsx-runtime";
 import { useOfflineSessionList } from "./offline-session-list.js";
 import { QueryClient, QueryClientProvider, useQueryClient } from "@tanstack/react-query";
-import { useAuiState } from "@assistant-ui/react";
+import { useAuiState, useAuiEvent } from "@assistant-ui/react";
 import { useContext, createContext, useState, useRef, useCallback, useMemo, useEffect } from "react";
 import { useAccountIdentityQuery } from "../auth/account-identity.js";
 import { AthenaAuthContext } from "../auth/AthenaAuthContext.js";
@@ -13,6 +13,7 @@ import { mintThreadId } from "./mint-thread-id.js";
 import { useAthenaStatewireThreadId } from "./thread-id-context.js";
 import { createThreadPrefetcher } from "./thread-prefetcher.js";
 import { useIndexTriggeredRefresh } from "./index-triggered-refresh.js";
+import { isAgentHomeIndexEntry, sortNewestFirst, withIndexedSessions, localSessionSend, isLocalSendReported, settledLocalSends, LOCAL_SEND_TTL_MS, inSessionListScope, withLocalSends, withoutLocalSends as withoutLocalSends$1 } from "./live-session-rows.js";
 import { useSessionListTabShare, sessionListShareKey } from "./session-list-tab-share.js";
 import { normalizeThreadTitle, isSerializedContentPartsTitle } from "./title.js";
 import { useRestartAfterAuthRefusal } from "../runtime/auth-refusal-restart.js";
@@ -21,7 +22,7 @@ import { useThreadArchive } from "./use-thread-archive.js";
 import { useThreadFavorite, withFavoriteOverrides, withFavoriteOverride } from "./use-thread-favorite.js";
 import { useThreadReadState, withReadStateOverrides } from "./use-thread-read-state.js";
 import { useThreadRename, withTitleOverrides, withTitleOverride } from "./use-thread-rename.js";
-import { deriveConversationAttention, summaryFromConversationIndexEntry, patchConversationListCaches, conversationListQueryKeys, conversationListFilterKey, removeConversationFromListCaches } from "../agent-runtime/src/conversation-list.js";
+import { deriveConversationAttention, summaryFromConversationIndexEntry, patchConversationListCaches, conversationListFilterKey, conversationListQueryKeys, removeConversationFromListCaches } from "../agent-runtime/src/conversation-list.js";
 import { useConversationList } from "../agent-runtime/src/use-conversation-list.js";
 import { useConversationIndex, conversationIndexUrlFromStatewireUrl } from "../agent-runtime/src/conversation-index.js";
 const AthenaStatewireThreadListContext = createContext(null);
@@ -32,14 +33,6 @@ function useRefreshThreadList() {
   var _a;
   return ((_a = useAthenaStatewireThreadList()) == null ? void 0 : _a.refresh) ?? null;
 }
-const AGENT_HOME_EXCLUDED_START_CHANNELS = /* @__PURE__ */ new Set([
-  "task",
-  "agent_orchestration"
-]);
-function isAgentHomeIndexEntry(entry, homeAgentId) {
-  if (!homeAgentId || entry.collab_agent_id !== homeAgentId) return false;
-  return !AGENT_HOME_EXCLUDED_START_CHANNELS.has(entry.start_channel ?? "");
-}
 function indexedThreadsToRefreshFor({
   entries,
   listedThreadIds,
@@ -98,9 +91,16 @@ function withPinnedConversations({
   return unlisted.length === 0 ? conversations : [...conversations, ...unlisted];
 }
 const AthenaSessionsLiveContext = createContext(null);
+const AthenaThreadListSendContext = createContext(null);
+const POLL_REFRESH_AFTER_SEND_MS = [3e3, 1e4];
 const THREAD_LIST_PAGE_SIZE = 50;
 const EMPTY_THREAD_IDS = /* @__PURE__ */ new Set();
+const EMPTY_LOCAL_SENDS = /* @__PURE__ */ new Map();
 const MAX_PINNED_PAGES = 10;
+function withoutLocalSends(state, threadIds) {
+  const sends = withoutLocalSends$1(state.sends, threadIds);
+  return sends === state.sends ? state : { ...state, sends };
+}
 function StatewireThreadListState({
   backendUrl,
   syncUrl,
@@ -249,6 +249,34 @@ function StatewireThreadListState({
     () => foldConversationIndex(listedConversations, indexByThread),
     [listedConversations, indexByThread]
   );
+  const homeAgentApplied = list.homeAgentApplied;
+  const scope = useMemo(
+    () => ({ appId, additionalAppIds: listedAppIds, homeAgentId, homeAgentApplied }),
+    [appId, listedAppIds, homeAgentId, homeAgentApplied]
+  );
+  const searching = trimmedSearch.length > 0;
+  const serverConversations = useMemo(
+    () => sortNewestFirst(
+      withIndexedSessions({
+        conversations: mergedConversations,
+        loadedConversations: list.conversations,
+        entries: index.entries,
+        scope,
+        searching,
+        hasResolvedFirstPage: list.hasResolvedFirstPage,
+        hasNextPage: list.hasNextPage
+      })
+    ),
+    [
+      mergedConversations,
+      list.conversations,
+      index.entries,
+      scope,
+      searching,
+      list.hasResolvedFirstPage,
+      list.hasNextPage
+    ]
+  );
   const queryClient = useQueryClient();
   const patchEveryList = useCallback(
     (threadId, patch) => patchConversationListCaches(queryClient, {
@@ -262,7 +290,7 @@ function StatewireThreadListState({
     backendUrl,
     principalId: principal,
     getAuth,
-    rows: mergedConversations,
+    rows: serverConversations,
     onServerTruth: ({ threadId, isFavorited }) => {
       patchEveryList(threadId, { isFavorited });
       if (isFavorited) void pinnedListRefresh();
@@ -273,7 +301,7 @@ function StatewireThreadListState({
     backendUrl,
     principalId: principal,
     getAuth,
-    rows: mergedConversations,
+    rows: serverConversations,
     onServerTruth: ({ threadId, title }) => patchEveryList(threadId, { title })
   });
   const titleOverrides = rename.overrides;
@@ -281,7 +309,7 @@ function StatewireThreadListState({
     backendUrl,
     principalId: principal,
     getAuth,
-    rows: mergedConversations,
+    rows: serverConversations,
     onServerTruth: ({ threadId, isUnread }) => patchEveryList(threadId, { isUnread })
   });
   const readOverrides = readState.overrides;
@@ -314,9 +342,110 @@ function StatewireThreadListState({
     ).filter((conversation) => !hiddenThreadIds.has(conversation.threadId)),
     [favoriteOverrides, titleOverrides, readOverrides, readCoveredUpdatedAt, hiddenThreadIds]
   );
+  const sendsScopeKey = `${listScopeKey}::${credentialGeneration}::${conversationListFilterKey({
+    ...appId ? { appId } : {},
+    ...listedAppIds.length > 0 ? { appIds: listedAppIds } : {},
+    ...homeAgentId ? { homeAgentId } : {}
+  })}`;
+  const [localSendState, setLocalSendState] = useState(() => ({ scopeKey: sendsScopeKey, sends: EMPTY_LOCAL_SENDS }));
+  const localSends = localSendState.scopeKey === sendsScopeKey ? localSendState.sends : EMPTY_LOCAL_SENDS;
+  const serverConversationsRef = useRef(serverConversations);
+  serverConversationsRef.current = serverConversations;
+  const sendsScopeKeyRef = useRef(sendsScopeKey);
+  sendsScopeKeyRef.current = sendsScopeKey;
+  const activeRef = useRef(active);
+  activeRef.current = active;
+  const indexLiveRef = useRef(index.entries !== null);
+  indexLiveRef.current = index.entries !== null;
+  const refreshAfterSendRef = useRef(() => void 0);
+  const pollRefreshTimersRef = useRef(/* @__PURE__ */ new Set());
+  useEffect(() => {
+    const timers = pollRefreshTimersRef.current;
+    return () => {
+      for (const timer of timers) clearTimeout(timer);
+      timers.clear();
+    };
+  }, []);
+  const noteSend = useCallback(({ threadId, text, startChannel }) => {
+    if (!threadId) return;
+    const rows = serverConversationsRef.current;
+    const send = localSessionSend({
+      threadId,
+      text,
+      sentAtMs: Date.now(),
+      newChat: activeRef.current.id === threadId && activeRef.current.isNew,
+      startChannel,
+      serverConversations: rows
+    });
+    const scopeKey = sendsScopeKeyRef.current;
+    setLocalSendState((current) => {
+      const sends = new Map(current.scopeKey === scopeKey ? current.sends : EMPTY_LOCAL_SENDS);
+      const previous = sends.get(threadId);
+      const serverRow = rows.find((row) => row.threadId === threadId);
+      sends.set(
+        threadId,
+        previous && !isLocalSendReported(previous, serverRow) ? {
+          ...send,
+          baselineUpdatedAt: previous.baselineUpdatedAt,
+          newChat: previous.newChat,
+          title: previous.title ?? send.title,
+          startChannel: previous.startChannel
+        } : send
+      );
+      return { scopeKey, sends };
+    });
+    if (!indexLiveRef.current && !send.newChat) {
+      for (const delay of POLL_REFRESH_AFTER_SEND_MS) {
+        const timer = setTimeout(() => {
+          pollRefreshTimersRef.current.delete(timer);
+          refreshAfterSendRef.current();
+        }, delay);
+        pollRefreshTimersRef.current.add(timer);
+      }
+    }
+  }, []);
+  useEffect(() => {
+    if (localSends.size === 0) return;
+    const settled = settledLocalSends({ sends: localSends, serverConversations, nowMs: Date.now() });
+    if (settled.length > 0) {
+      setLocalSendState((current) => withoutLocalSends(current, settled));
+      return;
+    }
+    let oldestMs = Number.POSITIVE_INFINITY;
+    for (const send of localSends.values()) oldestMs = Math.min(oldestMs, send.sentAtMs);
+    const timer = setTimeout(
+      () => setLocalSendState(
+        (current) => withoutLocalSends(
+          current,
+          [...current.sends.values()].filter((send) => Date.now() - send.sentAtMs >= LOCAL_SEND_TTL_MS).map((send) => send.threadId)
+        )
+      ),
+      Math.max(0, oldestMs + LOCAL_SEND_TTL_MS - Date.now()) + 1
+    );
+    return () => clearTimeout(timer);
+  }, [localSends, serverConversations]);
+  const showPendingSessions = useMemo(
+    () => !searching && inSessionListScope(
+      { app_id: appId ?? null, collab_agent_id: homeAgentId ?? null, start_channel: null },
+      scope
+    ),
+    [searching, appId, homeAgentId, scope]
+  );
+  const liveRows = useMemo(
+    () => withLocalSends({
+      conversations: serverConversations,
+      sends: localSends,
+      appId,
+      showPendingSessions
+    }),
+    [serverConversations, localSends, appId, showPendingSessions]
+  );
   const data = useMemo(
-    () => applyOverrides(mergedConversations).map(threadSummaryFromConversation),
-    [applyOverrides, mergedConversations]
+    () => sortNewestFirst(applyOverrides(liveRows.conversations)).map((conversation) => {
+      const summary = threadSummaryFromConversation(conversation);
+      return liveRows.pendingThreadIds.has(conversation.threadId) ? { ...summary, isPending: true } : summary;
+    }),
+    [applyOverrides, liveRows]
   );
   const isLoading = list.isLoading;
   const lastActiveConversationRef = useRef(null);
@@ -352,6 +481,7 @@ function StatewireThreadListState({
     void listRefresh();
     if (!pinnedFetchingRef.current) void pinnedListRefresh();
   }, [listRefresh, pinnedListRefresh]);
+  refreshAfterSendRef.current = refresh;
   const listFetchNextPage = list.fetchNextPage;
   const fetchNextPage = useCallback(() => {
     void listFetchNextPage();
@@ -360,7 +490,6 @@ function StatewireThreadListState({
     () => new Set(list.conversations.map((conversation) => conversation.threadId)),
     [list.conversations]
   );
-  const homeAgentApplied = list.homeAgentApplied;
   const unlistedIndexedThreads = useMemo(
     () => index.entries === null ? [] : indexedThreadsToRefreshFor({
       entries: index.entries,
@@ -404,14 +533,14 @@ function StatewireThreadListState({
   prefetchScopeRef.current = prefetchScope;
   const prefetcher = useMemo(() => {
     if (!snapshotCache) return null;
-    const scope = prefetchScope;
+    const scope2 = prefetchScope;
     return createThreadPrefetcher({
       cache: snapshotCache,
       fetchState: async (threadId) => withCredentialRenewal(
         getAuth,
         (auth) => getThreadState(backendUrl, auth, threadId, { skipCache: false })
       ),
-      isCurrent: () => prefetchScopeRef.current === scope
+      isCurrent: () => prefetchScopeRef.current === scope2
     });
   }, [snapshotCache, prefetchScope, backendUrl, getAuth]);
   const activeIdRef = useRef(active.id);
@@ -426,7 +555,7 @@ function StatewireThreadListState({
   const newThread = useCallback(() => {
     setActive({ id: mintThreadId(), isNew: true });
   }, []);
-  const isPersisted = data.some((thread) => thread.id === active.id) || indexByThread.has(active.id);
+  const isPersisted = mergedConversations.some((conversation) => conversation.threadId === active.id) || indexByThread.has(active.id);
   useEffect(() => {
     if (isPersisted) {
       setActive((current) => current.isNew ? { ...current, isNew: false } : current);
@@ -534,7 +663,7 @@ function StatewireThreadListState({
       tabShare
     ]
   );
-  return /* @__PURE__ */ jsx(AthenaStatewireThreadListContext.Provider, { value, children: /* @__PURE__ */ jsx(AthenaSessionsLiveContext.Provider, { value: live, children }) });
+  return /* @__PURE__ */ jsx(AthenaStatewireThreadListContext.Provider, { value, children: /* @__PURE__ */ jsx(AthenaSessionsLiveContext.Provider, { value: live, children: /* @__PURE__ */ jsx(AthenaThreadListSendContext.Provider, { value: noteSend, children }) }) });
 }
 function AthenaStatewireThreadListProvider({
   backendUrl,
@@ -583,7 +712,7 @@ function StatewireThreadTitlePoller() {
     return (((_a2 = s.optional.thread) == null ? void 0 : _a2.messages.length) ?? 0) > 0;
   });
   const titleSource = (list == null ? void 0 : list.titleSource) ?? "poll";
-  const entry = list == null ? void 0 : list.threads.find((thread) => thread.id === activeThreadId);
+  const entry = list == null ? void 0 : list.threads.find((thread) => thread.id === activeThreadId && !thread.isPending);
   const title = ((_a = entry == null ? void 0 : entry.title) == null ? void 0 : _a.trim()) ?? "";
   const hasTitle = title.length > 0 && !isSerializedContentPartsTitle(title);
   const visitRef = useRef(null);
@@ -618,10 +747,34 @@ function StatewireThreadTitlePoller() {
   }, [activeThreadId, hasMessages, hasTitle, titleSource]);
   return null;
 }
+function StatewireThreadListSendSync({
+  startChannel = null
+} = {}) {
+  const noteSend = useContext(AthenaThreadListSendContext);
+  const threadId = useAthenaStatewireThreadId();
+  const typedRef = useRef(/* @__PURE__ */ new Map());
+  useEffect(() => {
+    typedRef.current.clear();
+  }, [threadId]);
+  useAuiEvent({ scope: "thread", event: "composer.textChange" }, (event) => {
+    if (event.origin === "send") return;
+    typedRef.current.set(event.messageId ?? "", event.text ?? "");
+  });
+  useAuiEvent({ scope: "thread", event: "composer.send" }, (event) => {
+    const composer = event.messageId ?? "";
+    const typed = typedRef.current.get(composer) ?? null;
+    typedRef.current.delete(composer);
+    if (!noteSend || !threadId) return;
+    noteSend({ threadId, text: event.chars > 0 ? typed : null, startChannel });
+  });
+  return null;
+}
 export {
   AthenaSessionsLiveContext,
   AthenaStatewireThreadListContext,
   AthenaStatewireThreadListProvider,
+  AthenaThreadListSendContext,
+  StatewireThreadListSendSync,
   StatewireThreadTitlePoller,
   TITLE_POLL_DELAY_MS,
   foldConversationIndex,