d0zingcat/solana-agent-kit
1
0
Fork 0
mirror of https://github.com/d0zingcat/solana-agent-kit.git synced 2026-05-13 23:16:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add user chat history memory store

Browse Source
This commit is contained in:
Damjan
2025-01-04 11:14:21 +01:00
parent 6b2fad538b
commit a6856409bc
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
examples/discord-bot-starter/src/index.ts
View File

@@ -11,6 +11,8 @@ const client = new Client({
partials: [Partials.Message, Partials.Channel],
});
const chatHistory = new Map();
async function initializeAgent() {
try {
const llm = new ChatOpenAI({
@@ -68,13 +70,22 @@ client.on(Events.MessageCreate, async (message) => {
const { agent, config } = await initializeAgent();
const stream = await agent.stream({ messages: [new HumanMessage(message.content)] }, config);
const userId = message.author.id;
if (!chatHistory.has(userId)) {
chatHistory.set(userId, []);
}
const userChatHistory = chatHistory.get(userId);
userChatHistory.push(new HumanMessage(message.content));
const stream = await agent.stream({ messages: userChatHistory }, config);
const replyIfNotEmpty = async (content: string) => content.trim() !== '' && message.reply(content);
for await (const chunk of stream) {
if ('agent' in chunk) {
await replyIfNotEmpty(chunk.agent.messages[0].content);
const agentMessage = chunk.agent.messages[0].content;
await replyIfNotEmpty(agentMessage);
userChatHistory.push(new HumanMessage(agentMessage));
}
}
} catch (error) {