diff --git a/cmd/openteam/main.go b/cmd/openteam/main.go index e64aa9a..2e28247 100644 --- a/cmd/openteam/main.go +++ b/cmd/openteam/main.go @@ -158,6 +158,7 @@ func main() { } go func() { + fmt.Println("Starting server at port:", cfg.Port) // 服务启动 if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed { log.Fatalf("listen: %s\n", err) diff --git a/middleware/auth_team.go b/middleware/auth_team.go index 7c6d44c..a01757a 100644 --- a/middleware/auth_team.go +++ b/middleware/auth_team.go @@ -63,7 +63,7 @@ func AuthLLM(db *gorm.DB) gin.HandlerFunc { } auth_token = auth_token[7:] 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") c.Abort() return