From 9fd3ac3923d6a2075b22645c1ec1208781dc82b6 Mon Sep 17 00:00:00 2001 From: lvxiu_ext Date: Thu, 18 May 2023 13:00:33 +0800 Subject: [PATCH] =?UTF-8?q?key=E6=88=AA=E5=8F=96=E9=94=99=E8=AF=AF?= =?UTF-8?q?=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) 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 {