diff --git a/main.go b/main.go index eea0aed..2f4cc28 100644 --- a/main.go +++ b/main.go @@ -188,7 +188,8 @@ func embeddings(c *gin.Context) { } auth := c.Request.Header.Get("Authorization") - key := strings.Trim(auth, "Bearer ") + //key := strings.Trim(auth, "Bearer ") + key := auth[7:] //根据KEY调用用户余额接口,判断是否有足够的余额, 后期可考虑判断max_tokens参数来调整 serverInfo, err := checkBlance(key, chatRequest.Model) if err != nil { @@ -271,7 +272,7 @@ func completions(c *gin.Context) { } auth := c.Request.Header.Get("Authorization") - key := strings.Trim(auth, "Bearer ") + key := auth[7:] //根据KEY调用用户余额接口,判断是否有足够的余额, 后期可考虑判断max_tokens参数来调整 serverInfo, err := checkBlance(key, chatRequest.Model) if err != nil { @@ -480,7 +481,8 @@ func checkKeyMid() gin.HandlerFunc { if auth == "" { c.AbortWithStatusJSON(401, gin.H{"code": 40001}) } else { - key := strings.Trim(auth, "Bearer ") + //key := strings.Trim(auth, "Bearer ") + key := auth[7:] log.Printf("key: %v", key) msg, err := checkKeyAndTimeCount(key) if err != nil {