xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-12-26 08:35:58 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复多路复用bug

Browse Source
This commit is contained in:
CaIon
2023-08-31 00:44:16 +08:00
parent d9d5b001a7
commit fe94656260
5 changed files with 42 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
controller/relay-mj.go
View File

@@ -248,6 +248,10 @@ func relayMidjourneySubmit(c *gin.Context, relayMode int) *MidjourneyResponse {
req.Header.Set("Content-Type", c.Request.Header.Get("Content-Type"))
req.Header.Set("Accept", c.Request.Header.Get("Accept"))
//mjToken := ""
//if c.Request.Header.Get("Authorization") != "" {
// mjToken = strings.Split(c.Request.Header.Get("Authorization"), " ")[1]
//}
req.Header.Set("mj-api-secret", strings.Split(c.Request.Header.Get("Authorization"), " ")[1])
// print request header
log.Printf("request header: %s", req.Header)
@@ -353,7 +357,7 @@ func relayMidjourneySubmit(c *gin.Context, relayMode int) *MidjourneyResponse {
Progress: "0%",
FailReason: "",
}
if midjResponse.Code == 4 {
if midjResponse.Code == 4 || midjResponse.Code == 24 {
midjourneyTask.FailReason = midjResponse.Description
}
err = midjourneyTask.Insert()