From 14f29f813fe73bb8112b79f274a486a1c2854899 Mon Sep 17 00:00:00 2001
From: Gyubong Lee <jopemachine@naver.com>
Date: Wed, 4 Sep 2024 07:50:09 +0000
Subject: [PATCH] fix: Handle when entity_type not provided

---
 src/ai/backend/manager/container_registry/github.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/ai/backend/manager/container_registry/github.py b/src/ai/backend/manager/container_registry/github.py
index 5904698c99..08b4216972 100644
--- a/src/ai/backend/manager/container_registry/github.py
+++ b/src/ai/backend/manager/container_registry/github.py
@@ -19,7 +19,10 @@ async def fetch_repositories(
     ) -> AsyncIterator[str]:
         username = self.registry_info.username
         access_token = self.registry_info.password
-        entity_type = self.registry_info.extra.get("entity_type")
+        entity_type = self.registry_info.extra.get("entity_type", None)
+
+        if entity_type is None:
+            raise RuntimeError("Entity type is not provided for GitHub registry!")
 
         base_url = f"https://api.github.com/{entity_type}/{username}/packages"