Optimize summarization on query

69283f511f518eecfdda192345908e1fd74b4115

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

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

  * @returns {Promise<string>} Text to query
  */
 async function getQueryText(chat, initiator) {
-    let queryText = '';
+    let hashedMessages = chat
-    let i = 0;
+        .map(x => ({ text: String(substituteParams(x.mes)), hash: getStringHash(substituteParams(x.mes)) }))
+        .filter(x => x.text)
-    let hashedMessages = chat.map(x => ({ text: String(substituteParams(x.mes)), hash: getStringHash(substituteParams(x.mes)) }));
+        .reverse()
+        .slice(0, settings.query);
     if (initiator === 'chat' && settings.enabled_chats && settings.summarize && settings.summarize_sent) {
         hashedMessages = await summarize(hashedMessages, settings.summary_source);
+    }
-    for (const message of hashedMessages.slice().reverse()) {
+    const queryText = hashedMessages.map(x => x.text).join('\n');
-        if (message.text) {
-            queryText += message.text + '\n';
-            i++;
-        }
-        if (i === settings.query) {
-            break;
-        }
-    }
     return collapseNewlines(queryText).trim();
+}