diff --git a/bot-modules/4chat/module.js b/bot-modules/4chat/module.js deleted file mode 100644 index e622665..0000000 --- a/bot-modules/4chat/module.js +++ /dev/null @@ -1,156 +0,0 @@ -const Discord = require("discord.js"); -const fetch = require("node-fetch"); -const fs = require('fs'); - -const server = "192.168.1.70"; - -class Fourchat { - - - async random_prompt() { - const response = await fetch(`http://${server}:7860/run/textgen`, { - method: "POST", - headers: { "Content-Type": "application/json" }, - body: JSON.stringify({ - data: [ -`----- ---- 865467536 -`, - 250, //max tokens - true, //do_sample - 0.63, //temperature - 0.98, //top_p - 1, //typical_p - 1.05, //rep penalty - 0, //top_k - 0, //min_length - 0, //no_repeat_ngram_size - 1, //num_beams - 0, //penalty_alpha - 1, //length_penalty - false, //early_stopping - ] - }) - }); - - const response_json = await response.json(); - return response_json; - } - - async prompt(input_text) { - const response = await fetch(`http://${server}:7860/run/textgen`, { - method: "POST", - headers: { "Content-Type": "application/json" }, - body: JSON.stringify({ - data: [ -`----- ---- 865467536 -${input_text} ---- 865467537 ->>865467536 -`, - 250, //max tokens - true, //do_sample - 0.63, //temperature - 0.98, //top_p - 1, //typical_p - 1.05, //rep penalty - 0, //top_k - 0, //min_length - 0, //no_repeat_ngram_size - 1, //num_beams - 0, //penalty_alpha - 1, //length_penalty - false, //early_stopping - ] - }) - }); - - const response_json = await response.json(); - return response_json; - } - - async prompt_reply(input_text1,input_text2) { - const response = await fetch(`http://${server}:7860/run/textgen`, { - method: "POST", - headers: { "Content-Type": "application/json" }, - body: JSON.stringify({ - data: [ -`----- ---- 865467535 -${input_text1} ---- 865467536 ->>865467535 -${input_text2} ---- 865467537 ->>865467536 -`, - 250, //max tokens - true, //do_sample - 0.63, //temperature - 0.98, //top_p - 1, //typical_p - 1.05, //rep penalty - 0, //top_k - 0, //min_length - 0, //no_repeat_ngram_size - 1, //num_beams - 0, //penalty_alpha - 1, //length_penalty - false, //early_stopping - ] - }) - }); - - const response_json = await response.json(); - return response_json; - } - - find_response(res) { - const startIndex = res.indexOf("865467537"); - const output_only = res.slice(startIndex+9); - const responseIndex = output_only.indexOf(">>865467536"); - const s = output_only.slice(responseIndex+12).split("---")[0]; - return s.trim(); -} - - constructor() { - this.matches = [ "4chat", "botmention", "botrandom" ]; - } - - async action(msg, botrandom) { - let chan = msg.channel; - try { - const input_text = msg.content.replace("!4chat","").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 = Fourchat;