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

feat: support omni-moderation-latest

Browse Source
This commit is contained in:
WqyJh
2024-11-15 20:00:07 +08:00
parent 3915ce9814
commit 3e6c785955
5 changed files with 70 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
middleware/utils.go
View File

@@ -2,11 +2,12 @@ package middleware
import (
"fmt"
"strings"
"github.com/gin-gonic/gin"
"github.com/songquanpeng/one-api/common"
"github.com/songquanpeng/one-api/common/helper"
"github.com/songquanpeng/one-api/common/logger"
"strings"
)
func abortWithMessage(c *gin.Context, statusCode int, message string) {
@@ -28,7 +29,7 @@ func getRequestModel(c *gin.Context) (string, error) {
}
if strings.HasPrefix(c.Request.URL.Path, "/v1/moderations") {
if modelRequest.Model == "" {
modelRequest.Model = "text-moderation-stable"
modelRequest.Model = "omni-moderation-latest"
}
}
if strings.HasSuffix(c.Request.URL.Path, "embeddings") {