Google Imagen: Add personGeneration setting to config.yaml

2b7003a6da2d43e5198b5d0f175519ecc1e1afbe

Cohee <18619528+Cohee1207@users.noreply.github.com>

default/config.yaml+4 -0 Viewed

 gemini:
   # API endpoint version ("v1beta" or "v1alpha")
   apiVersion: 'v1beta'
+  # https://ai.google.dev/gemini-api/docs/imagen#imagen-configuration
+  image:
+    # Leave empty to use the API-default value.
+    personGeneration: 'allow_adult'
 # -- SERVER PLUGIN CONFIGURATION --
 enableServerPlugins: false
 # Attempt to automatically update server plugins on startup

src/endpoints/google.js+3 -1 Viewed

         const isVertex = request.body.api === 'vertexai';
         // Is it even worth it?
         const isDeprecated = model.startsWith('imagegeneration');
+        // Get person generation setting from config
+        const personGeneration = getConfigValue('gemini.image.personGeneration', 'allow_adult');
         const requestBody = {
             instances: [{

                 enhancePrompt: isVertex ? Boolean(request.body.enhance ?? false) : undefined,
                 negativePrompt: isVertex ? (request.body.negative_prompt || undefined) : undefined,
                 aspectRatio: String(request.body.aspect_ratio || '1:1'),
-                personGeneration: !isDeprecated ? 'allow_adult' : undefined,
+                personGeneration: !isDeprecated && personGeneration ? personGeneration : undefined,
                 language: isVertex ? 'auto' : undefined,
                 safetySetting: !isDeprecated ? (isVertex ? 'block_only_high' : 'block_low_and_above') : undefined,
                 addWatermark: isVertex ? false : undefined,