reflect mono2 changes (c4e83892)

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

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",

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", {}, {