Caching a user session with multiple relations (drizzle-orm postgresql)

export const user = createTable("user", {
  createdAt: timestamp("created_at", { withTimezone: true })
    .default(sql`CURRENT_TIMESTAMP`)
    .notNull(),
  email: text("email").notNull().unique(),
  emailVerified: boolean("email_verified").notNull(),
  id: text("id").primaryKey(),
  image: text("image"),
  musicVolume: real("music_volume").default(0.25).notNull(),
  name: text("name").notNull(),
  role: varchar("role", { length: 255 }).notNull().default("user"),
  updatedAt: timestamp("updated_at", { withTimezone: true }).$onUpdate(
    () => new Date(),
  ),
});

export const userRelations = relations(user, ({ one, many }) => ({
  accounts: many(account),
  guildMembership: one(musicGuildMembers),
  tutorialProgress: many(musicTutorialUserProgress),
}));

 customSession(async ({ user, session }) => {
      if (!user) {
        return { session };
      }
      const customUser = await db.query.user.findFirst({
        where: eq(users.id, user.id),
        with: {
          guildMembership: {
            with: {
              guild: true,
            },
          },
          tutorialProgress: true,
        },
      });
      return {
        session,
        user: {
          ...customUser,
        },
      };
    }),
  ],