MurasameNoa/chatgpt-plugin
1
0
Fork 0
mirror of https://github.com/ikechan8370/chatgpt-plugin.git synced 2025-12-16 13:27:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

试验性的记忆功能 (#812)

Browse source 
* feat: memory basic

* fix: chaite ver

* fix: update prompt

* fix: memory cursor and extract prompt

* fix: memory retrieval bug

* fix: memory retrieval bug

* fix: one more attempt by codex

* fix: messages prompt error

* fix: one more time by codex

* fix: metrics by codex

* fix: memory forward

* fix: memory show update time
This commit is contained in:
ikechan8370 2025-11-07 16:40:26 +08:00 • committed by GitHub
parent db386ccaf2
commit 8bfce5402f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
19 changed files with 4382 additions and 103 deletions
Download patch file Download diff file Expand all files Collapse all files

39
apps/chat.js
View file

@ -2,7 +2,9 @@ import Config from '../config/config.js'
import { Chaite, SendMessageOption } from 'chaite'
import { getPreset, intoUserMessage, toYunzai } from '../utils/message.js'
import { YunzaiUserState } from '../models/chaite/storage/lowdb/user_state_storage.js'
import { getGroupContextPrompt, getGroupHistory } from '../utils/group.js'
import { getGroupContextPrompt } from '../utils/group.js'
import { buildMemoryPrompt } from '../models/memory/prompt.js'
import { extractTextFromUserMessage, processUserMemory } from '../models/memory/userMemoryManager.js'
import * as crypto from 'node:crypto'
export class Chat extends plugin {
@ -11,7 +13,8 @@ export class Chat extends plugin {
name: 'ChatGPT-Plugin对话',
dsc: 'ChatGPT-Plugin对话',
event: 'message',
priority: 500,
// 应🥑要求降低优先级
priority: 555500,
rule: [
{
reg: '^[^#][sS]*',
@ -63,12 +66,34 @@ export class Chat extends plugin {
toggleMode: Config.basic.toggleMode,
togglePrefix: Config.basic.togglePrefix
})
const userText = extractTextFromUserMessage(userMessage) || e.msg || ''
sendMessageOptions.conversationId = state?.current?.conversationId
sendMessageOptions.parentMessageId = state?.current?.messageId || state?.conversations.find(c => c.id === sendMessageOptions.conversationId)?.lastMessageId
const systemSegments = []
const baseSystem = sendMessageOptions.systemOverride || preset.sendMessageOption?.systemOverride || ''
if (baseSystem) {
systemSegments.push(baseSystem)
}
if (userText) {
const memoryPrompt = await buildMemoryPrompt({
userId: e.sender.user_id + '',
groupId: e.isGroup ? e.group_id + '' : null,
queryText: userText
})
if (memoryPrompt) {
systemSegments.push(memoryPrompt)
logger.debug(`[Memory] memory prompt: ${memoryPrompt}`)
}
}
const enableGroupContext = (preset.groupContext === 'use_system' || !preset.groupContext) ? Config.llm.enableGroupContext : (preset.groupContext === 'enabled')
if (enableGroupContext && e.isGroup) {
const contextPrompt = await getGroupContextPrompt(e, Config.llm.groupContextLength)
sendMessageOptions.systemOverride = sendMessageOptions.systemOverride ? sendMessageOptions.systemOverride + '\n' + contextPrompt : (preset.sendMessageOption.systemOverride + contextPrompt)
if (contextPrompt) {
systemSegments.push(contextPrompt)
}
}
if (systemSegments.length > 0) {
sendMessageOptions.systemOverride = systemSegments.join('\n\n')
}
const response = await Chaite.getInstance().sendMessage(userMessage, e, {
...sendMessageOptions,
@ -95,5 +120,13 @@ export class Chat extends plugin {
for (let forwardElement of forward) {
this.reply(forwardElement)
}
await processUserMemory({
event: e,
userMessage,
userText,
conversationId: sendMessageOptions.conversationId,
assistantContents: response.contents,
assistantMessageId: response.id
})
}
}