fix: Regression of AgentSummary GQL resolver (#3045)

Backported-from: main Backported-to: 23.09 Backported-of: 3045
lablup · Nov 28, 2024 · 66b0a99 · 66b0a99
1 parent 3be18ea
commit 66b0a99
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/changes/3045.fix.md b/changes/3045.fix.md
@@ -0,0 +1 @@
+Fix regression of the `AgentSummary` resolver caused by an incorrect `batch_load_func` assignment.
diff --git a/src/ai/backend/manager/models/gql.py b/src/ai/backend/manager/models/gql.py
@@ -742,9 +742,9 @@ async def resolve_agent_summary(
         if ctx.local_config["manager"]["hide-agents"]:
             raise ObjectNotFound(object_name="agent")
 
-        loader = ctx.dataloader_manager.get_loader(
+        loader = ctx.dataloader_manager.get_loader_by_func(
             ctx,
-            "Agent",
+            AgentSummary.batch_load,
             raw_status=None,
             scaling_group=scaling_group,
             domain_name=domain_name,