mirrors/pissbot-9000
1
0
Fork 0
mirror of https://gitlab.com/bignutty/labscore.git synced 2025-06-17 10:27:05 -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)
})
}

13
commands/message/limited/clyde.js
View file

@ -9,6 +9,7 @@ const { iconPill, stringwrap } = require('../../../labscore/utils/markdown')
const { Permissions } = require("detritus-client/lib/constants");
const { getUser } = require('../../../labscore/utils/users');
const { chatgpt } = require('../../../labscore/api/obelisk');
const LOADING_QUIPS = [
"Crunching the data, one byte at a time...",
@ -86,16 +87,8 @@ Current time: ${new Date().toLocaleDateString('en-us', { weekday:"long", year:"n
await editOrReply(context, e)
let res = await superagent.post(`${process.env.AI_SERVER}/openai`)
.set({
Authorization: process.env.AI_SERVER_KEY
})
.send({
prompt,
input: [input],
temperature: "0.75",
model: "CHATGPT"
})
let res = await chatgpt(context, prompt, input);
res = res.response;
let description = []
let files = [];

18
commands/message/limited/disstrack.js
View file

@ -6,7 +6,8 @@ const { codeblock, iconPill, smallIconPill } = require('../../../labscore/utils/
const { Permissions } = require("detritus-client/lib/constants");
const { canUseLimitedTestCommands } = require('../../../labscore/utils/testing')
const { STATICS } = require('../../../labscore/utils/statics')
const { STATICS } = require('../../../labscore/utils/statics');
const { chatgpt } = require('../../../labscore/api/obelisk');
module.exports = {
name: 'disstrack',
@ -24,18 +25,11 @@ module.exports = {
context.triggerTyping();
if(!args.text) return editOrReply(context, createEmbed("warning", context, `Missing Parameter (text).`))
try{
await editOrReply(context, createEmbed("ai", context, "Generating response..."))
await editOrReply(context, createEmbed("ai", context, "Spitting bars..."))
let res = await superagent.post(`${process.env.AI_SERVER}/openai`)
.set({
Authorization: process.env.AI_SERVER_KEY
})
.send({
prompt: "Write a disstrack about the subject the user supplies. The disstrack should have at least two verses and a chorus.",
input: [args.text],
temperature: 0.6,
model: "CHATGPT"
})
let res = await chatgpt(context, "Write a disstrack about the subject the user supplies. The disstrack should have at least one verse and a chorus.", args.text);
res = res.response;
return editOrReply(context, createEmbed("default", context, {
description: smallIconPill("generative_ai", args.text) + '\n' + codeblock("ansi", [res.body.output.substr(0, 2020 - args.text.length)]),
footer: {

15
commands/message/limited/palm.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 { palm2 } = require('../../../labscore/api/obelisk');
module.exports = {
name: 'palm',
@ -56,17 +57,9 @@ module.exports = {
try{
await editOrReply(context, createEmbed("ai_custom", context, STATIC_ICONS.ai_palm_idle))
let res = await superagent.post(`${process.env.AI_SERVER}/google/palm2/chat`)
.set({
Authorization: process.env.AI_SERVER_KEY
})
.send({
prompt,
input: [input],
temperature,
model
})
let res = await palm2(context, prompt, input)
res = res.response;
let description = []
let files = [];