diff --git a/models/ai_model_manage.go b/models/ai_model_manage.go index 98c67274a..c1ab3afab 100644 --- a/models/ai_model_manage.go +++ b/models/ai_model_manage.go @@ -237,15 +237,19 @@ func QueryModelConvertByName(name string, repoId int64) ([]*AiModelConvert, erro func QueryModelConvertById(id string) (*AiModelConvert, error) { sess := x.NewSession() defer sess.Close() - sess.Table(new(AiModelConvert)).ID(id) - //sess.Select("*").Table(new(AiModelConvert)).Where("id = ?", id) - aiModelManageConvertList := make([]*AiModelConvert, 0) - err := sess.Find(&aiModelManageConvertList) - if err == nil { - if len(aiModelManageConvertList) == 1 { - return aiModelManageConvertList[0], nil - } + re := new(AiModelConvert) + isExist, err := sess.Table(new(AiModelConvert)).ID(id).Get(re) + if isExist { + return re, nil } + //sess.Select("*").Table(new(AiModelConvert)).Where("id = ?", id) + //aiModelManageConvertList := make([]*AiModelConvert, 0) + //err := sess.Find(&aiModelManageConvertList) + //if err == nil { + // if len(aiModelManageConvertList) == 1 { + // return aiModelManageConvertList[0], nil + // } + //} return nil, err }