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

Update hypav2.ts

Browse Source 
fuck chatgpt, today is the last day and I will finish this.
This commit is contained in:
HyperBlaze
2024-06-17 15:07:07 -07:00
parent ff52e8dd37
commit 289fe90027
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/ts/process/memory/hypav2.ts
View File

@@ -169,7 +169,7 @@ export async function hypaMemoryV2(
halfData.push(chat); halfData.push(chat);
idx++; idx++;
targetId = chat.memo; targetId = chat.memo;
console.log("current target chat Id:", targetId); console.log("current target chat: ", chat);
} }
// Avoid summarizing the last two chats // Avoid summarizing the last two chats
@@ -225,7 +225,14 @@ export async function hypaMemoryV2(
// Fetch additional memory from chunks // Fetch additional memory from chunks
const processor = new HypaProcesser(db.hypaModel); const processor = new HypaProcesser(db.hypaModel);
processor.oaikey = db.supaMemoryKey; processor.oaikey = db.supaMemoryKey;
await processor.addText(data.chunks.filter(v => v.text.trim().length > 0).map(v => "search_document: " + v.text.trim()));
// Filter chunks to only include those older than the last mainChunk's targetId
const lastMainChunkTargetId = data.mainChunks.length > 0 ? data.mainChunks[0].targetId : null;
const olderChunks = lastMainChunkTargetId
? data.chunks.filter(chunk => getValidChatIndex(chunk.targetId) < getValidChatIndex(lastMainChunkTargetId))
: data.chunks;
await processor.addText(olderChunks.filter(v => v.text.trim().length > 0).map(v => "search_document: " + v.text.trim()));
let scoredResults: { [key: string]: number } = {}; let scoredResults: { [key: string]: number } = {};
for (let i = 0; i < 3; i++) { for (let i = 0; i < 3; i++) {
@@ -260,12 +267,20 @@ export async function hypaMemoryV2(
const lastTargetId = data.mainChunks.length > 0 ? data.mainChunks[0].targetId : null; const lastTargetId = data.mainChunks.length > 0 ? data.mainChunks[0].targetId : null;
if (lastTargetId) { if (lastTargetId) {
const lastIndex = getValidChatIndex(lastTargetId); const lastIndex = getValidChatIndex(lastTargetId);
console.log(chats[lastIndex])
if (lastIndex !== -1) { if (lastIndex !== -1) {
const remainingChats = chats.slice(lastIndex + 1); const remainingChats = chats.slice(lastIndex);
chats = [chats[0], ...remainingChats]; chats = [chats[0], ...remainingChats];
} }
} }
// Add last two chats if they exist
const lastTwoChats = chats.slice(-2);
if (lastTwoChats.length === 2) {
chats.push(lastTwoChats[0]);
chats.push(lastTwoChats[1]);
}
console.log("model being used: ", db.hypaModel, db.supaModelType, "\nCurrent session tokens: ", currentTokens, "\nAll chats, including memory system prompt: ", chats, "\nMemory data, with all the chunks: ", data); console.log("model being used: ", db.hypaModel, db.supaModelType, "\nCurrent session tokens: ", currentTokens, "\nAll chats, including memory system prompt: ", chats, "\nMemory data, with all the chunks: ", data);
return { return {
currentTokens: currentTokens, currentTokens: currentTokens,