fix: Reverse the dependee/dependent semantics

lablup · Sep 19, 2024 · bdde3c6 · bdde3c6
1 parent 481c748
commit bdde3c6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/ai/backend/manager/models/gql_models/session.py b/src/ai/backend/manager/models/gql_models/session.py
@@ -285,7 +285,8 @@ async def resolve_dependees(
         info: graphene.ResolveInfo,
     ) -> ConnectionResolverResult[Self]:
         ctx: GraphQueryContext = info.context
-        loader = ctx.dataloader_manager.get_loader(ctx, "ComputeSessionNode.by_dependee_id")
+        # Get my dependees (myself is the dependent)
+        loader = ctx.dataloader_manager.get_loader(ctx, "ComputeSessionNode.by_dependent_id")
         sessions = await loader.load(self.row_id)
         return ConnectionResolverResult(
             sessions,
@@ -300,7 +301,8 @@ async def resolve_dependents(
         info: graphene.ResolveInfo,
     ) -> ConnectionResolverResult[Self]:
         ctx: GraphQueryContext = info.context
-        loader = ctx.dataloader_manager.get_loader(ctx, "ComputeSessionNode.by_dependent_id")
+        # Get my dependents (myself is the dependee)
+        loader = ctx.dataloader_manager.get_loader(ctx, "ComputeSessionNode.by_dependee_id")
         sessions = await loader.load(self.row_id)
         return ConnectionResolverResult(
             sessions,