diff --git a/src/ai/backend/manager/models/gql_models/session.py b/src/ai/backend/manager/models/gql_models/session.py index 2f40cdb7bf..d760d9a387 100644 --- a/src/ai/backend/manager/models/gql_models/session.py +++ b/src/ai/backend/manager/models/gql_models/session.py @@ -278,7 +278,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, @@ -293,7 +294,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,