diff --git a/service/model_service.go b/service/model_service.go index 23bc5c5..3691a47 100644 --- a/service/model_service.go +++ b/service/model_service.go @@ -358,7 +358,14 @@ func (s *modelService) UpdateChatModel(ctx context.Context, req *dto.UpdateChatM } func (s *modelService) GetIsChatModel(ctx context.Context) (*dto.GetIsChatModelRes, error) { + user, err := utils.GetUserInfo(ctx) + if err != nil { + return nil, err + } model, err := dao.Model.Get(ctx, &entity.AsynchModel{ + SQLBaseDO: beans.SQLBaseDO{ + Creator: user.UserName, + }, IsChatModel: new(1), }) if err != nil {