Possible to use zod effects inside union elements?

[phoneticallySAARTHaK]

I'm trying to use .refine on a discriminated union element, but it's probably not possible.

const GetRegionRequestSchema = y.discriminatedUnion('type', [
	z.object({ type: z.literal('COUNTRY') }),
	z.object({
		type: z.literal('STATE'),
		country_id: z.string().optional(),
		coutry_name: z.string().optional(),
	}).refine((val) => !val.coutry_name && !val.country_id ? false : val),
	z.object({
	type: z.literal('CITY'),
	state_id: z.string(),
	})
])

I can do refine on the discriminatedUnion itself, but just wanted to know if this usecase is possible