xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-17 05:33:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🔖 chore: Rename relay/util to relay/relay_util package and add utils package

Browse Source
This commit is contained in:
MartialBE
2024-05-29 00:36:54 +08:00
parent 853f2681f4
commit 79524108a3
61 changed files with 309 additions and 265 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
relay/midjourney/relay-mj.go
View File

@@ -14,7 +14,7 @@ import (
"one-api/model"
provider "one-api/providers/midjourney"
"one-api/relay"
"one-api/relay/util"
"one-api/relay/relay_util"
"one-api/types"
"strconv"
"strings"
@@ -539,10 +539,10 @@ func getMjRequestPath(path string) string {
return requestURL
}
func getQuota(c *gin.Context, action string) (*util.Quota, *types.OpenAIErrorWithStatusCode) {
func getQuota(c *gin.Context, action string) (*relay_util.Quota, *types.OpenAIErrorWithStatusCode) {
modelName := CoverActionToModelName(action)
return util.NewQuota(c, modelName, 1000)
return relay_util.NewQuota(c, modelName, 1000)
}
func getMJProviderWithRequest(c *gin.Context, relayMode int, request *provider.MidjourneyRequest) (*provider.MidjourneyProvider, *provider.MidjourneyResponse) {