mirrors/pissbot-9000
1
0
Fork 0
mirror of https://gitlab.com/bignutty/labscore.git synced 2025-06-17 18:37:07 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

migrate ai commands to obelisk

Browse source
This commit is contained in:
derpystuff 2023-12-14 21:46:33 +01:00
parent 3b478bd69e
commit 641ab48053
6 changed files with 52 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

23
commands/message/limited/chat.js
View file

@ -8,6 +8,7 @@ const superagent = require('superagent')
const { iconPill, stringwrap, smallIconPill } = require('../../../labscore/utils/markdown')
const { Permissions } = require("detritus-client/lib/constants");
const { chatgpt } = require('../../../labscore/api/obelisk');
const MODELS = {
"chatgpt": {
@ -78,27 +79,23 @@ module.exports = {
try{
await editOrReply(context, createEmbed("ai", context, "Generating response..."))
let res = await superagent.post(`${process.env.AI_SERVER}/openai`)
.set({
Authorization: process.env.AI_SERVER_KEY
})
.send({
prompt,
input: [input],
temperature,
model: MODELS[model.toLowerCase()].id
})
let res;
if(model.toLowerCase() == "chatgpt"){
res = await chatgpt(context, prompt, input)
} else if (model.toLowerCase() == "gpt4"){
res = await chatgpt(context, prompt, input)
}
let description = []
let files = [];
if(!res.body.output) throw "Unable to generate response"
if(!res.response.body.output) throw "Unable to generate response"
if(res.body.output.length <= 4000) description.push(res.body.output)
if(res.response.body.output.length <= 4000) description.push(res.response.body.output)
else {
files.push({
filename: `chat.${Date.now().toString(36)}.txt`,
value: Buffer.from(res.body.output)
value: Buffer.from(res.response.body.output)
})
}