oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 火山引擎增加文生图

Browse Source
This commit is contained in:
Adam.Wang
2025-05-22 13:58:05 +08:00
parent 1f9fc09989
commit 5281f2ba64
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dto/dalle.go
View File

@@ -15,6 +15,7 @@ type ImageRequest struct {
Background string `json:"background,omitempty"` Background string `json:"background,omitempty"`
Moderation string `json:"moderation,omitempty"` Moderation string `json:"moderation,omitempty"`
OutputFormat string `json:"output_format,omitempty"` OutputFormat string `json:"output_format,omitempty"`
Watermark *bool `json:"watermark,omitempty"`
} }
type ImageResponse struct { type ImageResponse struct {

6
relay/relay-image.go
View File

@@ -18,6 +18,7 @@ import (
"strings" "strings"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"one-api/relay/constant"
) )
func getAndValidImageRequest(c *gin.Context, info *relaycommon.RelayInfo) (*dto.ImageRequest, error) { 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" imageRequest.Quality = "standard"
} }
} }
if info.ApiType == constant.APITypeVolcEngine {
watermark := formData.Has("watermark")
imageRequest.Watermark = &watermark
}
default: default:
err := common.UnmarshalBodyReusable(c, imageRequest) err := common.UnmarshalBodyReusable(c, imageRequest)
if err != nil { if err != nil {