deal with nested config structure and types

lost-pixel · Nov 18, 2023 · 49c6acb · 49c6acb
1 parent 8f72fca
commit 49c6acb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/config.ts b/src/config.ts
@@ -455,9 +455,15 @@ export const ConfigSchema = z.union([
 export const FlexibleConfigSchema = z.union([
   PlatformModeConfigSchema.extend({
     timeouts: TimeoutsSchema.partial(),
+    pageShots: PageShotsSchema.extend({
+      pages: z.array(PageScreenshotParameterSchema.partial()),
+    }),
   }).partial(),
   GenerateOnlyModeConfigSchema.extend({
     timeouts: TimeoutsSchema.partial(),
+    pageShots: PageShotsSchema.extend({
+      pages: z.array(PageScreenshotParameterSchema.partial()),
+    }),
   }).partial(),
 ]);