Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
igophper 2023-08-12 22:09:37 +08:00
parent 79b6769c7f
commit c92a48b48b
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
controller/relay-baidu.go
View File

@ -323,7 +323,8 @@ func baiduEmbeddingHandler(c *gin.Context, resp *http.Response) (*OpenAIErrorWit
func getBaiduAccessToken(apiKey string) (string, error) { func getBaiduAccessToken(apiKey string) (string, error) {
var accessToken BaiduAccessToken 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 { if accessToken, ok = val.(BaiduAccessToken); ok {
// 提前1小时刷新 // 提前1小时刷新
if time.Now().Add(time.Hour).After(accessToken.ExpiresAt) { if time.Now().Add(time.Hour).After(accessToken.ExpiresAt) {
@ -337,6 +338,7 @@ func getBaiduAccessToken(apiKey string) (string, error) {
if len(splits) == 1 { if len(splits) == 1 {
accessToken.AccessToken = apiKey accessToken.AccessToken = apiKey
accessToken.ExpiresAt = time.Now().Add(30 * 24 * time.Hour) accessToken.ExpiresAt = time.Now().Add(30 * 24 * time.Hour)
baiduAccessTokens.Store(md5Key, accessToken)
return apiKey, nil return apiKey, nil
} }