bb7e7b64 · permissionBRICK/SillyTavern

 /**
  * @typedef {object} HashedMessage
  * @property {string} text - The hashed message text
+ * @property {number} hash - The hash used as the vector key
  */
 const MODULE_NAME = 'vectors';

 const moduleWorker = new ModuleWorkerWrapper(synchronizeChat);
+const cachedSummaries = new Map();
 /**
  * Gets the Collection ID for a file embedded in the chat.
  * @param {string} fileUrl URL of the file

             return;
+        }
+        // Clear all cached summaries to ensure that new ones are created
+        // upon request of a full vectorise
+        cachedSummaries.clear();
         const batchSize = 5;
         const elapsedLog = [];
         let finished = false;

 /**
  * Summarizes messages using the Extras API method.
- * @param {HashedMessage[]} hashedMessages Array of hashed messages
+ * @param {HashedMessage} element hashed message
- * @returns {Promise<HashedMessage[]>} Summarized messages
+ * @returns {Promise<boolean>} Sucess
  */
-async function summarizeExtra(hashedMessages) {
+async function summarizeExtra(element) {
-    for (const element of hashedMessages) {
+    try {
-        try {
+        const url = new URL(getApiUrl());
-            const url = new URL(getApiUrl());
+        url.pathname = '/api/summarize';
-            url.pathname = '/api/summarize';
-            const apiResult = await doExtrasFetch(url, {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                    'Bypass-Tunnel-Reminder': 'bypass',
-                },
-                body: JSON.stringify({
-                    text: element.text,
-                    params: {},
-                }),
-            });
-            if (apiResult.ok) {
+        const apiResult = await doExtrasFetch(url, {
-                const data = await apiResult.json();
+            method: 'POST',
-                element.text = data.summary;
+            headers: {
-            }
+                'Content-Type': 'application/json',
-        }
+                'Bypass-Tunnel-Reminder': 'bypass',
-        catch (error) {
+            },
-            console.log(error);
+            body: JSON.stringify({
+                text: element.text,
+                params: {},
+            }),
+        });
+        if (apiResult.ok) {
+            const data = await apiResult.json();
+            element.text = data.summary;
+        }
+    }
+    catch (error) {
+        console.log(error);
+        return false;
+    }
-    return hashedMessages;
+    return true;
+}
 /**
  * Summarizes messages using the main API method.
- * @param {HashedMessage[]} hashedMessages Array of hashed messages
+ * @param {HashedMessage} element hashed message
- * @returns {Promise<HashedMessage[]>} Summarized messages
+ * @returns {Promise<boolean>} Sucess
  */
-async function summarizeMain(hashedMessages) {
+async function summarizeMain(element) {
-    for (const element of hashedMessages) {
+    element.text = await generateRaw(element.text, '', false, false, settings.summary_prompt);
-        element.text = await generateRaw(element.text, '', false, false, settings.summary_prompt);
+    return true;
-    }
-    return hashedMessages;
+}
 /**
  * Summarizes messages using WebLLM.
- * @param {HashedMessage[]} hashedMessages Array of hashed messages
+ * @param {HashedMessage} element hashed message
- * @returns {Promise<HashedMessage[]>} Summarized messages
+ * @returns {Promise<boolean>} Sucess
  */
-async function summarizeWebLLM(hashedMessages) {
+async function summarizeWebLLM(element) {
     if (!isWebLlmSupported()) {
         console.warn('Vectors: WebLLM is not supported');
-        return hashedMessages;
+        return false;
+    }
-    for (const element of hashedMessages) {
+    const messages = [{ role:'system', content: settings.summary_prompt }, { role:'user', content: element.text }];
-        const messages = [{ role:'system', content: settings.summary_prompt }, { role:'user', content: element.text }];
+    element.text = await generateWebLlmChatPrompt(messages);
-        element.text = await generateWebLlmChatPrompt(messages);
-    }
-    return hashedMessages;
+    return true;
+}
 /**

  * @returns {Promise<HashedMessage[]>} Summarized messages
  */
 async function summarize(hashedMessages, endpoint = 'main') {
-    switch (endpoint) {
+    for (const element of hashedMessages) {
-        case 'main':
+        const cachedSummary = cachedSummaries.get(element.hash)
-            return await summarizeMain(hashedMessages);
+        if (!cachedSummary) {
-        case 'extras':
+            let sucess = true;
-            return await summarizeExtra(hashedMessages);
+            switch (endpoint) {
-        case 'webllm':
+                case 'main':
-            return await summarizeWebLLM(hashedMessages);
+                    sucess = await summarizeMain(element);
-        default:
+                    break;
-            console.error('Unsupported endpoint', endpoint);
+                case 'extras':
+                    sucess = await summarizeExtra(element);
+                    break;
+                case 'webllm':
+                    sucess = await summarizeWebLLM(element);
+                    break;
+                default:
+                    console.error('Unsupported endpoint', endpoint);
+                    sucess = false;
+                    break;
+            }
+            if (sucess) {
+                cachedSummaries.set(element.hash, element.text);
+            } else {
+                break;
+            }
+        } else {
+            element.text = cachedSummary;
+        }
+    }
+    return hashedMessages;
+}
 async function synchronizeChat(batchSize = 5) {

             return -1;
+        }
-        let hashedMessages = context.chat.filter(x => !x.is_system).map(x => ({ text: String(substituteParams(x.mes)), hash: getStringHash(substituteParams(x.mes)), index: context.chat.indexOf(x) }));
+        const hashedMessages = context.chat.filter(x => !x.is_system).map(x => ({ text: String(substituteParams(x.mes)), hash: getStringHash(substituteParams(x.mes)), index: context.chat.indexOf(x) }));
         const hashesInCollection = await getSavedHashes(chatId);
-        if (settings.summarize) {
+        let newVectorItems = hashedMessages.filter(x => !hashesInCollection.includes(x.hash));
-            hashedMessages = await summarize(hashedMessages, settings.summary_source);
-        }
-        const newVectorItems = hashedMessages.filter(x => !hashesInCollection.includes(x.hash));
         const deletedHashes = hashesInCollection.filter(x => !hashedMessages.some(y => y.hash === x));
+        if (settings.summarize) {
+            newVectorItems = await summarize(newVectorItems, settings.summary_source);
+        }
         if (newVectorItems.length > 0) {
             const chunkedBatch = splitByChunks(newVectorItems.slice(0, batchSize));

Add client side cacheing of vector summaries