improve gemini-pro command

This commit is contained in:
bignutty 2025-06-10 19:55:59 +02:00
parent ab2ddea506
commit 475d83987d
4 changed files with 41 additions and 13 deletions

View file

@ -8,6 +8,7 @@ const Api = Object.freeze({
HOST: Hosts.prod,
GOOGLE_GENERATIVEAI_EDIT_IMAGE: '/google/generativeai/edit-image',
GOOGLE_GENERATIVEAI_GEMINI_API: '/google/generativeai/gemini',
GOOGLE_GENERATIVEAI_IMAGEN: '/google/generativeai/imagen',
GOOGLE_PERSPECTIVE: '/google/perspective/analyze',
GOOGLE_SPEECH_RECOGNIZE: '/google/speech/recognize',

View file

@ -49,6 +49,14 @@ module.exports.googleGenaiEditImage = async function(context, prompt, url){
})
}
module.exports.googleGenaiGeminiApi = async function(context, model, input, prompt){
return await request(Api.GOOGLE_GENERATIVEAI_GEMINI_API, "GET", {}, {
prompt,
input,
model
})
}
module.exports.googleGenaiImagen = async function(context, prompt, imageCount = 2){
return await request(Api.GOOGLE_GENERATIVEAI_IMAGEN, "GET", {}, {
prompt: prompt,