Get Correct Return Type when using Prisma ORM inside a method that accepts the client's args

model ExampleItem {
  id            Int              @id @default(autoincrement())
  createdAt     DateTime         @default(now())
  updatedAt     DateTime         @default(now()) @updatedAt
  //
  name          String
  optionalField String?
  messages      ExampleMessage[]
  //
  shop          String
  shopifyStore  ShopifyStore     @relation(fields: [shop], references: [shop], onDelete: Cascade, onUpdate: Cascade)
}

model ExampleMessage {
  id            Int         @id @default(autoincrement())
  createdAt     DateTime    @default(now())
  updatedAt     DateTime    @default(now()) @updatedAt
  //
  message       String
  //
  exampleItemId Int
  exampleItem   ExampleItem @relation(fields: [exampleItemId], references: [id])
}

model ExampleItem {
  id            Int              @id @default(autoincrement())
  createdAt     DateTime         @default(now())
  updatedAt     DateTime         @default(now()) @updatedAt
  //
  name          String
  optionalField String?
  messages      ExampleMessage[]
  //
  shop          String
  shopifyStore  ShopifyStore     @relation(fields: [shop], references: [shop], onDelete: Cascade, onUpdate: Cascade)
}

model ExampleMessage {
  id            Int         @id @default(autoincrement())
  createdAt     DateTime    @default(now())
  updatedAt     DateTime    @default(now()) @updatedAt
  //
  message       String
  //
  exampleItemId Int
  exampleItem   ExampleItem @relation(fields: [exampleItemId], references: [id])
}

// Method
export async function getExampleItem(
  prismaArgs: Prisma.ExampleItemFindFirstArgs<DefaultArgs>,
  db: PrismaClient
) {
  try {
    const exampleItem = await db.exampleItem.findFirst(prismaArgs);

    // Do something else...

    return exampleItem;
  } catch (error) {
    await handleError({ operation: 'catch', error });
  }

  return null;
}

const exampleItem = await getExampleItem({ where: { id: 1 }, include: { messages: true } }, db);

// Method
export async function getExampleItem(
  prismaArgs: Prisma.ExampleItemFindFirstArgs<DefaultArgs>,
  db: PrismaClient
) {
  try {
    const exampleItem = await db.exampleItem.findFirst(prismaArgs);

    // Do something else...

    return exampleItem;
  } catch (error) {
    await handleError({ operation: 'catch', error });
  }

  return null;
}

const exampleItem = await getExampleItem({ where: { id: 1 }, include: { messages: true } }, db);