prompts-core

dao/model_dao.go

@@ -7,6 +7,7 @@ import (
 	"prompts-core/model/entity"
 
 	"gitea.com/red-future/common/db/gfdb"
+	"gitea.com/red-future/common/utils"
 )
 
 var Model = &modelDao{}
@@ -28,8 +29,13 @@ func (d *modelDao) GetByModelName(ctx context.Context, modelName string) (m *ent
 }
 
 func (d *modelDao) GetByIsChatModel(ctx context.Context) (m *entity.AsynchModel, err error) {
+	userInfo, err := utils.GetUserInfo(ctx)
+	if err != nil {
+		return nil, err
+	}
 	r, err := gfdb.DB(ctx).Model(ctx, public.TableNameModel).
 		Where(entity.AsynchModelCol.IsChatModel, 1).
+		Where(entity.AsynchModelCol.Creator, userInfo.UserName).
 		One()
 	if err != nil {
 		return nil, err