made it pretty
This commit is contained in:
Raven Scott
parent
875ab7fdd2
commit
451205cb7d
243
llamabot.js
243
llamabot.js
@ -6,17 +6,22 @@ const errorMessages = require('./assets/errorMessages.js');
|
||||
|
||||
require('dotenv').config()
|
||||
// hi
|
||||
const { Client, GatewayIntentBits, ActivityType, Partials } = require('discord.js');
|
||||
const {
|
||||
Client,
|
||||
GatewayIntentBits,
|
||||
ActivityType,
|
||||
Partials
|
||||
} = require('discord.js');
|
||||
|
||||
const client = new Client({
|
||||
intents: [
|
||||
GatewayIntentBits.DirectMessages,
|
||||
GatewayIntentBits.Guilds,
|
||||
GatewayIntentBits.GuildBans,
|
||||
GatewayIntentBits.GuildMessages,
|
||||
GatewayIntentBits.MessageContent,
|
||||
],
|
||||
partials: [Partials.Channel],
|
||||
intents: [
|
||||
GatewayIntentBits.DirectMessages,
|
||||
GatewayIntentBits.Guilds,
|
||||
GatewayIntentBits.GuildBans,
|
||||
GatewayIntentBits.GuildMessages,
|
||||
GatewayIntentBits.MessageContent,
|
||||
],
|
||||
partials: [Partials.Channel],
|
||||
});
|
||||
|
||||
const channelID = '1094494101631680653'; // Replace with your channel ID
|
||||
@ -25,114 +30,142 @@ const channelID2 = '1094628334727614605'; // Replace with your channel ID
|
||||
const conversations = new Map();
|
||||
|
||||
client.once('ready', () => {
|
||||
console.log('Bot is ready.');
|
||||
client.user.setPresence({
|
||||
activities: [{ name: `AI`, type: ActivityType.Playing }],
|
||||
status: 'dnd',
|
||||
});
|
||||
console.log('Bot is ready.');
|
||||
client.user.setPresence({
|
||||
activities: [{
|
||||
name: `AI`,
|
||||
type: ActivityType.Playing
|
||||
}],
|
||||
status: 'dnd',
|
||||
});
|
||||
});
|
||||
|
||||
client.on('messageCreate', async (message) => {
|
||||
// Only respond in the specified channels
|
||||
if (message.channel.id !== channelID && message.channel.id !== channelID2) {
|
||||
return;
|
||||
}
|
||||
if (message.author.bot) return; // Ignore messages from bots
|
||||
|
||||
if (conversations.get(message.author.id)?.busy) {
|
||||
message.delete()
|
||||
await message.author.send("I'm currently busy with another process. Please try again later.");
|
||||
return;
|
||||
}
|
||||
|
||||
const userID = message.author.id;
|
||||
let conversation = conversations.get(userID) || { messages: [], busy: false };
|
||||
|
||||
|
||||
if (conversation.messages.length === 0) {
|
||||
conversation.messages.push({ role: 'user', content: `You are the smartest AI and help with anything I ask. You are great at coding! When giving a list always use one line per item. My name is ${message.author.username}.` });
|
||||
conversation.messages.push({ role: 'user', content: `My name is ${message.author.username}.` });
|
||||
conversation.messages.push({ role: 'assistant', content: `Hello, ${message.author.username}, how may I help you?` });
|
||||
conversation.messages.push({ role: 'user', content: message.cleanContent });
|
||||
|
||||
} else {
|
||||
// Append user message to conversation history
|
||||
conversation.messages.push({ role: 'user', content: message.cleanContent });
|
||||
}
|
||||
if (message.content === '!reset' || message.content === '!r') {
|
||||
conversations.delete(userID); // Delete user's conversation map if they request reset
|
||||
await message.channel.send('Conversation reset.');
|
||||
return;
|
||||
}
|
||||
|
||||
// Append user message to conversation history
|
||||
conversation.messages.push({ role: 'user', content: message.cleanContent });
|
||||
|
||||
try {
|
||||
conversation.busy = true;
|
||||
|
||||
const response = await generateResponse(conversation);
|
||||
|
||||
// Append bot message to conversation history
|
||||
conversation.messages.push({ role: 'assistant', content: response });
|
||||
|
||||
if (response && response.trim()) {
|
||||
// Send response to user if it's not empty
|
||||
await message.channel.send(response);
|
||||
conversation.busy = false;
|
||||
} else {
|
||||
// Handle empty response here
|
||||
const randomResponse = emptyResponses[Math.floor(Math.random() * emptyResponses.length)];
|
||||
// We had an empty response, create a new memory map for sanity.
|
||||
await message.channel.send(randomResponse); // give the user a human like reponse about the error
|
||||
conversations.delete(userID); // Delete user's conversation map if they request reset
|
||||
const resetMessage = resetResponses[Math.floor(Math.random() * resetResponses.length)];
|
||||
await message.channel.send(resetMessage); // give a notification of reset using a human like response.
|
||||
conversation.busy = false;
|
||||
// Only respond in the specified channels
|
||||
if (message.channel.id !== channelID && message.channel.id !== channelID2) {
|
||||
return;
|
||||
}
|
||||
if (message.author.bot) return; // Ignore messages from bots
|
||||
|
||||
if (conversations.get(message.author.id)?.busy) {
|
||||
message.delete()
|
||||
await message.author.send("I'm currently busy with another process. Please try again later.");
|
||||
return;
|
||||
}
|
||||
|
||||
const userID = message.author.id;
|
||||
let conversation = conversations.get(userID) || {
|
||||
messages: [],
|
||||
busy: false
|
||||
};
|
||||
|
||||
|
||||
if (conversation.messages.length === 0) {
|
||||
conversation.messages.push({
|
||||
role: 'user',
|
||||
content: `You are the smartest AI and help with anything I ask. You are great at coding! When giving a list always use one line per item. My name is ${message.author.username}.`
|
||||
});
|
||||
conversation.messages.push({
|
||||
role: 'user',
|
||||
content: `My name is ${message.author.username}.`
|
||||
});
|
||||
conversation.messages.push({
|
||||
role: 'assistant',
|
||||
content: `Hello, ${message.author.username}, how may I help you?`
|
||||
});
|
||||
conversation.messages.push({
|
||||
role: 'user',
|
||||
content: message.cleanContent
|
||||
});
|
||||
|
||||
} else {
|
||||
// Append user message to conversation history
|
||||
conversation.messages.push({
|
||||
role: 'user',
|
||||
content: message.cleanContent
|
||||
});
|
||||
}
|
||||
if (message.content === '!reset' || message.content === '!r') {
|
||||
conversations.delete(userID); // Delete user's conversation map if they request reset
|
||||
await message.channel.send('Conversation reset.');
|
||||
return;
|
||||
}
|
||||
|
||||
// Append user message to conversation history
|
||||
conversation.messages.push({
|
||||
role: 'user',
|
||||
content: message.cleanContent
|
||||
});
|
||||
|
||||
try {
|
||||
conversation.busy = true;
|
||||
|
||||
const response = await generateResponse(conversation);
|
||||
|
||||
// Append bot message to conversation history
|
||||
conversation.messages.push({
|
||||
role: 'assistant',
|
||||
content: response
|
||||
});
|
||||
|
||||
if (response && response.trim()) {
|
||||
// Send response to user if it's not empty
|
||||
await message.channel.send(response);
|
||||
conversation.busy = false;
|
||||
} else {
|
||||
// Handle empty response here
|
||||
const randomResponse = emptyResponses[Math.floor(Math.random() * emptyResponses.length)];
|
||||
// We had an empty response, create a new memory map for sanity.
|
||||
await message.channel.send(randomResponse); // give the user a human like reponse about the error
|
||||
conversations.delete(userID); // Delete user's conversation map if they request reset
|
||||
const resetMessage = resetResponses[Math.floor(Math.random() * resetResponses.length)];
|
||||
await message.channel.send(resetMessage); // give a notification of reset using a human like response.
|
||||
conversation.busy = false;
|
||||
}
|
||||
conversations.set(userID, conversation); // Update user's conversation map in memory
|
||||
} catch (err) {
|
||||
console.error(err);
|
||||
const errorMessage = errorMessages[Math.floor(Math.random() * errorMessages.length)];
|
||||
await message.channel.send(errorMessage); // give a notification of reset using a human like response. } finally {
|
||||
conversation.busy = false;
|
||||
}
|
||||
conversations.set(userID, conversation); // Update user's conversation map in memory
|
||||
} catch (err) {
|
||||
console.error(err);
|
||||
const errorMessage = errorMessages[Math.floor(Math.random() * errorMessages.length)];
|
||||
await message.channel.send(errorMessage); // give a notification of reset using a human like response. } finally {
|
||||
conversation.busy = false;
|
||||
}
|
||||
});
|
||||
|
||||
async function generateResponse(conversation) {
|
||||
const controller = new AbortController();
|
||||
const timeout = setTimeout(() => {
|
||||
controller.abort();
|
||||
}, 900000);
|
||||
const controller = new AbortController();
|
||||
const timeout = setTimeout(() => {
|
||||
controller.abort();
|
||||
}, 900000);
|
||||
|
||||
try {
|
||||
const response = await fetch('http://192.168.0.15/v1/chat/completions', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'accept': 'application/json',
|
||||
'Content-Type': 'application/json'
|
||||
},
|
||||
body: JSON.stringify({ messages: conversation.messages }),
|
||||
signal: controller.signal
|
||||
});
|
||||
try {
|
||||
const response = await fetch('http://192.168.0.15/v1/chat/completions', {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'accept': 'application/json',
|
||||
'Content-Type': 'application/json'
|
||||
},
|
||||
body: JSON.stringify({
|
||||
messages: conversation.messages
|
||||
}),
|
||||
signal: controller.signal
|
||||
});
|
||||
|
||||
const responseData = await response.json();
|
||||
const choice = responseData.choices[0];
|
||||
const responseData = await response.json();
|
||||
const choice = responseData.choices[0];
|
||||
|
||||
// Remove "user None:" and any text after it from the response
|
||||
const responseText = choice.message.content.trim();
|
||||
const startIndex = responseText.indexOf('user None:');
|
||||
const sanitizedResponse = startIndex === -1 ? responseText : responseText.substring(0, startIndex);
|
||||
// Remove "user None:" and any text after it from the response
|
||||
const responseText = choice.message.content.trim();
|
||||
const startIndex = responseText.indexOf('user None:');
|
||||
const sanitizedResponse = startIndex === -1 ? responseText : responseText.substring(0, startIndex);
|
||||
|
||||
return sanitizedResponse;
|
||||
} catch (err) {
|
||||
console.error(err);
|
||||
return 'Oops, something went wrong!';
|
||||
} finally {
|
||||
clearTimeout(timeout);
|
||||
}
|
||||
return sanitizedResponse;
|
||||
} catch (err) {
|
||||
console.error(err);
|
||||
return 'Oops, something went wrong!';
|
||||
} finally {
|
||||
clearTimeout(timeout);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
client.login(process.env.THE_TOKEN); // Replace with your bot token
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user