key截取错误修复

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 {