From 336f0d3fd084ca44f16c2c57864484e34a4675ef Mon Sep 17 00:00:00 2001
From: MDude <mdude1324@gmail.com>
Date: Sun, 30 Apr 2023 01:39:29 -0500
Subject: [PATCH] Store memory on most recent, if only one message

---
 public/scripts/extensions/memory/index.js | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/public/scripts/extensions/memory/index.js b/public/scripts/extensions/memory/index.js
index 96399d177..0b3e1a38e 100644
--- a/public/scripts/extensions/memory/index.js
+++ b/public/scripts/extensions/memory/index.js
@@ -305,8 +305,9 @@ function setMemoryContext(value, saveToMessage) {
     context.setExtensionPrompt(MODULE_NAME, formatMemoryValue(value), extension_prompt_types.AFTER_SCENARIO);
     $('#memory_contents').val(value);
 
-    if (saveToMessage && context.chat.length > 1) {
-        const mes = context.chat[context.chat.length - 2];
+    if (saveToMessage && context.chat.length) {
+        const idx = context.chat.length - 2;
+        const mes = context.chat[idx < 0 ? 0 : idx];
 
         if (!mes.extra) {
             mes.extra = {};