Skip to content

Commit 380bd73

Browse files
authored
Merge pull request #52 from yokowu/fix-model-create
fix(model): 添加首个模型自动激活
2 parents bcd417d + 5b5b368 commit 380bd73

1 file changed

Lines changed: 11 additions & 1 deletion

File tree

backend/internal/model/repo/model.go

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,24 @@ func (r *ModelRepo) Create(ctx context.Context, m *domain.CreateModelReq) (*db.M
3030
if err != nil {
3131
return nil, err
3232
}
33+
34+
n, err := r.db.Model.Query().Where(model.ModelType(m.ModelType)).Count(ctx)
35+
if err != nil {
36+
return nil, err
37+
}
38+
status := consts.ModelStatusInactive
39+
if n == 0 {
40+
status = consts.ModelStatusActive
41+
}
42+
3343
return r.db.Model.Create().
3444
SetUserID(uid).
3545
SetModelName(m.ModelName).
3646
SetProvider(m.Provider).
3747
SetAPIBase(m.APIBase).
3848
SetAPIKey(m.APIKey).
3949
SetModelType(m.ModelType).
40-
SetStatus(consts.ModelStatusInactive).
50+
SetStatus(status).
4151
Save(ctx)
4252
}
4353

0 commit comments

Comments
 (0)