mirror of
https://gitlab.com/bignutty/labscore.git
synced 2025-06-09 06:33:05 -04:00
reflect mono2 changes (c4e83892)
This commit is contained in:
parent
97204fd33b
commit
f91f4dff7b
3 changed files with 12 additions and 2 deletions
|
@ -1,4 +1,4 @@
|
||||||
const { imagen } = require("../../../labscore/api/obelisk");
|
const { GenerativeImagesModelsImagen } = require("../../../labscore/api/obelisk");
|
||||||
const { createEmbed } = require("../../../labscore/utils/embed");
|
const { createEmbed } = require("../../../labscore/utils/embed");
|
||||||
const { editOrReply } = require("../../../labscore/utils/message");
|
const { editOrReply } = require("../../../labscore/utils/message");
|
||||||
|
|
||||||
|
@ -41,7 +41,7 @@ module.exports = {
|
||||||
|
|
||||||
await editOrReply(context, {embeds: loadingEmbeds});
|
await editOrReply(context, {embeds: loadingEmbeds});
|
||||||
|
|
||||||
let res = await imagen(context, args.text);
|
let res = await GenerativeImagesModelsImagen(context, args.text);
|
||||||
|
|
||||||
// Construct Embeds
|
// Construct Embeds
|
||||||
let files = [];
|
let files = [];
|
||||||
|
|
|
@ -10,6 +10,8 @@ const ObeliskApi = Object.freeze({
|
||||||
LLM_PRIVATE_BARD: "/llm/v1/_private:bard",
|
LLM_PRIVATE_BARD: "/llm/v1/_private:bard",
|
||||||
LLM_MODELS_GENERATE: "/llm/v1/generate",
|
LLM_MODELS_GENERATE: "/llm/v1/generate",
|
||||||
|
|
||||||
|
GENERATIVEIMAGES_MODELS_IMAGEN: "/generativeimages/v1/models/imagen",
|
||||||
|
|
||||||
GOOGLE_BARD: "/parrot/v1/google:bard",
|
GOOGLE_BARD: "/parrot/v1/google:bard",
|
||||||
GOOGLE_GEMINI_PRO: "/parrot/v1/google:gemini",
|
GOOGLE_GEMINI_PRO: "/parrot/v1/google:gemini",
|
||||||
GOOGLE_GEMINI_PRO_VISION: "/parrot/v1/google:geminiVision",
|
GOOGLE_GEMINI_PRO_VISION: "/parrot/v1/google:geminiVision",
|
||||||
|
|
|
@ -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
|
// GENERATIVEAI
|
||||||
module.exports.bard = async function(context, input){
|
module.exports.bard = async function(context, input){
|
||||||
return await request(ObeliskApi.GOOGLE_BARD, "POST", {}, {
|
return await request(ObeliskApi.GOOGLE_BARD, "POST", {}, {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue