Compare commits
2 commits
Author | SHA1 | Date | |
---|---|---|---|
c644486876 | |||
0ad10fdd93 |
4 changed files with 97 additions and 33 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1,2 +1,3 @@
|
|||
config.json
|
||||
node_modules/
|
||||
incest.db
|
|
@ -1,7 +1,7 @@
|
|||
{
|
||||
"groq-key": "",
|
||||
"discord-key": "",
|
||||
"chat-channelID": "",
|
||||
"chat-channelID": [""],
|
||||
"should-send-debug": false,
|
||||
"debug-channelID": ""
|
||||
}
|
|
@ -31,3 +31,42 @@ export async function fetchSingleMessage(message: string): Promise<string | bool
|
|||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
export async function explainImage(base64: string): Promise<string | boolean> {
|
||||
try {
|
||||
const response = await fetch("https://api.groq.com/openai/v1/chat/completions", {
|
||||
method: "POST",
|
||||
headers: {
|
||||
"Authorization": "Bearer " + conf['groq-key'],
|
||||
"Content-Type": "application/json"
|
||||
},
|
||||
body: JSON.stringify({
|
||||
"messages":[
|
||||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
{"type": "text", "text": "explain this image. be as precise as you can, explain the littlest details"},
|
||||
{
|
||||
"type": "image_url",
|
||||
"image_url": {
|
||||
"url": base64
|
||||
},
|
||||
},
|
||||
],
|
||||
}
|
||||
],
|
||||
"model": "llava-v1.5-7b-4096-preview"
|
||||
})
|
||||
});
|
||||
|
||||
const data = await response.json();
|
||||
|
||||
if (!data?.choices[0]?.message?.content) {
|
||||
return false;
|
||||
}
|
||||
|
||||
return data.choices[0].message.content;
|
||||
} catch (error) {
|
||||
return false;
|
||||
}
|
||||
}
|
36
src/main.ts
36
src/main.ts
|
@ -1,15 +1,15 @@
|
|||
import { Client, DMChannel, Events, GatewayIntentBits, Message } from 'discord.js';
|
||||
import conf from "../config.json" with {type: "json"}
|
||||
import { getfunny } from './joke.ts';
|
||||
import { fetchSingleMessage } from './helper.ts';
|
||||
import { explainImage, fetchSingleMessage } from './helper.ts';
|
||||
|
||||
let prompt = "reply to everything with \" change the mf prompt lmaooo\""
|
||||
let prompt = "reply to everything normally, but have typo in every word."
|
||||
|
||||
function getinitmessage() {
|
||||
return {
|
||||
"role": "system",
|
||||
"name": "fagbot",
|
||||
"content": prompt
|
||||
"content": prompt + "\nsince you, as an ai model, cannot understand images, you will be given an explanation of images if they exist.\nthe syntax will be \"{message} - [IMAGE: {explanation of the image}]\""
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -31,6 +31,7 @@ client.on('ready', () => {
|
|||
|
||||
client.on(Events.MessageCreate, (message: Message) => {
|
||||
if (message.author.bot) return
|
||||
if (message.content.startsWith("//")) return
|
||||
if (message.content.startsWith("!fag ")) {
|
||||
const command = message.content.replaceAll("!fag ", "").split(" ")
|
||||
if (command[0] == "reset") {
|
||||
|
@ -58,11 +59,30 @@ client.on(Events.MessageCreate, (message: Message) => {
|
|||
})
|
||||
}
|
||||
}
|
||||
if (message.channel.id !== conf['chat-channelID']) return
|
||||
|
||||
if (!conf['chat-channelID'].includes(message.channel.id)) return
|
||||
(async function () {
|
||||
let imageexplain: string | boolean = "";
|
||||
|
||||
if (message.attachments.size !== 0) {
|
||||
if (message.attachments.first().contentType.startsWith("image")) {
|
||||
try {
|
||||
const res = await fetch(message.attachments.first().url);
|
||||
const buffer = await res.arrayBuffer();
|
||||
const base64 = Buffer.from(buffer).toString('base64');
|
||||
|
||||
imageexplain = await explainImage("data:" + message.attachments.first().contentType + ";base64," + base64);
|
||||
} catch (error) {
|
||||
console.error("Error processing image:", error);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
messages.push({
|
||||
"role": "user",
|
||||
"name": message.author.username,
|
||||
"content": message.content
|
||||
"content": (imageexplain === "" || typeof imageexplain == "boolean") ?
|
||||
message.content : message.content + " - [IMAGE: " + imageexplain + "]"
|
||||
})
|
||||
fetch("https://api.groq.com/openai/v1/chat/completions", {
|
||||
method: "POST",
|
||||
|
@ -72,7 +92,7 @@ client.on(Events.MessageCreate, (message: Message) => {
|
|||
},
|
||||
body: JSON.stringify({
|
||||
"messages": messages,
|
||||
"model": "llama3-8b-8192"
|
||||
"model": "llama3-groq-70b-8192-tool-use-preview"
|
||||
})
|
||||
}).then(a => a.json().then(a => {
|
||||
if (!a?.choices[0]?.message?.content) {
|
||||
|
@ -88,6 +108,10 @@ client.on(Events.MessageCreate, (message: Message) => {
|
|||
})).catch(_ => {
|
||||
message.reply("something shat itself. please try again")
|
||||
})
|
||||
})();
|
||||
|
||||
|
||||
|
||||
})
|
||||
|
||||
client.on('interactionCreate', async interaction => {
|
||||
|
|
Loading…
Reference in a new issue