Added working 4chan-gpt integration
This commit is contained in:
1 parent
133393bf38
commit
6f0040e9a5
4 files changed
+94
-14
No files matched your search
@@ -0,0 +1,71 @@
|
||||
const Discord = require("discord.js");
|
||||
const fetch = require("node-fetch");
|
||||
const fs = require('fs');
|
||||
|
||||
const server = "192.168.1.70";
|
||||
|
||||
class Fourchat {
|
||||
|
||||
|
||||
|
||||
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
|
||||
`,
|
||||
200,
|
||||
true,
|
||||
1.99,
|
||||
0.18,
|
||||
1,
|
||||
1.15,
|
||||
30,
|
||||
0,
|
||||
0,
|
||||
1,
|
||||
0,
|
||||
1,
|
||||
false,
|
||||
]
|
||||
})
|
||||
});
|
||||
|
||||
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" ];
|
||||
}
|
||||
|
||||
async action(msg) {
|
||||
let chan = msg.channel;
|
||||
try {
|
||||
const input_text = msg.content.replace("!4chat","").replace(/<@.+?>/,"anon").trim();
|
||||
console.log(`Input Text: ${input_text}`);
|
||||
const output_json = await this.prompt(input_text);
|
||||
const output_text = (output_json.data[0]);
|
||||
const trimmed_text = this.find_response(output_text);
|
||||
chan.send(trimmed_text);
|
||||
} catch (e) {
|
||||
chan.send(`Error: ${e}`);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
module.exports = Fourchat;
|
||||
Reference in new issue
Block a user