From e5c44406fbb0165d54b3376fec5a9f8afeb8816d Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Wed, 17 May 2023 01:17:45 +0900
Subject: [PATCH] [fix] supamemory chat overflow

---
 src/ts/process/supaMemory.ts | 23 +++++++++++++----------
 1 file changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/ts/process/supaMemory.ts b/src/ts/process/supaMemory.ts
index cf48284a..c35a00ab 100644
--- a/src/ts/process/supaMemory.ts
+++ b/src/ts/process/supaMemory.ts
@@ -7,6 +7,8 @@ import { requestChatData } from "./request";
 
 export async function supaMemory(chats:OpenAIChat[],currentTokens:number,maxContextTokens:number,room:Chat,char:character|groupChat): Promise<{ currentTokens: number; chats: OpenAIChat[]; error?:string; memory?:string}>{
     const db = get(DataBase)
+    console.log("Memory: " + currentTokens)
+
     if(currentTokens > maxContextTokens){
         let coIndex = -1
         for(let i=0;i<chats.length;i++){
@@ -29,26 +31,28 @@ export async function supaMemory(chats:OpenAIChat[],currentTokens:number,maxCont
             const id = splited.splice(0,1)[0]
             const data = splited.join('\n')
 
-            for(let i=0;i<chats.length;i++){
+            let i =0;
+            while(true){
+                if(chats.length === 0){
+                    return {
+                        currentTokens: currentTokens,
+                        chats: chats,
+                        error: "SupaMemory: chat ID not found"
+                    }
+                }
                 if(chats[0].memo === id){
                     break
                 }
                 currentTokens -= (await tokenize(chats[0].content) + 1)
                 chats.splice(0, 1)
-            }
-
-            if(chats.length === 0){
-                return {
-                    currentTokens: currentTokens,
-                    chats: chats,
-                    error: "SupaMemory: chat ID not found"
-                }
+                i += 1
             }
 
             supaMemory = data
             currentTokens += await tokenize(supaMemory) + 1
         }
 
+
         if(currentTokens < maxContextTokens){
             chats.unshift({
                 role: "system",
@@ -134,7 +138,6 @@ export async function supaMemory(chats:OpenAIChat[],currentTokens:number,maxCont
             const tokenz = await tokenize(result + '\n\n') + 5
             currentTokens += tokenz
             supaMemory += result + '\n\n'
-            console.log(tokenz)
         }
 
         chats.unshift({