changed mainprompt
made mainprompt only have most recent summarized data, which will be correctly appended on mainChunks[0]
This commit is contained in:
@@ -714,7 +714,7 @@ export async function sendChat(chatProcessIndex = -1,arg:{chatAdditonalTokens?:n
|
|||||||
currentTokens += await tokenizer.tokenizeChat(chat)
|
currentTokens += await tokenizer.tokenizeChat(chat)
|
||||||
}
|
}
|
||||||
|
|
||||||
if(nowChatroom.supaMemory && (db.supaModelType !== 'none' || db.hanuraiEnable)){
|
if(nowChatroom.supaMemory && (db.supaModelType !== 'none' || db.hanuraiEnable || db.hypav2)){
|
||||||
chatProcessStage.set(2)
|
chatProcessStage.set(2)
|
||||||
if(db.hanuraiEnable){
|
if(db.hanuraiEnable){
|
||||||
const hn = await hanuraiMemory(chats, {
|
const hn = await hanuraiMemory(chats, {
|
||||||
|
|||||||
@@ -126,7 +126,6 @@ export async function hypaMemoryV2(
|
|||||||
): Promise<{ currentTokens: number; chats: OpenAIChat[]; error?: string; memory?: HypaV2Data; }> {
|
): Promise<{ currentTokens: number; chats: OpenAIChat[]; error?: string; memory?: HypaV2Data; }> {
|
||||||
|
|
||||||
const db = get(DataBase);
|
const db = get(DataBase);
|
||||||
|
|
||||||
const data: HypaV2Data = room.hypaV2Data ?? {
|
const data: HypaV2Data = room.hypaV2Data ?? {
|
||||||
chunks: [],
|
chunks: [],
|
||||||
mainChunks: []
|
mainChunks: []
|
||||||
@@ -134,18 +133,14 @@ export async function hypaMemoryV2(
|
|||||||
|
|
||||||
let allocatedTokens = db.hypaAllocatedTokens;
|
let allocatedTokens = db.hypaAllocatedTokens;
|
||||||
let chunkSize = db.hypaChunkSize;
|
let chunkSize = db.hypaChunkSize;
|
||||||
currentTokens += allocatedTokens;
|
currentTokens += allocatedTokens + 50;
|
||||||
currentTokens += 50; // this is for the template prompt
|
|
||||||
let mainPrompt = "";
|
let mainPrompt = "";
|
||||||
|
|
||||||
while (data.mainChunks.length > 0) {
|
// Processing mainChunks
|
||||||
|
if (data.mainChunks.length > 0) {
|
||||||
const chunk = data.mainChunks[0];
|
const chunk = data.mainChunks[0];
|
||||||
const ind = chats.findIndex(e => e.memo === chunk.targetId);
|
const ind = chats.findIndex(e => e.memo === chunk.targetId);
|
||||||
if (ind === -1) {
|
if (ind !== -1) {
|
||||||
data.mainChunks.shift();
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
|
|
||||||
const removedChats = chats.splice(0, ind);
|
const removedChats = chats.splice(0, ind);
|
||||||
for (const chat of removedChats) {
|
for (const chat of removedChats) {
|
||||||
currentTokens -= await tokenizer.tokenizeChat(chat);
|
currentTokens -= await tokenizer.tokenizeChat(chat);
|
||||||
@@ -154,20 +149,19 @@ export async function hypaMemoryV2(
|
|||||||
mainPrompt = chunk.text;
|
mainPrompt = chunk.text;
|
||||||
const mpToken = await tokenizer.tokenizeChat({ role: 'system', content: mainPrompt });
|
const mpToken = await tokenizer.tokenizeChat({ role: 'system', content: mainPrompt });
|
||||||
allocatedTokens -= mpToken;
|
allocatedTokens -= mpToken;
|
||||||
break;
|
}
|
||||||
|
data.mainChunks.shift();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Token management loop
|
||||||
while (currentTokens >= maxContextTokens) {
|
while (currentTokens >= maxContextTokens) {
|
||||||
let idx = 0;
|
let idx = 0;
|
||||||
let targetId = '';
|
let targetId = '';
|
||||||
const halfData: OpenAIChat[] = [];
|
const halfData: OpenAIChat[] = [];
|
||||||
|
|
||||||
let halfDataTokens = 0;
|
let halfDataTokens = 0;
|
||||||
while (halfDataTokens < chunkSize) {
|
while (halfDataTokens < chunkSize && chats[idx]) {
|
||||||
const chat = chats[idx];
|
const chat = chats[idx];
|
||||||
if (!chat) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
halfDataTokens += await tokenizer.tokenizeChat(chat);
|
halfDataTokens += await tokenizer.tokenizeChat(chat);
|
||||||
halfData.push(chat);
|
halfData.push(chat);
|
||||||
idx++;
|
idx++;
|
||||||
@@ -175,7 +169,6 @@ export async function hypaMemoryV2(
|
|||||||
}
|
}
|
||||||
|
|
||||||
const stringlizedChat = halfData.map(e => `${e.role}: ${e.content}`).join('\n');
|
const stringlizedChat = halfData.map(e => `${e.role}: ${e.content}`).join('\n');
|
||||||
|
|
||||||
const summaryData = await summary(stringlizedChat);
|
const summaryData = await summary(stringlizedChat);
|
||||||
|
|
||||||
if (!summaryData.success) {
|
if (!summaryData.success) {
|
||||||
@@ -187,7 +180,7 @@ export async function hypaMemoryV2(
|
|||||||
}
|
}
|
||||||
|
|
||||||
const summaryDataToken = await tokenizer.tokenizeChat({ role: 'system', content: summaryData.data });
|
const summaryDataToken = await tokenizer.tokenizeChat({ role: 'system', content: summaryData.data });
|
||||||
mainPrompt += `\n\n${summaryData.data}`;
|
mainPrompt = summaryData.data; // Ensure mainPrompt only contains the latest summary
|
||||||
currentTokens -= halfDataTokens;
|
currentTokens -= halfDataTokens;
|
||||||
allocatedTokens -= summaryDataToken;
|
allocatedTokens -= summaryDataToken;
|
||||||
|
|
||||||
@@ -196,8 +189,11 @@ export async function hypaMemoryV2(
|
|||||||
targetId: targetId
|
targetId: targetId
|
||||||
});
|
});
|
||||||
|
|
||||||
if (allocatedTokens < 1500) {
|
if (allocatedTokens < 1000) {
|
||||||
|
console.log("Currently allocatedTokens for HypaMemoryV2 is short, thus summarizing mainPrompt twice.", allocatedTokens);
|
||||||
|
console.log("This is mainPrompt(summarized data): ", mainPrompt)
|
||||||
const summarizedMp = await summary(mainPrompt);
|
const summarizedMp = await summary(mainPrompt);
|
||||||
|
console.log("Re-summarized, expected behavior: ", summarizedMp.data)
|
||||||
const mpToken = await tokenizer.tokenizeChat({ role: 'system', content: mainPrompt });
|
const mpToken = await tokenizer.tokenizeChat({ role: 'system', content: mainPrompt });
|
||||||
const summaryToken = await tokenizer.tokenizeChat({ role: 'system', content: summarizedMp.data });
|
const summaryToken = await tokenizer.tokenizeChat({ role: 'system', content: summarizedMp.data });
|
||||||
|
|
||||||
@@ -217,50 +213,32 @@ export async function hypaMemoryV2(
|
|||||||
|
|
||||||
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 => {
|
await processor.addText(data.chunks.filter(v => v.text.trim().length > 0).map(v => "search_document: " + v.text.trim()));
|
||||||
return v.text.trim().length > 0;
|
|
||||||
}).map((v) => {
|
|
||||||
return "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++) {
|
||||||
const pop = chats[chats.length - i - 1];
|
const pop = chats[chats.length - i - 1];
|
||||||
if (!pop) {
|
if (!pop) break;
|
||||||
break;
|
|
||||||
}
|
|
||||||
const searched = await processor.similaritySearchScored(`search_query: ${pop.content}`);
|
const searched = await processor.similaritySearchScored(`search_query: ${pop.content}`);
|
||||||
for (const result of searched) {
|
for (const result of searched) {
|
||||||
const score = result[1] / (i + 1);
|
const score = result[1] / (i + 1);
|
||||||
if (scoredResults[result[0]]) {
|
scoredResults[result[0]] = (scoredResults[result[0]] || 0) + score;
|
||||||
scoredResults[result[0]] += score;
|
|
||||||
} else {
|
|
||||||
scoredResults[result[0]] = score;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const scoredArray = Object.entries(scoredResults).sort((a, b) => b[1] - a[1]);
|
const scoredArray = Object.entries(scoredResults).sort((a, b) => b[1] - a[1]);
|
||||||
|
|
||||||
let chunkResultPrompts = "";
|
let chunkResultPrompts = "";
|
||||||
while (allocatedTokens > 0) {
|
while (allocatedTokens > 0 && scoredArray.length > 0) {
|
||||||
const target = scoredArray.shift();
|
const target = scoredArray.shift();
|
||||||
if (!target) {
|
const tokenized = await tokenizer.tokenizeChat({ role: 'system', content: target[0].substring(14) });
|
||||||
break;
|
if (tokenized > allocatedTokens) break;
|
||||||
}
|
|
||||||
const tokenized = await tokenizer.tokenizeChat({
|
|
||||||
role: 'system',
|
|
||||||
content: target[0].substring(14)
|
|
||||||
});
|
|
||||||
if (tokenized > allocatedTokens) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
chunkResultPrompts += target[0].substring(14) + '\n\n';
|
chunkResultPrompts += target[0].substring(14) + '\n\n';
|
||||||
allocatedTokens -= tokenized;
|
allocatedTokens -= tokenized;
|
||||||
}
|
}
|
||||||
|
|
||||||
const fullResult = `<Past Events Summary>${mainPrompt}</Past Events Summary>\n<Past Events Details>${chunkResultPrompts}</Past Events Details>`;
|
const fullResult = `<Past Events Summary>${mainPrompt}</Past Events Summary>\n<Past Events Details>${chunkResultPrompts}</Past Events Details>`;
|
||||||
|
console.log(fullResult, " This chat is added to system prompt for memory. ");
|
||||||
chats.unshift({
|
chats.unshift({
|
||||||
role: "system",
|
role: "system",
|
||||||
content: fullResult,
|
content: fullResult,
|
||||||
|
|||||||
Reference in New Issue
Block a user