diff --git a/apps/api/src/controllers/v1/types.ts b/apps/api/src/controllers/v1/types.ts index 9e212d4d..8ecaf899 100644 --- a/apps/api/src/controllers/v1/types.ts +++ b/apps/api/src/controllers/v1/types.ts @@ -379,11 +379,14 @@ export const scrapeOptions = baseScrapeOptions .object({ model: z.string().default(agentExtractModelValue), prompt: z.string().optional(), + sessionId: z.string().optional(), + waitBeforeClosingMs: z.number().optional(), }) .optional(), extract: extractOptionsWithAgent.optional(), jsonOptions: extractOptionsWithAgent.optional(), }) + .strict(strictMessage) .refine( (obj) => { if (!obj.actions) return true; @@ -400,7 +403,16 @@ export const scrapeOptions = baseScrapeOptions .refine(fire1Refine, fire1RefineOpts) .transform(extractTransform); -export type ScrapeOptions = z.infer; +export type ScrapeOptions = z.infer & { + extract: z.infer, + jsonOptions: z.infer, + agent: { + model: string, + prompt: string, + sessionId?: string, + waitBeforeClosingMs?: number, + }, +}; import Ajv from "ajv"; import type { CostTracking } from "../../lib/extract/extraction-service";