tweak memorize feature

Leask · Jul 24, 2024 · a1f4d6a · a1f4d6a
1 parent 49c28b0
commit a1f4d6a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/bot.mjs b/lib/bot.mjs
@@ -288,7 +288,7 @@ const memorize = async (ctx) => {
             received_text, response_text, ...collected.map(x => x.content)
         ])),
     };
-    await ignoreErrFunc(async () => {
+    return await ignoreErrFunc(async () => {
         event.distilled_vector = bot._.embedding
             ? await bot._.embedding(event.distilled) : [];
         switch (bot._.database.provider) {
@@ -297,7 +297,7 @@ const memorize = async (ctx) => {
                 break;
         }
         await bot._.database?.client?.upsert?.(table, event, { skipEcho: true });
-        bot._.memorize && await bot._.memorize(event);
+        return bot._.memorize && await bot._.memorize(event);
     }, logOptions);
 };