ChrisChrome/openai-bot
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Add session timeout to keep things fresh when you go to an existing channel

Browse source
This commit is contained in:
Christopher Cookman 2023-03-09 19:41:39 -07:00
parent 678db465bf
commit 15fb961e79
Signed by: ChrisChrome
GPG key ID: A023A26E42C33A42
1 changed files with 22 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
index.js
View file

@ -24,9 +24,19 @@ const client = new Discord.Client({
intents: ["MessageContent", "GuildMessages", "Guilds"]
});
var sessions = {}; // Keep track of sessions, not really used right now, but if I wanted to allow multiple sessions, I could
var timers = {}; // Keep track of reset timers
var basePrompt = config.openai.basePrompt
basePrompt.content = fs.readFileSync("./basePrompt.txt", "utf8").toString();
const resetSession = async (id) => {
if (timers[id]) {
await clearTimeout(timers[id]);
delete timers[id];
};
await delete sessions[id];
return true;
}
client.on("ready", () => {
console.log(`${colors.cyan("[INFO]")} Logged in as ${colors.green(client.user.tag)}`)
// Log startup time in seconds
@ -60,7 +70,7 @@ client.on('interactionCreate', async (interaction) => {
switch (interaction.commandName) {
case "reset":
// Remove the session
await delete sessions[interaction.channelId];
await resetSession(interaction.channelId);
interaction.reply(lang.reset);
break;
case "info": // Info about the current session
@ -118,6 +128,12 @@ client.on('messageCreate', async (message) => {
started: new Date(),
};
}
// If the session already exists, reset the timer
if (timers[message.channelId]) {
await clearTimeout(timers[message.channelId]);
delete timers[message.channelId];
}
message.channel.sendTyping();
var typing = setInterval(() => {
message.channel.sendTyping();
@ -150,6 +166,11 @@ client.on('messageCreate', async (message) => {
} else {
message.channel.send(output.content);
}
// Set the reset timer
timers[message.channelId] = setTimeout(() => {
resetSession(message.channelId);
message.channel.send(lang.timeout)
}, config.openai.resetTime);
}).catch((err) => {
message.channel.send({
"embeds": [{