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

reflect mono2 changes (c4e83892)

Browse source
This commit is contained in:
bignutty 2024-05-13 15:30:29 +02:00
parent 97204fd33b
commit f91f4dff7b
3 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
commands/message/genai/imagen.js
View file

@ -1,4 +1,4 @@
const { imagen } = require("../../../labscore/api/obelisk");
const { GenerativeImagesModelsImagen } = require("../../../labscore/api/obelisk");
const { createEmbed } = require("../../../labscore/utils/embed");
const { editOrReply } = require("../../../labscore/utils/message");
@ -41,7 +41,7 @@ module.exports = {
await editOrReply(context, {embeds: loadingEmbeds});
let res = await imagen(context, args.text);
let res = await GenerativeImagesModelsImagen(context, args.text);
// Construct Embeds
let files = [];

2
labscore/api/obelisk/endpoints.js
View file

@ -10,6 +10,8 @@ const ObeliskApi = Object.freeze({
LLM_PRIVATE_BARD: "/llm/v1/_private:bard",
LLM_MODELS_GENERATE: "/llm/v1/generate",
GENERATIVEIMAGES_MODELS_IMAGEN: "/generativeimages/v1/models/imagen",
GOOGLE_BARD: "/parrot/v1/google:bard",
GOOGLE_GEMINI_PRO: "/parrot/v1/google:gemini",
GOOGLE_GEMINI_PRO_VISION: "/parrot/v1/google:geminiVision",

8
labscore/api/obelisk/index.js
View file

@ -56,6 +56,14 @@ module.exports.LlmModelsGenerate = async function(context, model, prompt, harmLe
})
}
module.exports.GenerativeImagesModelsImagen = async function(context, prompt){
return await request(ObeliskApi.GENERATIVEIMAGES_MODELS_IMAGEN, "POST", {}, {
image_prompt: prompt
})
}
// GENERATIVEAI
module.exports.bard = async function(context, input){
return await request(ObeliskApi.GOOGLE_BARD, "POST", {}, {