From 5281f2ba64092097e7dd1acb87857d1a00208393 Mon Sep 17 00:00:00 2001
From: "Adam.Wang" <bddiudiu@gmail.com>
Date: Thu, 22 May 2025 13:58:05 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=81=AB=E5=B1=B1=E5=BC=95=E6=93=8E?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E6=96=87=E7=94=9F=E5=9B=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dto/dalle.go         | 1 +
 relay/relay-image.go | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/dto/dalle.go b/dto/dalle.go
index a1309b6c..ce2f6361 100644
--- a/dto/dalle.go
+++ b/dto/dalle.go
@@ -15,6 +15,7 @@ type ImageRequest struct {
 	Background     string          `json:"background,omitempty"`
 	Moderation     string          `json:"moderation,omitempty"`
 	OutputFormat   string          `json:"output_format,omitempty"`
+	Watermark      *bool           `json:"watermark,omitempty"`
 }
 
 type ImageResponse struct {
diff --git a/relay/relay-image.go b/relay/relay-image.go
index daed3d80..9b1515c4 100644
--- a/relay/relay-image.go
+++ b/relay/relay-image.go
@@ -18,6 +18,7 @@ import (
 	"strings"
 
 	"github.com/gin-gonic/gin"
+	"one-api/relay/constant"
 )
 
 func getAndValidImageRequest(c *gin.Context, info *relaycommon.RelayInfo) (*dto.ImageRequest, error) {
@@ -41,6 +42,11 @@ func getAndValidImageRequest(c *gin.Context, info *relaycommon.RelayInfo) (*dto.
 				imageRequest.Quality = "standard"
 			}
 		}
+
+		if info.ApiType == constant.APITypeVolcEngine {
+			watermark := formData.Has("watermark")
+			imageRequest.Watermark = &watermark
+		}
 	default:
 		err := common.UnmarshalBodyReusable(c, imageRequest)
 		if err != nil {