From 5b5b36822f670c4a90e1b75561f13b5aab40793a Mon Sep 17 00:00:00 2001
From: yokowu <18836617@qq.com>
Date: Fri, 4 Jul 2025 11:09:51 +0800
Subject: [PATCH] =?UTF-8?q?fix(model):=20=E6=B7=BB=E5=8A=A0=E9=A6=96?=
 =?UTF-8?q?=E4=B8=AA=E6=A8=A1=E5=9E=8B=E8=87=AA=E5=8A=A8=E6=BF=80=E6=B4=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/internal/model/repo/model.go | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/backend/internal/model/repo/model.go b/backend/internal/model/repo/model.go
index c12d6e9..cbc1616 100644
--- a/backend/internal/model/repo/model.go
+++ b/backend/internal/model/repo/model.go
@@ -30,6 +30,16 @@ func (r *ModelRepo) Create(ctx context.Context, m *domain.CreateModelReq) (*db.M
 	if err != nil {
 		return nil, err
 	}
+
+	n, err := r.db.Model.Query().Where(model.ModelType(m.ModelType)).Count(ctx)
+	if err != nil {
+		return nil, err
+	}
+	status := consts.ModelStatusInactive
+	if n == 0 {
+		status = consts.ModelStatusActive
+	}
+
 	return r.db.Model.Create().
 		SetUserID(uid).
 		SetModelName(m.ModelName).
@@ -37,7 +47,7 @@ func (r *ModelRepo) Create(ctx context.Context, m *domain.CreateModelReq) (*db.M
 		SetAPIBase(m.APIBase).
 		SetAPIKey(m.APIKey).
 		SetModelType(m.ModelType).
-		SetStatus(consts.ModelStatusInactive).
+		SetStatus(status).
 		Save(ctx)
 }