fix: Broken ComputeContainer GQL (#3042)

Co-authored-by: Joongi Kim <[email protected]> Backported-from: main (24.12) Backported-to: 23.09 Backport-of: 3042
lablup · Nov 12, 2024 · 919802e · 919802e
1 parent fccaf6d
commit 919802e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/changes/3042.fix.md b/changes/3042.fix.md
@@ -0,0 +1 @@
+Fix regression of `ComputeContainer` GraphQL queries due to newly introduced relationship fields
diff --git a/src/ai/backend/manager/models/kernel.py b/src/ai/backend/manager/models/kernel.py
@@ -1022,6 +1022,14 @@ async def batch_load_detail(
             .where(
                 (kernels.c.id.in_(container_ids)),
             )
+            .options(
+                noload("*"),
+                selectinload(
+                    KernelRow.group_row,
+                    KernelRow.user_row,
+                    KernelRow.image_row,
+                ),
+            )
         )
         if domain_name is not None:
             query = query.where(kernels.c.domain_name == domain_name)