mirrors/pissbot-9000
1
0
Fork 0
mirror of https://gitlab.com/bignutty/labscore.git synced 2025-06-08 14:13:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

wip gpt command

Browse source
This commit is contained in:
derpystuff 2023-04-20 00:16:59 +02:00
parent 0bad32c185
commit dfc8d20c84
4 changed files with 107 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

57
commands/interaction/slash/gpt.js Normal file
View file

@ -0,0 +1,57 @@
const { Constants } = require('detritus-client');
const { InteractionCallbackTypes, ApplicationCommandOptionTypes } = Constants;
const { AI_GPT_MODELS, AI_GPT_MODEL_CONFIG } = require("../../../labscore/constants");
const superagent = require('superagent')
const { createEmbed } = require('../../../labscore/utils/embed');
const { icon, highlight, codeblock } = require('../../../labscore/utils/markdown');
module.exports = {
description: 'Access a wide selection of Large Language Models.',
name: 'gpt',
options: [
{
name: 'model',
description: 'LLM to use',
choices: AI_GPT_MODELS,
required: true,
},
{
name: 'prompt',
description: 'Prompt',
type: ApplicationCommandOptionTypes.STRING,
required: true,
maxLength: 256
}
],
run: async (context, args) => {
try{
let s = Date.now()
await context.respond({data: {}, type: InteractionCallbackTypes.DEFERRED_CHANNEL_MESSAGE_WITH_SOURCE})
await context.editOrRespond({
embeds: [createEmbed("loading_ai", context)]
})
let res = await superagent.get(`${process.env.AI_SERVER}/gpt`)
.query({
model: args.model,
prompt: args.prompt
})
await context.editOrRespond({
embeds: [createEmbed("default", context, {
footer: {
iconUrl: `https://derpystuff.gitlab.io/webstorage4/v2/assets/icons/ai/ico_ai_${AI_GPT_MODEL_CONFIG[args.model].icon}.png`,
text: `${AI_GPT_MODEL_CONFIG[args.model].name} • ${context.application.name}`,
},
description: codeblock("ansi", [res.body.response.substr(0, 1024).replace(/\\n/g,'\n')])
})]
})
}catch(e){
console.log(e)
await context.editOrRespond({
embeds: [createEmbed("error", context, "Unable to generate response. Try again in a bit.")]
})
}
},
};

36
labscore/constants.js
View file

@ -298,6 +298,42 @@ module.exports.MICROSOFT_VOICE_CONFIG = {
"Male Whisper": { pitch: 113, speed: 170 }
}
module.exports.AI_GPT_MODEL_CONFIG = {
"anthropic:claude-instant-v1": { name: "Anthropic claude-instant-v1", icon: "anthropic" },
"anthropic:claude-v1": { name: "Anthropic claude-v1", icon: "anthropic" },
"replicate:alpaca-7b": { name: "Replicate alpaca-7b", icon: "replicate" },
"huggingface:bigscience/bloomz": { name: "HuggingFace bigscience/bloomz", icon: "huggingface" },
"huggingface:google/flan-t5-xxl": { name: "HuggingFace google/flan-t5-xxl", icon: "huggingface" },
"huggingface:google/flan-ul2": { name: "HuggingFace google/flan-ul2", icon: "huggingface" },
"cohere:command-medium-nightly": { name: "Cohere command-medium-nightly", icon: "cohere" },
"cohere:command-xlarge-nightly": { name: "Cohere command-medium-nightly", icon: "cohere" },
"openai:gpt-3.5-turbo": { name: "OpenAI gpt-3.5-turbo", icon: "openai" },
"openai:text-ada-001": { name: "OpenAI text-ada-001", icon: "openai" },
"openai:text-babbage-001": { name: "OpenAI text-babbage-001", icon: "openai" },
"openai:text-curie-001": { name: "OpenAI text-curie-001", icon: "openai" },
"openai:text-davinci-002": { name: "OpenAI text-davinci-002", icon: "openai" },
"openai:text-davinci-003": { name: "OpenAI text-davinci-003", icon: "openai" }
}
module.exports.AI_GPT_MODELS = [
/*
{ name: "Anthropic claude-instant-v1", value: "anthropic:claude-instant-v1" },
{ name: "HuggingFace bigscience/bloomz", value: "huggingface:bigscience/bloomz" },
{ name: "HuggingFace google/flan-t5-xxl", value: "huggingface:google/flan-t5-xxl" },
{ name: "HuggingFace google/flan-ul2", value: "huggingface:google/flan-ul2" },
{ name: "Cohere command-medium-nightly", value: "cohere:command-medium-nightly" },
{ name: "Cohere command-medium-nightly", value: "cohere:command-xlarge-nightly" },
*/
{ name: "OpenAI gpt-3.5-turbo (ChatGPT)", value: "openai:gpt-3.5-turbo" },
{ name: "Anthropic claude-v1", value: "anthropic:claude-v1" },
{ name: "Replicate alpaca-7b", value: "replicate:alpaca-7b" },
{ name: "OpenAI text-ada-001", value: "openai:text-ada-001" },
{ name: "OpenAI text-babbage-001", value: "openai:text-babbage-001" },
{ name: "OpenAI text-curie-001", value: "openai:text-curie-001" },
{ name: "OpenAI text-davinci-002", value: "openai:text-davinci-002" },
{ name: "OpenAI text-davinci-003", value: "openai:text-davinci-003" }
]
module.exports.TRANSLATE_LANGUAGE_MAPPINGS = Object.freeze({
"af": "🇿🇦",
"sq": "🇦🇱",

9
labscore/utils/embed.js
View file

@ -69,6 +69,15 @@ const embedTypes = Object.freeze({
},
color: COLORS.embed
}
},
"loading_ai": (context) => {
return {
author: {
iconUrl: STATIC_ICONS.loading_ai,
name: `Generating...`
},
color: COLORS.brand
}
}
})

5
labscore/utils/statics.js
View file

@ -82,6 +82,10 @@ const Statics = Object.freeze({
file: "icons/core/ico_notice_loading.gif",
revision: 0
},
loading_ai: {
file: "icons/ai/ico_ai_generating.gif",
revision: 0
},
warning: {
file: "icons/core/ico_notice_warning.png",
revision: 0
@ -120,5 +124,6 @@ module.exports.STATIC_ICONS = Object.freeze({
adult: staticAsset(Statics.icons.adult),
error: staticAsset(Statics.icons.error),
loading: staticAsset(Statics.icons.loading),
loading_ai: staticAsset(Statics.icons.loading_ai),
warning: staticAsset(Statics.icons.warning)
})