fix auth bug
This commit is contained in:
@@ -158,6 +158,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
|
fmt.Println("Starting server at port:", cfg.Port)
|
||||||
// 服务启动
|
// 服务启动
|
||||||
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||||
log.Fatalf("listen: %s\n", err)
|
log.Fatalf("listen: %s\n", err)
|
||||||
|
|||||||
@@ -63,7 +63,7 @@ func AuthLLM(db *gorm.DB) gin.HandlerFunc {
|
|||||||
}
|
}
|
||||||
auth_token = auth_token[7:]
|
auth_token = auth_token[7:]
|
||||||
token := model.Token{}
|
token := model.Token{}
|
||||||
if err := db.Preload("Users").First(&token, "key = ?", auth_token).Error; err != nil {
|
if err := db.Preload("User").Where("key = ?", auth_token).First(&token).Error; err != nil {
|
||||||
dto.WrapErrorAsOpenAI(c, http.StatusUnauthorized, "invalid_api_key")
|
dto.WrapErrorAsOpenAI(c, http.StatusUnauthorized, "invalid_api_key")
|
||||||
c.Abort()
|
c.Abort()
|
||||||
return
|
return
|
||||||
|
|||||||
Reference in New Issue
Block a user