From 87778326ea5743d73e958155e5029df8be11e225 Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Thu, 17 Apr 2025 19:58:46 +0800
Subject: [PATCH] fix auth bug

---
 cmd/openteam/main.go    | 1 +
 middleware/auth_team.go | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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