91 lines
3.1 KiB
JavaScript
91 lines
3.1 KiB
JavaScript
const Discord = require("discord.js");
|
|
const fetch = require("node-fetch");
|
|
const fs = require('fs');
|
|
|
|
const server = "192.168.1.213";
|
|
|
|
class BotChat {
|
|
|
|
async get_response(history, input_text) {
|
|
const complete_prompt = this.bot_prompt.replace("<CONVHISTORY>",history).replace("<INTEXT>",input_text);
|
|
|
|
console.log("PROMPT:" + complete_prompt);
|
|
|
|
const response = await fetch(`http://${server}:8080/v1/engines/llama_13B_q4/completions`, {
|
|
method: "POST",
|
|
headers: { "Content-Type": "application/json" },
|
|
body: JSON.stringify({
|
|
prompt: complete_prompt
|
|
})
|
|
});
|
|
|
|
const response_json = await response.json();
|
|
console.log(response_json);
|
|
return response_json.text;
|
|
}
|
|
|
|
constructor() {
|
|
this.matches = [ "botchat", "botmention", "botrandom" ];
|
|
this.bot_prompt = fs.readFileSync("bot-modules/botchat/prompts/default.txt").toString();
|
|
}
|
|
|
|
async getRecentChannelHistory(chan) {
|
|
let history = "";
|
|
// Fetch the last 5 messages in the channel
|
|
const messages = await chan.messages.fetch({ limit: 5 });
|
|
// Reverse the order of the messages so that the oldest one is first
|
|
const reversed = messages.reverse();
|
|
// Loop through the messages and print their content and author
|
|
for (let msg of reversed) {
|
|
const user = msg[1].author.username;
|
|
const content = msg[1].content;
|
|
history += `${user}: ${content}\n`;
|
|
}
|
|
|
|
return history;
|
|
}
|
|
|
|
async action(msg, botrandom) {
|
|
let chan = msg.channel;
|
|
try {
|
|
const input_text = msg.content.replace("!botchat","").replace(/<@.+?>/,"bot,").trim();
|
|
console.log(`Input Text: ${input_text}`);
|
|
const history = await this.getRecentChannelHistory(chan);
|
|
const trimmed_text = await this.get_response(history,input_text);
|
|
chan.send(trimmed_text);
|
|
// chan.send(trimmed_text);
|
|
/*
|
|
const input_text = msg.content.replace("!botchat","").replace(/<@.+?>/,"anon").trim();
|
|
console.log(`Input Text: ${input_text}`);
|
|
await chan.sendTyping();
|
|
if (botrandom == "botrandom" || msg.content.startsWith('!botrandom')) {
|
|
const output_json = await this.random_prompt();
|
|
const output_text = (output_json.data[0]);
|
|
const startIndex = output_text.indexOf("865467536");
|
|
const output_only = output_text.slice(startIndex+9).split("---")[0].trim();
|
|
console.log(`Generating random nonsense: ${output_only}`);
|
|
chan.send(output_only);
|
|
} else if (msg.reference) {
|
|
const repliedTo = await msg.channel.messages.fetch(msg.reference.messageId);
|
|
const oldReply = repliedTo.content;
|
|
const output_json = await this.prompt_reply(oldReply,input_text);
|
|
const output_text = (output_json.data[0]);
|
|
const trimmed_text = this.find_response(output_text);
|
|
console.log(`Generated Reply: ${trimmed_text}`);
|
|
chan.send(trimmed_text);
|
|
} else {
|
|
const output_json = await this.prompt(input_text);
|
|
const output_text = (output_json.data[0]);
|
|
const trimmed_text = this.find_response(output_text);
|
|
console.log(`Generated Message: ${trimmed_text}`);
|
|
chan.send(trimmed_text);
|
|
}*/
|
|
} catch (e) {
|
|
chan.send(`Error: ${e}`);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
module.exports = BotChat;
|