fix: local generations ignoring regenCache on continue

app/constants/APIState/LocalAPI.ts

@@ -85,7 +85,8 @@ class LocalAPI extends APIBase {
         }
 
         const outputCompleted = (text: string) => {
-            Chats.useChat.getState().setBuffer(text.replaceAll(replace, ''))
+            const regenCache = Chats.useChat.getState().getRegenCache()
+            Chats.useChat.getState().setBuffer((regenCache + text).replaceAll(replace, ''))
             if (mmkv.getBoolean(AppSettings.PrintContext)) Logger.log(`Completion Output:\n${text}`)
         }
 

app/constants/Chat.ts

@@ -6,11 +6,11 @@ import { create } from 'zustand'
 import { Characters } from './Characters'
 import { AppSettings, Global } from './GlobalValues'
 import { Logger } from './Logger'
+import { mmkv } from './MMKV'
 import { RecentMessages } from './RecentMessages'
 import { convertToFormatInstruct } from './TextFormat'
 import { Tokenizer } from './Tokenizer'
 import { replaceMacros } from './Utils'
-import { mmkv } from './MMKV'
 
 export type ChatSwipe = {
     id: number
@@ -66,6 +66,7 @@ export interface ChatState {
     updateFromBuffer: () => Promise<void>
     insertLastToBuffer: () => void
     setRegenCache: () => void
+    getRegenCache: () => string
     stopGenerating: () => void
     startGenerating: () => void
     abortFunction: undefined | AbortFunction
@@ -279,6 +280,12 @@ export namespace Chats {
                 data: state?.data ? { ...state.data, messages: messages } : state.data,
             }))
         },
+        getRegenCache: () => {
+            const messages = get()?.data?.messages
+            const message = messages?.[messages.length - 1]
+            if (!messages || !message) return ''
+            return message.swipes[message.swipe_id].regen_cache ?? ''
+        },
     }))
 
     export namespace db {