Browse Source

Merge branch 'fix-2392' of git.openi.org.cn:OpenI/aiforge into fix-2392

pull/2758/head
zhoupzh 2 years ago
parent
commit
5122e7a922
1 changed files with 20 additions and 16 deletions
  1. +20
    -16
      routers/repo/dataset.go

+ 20
- 16
routers/repo/dataset.go View File

@@ -565,22 +565,26 @@ func ReferenceDataset(ctx *context.Context) {
func ReferenceDatasetData(ctx *context.Context) { func ReferenceDatasetData(ctx *context.Context) {
MustEnableDataset(ctx) MustEnableDataset(ctx)
datasetIds := models.GetDatasetIdsByRepoID(ctx.Repo.Repository.ID) datasetIds := models.GetDatasetIdsByRepoID(ctx.Repo.Repository.ID)
opts := &models.SearchDatasetOptions{
DatasetIDs: datasetIds,
NeedAttachment: false,
CloudBrainType: models.TypeCloudBrainAll,
ListOptions: models.ListOptions{
Page: 1,
PageSize: setting.RepoMaxReferenceDatasetNum,
},
SearchOrderBy: getSearchOrderByInValues(datasetIds),
QueryReference: true,
}
datasets, _, err := models.SearchDataset(opts)

if err != nil {
ctx.ServerError("SearchDatasets", err)
return
var datasets models.DatasetList
var err error
if len(datasetIds) > 0 {

opts := &models.SearchDatasetOptions{
DatasetIDs: datasetIds,
NeedAttachment: false,
CloudBrainType: models.TypeCloudBrainAll,
ListOptions: models.ListOptions{
Page: 1,
PageSize: setting.RepoMaxReferenceDatasetNum,
},
SearchOrderBy: getSearchOrderByInValues(datasetIds),
QueryReference: true,
}
datasets, _, err = models.SearchDataset(opts)
if err != nil {
ctx.ServerError("SearchDatasets", err)
return
}
} }


ctx.JSON(http.StatusOK, repository.ConvertToDatasetWithStar(ctx, datasets)) ctx.JSON(http.StatusOK, repository.ConvertToDatasetWithStar(ctx, datasets))


Loading…
Cancel
Save