From d0ed6ba4b42c4fa7e96d5bda23fd180d5cb93978 Mon Sep 17 00:00:00 2001 From: zouap Date: Mon, 12 Dec 2022 11:44:05 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E9=A1=B9=E7=9B=AE=E6=9D=83?= =?UTF-8?q?=E9=99=90=E5=9B=A2=E9=98=9F=E7=A7=81=E6=9C=89=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/ai_model_manage.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 1155f3239..205660402 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -1004,7 +1004,11 @@ func isCanDownload(ctx *context.Context, task *models.AiModelManage) bool { if err != nil { log.Info("query error.") } - if ctx.User.IsAdmin || ctx.User.ID == task.UserId || isCollaborator { + isTeamMember, err := ctx.Repo.Repository.IsInRepoTeam(ctx.User.ID) + if err != nil { + log.Info("query IsInRepoTeam error." + err.Error()) + } + if ctx.User.IsAdmin || ctx.User.ID == task.UserId || isCollaborator || isTeamMember { return true } if ctx.Repo.IsOwner() {