@adminide-stack/yantra-mobile 12.0.53-alpha.2 → 12.0.53-alpha.21

package/lib/hooks/useChatApi.js

@@ -1,4 +1,4 @@
-import {useApolloClient}from'@apollo/client/index.js';import {SortEnum,RoomType,PostTypeEnum,AiAgentMessageRole}from'common';import {useGetChannelsByUserWithLastMessageQuery,useMessagesQuery,useAddChannelMutation,useSendMessagesMutation,MessagesDocument,GetChannelsByUserWithLastMessageDocument}from'common/graphql';import {useMemo,useCallback}from'react';import {v4}from'uuid';import {isAttachedCaption,historyAttachmentTitle,attachmentsFromUserContent}from'../features/attachments/historyAttachmentLabel.js';var __defProp = Object.defineProperty;
+import {useApolloClient}from'@apollo/client/index.js';import {SortEnum,RoomType,PostTypeEnum,AiAgentMessageRole}from'common';import {useGetChannelsByUserWithLastMessageQuery,useMessagesQuery,useAddChannelMutation,useSendMessagesMutation,OnChatMessageAddedDocument,MessagesDocument,GetChannelsByUserWithLastMessageDocument}from'common/graphql';import {useMemo,useCallback,useEffect}from'react';import {v4}from'uuid';import {isAttachedCaption,historyAttachmentTitle,attachmentsFromUserContent}from'../features/attachments/historyAttachmentLabel.js';import {stripAskUser}from'../features/chat/askUser.js';import {parseToolActivity}from'../features/chat/toolActivity.js';var __defProp = Object.defineProperty;
 var __defProps = Object.defineProperties;
 var __getOwnPropDescs = Object.getOwnPropertyDescriptors;
 var __getOwnPropSymbols = Object.getOwnPropertySymbols;
@@ -81,11 +81,7 @@ function getThreadMessagesQueryVariables(sessionId) {
   return {
     channelId: sessionId,
     limit: MESSAGES_PAGE_LIMIT,
-    skip: 0,
-    sort: {
-      key: "createdAt",
-      value: SortEnum.Asc
-    }
+    skip: 0
   };
 }
 function getChatHistoryMessagesQueryVariables(accountUserId, options) {
@@ -215,7 +211,7 @@ function cleanMessageText(raw) {
   return raw.replace(ACTIVE_CONNECTORS_PREFIX_RE, "").replace(/\s+/g, " ").trim();
 }
 function isTransientHistoryText(text) {
-  const t = cleanMessageText(text);
+  const t = cleanHistoryPreview(text) || cleanMessageText(text);
   if (!t) return true;
   if (/^thinking[.…]*$/i.test(t)) return true;
   if (/^let me check on that requested skill/i.test(t)) return true;
@@ -232,10 +228,14 @@ function looksLikeAssistantReply(text, role) {
   return false;
 }
 function cleanHistoryPreview(raw) {
-  let t = cleanMessageText(raw);
+  const withoutAsk = stripAskUser(raw);
+  const {
+    text: withoutTools
+  } = parseToolActivity(withoutAsk, false);
+  let t = cleanMessageText(withoutTools);
   t = t.replace(/^(thinking[.…]*\s*)+/i, "").trim();
   t = t.replace(/^let me check on that requested skill[^\n.!?]*[.!?-]?\s*/i, "").trim();
-  t = t.replace(/^(?:⚙️[^\n]*\n)+\s*/g, "").trim();
+  t = t.replace(/^(?:[⚙🔧⚙️⚠][^\n]*\n)+\s*/gu, "").trim();
   return t;
 }
 function buildSessionFromChannel(channel) {
@@ -298,26 +298,151 @@ function chatHistorySessionsFromChannels(data) {
   return channels.filter((c) => Boolean(c == null ? void 0 : c.id) && (!(c == null ? void 0 : c.type) || c.type === RoomType.Aiassistant)).map((c) => buildSessionFromChannel(c)).filter((row) => row !== null).sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
 }
 const rememberedHistoryTitles = /* @__PURE__ */ new Map();
+const rememberedHistoryListeners = /* @__PURE__ */ new Set();
+function subscribeRememberedHistory(listener) {
+  rememberedHistoryListeners.add(listener);
+  return () => {
+    rememberedHistoryListeners.delete(listener);
+  };
+}
+function mergeRememberedHistory(channelId, patch, notify = true) {
+  var _a, _b;
+  const existing = (_a = rememberedHistoryTitles.get(channelId)) != null ? _a : {};
+  const next = {
+    title: patch.title || existing.title,
+    preview: patch.preview || existing.preview,
+    isAttachment: (_b = patch.isAttachment) != null ? _b : existing.isAttachment,
+    running: patch.running !== void 0 ? patch.running : existing.running
+  };
+  rememberedHistoryTitles.set(channelId, next);
+  if (notify && (next.title !== existing.title || next.preview !== existing.preview || next.running !== existing.running)) {
+    rememberedHistoryListeners.forEach((fn) => fn());
+  }
+  return next;
+}
 function rememberHistoryTitle(session) {
-  var _a;
-  if (session.isPlaceholder) return;
-  const title = (_a = session.title) == null ? void 0 : _a.trim();
-  if (!title || /^new chat$/i.test(title) || isTransientHistoryText(title)) return;
-  rememberedHistoryTitles.set(session.channelId, {
-    title,
+  var _a, _b;
+  const existing = rememberedHistoryTitles.get(session.channelId);
+  const title = session.isPlaceholder || /^new chat$/i.test(session.title) ? "" : (_a = session.title) == null ? void 0 : _a.trim();
+  if (existing == null ? void 0 : existing.running) {
+    if (title && !isTransientHistoryText(title)) {
+      mergeRememberedHistory(session.channelId, {
+        title
+      }, false);
+    }
+    return;
+  }
+  const preview = ((_b = session.preview) == null ? void 0 : _b.trim()) && !isTransientHistoryText(session.preview) ? session.preview.trim() : "";
+  if (title && isTransientHistoryText(title) && !preview) return;
+  if (!title && !preview) return;
+  mergeRememberedHistory(session.channelId, __spreadProps(__spreadValues(__spreadValues({}, title && !isTransientHistoryText(title) ? {
+    title
+  } : {}), preview ? {
+    preview
+  } : {}), {
     isAttachment: session.isAttachment
+  }), false);
+}
+function deriveHistoryTitleFromPrompt(rawPrompt) {
+  const cleaned = cleanMessageText(rawPrompt);
+  if (!cleaned || isTransientHistoryText(cleaned) || /^new chat$/i.test(cleaned)) return "";
+  if (cleaned.length <= 64) return cleaned;
+  return `${cleaned.slice(0, 64).trim()}...`;
+}
+function patchChannelHistoryTitle(client, channelId, rawPrompt) {
+  const cleaned = deriveHistoryTitleFromPrompt(rawPrompt);
+  if (!channelId || !cleaned) return;
+  mergeRememberedHistory(channelId, {
+    title: cleaned,
+    isAttachment: isAttachedCaption(cleaned)
+  });
+  const channelCacheId = client.cache.identify({
+    __typename: "Channel",
+    id: channelId
+  });
+  if (!channelCacheId) return;
+  try {
+    client.cache.modify({
+      id: channelCacheId,
+      fields: {
+        title(existing) {
+          return cleanChannelTitle(existing) ? existing : cleaned;
+        }
+        // Do not touch updatedAt here. Bumping it on open/hydrate moves the
+        // row to TODAY / "now". Activity time is only written when the user
+        // actually sends (touchChannelHistoryUpdatedAt).
+      }
+    });
+  } catch (err) {
+    console.warn("[useChatApi] patchChannelHistoryTitle failed:", err);
+  }
+}
+function touchChannelHistoryUpdatedAt(client, channelId) {
+  if (!channelId) return;
+  const channelCacheId = client.cache.identify({
+    __typename: "Channel",
+    id: channelId
+  });
+  if (!channelCacheId) return;
+  try {
+    client.cache.modify({
+      id: channelCacheId,
+      fields: {
+        updatedAt() {
+          return (/* @__PURE__ */ new Date()).toISOString();
+        }
+      }
+    });
+  } catch (err) {
+    console.warn("[useChatApi] touchChannelHistoryUpdatedAt failed:", err);
+  }
+}
+function patchChannelHistoryPreview(channelId, rawPreview) {
+  var _a, _b;
+  if (!channelId) return;
+  const preview = cleanHistoryPreview(rawPreview);
+  if (!preview || isTransientHistoryText(preview) || isAttachedCaption(preview)) return;
+  const title = (_b = (_a = rememberedHistoryTitles.get(channelId)) == null ? void 0 : _a.title) != null ? _b : "";
+  if (title && preview === title) return;
+  mergeRememberedHistory(channelId, {
+    preview
+  });
+}
+function markChannelHistoryRunning(channelId, rawPreview) {
+  if (!channelId) return;
+  const preview = rawPreview ? cleanHistoryPreview(rawPreview) : "";
+  const usable = preview && !isTransientHistoryText(preview) && !isAttachedCaption(preview) ? preview : void 0;
+  mergeRememberedHistory(channelId, __spreadValues({
+    running: true
+  }, usable ? {
+    preview: usable
+  } : {}));
+}
+function clearChannelHistoryRunning(channelId) {
+  if (!channelId) return;
+  const existing = rememberedHistoryTitles.get(channelId);
+  if (!(existing == null ? void 0 : existing.running)) return;
+  mergeRememberedHistory(channelId, {
+    running: false
   });
 }
 function applyRememberedHistoryTitles(rows) {
   return rows.map((row) => {
+    var _a;
     rememberHistoryTitle(row);
-    if (!row.isPlaceholder && !/^new chat$/i.test(row.title)) return row;
     const mem = rememberedHistoryTitles.get(row.channelId);
     if (!mem) return row;
+    const titleMissing = row.isPlaceholder || /^new chat$/i.test(row.title);
+    const previewMissing = !((_a = row.preview) == null ? void 0 : _a.trim());
+    const nextTitle = titleMissing && mem.title ? mem.title : row.title;
+    const livePreview = mem.running && mem.preview && mem.preview !== nextTitle ? mem.preview : previewMissing && mem.preview && mem.preview !== nextTitle ? mem.preview : row.preview;
+    if (!titleMissing && !previewMissing && !mem.running) return row;
     return __spreadProps(__spreadValues({}, row), {
-      title: mem.title,
-      isPlaceholder: false,
-      isAttachment: mem.isAttachment || row.isAttachment
+      title: nextTitle,
+      preview: livePreview,
+      isPlaceholder: titleMissing && mem.title ? false : row.isPlaceholder,
+      isAttachment: mem.isAttachment || row.isAttachment,
+      isRunning: Boolean(mem.running)
     });
   });
 }
@@ -490,19 +615,23 @@ function mapPostToChatMessageUI(post, fallbackChannelId) {
   };
 }
 function useChatMessages(sessionId, options) {
+  const client = useApolloClient();
   const {
     data,
     loading,
     error,
-    refetch
+    refetch,
+    subscribeToMore
   } = useMessagesQuery({
     variables: sessionId ? getThreadMessagesQueryVariables(sessionId) : void 0,
     skip: !sessionId || (void 0 ),
-    // cache-first + `messages` typePolicy (keyed by channelId). Opening a thread
-    // that was already loaded does not refetch. New channels still hit the
-    // network once. saveMessages writes new posts into this cache slot.
-    fetchPolicy: "cache-first",
+    // Same as browser: paint cache immediately, then hit the network so a thread
+    // opened from history is not stuck on a cache-first snapshot that only has
+    // the latest user post (assistant replies persist after the first query).
+    fetchPolicy: "cache-and-network",
     nextFetchPolicy: "cache-first",
+    errorPolicy: "all",
+    notifyOnNetworkStatusChange: true,
     /**
      * Cache key is per-channel so switching sessions doesn't read another channel's response.
      * Keeping this as a constant ('messages-list') used to cause cross-session bleed in the
@@ -512,12 +641,77 @@ function useChatMessages(sessionId, options) {
       cacheKey: sessionId ? `messages-list:${sessionId}` : "messages-list"
     }
   });
+  useEffect(() => {
+    if (!sessionId || (void 0 )) return;
+    const unsubscribe = subscribeToMore({
+      document: OnChatMessageAddedDocument,
+      variables: {
+        channelId: sessionId
+      },
+      updateQuery: (prev, {
+        subscriptionData
+      }) => {
+        var _a, _b, _c, _d, _e;
+        const post = (_a = subscriptionData == null ? void 0 : subscriptionData.data) == null ? void 0 : _a.chatMessageAdded;
+        if (!(post == null ? void 0 : post.id)) return prev;
+        if (!(prev == null ? void 0 : prev.messages)) return prev;
+        const existing = (_b = prev.messages.data) != null ? _b : [];
+        const idx = existing.findIndex((row) => (row == null ? void 0 : row.id) === post.id);
+        let nextData;
+        let nextTotal;
+        if (idx >= 0) {
+          nextData = [...existing.slice(0, idx), post, ...existing.slice(idx + 1)];
+          nextTotal = (_c = prev.messages.totalCount) != null ? _c : existing.length;
+        } else {
+          nextData = [...existing, post];
+          nextTotal = ((_d = prev.messages.totalCount) != null ? _d : existing.length) + 1;
+          const parentId = (_e = post.parentId) != null ? _e : null;
+          if (parentId) {
+            nextData = nextData.map((root) => {
+              var _a2, _b2, _c2, _d2, _e2, _f, _g;
+              if ((root == null ? void 0 : root.id) !== parentId) return root;
+              const replies = (_b2 = (_a2 = root.replies) == null ? void 0 : _a2.data) != null ? _b2 : [];
+              if (replies.some((reply) => (reply == null ? void 0 : reply.id) === post.id)) return root;
+              return __spreadProps(__spreadValues({}, root), {
+                replies: __spreadProps(__spreadValues({}, (_c2 = root.replies) != null ? _c2 : {}), {
+                  __typename: (_e2 = (_d2 = root.replies) == null ? void 0 : _d2.__typename) != null ? _e2 : "Messages",
+                  data: [...replies, post],
+                  totalCount: ((_g = (_f = root.replies) == null ? void 0 : _f.totalCount) != null ? _g : replies.length) + 1
+                })
+              });
+            });
+          }
+        }
+        const seen = /* @__PURE__ */ new Set();
+        nextData = nextData.filter((row) => {
+          if (!(row == null ? void 0 : row.id) || seen.has(row.id)) return false;
+          seen.add(row.id);
+          return true;
+        });
+        return __spreadProps(__spreadValues({}, prev), {
+          messages: __spreadProps(__spreadValues({}, prev.messages), {
+            data: nextData,
+            totalCount: nextTotal
+          })
+        });
+      },
+      onError: (err) => {
+        console.error("[useChatMessages] subscribeToMore error:", err);
+      }
+    });
+    return () => unsubscribe();
+  }, [sessionId, void 0 , subscribeToMore, client]);
   const messagesLoaded = data !== void 0;
   const messages = useMemo(() => {
     var _a, _b;
     if (!sessionId) return [];
     const rows = (_b = (_a = data == null ? void 0 : data.messages) == null ? void 0 : _a.data) != null ? _b : [];
-    return flattenPostsWithReplies(rows, sessionId);
+    const ownRows = rows.filter((post) => {
+      var _a2;
+      const postChannelId = (_a2 = post == null ? void 0 : post.channel) == null ? void 0 : _a2.id;
+      return !postChannelId || postChannelId === sessionId;
+    });
+    return flattenPostsWithReplies(ownRows, sessionId);
   }, [data, sessionId]);
   return {
     messages,
@@ -775,13 +969,18 @@ function useChatMutations() {
       }
     };
   }, [client, sendMessagesMutation]);
+  const patchChannelTitle = useCallback((channelId, rawPrompt, bumpActivity = false) => {
+    patchChannelHistoryTitle(client, channelId, rawPrompt);
+    if (bumpActivity) touchChannelHistoryUpdatedAt(client, channelId);
+  }, [client]);
   return {
     createChannel,
     createSession: createChannel,
     saveMessages,
+    patchChannelTitle,
     loading: {
       create: createChannelLoading,
       saveMessages: sendMessagesLoading
     }
   };
-}export{AI_ASSISTANT_CHANNELS_QUERY_VARS,HISTORY_PAGE_SIZE,HISTORY_QUERY_BASE,buildSessionFromChannel,chatHistorySessionsFromChannels,chatHistorySessionsFromMessages,enrichHistorySessionsWithUserPrompts,getChatHistoryChannelRefetchQueries,getChatHistoryMessagesQueryVariables,getHistoryChannelsQueryVariables,useChatHistorySessionsFromChannels,useChatHistorySessionsFromMessages,useChatMessages,useChatMutations,usePrefetchChatHistory};//# sourceMappingURL=useChatApi.js.map
+}export{AI_ASSISTANT_CHANNELS_QUERY_VARS,HISTORY_PAGE_SIZE,HISTORY_QUERY_BASE,buildSessionFromChannel,chatHistorySessionsFromChannels,chatHistorySessionsFromMessages,clearChannelHistoryRunning,enrichHistorySessionsWithUserPrompts,getChatHistoryChannelRefetchQueries,getChatHistoryMessagesQueryVariables,getHistoryChannelsQueryVariables,markChannelHistoryRunning,patchChannelHistoryPreview,patchChannelHistoryTitle,subscribeRememberedHistory,touchChannelHistoryUpdatedAt,useChatHistorySessionsFromChannels,useChatHistorySessionsFromMessages,useChatMessages,useChatMutations,usePrefetchChatHistory};//# sourceMappingURL=useChatApi.js.map