soaska/chatgpt-to-api
soaska/chatgpt-to-api
Archived
1
0
Fork 0
Code Pull Requests Activity

customAccessToken Closes #38

Browse Source
This commit is contained in:
Antonio 2023-05-28 20:12:52 +08:00
parent f63d4cc95c
commit 8a6f07b9eb
2 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
handlers.go
View File

@ -80,17 +80,16 @@ func nightmare(c *gin.Context) {
}
// Convert the chat request to a ChatGPT request
translated_request := chatgpt.ConvertAPIRequest(original_request)
// c.JSON(200, chatgpt_request)
// authHeader := c.GetHeader("Authorization")
authHeader := c.GetHeader("Authorization")
token := ACCESS_TOKENS.GetToken()
// if authHeader != "" {
// customAccessToken := strings.Replace(authHeader, "Bearer ", "", 1)
// if customAccessToken != "" {
// token = customAccessToken
// println("customAccessToken set:" + customAccessToken)
// }
// }
if authHeader != "" {
customAccessToken := strings.Replace(authHeader, "Bearer ", "", 1)
// Check if customAccessToken starts with sk-
if strings.HasPrefix(customAccessToken, "eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCIsImtpZCI6Ik1UaEVOVUpHTkVNMVFURTRNMEZCTWpkQ05UZzVNRFUxUlRVd1FVSkRNRU13UmtGRVFrRXpSZyJ9") {
token = customAccessToken
}
}
response, err := chatgpt.SendRequest(translated_request, token)
if err != nil {

2
middleware.go
View File

@ -49,7 +49,7 @@ func Authorization(c *gin.Context) {
}
}
}
if len(API_KEYS) != 0 && API_KEYS[c.Request.Header.Get("Authorization")] != true {
if len(API_KEYS) != 0 && !API_KEYS[c.Request.Header.Get("Authorization")] {
c.String(401, "Unauthorized")
c.Abort()
return