take ZodRawShape

Author: vsnig

package.json

@@ -1,6 +1,6 @@
 {
   "name": "zod-cdk-context",
-  "version": "0.0.9",
+  "version": "0.0.10",
   "description": "Validate AWS CDK context with Zod",
   "keywords": [
     "zod",

src/index.ts

@@ -1,18 +1,10 @@
 import type { Node } from 'constructs'
 
-import { z } from 'zod'
+import { z, type ZodRawShape } from 'zod'
 
 export { z } from 'zod'
 
-export const getValidatedContext = <
-  T extends {
-    [key: string]:
-      | z.ZodString
-      | z.ZodNumber
-      | z.ZodOptional<z.ZodString | z.ZodNumber>
-      | z.ZodEnum<[string, ...string[]]>
-  },
->(
+export const getValidatedContext = <T extends ZodRawShape>(
   node: Node,
   schemaObject: T,
 ): z.infer<z.ZodObject<T>> => {