feat: jimeng images base64 limit (#2032)
Some checks failed
Publish Docker image (Multi Registries, native amd64+arm64) / Build & push (amd64) [native] (push) Has been cancelled
Publish Docker image (Multi Registries, native amd64+arm64) / Build & push (arm64) [native] (push) Has been cancelled
Publish Docker image (Multi Registries, native amd64+arm64) / Create multi-arch manifests (Docker Hub) (push) Has been cancelled
Build Electron App / build (windows-latest) (push) Has been cancelled
Build Electron App / release (push) Has been cancelled
Release (Linux, macOS, Windows) / Linux Release (push) Has been cancelled
Release (Linux, macOS, Windows) / macOS Release (push) Has been cancelled
Release (Linux, macOS, Windows) / Windows Release (push) Has been cancelled
Some checks failed
Publish Docker image (Multi Registries, native amd64+arm64) / Build & push (amd64) [native] (push) Has been cancelled
Publish Docker image (Multi Registries, native amd64+arm64) / Build & push (arm64) [native] (push) Has been cancelled
Publish Docker image (Multi Registries, native amd64+arm64) / Create multi-arch manifests (Docker Hub) (push) Has been cancelled
Build Electron App / build (windows-latest) (push) Has been cancelled
Build Electron App / release (push) Has been cancelled
Release (Linux, macOS, Windows) / Linux Release (push) Has been cancelled
Release (Linux, macOS, Windows) / macOS Release (push) Has been cancelled
Release (Linux, macOS, Windows) / Windows Release (push) Has been cancelled
This commit is contained in:
parent
c6175c0b46
commit
7c543a9e19
@ -65,6 +65,11 @@ type responseTask struct {
|
|||||||
TimeElapsed string `json:"time_elapsed"`
|
TimeElapsed string `json:"time_elapsed"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
// 即梦限制单个文件最大4.7MB https://www.volcengine.com/docs/85621/1747301
|
||||||
|
MaxFileSize int64 = 4*1024*1024 + 700*1024 // 4.7MB (4MB + 724KB)
|
||||||
|
)
|
||||||
|
|
||||||
// ============================
|
// ============================
|
||||||
// Adaptor implementation
|
// Adaptor implementation
|
||||||
// ============================
|
// ============================
|
||||||
@ -127,15 +132,19 @@ func (a *TaskAdaptor) BuildRequestBody(c *gin.Context, info *relaycommon.RelayIn
|
|||||||
if files, exists := mf.File["input_reference"]; exists && len(files) > 0 {
|
if files, exists := mf.File["input_reference"]; exists && len(files) > 0 {
|
||||||
if len(files) == 1 {
|
if len(files) == 1 {
|
||||||
info.Action = constant.TaskActionGenerate
|
info.Action = constant.TaskActionGenerate
|
||||||
} else if len(files) == 2 {
|
} else if len(files) > 1 {
|
||||||
info.Action = constant.TaskActionFirstTailGenerate
|
info.Action = constant.TaskActionFirstTailGenerate
|
||||||
} else if len(files) > 2 {
|
|
||||||
info.Action = constant.TaskActionReferenceGenerate
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// 将上传的文件转换为base64格式
|
// 将上传的文件转换为base64格式
|
||||||
var images []string
|
var images []string
|
||||||
|
|
||||||
for _, fileHeader := range files {
|
for _, fileHeader := range files {
|
||||||
|
// 检查文件大小
|
||||||
|
if fileHeader.Size > MaxFileSize {
|
||||||
|
return nil, fmt.Errorf("文件 %s 大小超过限制,最大允许 %d MB", fileHeader.Filename, MaxFileSize/(1024*1024))
|
||||||
|
}
|
||||||
|
|
||||||
file, err := fileHeader.Open()
|
file, err := fileHeader.Open()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
continue
|
continue
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user