From c92a48b48bf8d235b80cde697e2e0ac383ff0507 Mon Sep 17 00:00:00 2001
From: igophper <admin@jialigu.cn>
Date: Sat, 12 Aug 2023 22:09:37 +0800
Subject: [PATCH] fix

---
 controller/relay-baidu.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/controller/relay-baidu.go b/controller/relay-baidu.go
index 79cf751e..d0e90510 100644
--- a/controller/relay-baidu.go
+++ b/controller/relay-baidu.go
@@ -323,7 +323,8 @@ func baiduEmbeddingHandler(c *gin.Context, resp *http.Response) (*OpenAIErrorWit
 
 func getBaiduAccessToken(apiKey string) (string, error) {
 	var accessToken BaiduAccessToken
-	if val, ok := baiduAccessTokens.Load(md5.Sum([]byte(apiKey))); ok {
+	md5Key := md5.Sum([]byte(apiKey))
+	if val, ok := baiduAccessTokens.Load(md5Key); ok {
 		if accessToken, ok = val.(BaiduAccessToken); ok {
 			// 提前1小时刷新
 			if time.Now().Add(time.Hour).After(accessToken.ExpiresAt) {
@@ -337,6 +338,7 @@ func getBaiduAccessToken(apiKey string) (string, error) {
 	if len(splits) == 1 {
 		accessToken.AccessToken = apiKey
 		accessToken.ExpiresAt = time.Now().Add(30 * 24 * time.Hour)
+		baiduAccessTokens.Store(md5Key, accessToken)
 		return apiKey, nil
 	}