pothos-native select in ModelExport

packages/hub/src/graphql/queries/exports.ts

@@ -45,9 +45,6 @@ builder.queryField("modelExports", (t) =>
               createdAt: "desc",
             },
           },
-          include: {
-            modelRevision: true,
-          },
         });
       },
     },

packages/hub/src/graphql/types/ModelExport.ts

@@ -16,39 +16,30 @@ export const ModelExport = builder.prismaNode("ModelExport", {
     title: t.exposeString("title", { nullable: true }),
     owner: t.field({
       type: Owner,
-      resolve: async (modelExport, _, context) => {
-        //There might be a way to do this using "select" or "resolve", but I couldn't get those to work.
-        const modelRevision = await prisma.modelRevision.findUnique({
-          where: { id: modelExport.modelRevisionId },
-          include: {
+      select: {
+        modelRevision: {
+          select: {
             model: {
-              include: {
+              select: {
                 owner: {
-                  include: {
+                  select: {
                     user: true,
                     group: true,
                   },
                 },
               },
             },
           },
-        });
-
-        if (!modelRevision || !modelRevision.model) {
-          throw new Error("Invalid model revision or model");
-        }
-
-        const owner = modelRevision.model.owner;
-        const result = owner.user ?? owner.group;
-
+        },
+      },
+      resolve: async ({ modelRevision: { model } }) => {
+        const result = model.owner.user ?? model.owner.group;
         if (!result) {
           throw new Error("Invalid owner object, missing user or group");
         }
-
         (result as any)["_owner"] = {
-          type: owner.user ? "User" : "Group",
+          type: model.owner.user ? "User" : "Group",
         };
-
         return result;
       },
     }),