Handling compound @@unique elements

[callajd]

Hello,

I'm working on integrating this generator into our workflow, we've been looking for something just like this.

I've run into one small problem that I can't figure out: correctly generating types from models that have compound @@unique elements referring to multiple model fields. For example:

model ProductVariant {
  productId Int
  name      String
  url       String
  imageUrl  String
  price     Float
  product   Product @relation(fields: [productId], references: [id])

  @@unique([productId, name]) <---- This element
}

After generation, I'll get an error on the "findUnique" definition for the ProductVariant type:

pothos/generated/ProductVariant/object.base.ts:12:3 - error TS2322: <snip>
        Property 'productId_name' is missing in type '{ name: string; imageUrl: string; productId: number; url: string; price: number; }' but required in type '{ productId_name: ProductVariantProductIdNameCompoundUniqueInput; }'.

   10 | export const ProductVariantObject = definePrismaObject('ProductVariant', {
   11 |   description: undefined,
 > 12 |   findUnique: (fields) => ({ ...fields }),
      |   ^
   13 |   fields: (t) => ({
   14 |     productId: t.field(ProductVariantProductIdFieldObject),
   15 |     name: t.field(ProductVariantNameFieldObject),

Any suggestions to get around this issue? Much appreciated!

EDIT: I should have mentioned, it's clear what the fix to ProductVariantObject->findUnique should be:

  findUnique: (fields) => ({ 
    ...fields,
    productId_name: {
      productId: fields.productId,
      name: fields.name,
    }
  }),

I'm specifically asking how to address this upstream of the code generation so that my team doesn't need to apply these fixes manually (there are about 13 different types that each have this same issue).