InferIdType and OptionalId are logically complex types it would be ideal to break these types down into more readable components and attempt to make use of Typescript's utility types as much as possible.

Potentially some combination of union types, and the Extract and/or Pick utility types: Extract<Pick<TSchema, '_id'> | ObjectId, NonNullable>