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

allow specifying how many images should be generated

Browse source
This commit is contained in:
bignutty 2025-03-04 12:30:45 +01:00
parent 90b4b0c6a8
commit 9b1e76ebc3
2 changed files with 9 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
labscore/api/index.js
View file

@ -42,9 +42,10 @@ async function request(path, type, headers, args, host) {
throw new Error("unsupported, must either use GET or POST");
}
module.exports.googleGenaiImagen = async function(context, prompt){
module.exports.googleGenaiImagen = async function(context, prompt, imageCount = 2){
return await request(Api.GOOGLE_GENERATIVEAI_IMAGEN, "GET", {}, {
prompt: prompt
prompt: prompt,
image_count: imageCount
})
}