diff --git a/src/index.ts b/src/index.ts index 410d97a..ea66772 100644 --- a/src/index.ts +++ b/src/index.ts @@ -39,6 +39,10 @@ interface AppConfig { imageApiKey?: string; imageBaseUrl?: string; imageModel?: string; + imageSize?: string; + imageQuality?: string; + imageStyle?: string; + imageN?: number; autoConfirm?: boolean; feishuWebhook?: string; feishuKeyword?: string; diff --git a/src/tools/image.ts b/src/tools/image.ts index ed89c82..a31e2b1 100644 --- a/src/tools/image.ts +++ b/src/tools/image.ts @@ -90,13 +90,14 @@ const handler = async (args: any, config: any): Promise => { prompt, image_path, mask_path, - n = 1, - size = "1024x1024", - quality = "standard", - style = "vivid", output_dir = "." } = args; + const n = args.n || config.imageN || 1; + const size = args.size || config.imageSize || "1024x1024"; + const quality = args.quality || config.imageQuality || "standard"; + const style = args.style || config.imageStyle || "vivid"; + let mode = args.mode; let model = args.model || config.imageModel || "dall-e-3";