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 {