From 79604b6a1e05c8fcfa733bfac04a64b56a9b0264 Mon Sep 17 00:00:00 2001 From: Gyubong Lee Date: Tue, 30 Jul 2024 02:06:03 +0000 Subject: [PATCH] feat: Support gitlab onprem instance --- src/ai/backend/manager/container_registry/gitlab.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/ai/backend/manager/container_registry/gitlab.py b/src/ai/backend/manager/container_registry/gitlab.py index 2351e237b27..70b5bfd76d2 100644 --- a/src/ai/backend/manager/container_registry/gitlab.py +++ b/src/ai/backend/manager/container_registry/gitlab.py @@ -15,16 +15,15 @@ class GitLabRegistry_v2(BaseContainerRegistry): async def fetch_repositories(self, sess: aiohttp.ClientSession) -> AsyncIterator[str]: - name, access_token, gitlab_project = ( + name, access_token, gitlab_url, gitlab_project = ( self.registry_info["username"], self.registry_info["password"], + self.registry_info["gitlab_url"], self.registry_info["gitlab_project"], ) encoded_project_id = urllib.parse.quote(f"{name}/{gitlab_project}", safe="") - repo_list_url = ( - f"https://gitlab.com/api/v4/projects/{encoded_project_id}/registry/repositories" - ) + repo_list_url = f"{gitlab_url}/api/v4/projects/{encoded_project_id}/registry/repositories" headers = { "Accept": "application/json",