json types are omitted in withRelations

[Schmell]

Hey everyone.
I am propbably not configured correctly but i have managed to get types for JSON column types.

model Event {
   // ...rest
   /// [fileInfo]
   fileInfo        Json? /// @zod.custom.use(fileInfoZod)
   /// [resultColumns]
   resultColumns   Json? /// @zod.custom.use(resultColumnsZod)
}

and this works great if i use Event schema in the models, and i can see there is properly generated zod in the file but, when i look at the EventWithRelations or even EventWithPartialRelations i get...

export type EventWithRelations = Omit<
        z.infer<typeof EventSchema>,
	'fileInfo' | 'resultColumns' | 'rest'
> & {
	fileInfo?: JsonValueType | null
	resultColumns?: JsonValueType | null
	rest?: JsonValueType | null
} & EventRelations

It seems a bit unintuitive that the omit clause is there as this can be fixed by

export type EventWithRelations = z.infer<typeof EventSchema> & EventRelations

which just seems simpler and it works.

Again I may just need to change some config options to make this work as expected.
here is my config options

generator zod {
  provider                         = "zod-prisma-types"
  output                           = "../src/lib/schemas/generated/zod"
  useMultipleFiles                 = true // default is false
  writeBarrelFiles                 = true // default is true
  createInputTypes                 = false // default is true
  createModelTypes                 = true // default is true
  addInputTypeValidation           = false // default is true
  addIncludeType                   = false // default is true
  addSelectType                    = false // default is true
  validateWhereUniqueInput         = false // default is false
  createOptionalDefaultValuesTypes = false // default is false
  createRelationValuesTypes        = true // default is false
  createPartialTypes               = true // default is false
  useDefaultValidators             = true // default is true
  coerceDate                       = false // default is true
  writeNullishInModelTypes         = true // default is false
}

Thanks everyone