mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] fixed hypermemory again

Browse Source
This commit is contained in:
kwaroran
2023-06-29 01:37:18 +09:00
parent 21ae693f08
commit 739bc2a4bf
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/ts/process/memory/supaMemory.ts
View File

@@ -60,8 +60,7 @@ export async function supaMemory(
error: "SupaMemory: Data saved in hypaMemory, loaded as SupaMemory." error: "SupaMemory: Data saved in hypaMemory, loaded as SupaMemory."
} }
} }
HypaData = JSON.parse(data.trim())
HypaData = JSON.parse(data.substring(0,5).trim())
if(!Array.isArray(HypaData)){ if(!Array.isArray(HypaData)){
return { return {
currentTokens: currentTokens, currentTokens: currentTokens,
@@ -71,7 +70,7 @@ export async function supaMemory(
} }
let indexSelected = -1 let indexSelected = -1
for(let i=0;i<HypaData.length;i++){ for(let j=0;j<HypaData.length;j++){
let i =0; let i =0;
let countTokens = currentTokens let countTokens = currentTokens
let countChats = cloneDeep(chats) let countChats = cloneDeep(chats)
@@ -79,11 +78,11 @@ export async function supaMemory(
if(countChats.length === 0){ if(countChats.length === 0){
break break
} }
if(countChats[0].memo === HypaData[i].id){ if(countChats[0].memo === HypaData[j].id){
lastId = HypaData[i].id lastId = HypaData[j].id
currentTokens = countTokens currentTokens = countTokens
chats = countChats chats = countChats
indexSelected = i indexSelected = j
break break
} }
countTokens -= await tokenizer.tokenizeChat(countChats[0]) countTokens -= await tokenizer.tokenizeChat(countChats[0])
@@ -168,6 +167,8 @@ export async function supaMemory(
"temperature": 0 "temperature": 0
} }
}) })
console.log(da)
result = (await da.data).choices[0].text.trim() result = (await da.data).choices[0].text.trim()
} }
@@ -203,6 +204,7 @@ export async function supaMemory(
if(arg.asHyper){ if(arg.asHyper){
const hypa = new HypaProcesser() const hypa = new HypaProcesser()
hypa.oaikey = db.supaMemoryKey hypa.oaikey = db.supaMemoryKey
hypa.vectors = []
await hypa.addText(hypaChunks) await hypa.addText(hypaChunks)
const filteredChat = chats.filter((r) => r.role !== 'system' && r.role !== 'function') const filteredChat = chats.filter((r) => r.role !== 'system' && r.role !== 'function')
const s = await hypa.similaritySearch(stringlizeChat(filteredChat.slice(0, 4))) const s = await hypa.similaritySearch(stringlizeChat(filteredChat.slice(0, 4)))
@@ -340,7 +342,7 @@ export async function supaMemory(
return { return {
currentTokens: currentTokens, currentTokens: currentTokens,
chats: chats, chats: chats,
memory: JSON.stringify(HypaData, null, 2), memory: "hypa:\n" + JSON.stringify(HypaData, null, 2),
lastId: lastId lastId: lastId
} }