Vector Storage: Fix query summarizing for WI and files

c88edaa4ed20c6b22ca89fff91b2001a2fb08c8e

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/scripts/extensions/vectors/index.js+7 -6 Viewed

         const dataBankCollectionIds = await ingestDataBankAttachments();
         if (dataBankCollectionIds.length) {
-            const queryText = await getQueryText(chat);
+            const queryText = await getQueryText(chat, 'file');
             await injectDataBankChunks(queryText, dataBankCollectionIds);
+        }

                 await vectorizeFile(fileText, fileName, collectionId, settings.chunk_size, settings.overlap_percent);
+            }
-            const queryText = await getQueryText(chat);
+            const queryText = await getQueryText(chat, 'file');
             const fileChunks = await retrieveFileChunks(queryText, collectionId);
             message.mes = `${fileChunks}\n\n${message.mes}`;

             return;
+        }
-        const queryText = await getQueryText(chat);
+        const queryText = await getQueryText(chat, 'chat');
         if (queryText.length === 0) {
             console.debug('Vectors: No text to query');

 /**
  * Gets the text to query from the chat
  * @param {object[]} chat Chat messages
+ * @param {'file'|'chat'|'world-info'} initiator Initiator of the query
  * @returns {Promise<string>} Text to query
  */
-async function getQueryText(chat) {
+async function getQueryText(chat, initiator) {
     let queryText = '';
     let i = 0;
     let hashedMessages = chat.map(x => ({ text: String(substituteParams(x.mes)) }));
-    if (settings.summarize && settings.summarize_sent) {
+    if (initiator === 'chat' && settings.enabled_chats && settings.summarize && settings.summarize_sent) {
         hashedMessages = await summarize(hashedMessages, settings.summary_source);
+    }

+    }
     // Perform a multi-query
-    const queryText = await getQueryText(chat);
+    const queryText = await getQueryText(chat, 'world-info');
     if (queryText.length === 0) {
         console.debug('Vectors: No text to query for WI');