feat: add safe_mode (& cfg_scale) for venice image generation (#2354)

* venice: add safe_mode & cfg_scale for images

* add missing types

for safeMode & cfgScale

* update lockfile

---------

Co-authored-by: Sayo <hi@sayo.wtf>

Author: proteanx

packages/core/src/generation.ts

@@ -1508,6 +1508,8 @@ export const generateImage = async (
         jobId?: string;
         stylePreset?: string;
         hideWatermark?: boolean;
+        safeMode?: boolean;
+        cfgScale?: number;
     },
     runtime: IAgentRuntime
 ): Promise<{
@@ -1714,10 +1716,12 @@ export const generateImage = async (
                     body: JSON.stringify({
                         model: model,
                         prompt: data.prompt,
+                        cfg_scale: data.guidanceScale,
                         negative_prompt: data.negativePrompt,
                         width: data.width,
                         height: data.height,
                         steps: data.numIterations,
+                        safe_mode: data.safeMode,
                         seed: data.seed,
                         style_preset: data.stylePreset,
                         hide_watermark: data.hideWatermark,

packages/core/src/types.ts

@@ -786,6 +786,7 @@ export type Character = {
             steps?: number;
             width?: number;
             height?: number;
+            cfgScale?: number;
             negativePrompt?: string;
             numIterations?: number;
             guidanceScale?: number;
@@ -795,6 +796,7 @@ export type Character = {
             count?: number;
             stylePreset?: string;
             hideWatermark?: boolean;
+            safeMode?: boolean;
         };
         voice?: {
             model?: string; // For VITS

packages/plugin-image-generation/src/index.ts

@@ -110,6 +110,7 @@ const imageGeneration: Action = {
             width?: number;
             height?: number;
             count?: number;
+            cfgScale?: number;
             negativePrompt?: string;
             numIterations?: number;
             guidanceScale?: number;
@@ -118,6 +119,7 @@ const imageGeneration: Action = {
             jobId?: string;
             stylePreset?: string;
             hideWatermark?: boolean;
+            safeMode?: boolean;
         },
         callback: HandlerCallback
     ) => {
@@ -231,18 +233,19 @@ Ensure that your prompt is detailed, vivid, and incorporates all the elements me
                     : {}),
                 ...(options.stylePreset != null ||
                 imageSettings.stylePreset != null
-                    ? {
-                          stylePreset:
-                              options.stylePreset || imageSettings.stylePreset,
-                      }
+                    ? { stylePreset: options.stylePreset ||
+                            imageSettings.stylePreset }
                     : {}),
                 ...(options.hideWatermark != null ||
                 imageSettings.hideWatermark != null
-                    ? {
-                          hideWatermark:
-                              options.hideWatermark ||
-                              imageSettings.hideWatermark,
-                      }
+                    ? { hideWatermark: options.hideWatermark ||
+                            imageSettings.hideWatermark }
+                    : {}),
+                ...(options.safeMode != null || imageSettings.safeMode != null
+                    ? { safeMode: options.safeMode || imageSettings.safeMode }
+                    : {}),
+                ...(options.cfgScale != null || imageSettings.cfgScale != null
+                    ? { cfgScale: options.cfgScale || imageSettings.cfgScale }
                     : {}),
             },
             runtime