feat: Support gitlab onprem instance

lablup · Aug 1, 2024 · 8994110 · 8994110
1 parent 229f764
commit 8994110
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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",