From 7ea768cb5bcd9e4013af7b0e1f736b8ab54e6ce0 Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Wed, 31 May 2023 05:08:46 +0900
Subject: [PATCH 1/2] [feat] better group chat

---
 src/lang/en.ts                     |   6 +-
 src/lang/ko.ts                     |  10 ++-
 src/lib/Others/Check.svelte        |   3 +-
 src/lib/SideBars/CharConfig.svelte |  73 ++++++++++----------
 src/ts/characters.ts               |  18 ++++-
 src/ts/process/group.ts            | 103 +++++++++++++++++++++++++++++
 src/ts/process/index.ts            |  45 +++++++++----
 src/ts/process/request.ts          |   2 +-
 src/ts/storage/database.ts         |   5 +-
 9 files changed, 208 insertions(+), 57 deletions(-)
 create mode 100644 src/ts/process/group.ts

diff --git a/src/lang/en.ts b/src/lang/en.ts
index 6b267ad8..896ddd99 100644
--- a/src/lang/en.ts
+++ b/src/lang/en.ts
@@ -286,5 +286,9 @@ export const languageEnglish = {
     setNodePassword: "Set your password to security",
     inputNodePassword: "Input your password. if you can't remember, remove save/__password.txt in your server files and restart the server.",
     simple:"Simple",
-    advanced: "Advanced"
+    advanced: "Advanced",
+    orderByOrder: "Talk by Order",
+    removeFromGroup: "Do you really want to remove {{char}} from group?",
+    talkness: "Talkativeness",
+    active: "Active"
 }
\ No newline at end of file
diff --git a/src/lang/ko.ts b/src/lang/ko.ts
index ec27b890..7553d90f 100644
--- a/src/lang/ko.ts
+++ b/src/lang/ko.ts
@@ -261,5 +261,13 @@ export const languageKorean = {
     globalRegexScript: "글로별 정규식",
     accessibility: "접근성",
     sendWithEnter: "엔터키로 메세지 보내기",
-    clickToEdit: "클릭해서 수정하기"
+    clickToEdit: "클릭해서 수정하기",
+    setNodePassword: "보안을 위해 비밀번호를 정해주세요",
+    inputNodePassword: "비밀번호를 입력해주세요. 기억이 안나신다면, save/__password를 지우고 서버를 재시작해주세요.",
+    simple:"간단",
+    advanced: "고급",
+    orderByOrder: "순서대로 말하기",
+    removeFromGroup: "정말로 {{char}}을 그룹에서 삭제시키겠습니까?",
+    talkness: "대화량",
+    active: "활성화를"
 }
\ No newline at end of file
diff --git a/src/lib/Others/Check.svelte b/src/lib/Others/Check.svelte
index e1cab220..f1d8c23e 100644
--- a/src/lib/Others/Check.svelte
+++ b/src/lib/Others/Check.svelte
@@ -3,9 +3,10 @@
 
     export let check = false
     export let onChange = (check) => {}
+    export let margin = true
 </script>
 
-<label class="mr-2">
+<label class:mr-2={margin}>
     <input type="checkbox" class="hidden" bind:checked={check} on:change={() => {
         onChange(check)
     }}>
diff --git a/src/lib/SideBars/CharConfig.svelte b/src/lib/SideBars/CharConfig.svelte
index 37a8d07e..ee3f5c19 100644
--- a/src/lib/SideBars/CharConfig.svelte
+++ b/src/lib/SideBars/CharConfig.svelte
@@ -3,7 +3,7 @@
     import { tokenize } from "../../ts/tokenizer";
     import { DataBase, saveImage as saveAsset, type Database, type character, type groupChat } from "../../ts/storage/database";
     import { selectedCharID } from "../../ts/stores";
-    import { PlusIcon, SmileIcon, TrashIcon, UserIcon, ActivityIcon, BookIcon, LoaderIcon, User, DnaIcon, CurlyBracesIcon, Volume2Icon } from 'lucide-svelte'
+    import { PlusIcon, SmileIcon, TrashIcon, UserIcon, ActivityIcon, BookIcon, LoaderIcon, User, DnaIcon, CurlyBracesIcon, Volume2Icon, XIcon } from 'lucide-svelte'
     import Check from "../Others/Check.svelte";
     import { addCharEmotion, addingEmotion, getCharImage, rmCharEmotion, selectCharImg, makeGroupImage } from "../../ts/characters";
     import LoreBook from "./LoreBookSetting.svelte";
@@ -17,6 +17,7 @@
     import { exportChar } from "src/ts/characterCards";
     import { getElevenTTSVoices, getWebSpeechTTSVoices, getVOICEVOXVoices } from "src/ts/process/tts";
     import { checkCharOrder } from "src/ts/storage/globalApi";
+  import { addGroupChar, rmCharFromGroup } from "src/ts/process/group";
 
     let subMenu = 0
     let subberMenu = 0
@@ -58,41 +59,6 @@
 
     }
 
-
-    async function addGroupChar(){
-        let group = currentChar.data
-        if(group.type === 'group'){
-            const res = await alertSelectChar()
-            if(res){
-                if(group.characters.includes(res)){
-                    alertError(language.errors.alreadyCharInGroup)
-                }
-                else{
-                    if(await alertConfirm(language.askLoadFirstMsg)){
-                        group.chats[group.chatPage].message.push({
-                            role:'char',
-                            data: findCharacterbyId(res).firstMessage,
-                            saying: res,
-                        })
-                    }
-
-                    group.characters.push(res)
-                    currentChar.data = group
-                }
-            }
-        }
-        currentChar = currentChar
-    }
-
-
-    function rmCharFromGroup(index:number){
-        let group = currentChar.data
-        if(group.type === 'group'){
-            group.characters.splice(index, 1)
-            currentChar.data = group
-        }
-    }
-
     let database:Database
     let currentChar:{
         type: 'character',
@@ -182,10 +148,13 @@
     {:else}
         <input class="text-neutral-200 mt-2 mb-4 p-2 bg-transparent input-text text-xl focus:bg-selected" placeholder="Group Name" bind:value={currentChar.data.name}>
         <span class="text-neutral-200">{language.character}</span>
-        <div class="p-2 flex gap-2">
+        <div class="p-4 gap-2 bg-bgcolor rounded-lg char-grid">
             {#if currentChar.data.characters.length === 0}
                 <span class="text-gray-500">No Character</span>
             {:else}
+                <div></div>
+                <div class="text-center">{language.talkness}</div>
+                <div class="text-center">{language.active}</div>
                 {#each currentChar.data.characters as char, i}
                     {#await getCharImage(findCharacterbyId(char).image, 'css')}
                         <BarIcon onClick={() => {
@@ -198,6 +167,24 @@
                             rmCharFromGroup(i)
                         }} additionalStyle={im} />
                     {/await}
+                    <div class="flex items-center px-2 py-3">
+                        {#each [1,2,3,4,5,6] as barIndex}
+                            <button class="bg-selected h-full flex-1 border-r-bgcolor border-r" 
+                                class:bg-green-500={currentChar.data.characterTalks[i] >= (1 / 6 * barIndex)}
+                                class:bg-selected={currentChar.data.characterTalks[i] < (1 / 6 * barIndex)}
+                                class:rounded-l-lg={barIndex === 1}
+                                class:rounded-r-lg={barIndex === 6}
+                                on:click={() => {
+                                    if(currentChar.data.type === 'group'){
+                                        currentChar.data.characterTalks[i] = (1 / 6 * barIndex)
+                                    }
+                                }}
+                            ></button>
+                        {/each}
+                    </div>
+                    <div class="flex items-center justify-center">
+                        <Check margin={false} bind:check={currentChar.data.characterActive[i]} />
+                    </div>
                 {/each}
             {/if}
         </div>
@@ -222,6 +209,13 @@
             <span class="text-neutral-200 ml-2">{language.ToggleSuperMemory}</span>
         </div>
     {/if}
+
+    {#if currentChar.type === 'group'}
+        <div class="flex mt-2 items-center">
+            <Check bind:check={currentChar.data.orderByOrder}/>
+            <span class="text-neutral-200 ml-2">{language.orderByOrder}</span>
+        </div>
+    {/if}
 {:else if subMenu === 1}
     <h2 class="mb-2 text-2xl font-bold mt-2">{language.characterDisplay}</h2>
     <span class="text-neutral-200 mt-2 mb-2">{currentChar.type !== 'group' ? language.charIcon : language.groupIcon}</span>
@@ -714,4 +708,9 @@
         overflow: hidden;
         text-overflow: ellipsis;
     }
+
+    .char-grid{
+        display: grid;
+        grid-template-columns: auto 1fr auto;
+    }
 </style>
\ No newline at end of file
diff --git a/src/ts/characters.ts b/src/ts/characters.ts
index 357197e2..5380f15f 100644
--- a/src/ts/characters.ts
+++ b/src/ts/characters.ts
@@ -38,7 +38,9 @@ export function createNewGroup(){
         emotionImages: [],
         customscript: [],
         chaId: uuidv4(),
-        firstMsgIndex: -1
+        firstMsgIndex: -1,
+        characterTalks: [],
+        characterActive: []
     })
     setDatabase(db)
     checkCharOrder()
@@ -300,6 +302,20 @@ export function characterFormatUpdate(index:number|character){
         }
 
     }
+    else{
+        if((!cha.characterTalks) || cha.characterTalks.length !== cha.characters.length){
+            cha.characterTalks = []
+            for(let i=0;i<cha.characters.length;i++){
+                cha.characterTalks.push(1 / 6 * 4)
+            }
+        }
+        if((!cha.characterActive) || cha.characterActive.length !== cha.characters.length){
+            cha.characterActive = []
+            for(let i=0;i<cha.characters.length;i++){
+                cha.characterActive.push(true)
+            }
+        }
+    }
     if(checkNullish(cha.customscript)){
         cha.customscript = []
     }
diff --git a/src/ts/process/group.ts b/src/ts/process/group.ts
new file mode 100644
index 00000000..86ed862d
--- /dev/null
+++ b/src/ts/process/group.ts
@@ -0,0 +1,103 @@
+import { shuffle } from "lodash";
+import { findCharacterbyId } from "../util";
+import { alertConfirm, alertError, alertSelectChar } from "../alert";
+import { language } from "src/lang";
+import { get } from "svelte/store";
+import { DataBase, setDatabase } from "../storage/database";
+import { selectedCharID } from "../stores";
+
+export async function addGroupChar(){
+    let db = get(DataBase)
+    let selectedId = get(selectedCharID)
+    let group = db.characters[selectedId]
+    if(group.type === 'group'){
+        const res = await alertSelectChar()
+        if(res){
+            if(group.characters.includes(res)){
+                alertError(language.errors.alreadyCharInGroup)
+            }
+            else{
+                if(await alertConfirm(language.askLoadFirstMsg)){
+                    group.chats[group.chatPage].message.push({
+                        role:'char',
+                        data: findCharacterbyId(res).firstMessage,
+                        saying: res,
+                    })
+                }
+
+                group.characters.push(res)
+                group.characterTalks.push(1 / 6 * 4)
+                group.characterActive.push(true)
+            }
+        }
+        setDatabase(db)
+    }
+}
+
+
+export function rmCharFromGroup(index:number){
+    let db = get(DataBase)
+    let selectedId = get(selectedCharID)
+    let group = db.characters[selectedId]
+    if(group.type === 'group'){
+        group.characters.splice(index, 1)
+        group.characterTalks.splice(index, 1)
+        group.characterActive.splice(index, 1)
+
+        setDatabase(db)
+    }
+}
+
+export type GroupOrder = {
+    id: string,
+    talkness: number,
+    index: number
+}
+
+export function groupOrder(chars:GroupOrder[], input:string):GroupOrder[] {
+    let order:GroupOrder[] = [];
+    if (input) {
+        const words = getWords(input)
+
+        for (const word of words) {
+            for (let char of chars) {
+                const charNameChunks = getWords(findCharacterbyId(char.id).name)
+                console.log(charNameChunks)
+
+                if (charNameChunks.includes(word)) {
+                    order.push(char);
+                    break;
+                }
+            }
+        }
+    }
+
+    const shuffled = shuffle(chars)
+    for (const char of shuffled) {
+        if(order.includes(char)){
+            continue
+        }
+
+        //TODO
+        const chance = 0.5
+
+        if (chance >= Math.random()) {
+            order.push(char);
+        }
+    }
+
+    while (order.length === 0) {
+        order.push(chars[Math.floor(Math.random() * chars.length)]);
+    }
+
+    return order;
+}
+
+function getWords(data:string){
+    const matches =  data.match(/\b\w+\b/gmi)
+    let words:string[] = []
+    for(const match of matches){
+        words.push(match.toLocaleLowerCase())
+    }
+    return words
+}
\ No newline at end of file
diff --git a/src/ts/process/index.ts b/src/ts/process/index.ts
index f795b5a1..a2a84b26 100644
--- a/src/ts/process/index.ts
+++ b/src/ts/process/index.ts
@@ -13,6 +13,8 @@ import { exampleMessage } from "./exampleMessages";
 import { sayTTS } from "./tts";
 import { supaMemory } from "./supaMemory";
 import { v4 } from "uuid";
+import { cloneDeep } from "lodash";
+import { groupOrder } from "./group";
 
 export interface OpenAIChat{
     role: 'system'|'user'|'assistant'
@@ -58,8 +60,25 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
 
     if(nowChatroom.type === 'group'){
         if(chatProcessIndex === -1){
-            for(let i=0;i<nowChatroom.characters.length;i++){
-                const r = await sendChat(i)
+            const messages = nowChatroom.chats[nowChatroom.chatPage].message
+            const lastMessage = messages[messages.length-1]
+            let order = nowChatroom.characters.map((v,i) => {
+                return {
+                    id: v,
+                    talkness: nowChatroom.characterActive[i] ? nowChatroom.characterTalks[i] : -1,
+                    index: i
+                }
+            })
+            if(!nowChatroom.orderByOrder){
+                order = groupOrder(order, lastMessage?.data).filter((v) => {
+                    if(v.id === lastMessage?.saying){
+                        return false
+                    }
+                    return true
+                })
+            }
+            for(let i=0;i<order.length;i++){
+                const r = await sendChat(order[i].index)
                 if(!r){
                     return false
                 }
@@ -103,6 +122,7 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
         'authorNote':([] as OpenAIChat[]),
         'lastChat':([] as OpenAIChat[]),
         'description':([] as OpenAIChat[]),
+        'postEverything':([] as OpenAIChat[]),
     }
 
     if(!currentChar.utilityBot){
@@ -149,6 +169,13 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
             content: description
         })
 
+        if(nowChatroom.type === 'group'){
+            const systemMsg = `[Write the next reply only as ${currentChar.name}]`
+            unformated.postEverything.push({
+                role: 'system',
+                content: systemMsg
+            })
+        }
     }
 
     unformated.lorebook.push({
@@ -220,15 +247,6 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
         currentTokens += (await tokenize(formedChat) + 5)
     }
 
-    if(nowChatroom.type === 'group'){
-        const systemMsg = `[Write the next reply only as ${currentChar.name}]`
-        chats.push({
-            role: 'system',
-            content: systemMsg
-        })
-        currentTokens += (await tokenize(systemMsg) + 5)
-    }
-
     if(nowChatroom.supaMemory && db.supaMemoryType !== 'none'){
         const sp = await supaMemory(chats, currentTokens, maxContextTokens, currentChat, nowChatroom)
         if(sp.error){
@@ -252,7 +270,6 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
             chats.splice(0, 1)
         }
         currentChat.lastMemory = chats[0].memo
-        console.log(currentChat.lastMemory)
     }
     let bias:{[key:number]:number} = {}
 
@@ -283,7 +300,8 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
     //make into one
 
     let formated:OpenAIChat[] = []
-    const formatOrder = db.formatingOrder
+    const formatOrder = cloneDeep(db.formatingOrder)
+    formatOrder.push('postEverything')
     let sysPrompts:string[] = []
     for(let i=0;i<formatOrder.length;i++){
         const cha = unformated[formatOrder[i]]
@@ -443,7 +461,6 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
             },
         ]
 
-        console.log('requesting chat')
         const rq = await requestChatData({
             formated: promptbody,
             bias: emobias,
diff --git a/src/ts/process/request.ts b/src/ts/process/request.ts
index 2a2897d9..5e46a878 100644
--- a/src/ts/process/request.ts
+++ b/src/ts/process/request.ts
@@ -61,7 +61,7 @@ export async function requestChatDataMain(arg:requestDataArgument, model:'model'
         case 'gpt4_32k':{
 
             for(let i=0;i<formated.length;i++){
-                if(arg.isGroupChat){
+                if(arg.isGroupChat && formated[i].name){
                     formated[i].content = formated[i].name + ": " + formated[i].content
                 }
                 formated[i].name = undefined
diff --git a/src/ts/storage/database.ts b/src/ts/storage/database.ts
index 37a49d68..e92484ba 100644
--- a/src/ts/storage/database.ts
+++ b/src/ts/storage/database.ts
@@ -351,6 +351,8 @@ export interface groupChat{
     name:string
     viewScreen: 'single'|'multiple'|'none'|'emp',
     characters:string[]
+    characterTalks:number[]
+    characterActive:boolean[]
     globalLore: loreBook[]
     autoMode: boolean
     useCharacterLore :boolean
@@ -364,6 +366,7 @@ export interface groupChat{
     loreSettings?:loreSettings
     supaMemory?:boolean
     ttsMode?:string
+    orderByOrder?:boolean
 }
 
 export interface botPreset{
@@ -517,7 +520,7 @@ interface sdConfig{
     hr_upscaler:string
 }
 
-export type FormatingOrderItem = 'main'|'jailbreak'|'chats'|'lorebook'|'globalNote'|'authorNote'|'lastChat'|'description'
+export type FormatingOrderItem = 'main'|'jailbreak'|'chats'|'lorebook'|'globalNote'|'authorNote'|'lastChat'|'description'|'postEverything'
 
 export interface Chat{
     message: Message[]

From f98764776ee4bed98558835d474bddadbd9d39ea Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Wed, 31 May 2023 05:26:10 +0900
Subject: [PATCH 2/2] [fix] tokenizer problems

---
 src/ts/process/index.ts      | 33 ++++++++++++++++++++++++++-------
 src/ts/process/stringlize.ts | 12 ++++++++++++
 src/ts/process/supaMemory.ts | 22 +++++++++++++++-------
 3 files changed, 53 insertions(+), 14 deletions(-)

diff --git a/src/ts/process/index.ts b/src/ts/process/index.ts
index a2a84b26..0c48d7e4 100644
--- a/src/ts/process/index.ts
+++ b/src/ts/process/index.ts
@@ -15,6 +15,7 @@ import { supaMemory } from "./supaMemory";
 import { v4 } from "uuid";
 import { cloneDeep } from "lodash";
 import { groupOrder } from "./group";
+import { getNameMaxTokens } from "./stringlize";
 
 export interface OpenAIChat{
     role: 'system'|'user'|'assistant'
@@ -25,7 +26,7 @@ export interface OpenAIChat{
 
 export const doingChat = writable(false)
 
-export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
+export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:number} = {}):Promise<boolean> {
 
     let findCharCache:{[key:string]:character} = {}
     function findCharacterbyIdwithCache(id:string){
@@ -57,9 +58,19 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
     let selectedChar = get(selectedCharID)
     const nowChatroom = db.characters[selectedChar]
     let currentChar:character
+    let caculatedChatTokens = 0
+    if(db.aiModel.startsWith('gpt')){
+        caculatedChatTokens += 5
+    }
+    else{
+        caculatedChatTokens += 3
+    }
 
     if(nowChatroom.type === 'group'){
         if(chatProcessIndex === -1){
+            const charNames =nowChatroom.characters.map((v) => findCharacterbyIdwithCache(v).name)
+            caculatedChatTokens += await getNameMaxTokens([...charNames, db.username])
+
             const messages = nowChatroom.chats[nowChatroom.chatPage].message
             const lastMessage = messages[messages.length-1]
             let order = nowChatroom.characters.map((v,i) => {
@@ -78,7 +89,9 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
                 })
             }
             for(let i=0;i<order.length;i++){
-                const r = await sendChat(order[i].index)
+                const r = await sendChat(order[i].index, {
+                    chatAdditonalTokens: caculatedChatTokens
+                })
                 if(!r){
                     return false
                 }
@@ -95,7 +108,13 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
     }
     else{
         currentChar = nowChatroom
+        if(!db.aiModel.startsWith('gpt')){
+            caculatedChatTokens += await getNameMaxTokens([currentChar.name, db.username])
+        }
+
     }
+
+    let chatAdditonalTokens = arg.chatAdditonalTokens ?? caculatedChatTokens
     
     let selectedChat = nowChatroom.chatPage
     let currentChat = nowChatroom.chats[selectedChat]
@@ -188,13 +207,13 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
         return (unformated[key] as OpenAIChat[]).map((d) => {
             return d.content
         }).join('\n\n')
-    }).join('\n\n')) + db.maxResponse) + 150
+    }).join('\n\n')) + db.maxResponse) + 100
 
     
     const examples = exampleMessage(currentChar)
 
     for(const example of examples){
-        currentTokens += await tokenize(example.content) + 5
+        currentTokens += await tokenize(example.content) + chatAdditonalTokens
     }
 
     let chats:OpenAIChat[] = examples
@@ -244,11 +263,11 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
             memo: msg.chatId,
             name: name
         })
-        currentTokens += (await tokenize(formedChat) + 5)
+        currentTokens += (await tokenize(formedChat) + chatAdditonalTokens)
     }
 
     if(nowChatroom.supaMemory && db.supaMemoryType !== 'none'){
-        const sp = await supaMemory(chats, currentTokens, maxContextTokens, currentChat, nowChatroom)
+        const sp = await supaMemory(chats, currentTokens, maxContextTokens, currentChat, nowChatroom, chatAdditonalTokens)
         if(sp.error){
             alertError(sp.error)
             return false
@@ -266,7 +285,7 @@ export async function sendChat(chatProcessIndex = -1):Promise<boolean> {
                 return false
             }
 
-            currentTokens -= (await tokenize(chats[0].content) + 5)
+            currentTokens -= (await tokenize(chats[0].content) + chatAdditonalTokens)
             chats.splice(0, 1)
         }
         currentChat.lastMemory = chats[0].memo
diff --git a/src/ts/process/stringlize.ts b/src/ts/process/stringlize.ts
index bc62b518..140389b6 100644
--- a/src/ts/process/stringlize.ts
+++ b/src/ts/process/stringlize.ts
@@ -1,4 +1,5 @@
 import type { OpenAIChat } from ".";
+import { tokenize } from "../tokenizer";
 
 export function multiChatReplacer(){
 
@@ -52,4 +53,15 @@ export function unstringlizeChat(text:string, formated:OpenAIChat[], char:string
     }
 
     return text
+}
+
+export async function getNameMaxTokens(names:string[]){
+    let maxCharNameTokens = 0
+    for(const name of names){
+        const tokens = await tokenize(name + ': ') + 1
+        if(maxCharNameTokens < tokens){
+            maxCharNameTokens = tokens
+        }
+    }
+    return maxCharNameTokens
 }
\ No newline at end of file
diff --git a/src/ts/process/supaMemory.ts b/src/ts/process/supaMemory.ts
index e58e1876..634fd447 100644
--- a/src/ts/process/supaMemory.ts
+++ b/src/ts/process/supaMemory.ts
@@ -5,9 +5,17 @@ import { tokenize } from "../tokenizer";
 import { findCharacterbyId } from "../util";
 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;lastId?:string}>{
+export async function supaMemory(
+        chats:OpenAIChat[],
+        currentTokens:number,
+        maxContextTokens:number,
+        room:Chat,
+        char:character|groupChat,
+        chatAdditonalTokens:number
+    ): Promise<{ currentTokens: number; chats: OpenAIChat[]; error?:string; memory?:string;lastId?:string}>{
     const db = get(DataBase)
-    console.log("Memory: " + currentTokens)
+
+    currentTokens += 10
 
     if(currentTokens > maxContextTokens){
         let coIndex = -1
@@ -19,7 +27,7 @@ export async function supaMemory(chats:OpenAIChat[],currentTokens:number,maxCont
         }
         if(coIndex !== -1){
             for(let i=0;i<coIndex;i++){
-                currentTokens -= (await tokenize(chats[0].content) + 1)
+                currentTokens -= (await tokenize(chats[0].content) + chatAdditonalTokens)
                 chats.splice(0, 1)
             }
         }
@@ -45,13 +53,13 @@ export async function supaMemory(chats:OpenAIChat[],currentTokens:number,maxCont
                     lastId = id
                     break
                 }
-                currentTokens -= (await tokenize(chats[0].content) + 1)
+                currentTokens -= (await tokenize(chats[0].content) + chatAdditonalTokens)
                 chats.splice(0, 1)
                 i += 1
             }
 
             supaMemory = data
-            currentTokens += await tokenize(supaMemory) + 1
+            currentTokens += await tokenize(supaMemory) + chatAdditonalTokens
         }
 
 
@@ -171,7 +179,7 @@ export async function supaMemory(chats:OpenAIChat[],currentTokens:number,maxCont
                     }
                     continue
                 }
-                const tokens = await tokenize(cont.content) + 5
+                const tokens = await tokenize(cont.content) + chatAdditonalTokens
                 if((chunkSize + tokens) > maxChunkSize){
                     if(stringlizedChat === ''){
                         stringlizedChat += `${cont.role === 'assistant' ? char.type === 'group' ? '' : char.name : db.username}: ${cont.content}\n\n`
@@ -193,7 +201,7 @@ export async function supaMemory(chats:OpenAIChat[],currentTokens:number,maxCont
                     return result
                 }
     
-                const tokenz = await tokenize(result + '\n\n') + 5
+                const tokenz = await tokenize(result + '\n\n') + chatAdditonalTokens
                 currentTokens += tokenz
                 supaMemory += result.replace(/\n+/g,'\n') + '\n\n'