diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 3b85a0b20..ad6f52aef 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -214,6 +214,7 @@ func DownloadSingleModelFile(ctx *context.Context) { func ShowSingleModel(ctx *context.Context) { id := ctx.Params(":ID") + parentDir := ctx.Query("parentDir") log.Info("Show single ModelInfo start.id=" + id) task, err := models.QueryModelById(id) if err != nil { @@ -222,7 +223,7 @@ func ShowSingleModel(ctx *context.Context) { return } log.Info("bucket=" + setting.Bucket + " key=" + task.Path[len(setting.Bucket)+1:]) - models, err := storage.GetObsListObjectByBucketAndPrefix(setting.Bucket, task.Path[len(setting.Bucket)+1:]) + models, err := storage.GetObsListObjectByBucketAndPrefix(setting.Bucket, task.Path[len(setting.Bucket)+1:], parentDir) if err != nil { log.Info("get model list failed:", err) ctx.ServerError("GetObsListObject:", err)