feature: followup uses groups ids in group mode

Author: cdxker

clients/search-component/src/utils/hooks/chat-context.tsx

@@ -139,6 +139,7 @@ function ChatProvider({ children }: { children: React.ReactNode }) {
   const [productsWithClicks, setProductsWithClicks] = useState<
     ChunkIdWithIndex[]
   >([]);
+  const [groupIdsInChat, setGroupIdsInChat] = useState<string[]>([]);
 
   const createTopic = async ({
     question,
@@ -573,7 +574,7 @@ function ChatProvider({ children }: { children: React.ReactNode }) {
         });
 
         const skipSearchPromise = retryOperation(async () => {
-          if (props.type === "ecommerce" && !curGroup && messages.length > 1) {
+          if (!curGroup && messages.length > 1) {
             return await trieveSDK.getToolCallFunctionParams({
               user_message_text: `Here's the previous message thread so far: ${messages.map(
                 (message) => {
@@ -1031,6 +1032,8 @@ function ChatProvider({ children }: { children: React.ReactNode }) {
             ],
           };
 
+          setGroupIdsInChat((prev) => [...prev, ...topGroupIds]);
+
           try {
             const topImageGroupIds = topGroupIds.slice(0, 3);
             const getChunksPromises = topImageGroupIds.map((groupId) =>
@@ -1151,17 +1154,25 @@ function ChatProvider({ children }: { children: React.ReactNode }) {
           createMessageFilters = filters;
         }
         if (skipSearch) {
-          createMessageFilters = {
-            must: [
-              {
-                field: "ids",
-                match_any: messages
-                  .filter((msg) => msg.type == "system")
-                  .flatMap((msg) => msg.additional ?? [])
-                  .map((chunk) => chunk.id),
-              },
-            ],
-          };
+          createMessageFilters = props.useGroupSearch
+            ? {
+              must: [
+                {
+                  field: "group_ids",
+                  match_any: groupIdsInChat,
+                },
+              ],
+            }
+            : {
+              must: [
+                {
+                  field: "ids",
+                  match_any: messages
+                    .flatMap((m) => m.additional ?? [])
+                    .map((chunk) => chunk.id),
+                },
+              ],
+            };
         }
         const systemPromptToUse =
           props.systemPrompt && props.systemPrompt !== ""