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

fix: index issues

Browse Source 
forgot to commit on my pc, so doing it on laptop
This commit is contained in:
HyperBlaze
2024-12-05 12:21:58 -08:00
parent ea6d7dada1
commit b283b4a126
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -160,14 +160,20 @@ function cleanInvalidChunks(
data.mainChunks = data.mainChunks.filter((mainChunk) => { data.mainChunks = data.mainChunks.filter((mainChunk) => {
return isSubset(mainChunk.chatMemos, currentChatMemos); return isSubset(mainChunk.chatMemos, currentChatMemos);
}); });
// chunk filtering based on mainChunk's id // chunk filtering based on mainChunk's id
const validMainChunkIds = new Set(data.mainChunks.map((mainChunk) => mainChunk.id)); const validMainChunkIds = new Set(data.mainChunks.map((mainChunk) => mainChunk.id));
data.chunks = data.chunks.filter((chunk) => data.chunks = data.chunks.filter((chunk) =>
validMainChunkIds.has(chunk.mainChunkID) validMainChunkIds.has(chunk.mainChunkID)
); );
data.lastMainChunkId = data.mainChunks[-1].id; // Quite literally the definition of lastMainChunkId. Didn't use .length, since middle chat context can be partially deleted. // Update lastMainChunkId
if (data.mainChunks.length > 0) {
data.lastMainChunkId = data.mainChunks[data.mainChunks.length - 1].id;
} else {
data.lastMainChunkId = 0;
}
} }
export async function regenerateSummary( export async function regenerateSummary(
chats: OpenAIChat[], chats: OpenAIChat[],
data: HypaV2Data, data: HypaV2Data,
@@ -206,7 +212,7 @@ export async function hypaMemoryV2(
const lastTwoChats = chats.slice(-2); const lastTwoChats = chats.slice(-2);
let summarizationFailures = 0; let summarizationFailures = 0;
const maxSummarizationFailures = 3; const maxSummarizationFailures = 3;
// Find the index to start summarizing from // Find the index to start summarizing from
let idx = 0; let idx = 0;
if (data.mainChunks.length > 0) { if (data.mainChunks.length > 0) {
@@ -216,7 +222,7 @@ export async function hypaMemoryV2(
if (lastChatIndex !== -1) { if (lastChatIndex !== -1) {
idx = lastChatIndex + 1; idx = lastChatIndex + 1;
} }
} }
// Starting chat index of new mainChunk to be generated // Starting chat index of new mainChunk to be generated
// Token management loop(where using of ) // Token management loop(where using of )
@@ -396,4 +402,4 @@ export async function hypaMemoryV2(
chats: unsummarizedChats, chats: unsummarizedChats,
memory: data, memory: data,
}; };
} }