feat(BA-3044): Add dependency verification system for web component (#6757)

Co-authored-by: Claude <noreply@anthropic.com>

Author: HyeockJinKim

changes/6757.feature.md

@@ -0,0 +1 @@
+Add dependency verification system for web component

src/ai/backend/common/exception.py

@@ -801,3 +801,19 @@ def error_code(self) -> ErrorCode:
             operation=ErrorOperation.READ,
             error_detail=ErrorDetail.UNAVAILABLE,
         )
+
+
+class DatabaseError(BackendAIError, web.HTTPServiceUnavailable):
+    """
+    Raised when a database operation fails.
+    """
+
+    error_type = "https://api.backend.ai/probs/database-error"
+    error_title = "Database Error"
+
+    def error_code(self) -> ErrorCode:
+        return ErrorCode(
+            domain=ErrorDomain.HEALTH_CHECK,
+            operation=ErrorOperation.READ,
+            error_detail=ErrorDetail.UNAVAILABLE,
+        )

src/ai/backend/common/leader/tasks/event_task.py

@@ -9,7 +9,7 @@
 from ai.backend.common.events.dispatcher import EventProducer
 from ai.backend.common.events.types import AbstractAnycastEvent
 from ai.backend.common.leader.tasks.base import PeriodicTask
-from ai.backend.common.logging import BraceStyleAdapter
+from ai.backend.logging import BraceStyleAdapter
 
 log = BraceStyleAdapter(logging.getLogger(__spec__.name))
 

src/ai/backend/common/leader/tasks/leader_cron.py

@@ -8,7 +8,7 @@
 
 from ai.backend.common.leader.base import LeadershipChecker, LeaderTask
 from ai.backend.common.leader.tasks.base import PeriodicTask
-from ai.backend.common.logging import BraceStyleAdapter
+from ai.backend.logging import BraceStyleAdapter
 
 log = BraceStyleAdapter(logging.getLogger(__spec__.name))
 

src/ai/backend/common/leader/valkey_leader_election.py

@@ -10,7 +10,7 @@
 from ai.backend.common.clients.valkey_client.valkey_leader.client import ValkeyLeaderClient
 from ai.backend.common.leader.base import AbstractLeaderElection, LeaderTask
 from ai.backend.common.leader.exceptions import AlreadyStartedError
-from ai.backend.common.logging import BraceStyleAdapter
+from ai.backend.logging import BraceStyleAdapter
 
 log = BraceStyleAdapter(logging.getLogger(__spec__.name))
 

src/ai/backend/manager/event_dispatcher/handlers/notification.py

@@ -5,7 +5,7 @@
 
 from ai.backend.common.data.notification import NotifiableMessage
 from ai.backend.common.events.event_types.notification import NotificationTriggeredEvent
-from ai.backend.common.logging import BraceStyleAdapter
+from ai.backend.logging import BraceStyleAdapter
 
 from ...data.notification import NotificationRuleType
 from ...services.notification.actions import ProcessNotificationAction

src/ai/backend/manager/models/utils.py

@@ -42,6 +42,7 @@
 )
 from yarl import URL
 
+from ai.backend.common.exception import DatabaseError
 from ai.backend.common.json import ExtendedJSONEncoder
 from ai.backend.logging import BraceStyleAdapter
 from ai.backend.manager.config.bootstrap import DatabaseConfig
@@ -163,10 +164,13 @@ async def ping(self) -> None:
         Ping the database to check if the connection is alive.
 
         Raises:
-            Exception: If the ping fails or connection is not available
+            DatabaseError: If the ping fails or connection is not available
         """
-        async with self.connect() as conn:
-            await conn.execute(sa.text("SELECT 1"))
+        async with self.begin_readonly_read_committed() as conn:
+            result = await conn.execute(sa.text("SELECT 1"))
+            scalar_result = result.scalar()
+            if scalar_result != 1:
+                raise DatabaseError("Database ping failed: unexpected result")
 
     @actxmgr
     async def begin_readonly(

src/ai/backend/manager/notification/channels/webhook/channel.py

@@ -10,7 +10,7 @@
 from ai.backend.common.clients.http_client import ClientPool
 from ai.backend.common.clients.http_client.client_pool import ClientKey
 from ai.backend.common.data.notification import WebhookConfig
-from ai.backend.common.logging import BraceStyleAdapter
+from ai.backend.logging import BraceStyleAdapter
 from ai.backend.manager.errors.notification import NotificationProcessingFailure
 
 from ...types import NotificationMessage, SendResult

src/ai/backend/manager/notification/notification_center.py

@@ -9,7 +9,7 @@
 
 from ai.backend.common.clients.http_client import ClientPool
 from ai.backend.common.data.notification import NotifiableMessage, NotificationChannelType
-from ai.backend.common.logging import BraceStyleAdapter
+from ai.backend.logging import BraceStyleAdapter
 from ai.backend.manager.data.notification import NotificationChannelData
 from ai.backend.manager.errors.notification import (
     InvalidNotificationChannelType,

src/ai/backend/manager/services/artifact_revision/service.py

@@ -25,7 +25,7 @@
     HuggingFaceImportModelsReq,
     ReservoirImportModelsReq,
 )
-from ai.backend.common.logging import BraceStyleAdapter
+from ai.backend.logging import BraceStyleAdapter
 from ai.backend.manager.client.artifact_registry.reservoir_client import ReservoirRegistryClient
 from ai.backend.manager.clients.storage_proxy.session_manager import StorageSessionManager
 from ai.backend.manager.config.provider import ManagerConfigProvider