@elevasis/core 0.18.0 → 0.19.0

package/src/auth/multi-tenancy/organizations/api-schemas.ts

@@ -1,128 +1,136 @@
-/**
- * Organizations Domain - Zod Validation Schemas
- *
- * Validation schemas for organization management endpoints.
- * Includes request bodies, query params, and path params.
- *
- * Security:
- * - All schemas use .strict() to prevent mass assignment attacks
- * - UUID/WorkOS ID validation prevents invalid references
- * - String length limits prevent DoS
- * - Domain and metadata size limits
- */
-
-import { z } from 'zod'
-import { UuidSchema } from '../../../platform/utils/validation'
-
-// ============================================================================
-// Shared Schemas
-// ============================================================================
-
-/**
- * Organization name validation
- * - Alphanumeric, spaces, hyphens, underscores only
- * - 2-100 characters
- *
- * Security: Prevents injection, DoS via long names
- */
-export const OrganizationNameSchema = z.string()
-  .min(2, 'Organization name must be at least 2 characters')
-  .max(100, 'Organization name must be at most 100 characters')
-  .trim()
-  .regex(/^[a-zA-Z0-9\s\-_]+$/, 'Organization name must contain only letters, numbers, spaces, hyphens, and underscores')
-
-/**
- * Organization ID validation
- * Supports both UUID and WorkOS org_ prefixed IDs
- */
-export const OrganizationIdSchema = z.union([
-  UuidSchema,
-  z.string().regex(/^org_[a-zA-Z0-9]+$/, 'Invalid WorkOS organization ID')
-])
-
-/**
- * Organization domain data schema
- */
-export const OrganizationDomainSchema = z.object({
-  domain: z.string().min(3).max(255),
-  state: z.enum(['verified', 'pending', 'failed']).optional()
-})
-
-// ============================================================================
-// Path Parameters
-// ============================================================================
-
-/**
- * Validate organization ID in URL path
- * Used by: GET/PUT/DELETE /organizations/:id
- */
-export const OrganizationIdParamSchema = z
-  .object({
-    id: OrganizationIdSchema
-  })
-  .strict()
-
-// ============================================================================
-// Request Bodies
-// ============================================================================
-
-/**
- * Create new organization
- * POST /organizations
- *
- * Security:
- * - Name format validated (alphanumeric + spaces + hyphens + underscores)
- * - Domain array size limited (max 10)
- * - Metadata size limited (10KB)
- * - Strict mode prevents unknown field injection
- */
-export const CreateOrganizationSchema = z
-  .object({
-    name: OrganizationNameSchema,
-    domainData: z.array(OrganizationDomainSchema).max(10).optional(),
-    metadata: z.record(z.string(), z.unknown())
-      .refine(
-        val => JSON.stringify(val).length <= 10240,
-        'Metadata must be under 10KB'
-      )
-      .optional()
-  })
-  .strict()
-
-/**
- * Update organization
- * PUT /organizations/:id
- *
- * Security:
- * - All fields optional (partial update)
- * - Same validation as create
- */
-export const UpdateOrganizationSchema = CreateOrganizationSchema.partial().strict()
-
-// ============================================================================
-// Query Parameters
-// ============================================================================
-
-/**
- * List organizations with filters
- * GET /organizations
- *
- * Security:
- * - Limit bounded (prevents DoS)
- * - WorkOS pagination cursors
- */
-export const ListOrganizationsQuerySchema = z.object({
-  limit: z.coerce.number().int().min(1).max(100).default(20),
-  before: z.string().optional(), // WorkOS pagination cursor
-  after: z.string().optional()   // WorkOS pagination cursor
-})
-
-// ============================================================================
-// TypeScript Type Exports
-// ============================================================================
-
-// Export inferred types for use in route handlers
-export type CreateOrganizationInput = z.infer<typeof CreateOrganizationSchema>
-export type UpdateOrganizationInput = z.infer<typeof UpdateOrganizationSchema>
-export type ListOrganizationsQuery = z.infer<typeof ListOrganizationsQuerySchema>
-export type OrganizationIdParam = z.infer<typeof OrganizationIdParamSchema>
+/**
+ * Organizations Domain - Zod Validation Schemas
+ *
+ * Validation schemas for organization management endpoints.
+ * Includes request bodies, query params, and path params.
+ *
+ * Security:
+ * - All schemas use .strict() to prevent mass assignment attacks
+ * - UUID/WorkOS ID validation prevents invalid references
+ * - String length limits prevent DoS
+ * - Domain and metadata size limits
+ */
+
+import { z } from 'zod'
+import { UuidSchema } from '../../../platform/utils/validation'
+
+// ============================================================================
+// Shared Schemas
+// ============================================================================
+
+/**
+ * Organization name validation
+ * - Alphanumeric, spaces, hyphens, underscores only
+ * - 2-100 characters
+ *
+ * Security: Prevents injection, DoS via long names
+ */
+export const OrganizationNameSchema = z
+  .string()
+  .min(2, 'Organization name must be at least 2 characters')
+  .max(100, 'Organization name must be at most 100 characters')
+  .trim()
+  .regex(
+    /^[a-zA-Z0-9\s\-_]+$/,
+    'Organization name must contain only letters, numbers, spaces, hyphens, and underscores'
+  )
+
+/**
+ * Organization ID validation
+ * Supports both UUID and WorkOS org_ prefixed IDs
+ */
+export const OrganizationIdSchema = z.union([
+  UuidSchema,
+  z.string().regex(/^org_[a-zA-Z0-9]+$/, 'Invalid WorkOS organization ID')
+])
+
+/**
+ * Organization domain data schema
+ */
+export const OrganizationDomainSchema = z.object({
+  domain: z.string().min(3).max(255),
+  state: z.enum(['verified', 'pending', 'failed']).optional()
+})
+
+// ============================================================================
+// Path Parameters
+// ============================================================================
+
+/**
+ * Validate organization ID in URL path
+ * Used by: GET/PUT/DELETE /organizations/:id
+ */
+export const OrganizationIdParamSchema = z
+  .object({
+    id: OrganizationIdSchema
+  })
+  .strict()
+
+// ============================================================================
+// Request Bodies
+// ============================================================================
+
+/**
+ * Create new organization
+ * POST /organizations
+ *
+ * Security:
+ * - Name format validated (alphanumeric + spaces + hyphens + underscores)
+ * - Domain array size limited (max 10)
+ * - Metadata size limited (10KB)
+ * - Strict mode prevents unknown field injection
+ */
+export const CreateOrganizationSchema = z
+  .object({
+    name: OrganizationNameSchema,
+    domainData: z.array(OrganizationDomainSchema).max(10).optional(),
+    metadata: z
+      .record(z.string(), z.unknown())
+      .refine((val) => JSON.stringify(val).length <= 10240, 'Metadata must be under 10KB')
+      .optional()
+  })
+  .strict()
+
+/**
+ * Update organization
+ * PUT /organizations/:id
+ *
+ * Security:
+ * - All fields optional (partial update)
+ * - Same validation as create
+ * - At least one field required (matches documented Update schema convention,
+ *   see .claude/rules/core-package.md → "api-schemas.ts Pattern")
+ */
+export const UpdateOrganizationSchema = CreateOrganizationSchema.partial()
+  .strict()
+  .refine((data) => Object.keys(data).length > 0, {
+    message: 'At least one field (name, domainData, or metadata) must be provided'
+  })
+
+// ============================================================================
+// Query Parameters
+// ============================================================================
+
+/**
+ * List organizations with filters
+ * GET /organizations
+ *
+ * Security:
+ * - Limit bounded (prevents DoS)
+ * - WorkOS pagination cursors
+ */
+export const ListOrganizationsQuerySchema = z.object({
+  limit: z.coerce.number().int().min(1).max(100).default(20),
+  before: z.string().optional(), // WorkOS pagination cursor
+  after: z.string().optional() // WorkOS pagination cursor
+})
+
+// ============================================================================
+// TypeScript Type Exports
+// ============================================================================
+
+// Export inferred types for use in route handlers
+export type CreateOrganizationInput = z.infer<typeof CreateOrganizationSchema>
+export type UpdateOrganizationInput = z.infer<typeof UpdateOrganizationSchema>
+export type ListOrganizationsQuery = z.infer<typeof ListOrganizationsQuerySchema>
+export type OrganizationIdParam = z.infer<typeof OrganizationIdParamSchema>

package/src/business/acquisition/api-schemas.test.ts

@@ -13,6 +13,7 @@ import {
   AcqContactStatusSchema,
   AcqEmailValidSchema,
   AcqListResponseSchema,
+  BuildPlanSnapshotSchema,
   CreateArtifactRequestSchema,
   CreateCompanyRequestSchema,
   CreateContactRequestSchema,
@@ -43,6 +44,7 @@ import {
   UpdateListRequestSchema,
   UpdateListStatusRequestSchema
 } from './api-schemas'
+import { createBuildPlanSnapshotFromTemplateId } from './build-templates'
 
 // ---------------------------------------------------------------------------
 // Helpers
@@ -416,6 +418,22 @@ describe('UpdateContactRequestSchema', () => {
     )
   })
 
+  it('accepts processingState keyed by the stage catalog', () => {
+    const result = UpdateContactRequestSchema.safeParse({
+      processingState: {
+        [LEAD_GEN_STAGE_CATALOG.verified.key]: {
+          status: 'no_result'
+        }
+      }
+    })
+
+    expect(result.success).toBe(true)
+  })
+
+  it('accepts deprecated pipelineStatus as a compatibility no-op', () => {
+    expect(UpdateContactRequestSchema.safeParse({ pipelineStatus: 'emailed' }).success).toBe(true)
+  })
+
   it('rejects unknown top-level fields (strict mode)', () => {
     expect(UpdateContactRequestSchema.safeParse({ firstName: 'Alice', unknown: 'x' }).success).toBe(false)
   })
@@ -743,7 +761,7 @@ describe('DealDetailResponseSchema (forward-compat)', () => {
         title: null,
         headline: null,
         linkedin_url: null,
-        pipeline_status: null,
+        processing_state: null,
         enrichment_data: null,
         company: null
       }
@@ -881,6 +899,58 @@ describe('PipelineStageSchema', () => {
   })
 })
 
+// ---------------------------------------------------------------------------
+// BuildPlanSnapshotSchema
+// ---------------------------------------------------------------------------
+
+describe('BuildPlanSnapshotSchema', () => {
+  const validSnapshot = createBuildPlanSnapshotFromTemplateId('dtc-subscription-apollo-clickup')
+
+  it('accepts a snapshot generated from a prospecting build template', () => {
+    expect(validSnapshot).not.toBeNull()
+    expect(BuildPlanSnapshotSchema.safeParse(validSnapshot).success).toBe(true)
+  })
+
+  it('rejects a step stageKey outside LEAD_GEN_STAGE_CATALOG', () => {
+    const result = BuildPlanSnapshotSchema.safeParse({
+      ...validSnapshot,
+      steps: [{ ...validSnapshot!.steps[0], stageKey: 'made-up-stage' }]
+    })
+
+    expect(result.success).toBe(false)
+  })
+
+  it('rejects a step capabilityKey outside CAPABILITY_REGISTRY', () => {
+    const result = BuildPlanSnapshotSchema.safeParse({
+      ...validSnapshot,
+      steps: [{ ...validSnapshot!.steps[0], capabilityKey: 'lead-gen.missing.capability' }]
+    })
+
+    expect(result.success).toBe(false)
+  })
+
+  it('rejects duplicate step ids', () => {
+    const first = validSnapshot!.steps[0]!
+    const second = validSnapshot!.steps[1]!
+    const rest = validSnapshot!.steps.slice(2)
+    const result = BuildPlanSnapshotSchema.safeParse({
+      ...validSnapshot,
+      steps: [first, { ...second, id: first.id }, ...rest]
+    })
+
+    expect(result.success).toBe(false)
+  })
+
+  it('rejects dependsOn references to unknown step ids', () => {
+    const result = BuildPlanSnapshotSchema.safeParse({
+      ...validSnapshot,
+      steps: [{ ...validSnapshot!.steps[0], dependsOn: ['missing-step'] }]
+    })
+
+    expect(result.success).toBe(false)
+  })
+})
+
 // ---------------------------------------------------------------------------
 // ScrapingConfigSchema
 // ---------------------------------------------------------------------------
@@ -1255,6 +1325,33 @@ describe('UpdateCompanyRequestSchema', () => {
     expect(UpdateCompanyRequestSchema.safeParse({ status: 'invalid' }).success).toBe(true)
   })
 
+  it('accepts processingState keyed by the stage catalog', () => {
+    const result = UpdateCompanyRequestSchema.safeParse({
+      processingState: {
+        [LEAD_GEN_STAGE_CATALOG.qualified.key]: {
+          status: 'success',
+          data: { score: 92 }
+        }
+      }
+    })
+
+    expect(result.success).toBe(true)
+  })
+
+  it('accepts deprecated pipelineStatus as a compatibility no-op', () => {
+    expect(UpdateCompanyRequestSchema.safeParse({ pipelineStatus: 'emailed' }).success).toBe(true)
+  })
+
+  it('rejects processingState keys outside the stage catalog', () => {
+    expect(
+      UpdateCompanyRequestSchema.safeParse({
+        processingState: {
+          madeUpStage: { status: 'success' }
+        }
+      }).success
+    ).toBe(false)
+  })
+
   it('accepts numEmployees of 0', () => {
     expect(UpdateCompanyRequestSchema.safeParse({ numEmployees: 0 }).success).toBe(true)
   })
@@ -1468,7 +1565,7 @@ describe('AcqContactResponseSchema (forward-compat)', () => {
     openingLine: null,
     source: null,
     sourceId: null,
-    pipelineStatus: null,
+    processingState: null,
     enrichmentData: null,
     attioPersonId: null,
     batchId: null,
@@ -1488,4 +1585,5 @@ describe('AcqContactResponseSchema (forward-compat)', () => {
   it('rejects an invalid emailValid value', () => {
     expect(AcqContactResponseSchema.safeParse({ ...baseContact, emailValid: 'BAD' }).success).toBe(false)
   })
+
 })

package/src/business/acquisition/api-schemas.ts

@@ -1,10 +1,36 @@
 import { z } from 'zod'
 import { UuidSchema, NonEmptyStringSchema } from '../../platform/utils/validation'
 import { LEAD_GEN_STAGE_CATALOG } from '../../organization-model/domains/sales'
+import { CAPABILITY_REGISTRY } from '../../organization-model/domains/prospecting'
 import { isProspectingBuildTemplateId } from './build-templates'
 export { CrmPriorityBucketKeySchema, CrmPriorityBucketOverrideSchema, CrmPriorityOverrideSchema } from './crm-priority'
 export type { CrmPriorityBucketOverride, CrmPriorityOverride, ResolvedCrmPriorityRuleConfig } from './crm-priority'
 
+export const ProcessingStageStatusSchema = z.enum(['success', 'no_result', 'skipped', 'error'])
+
+export const LeadGenStageKeySchema = z
+  .string()
+  .refine((value) => Object.prototype.hasOwnProperty.call(LEAD_GEN_STAGE_CATALOG, value), {
+    message: 'processing state key must match LEAD_GEN_STAGE_CATALOG'
+  })
+
+export const LeadGenCapabilityKeySchema = z
+  .string()
+  .refine((value) => Object.prototype.hasOwnProperty.call(CAPABILITY_REGISTRY, value), {
+    message: 'capabilityKey must match CAPABILITY_REGISTRY'
+  })
+
+export const ProcessingStateEntrySchema = z
+  .object({
+    status: ProcessingStageStatusSchema,
+    data: z.unknown().optional()
+  })
+  .passthrough()
+
+export const ProcessingStateSchema = z.record(LeadGenStageKeySchema, ProcessingStateEntrySchema)
+export const CompanyProcessingStateSchema = ProcessingStateSchema
+export const ContactProcessingStateSchema = ProcessingStateSchema
+
 /**
  * Deal Management API Schemas
  *
@@ -132,7 +158,7 @@ export const DealContactSummarySchema = z.object({
   title: z.string().nullable(),
   headline: z.string().nullable(),
   linkedin_url: z.string().nullable(),
-  pipeline_status: z.record(z.string(), z.unknown()).nullable(),
+  processing_state: ProcessingStateSchema.nullable(),
   enrichment_data: z.record(z.string(), z.unknown()).nullable(),
   company: z
     .object({
@@ -442,10 +468,10 @@ export const BuildPlanSnapshotStepSchema = z
     description: z.string().trim().min(1).max(2000).optional(),
     primaryEntity: z.enum(['company', 'contact']),
     outputs: z.array(z.enum(['company', 'contact', 'export'])).min(1),
-    stageKey: z.string().trim().min(1).max(100),
+    stageKey: LeadGenStageKeySchema,
     dependsOn: z.array(z.string().trim().min(1).max(100)).optional(),
     dependencyMode: z.literal('per-record-eligibility'),
-    capabilityKey: z.string().trim().min(1).max(100),
+    capabilityKey: LeadGenCapabilityKeySchema,
     defaultBatchSize: z.number().int().positive(),
     maxBatchSize: z.number().int().positive()
   })
@@ -454,11 +480,38 @@ export const BuildPlanSnapshotStepSchema = z
     path: ['defaultBatchSize']
   })
 
-export const BuildPlanSnapshotSchema = z.object({
-  templateId: z.string().trim().min(1).max(100),
-  templateLabel: z.string().trim().min(1).max(120),
-  steps: z.array(BuildPlanSnapshotStepSchema).min(1)
-})
+export const BuildPlanSnapshotSchema = z
+  .object({
+    templateId: z.string().trim().min(1).max(100),
+    templateLabel: z.string().trim().min(1).max(120),
+    steps: z.array(BuildPlanSnapshotStepSchema).min(1)
+  })
+  .superRefine((snapshot, ctx) => {
+    const stepIds = new Set<string>()
+
+    snapshot.steps.forEach((step, index) => {
+      if (stepIds.has(step.id)) {
+        ctx.addIssue({
+          code: z.ZodIssueCode.custom,
+          message: `duplicate build-plan step id "${step.id}"`,
+          path: ['steps', index, 'id']
+        })
+      }
+      stepIds.add(step.id)
+    })
+
+    snapshot.steps.forEach((step, index) => {
+      for (const dependencyId of step.dependsOn ?? []) {
+        if (!stepIds.has(dependencyId)) {
+          ctx.addIssue({
+            code: z.ZodIssueCode.custom,
+            message: `dependsOn references unknown build-plan step "${dependencyId}"`,
+            path: ['steps', index, 'dependsOn']
+          })
+        }
+      }
+    })
+  })
 
 export const AcqListMetadataSchema = z
   .object({
@@ -638,13 +691,6 @@ export const ListTelemetryResponseSchema = ListTelemetrySchema
 
 export const ListTelemetryListResponseSchema = z.array(ListTelemetrySchema)
 
-/**
- * Terminal row-level status for one lead-gen processing stage.
- * Missing key still means not attempted; legacy boolean `true` is normalized
- * to `success` by the API reader during rollout.
- */
-export const ProcessingStageStatusSchema = z.enum(['success', 'no_result', 'skipped', 'error'])
-
 /**
  * Per-stage progress aggregate for a single pipeline stage.
  * `attempted` counts terminal statuses, including success, no-result, skipped,
@@ -704,29 +750,6 @@ export const AcqCompanyStatusSchema = z.enum(['active', 'invalid'])
 export const AcqContactStatusSchema = z.enum(['active', 'invalid'])
 export const AcqEmailValidSchema = z.enum(['VALID', 'INVALID', 'RISKY', 'UNKNOWN'])
 
-/**
- * Zod schema mirroring the CompanyPipelineStatus interface from types.ts.
- * Used by operations-layer CompanyRecord schemas to validate pipelineStatus
- * against the canonical shape instead of using open passthrough().
- */
-export const CompanyPipelineStatusSchema = z
-  .object({
-    acquired: z.boolean().optional(),
-    enrichment: z
-      .record(
-        z.string(),
-        z
-          .object({
-            status: z.enum(['pending', 'complete', 'failed', 'skipped']),
-            completedAt: z.string().optional(),
-            error: z.string().optional()
-          })
-          .passthrough()
-      )
-      .optional()
-  })
-  .passthrough()
-
 export const CompanyIdParamsSchema = z.object({
   companyId: UuidSchema
 })
@@ -743,6 +766,7 @@ export const ListCompaniesQuerySchema = z
     website: z.string().trim().min(1).max(2048).optional(),
     segment: z.string().trim().min(1).max(255).optional(),
     category: z.string().trim().min(1).max(255).optional(),
+    pipelineStatus: z.unknown().optional(),
     batchId: z.string().trim().min(1).max(255).optional(),
     status: AcqCompanyStatusSchema.optional(),
     includeAll: QueryBooleanSchema.optional(),
@@ -776,6 +800,7 @@ export const CreateCompanyRequestSchema = z
     category: z.string().trim().min(1).max(255).optional(),
     source: z.string().trim().min(1).max(255).optional(),
     batchId: z.string().trim().min(1).max(255).optional(),
+    pipelineStatus: z.unknown().optional(),
     verticalResearch: z.string().trim().min(1).max(5000).optional()
   })
   .strict()
@@ -792,7 +817,8 @@ export const UpdateCompanyRequestSchema = z
     locationState: z.string().trim().min(1).max(255).optional(),
     category: z.string().trim().min(1).max(255).optional(),
     segment: z.string().trim().min(1).max(255).optional(),
-    pipelineStatus: z.record(z.string(), z.unknown()).optional(),
+    processingState: CompanyProcessingStateSchema.optional(),
+    pipelineStatus: z.unknown().optional(),
     enrichmentData: z.record(z.string(), z.unknown()).optional(),
     source: z.string().trim().min(1).max(255).optional(),
     batchId: z.string().trim().min(1).max(255).optional(),
@@ -812,6 +838,7 @@ export const UpdateCompanyRequestSchema = z
       data.locationState !== undefined ||
       data.category !== undefined ||
       data.segment !== undefined ||
+      data.processingState !== undefined ||
       data.pipelineStatus !== undefined ||
       data.enrichmentData !== undefined ||
       data.source !== undefined ||
@@ -833,7 +860,8 @@ export const CreateContactRequestSchema = z
     title: z.string().trim().min(1).max(255).optional(),
     source: z.string().trim().min(1).max(255).optional(),
     sourceId: z.string().trim().min(1).max(255).optional(),
-    batchId: z.string().trim().min(1).max(255).optional()
+    batchId: z.string().trim().min(1).max(255).optional(),
+    pipelineStatus: z.unknown().optional()
   })
   .strict()
 
@@ -848,7 +876,8 @@ export const UpdateContactRequestSchema = z
     headline: z.string().trim().min(1).max(5000).optional(),
     filterReason: z.string().trim().min(1).max(5000).optional(),
     openingLine: z.string().trim().min(1).max(5000).optional(),
-    pipelineStatus: z.record(z.string(), z.unknown()).optional(),
+    processingState: ContactProcessingStateSchema.optional(),
+    pipelineStatus: z.unknown().optional(),
     enrichmentData: z.record(z.string(), z.unknown()).optional(),
     status: AcqContactStatusSchema.optional()
   })
@@ -864,6 +893,7 @@ export const UpdateContactRequestSchema = z
       data.headline !== undefined ||
       data.filterReason !== undefined ||
       data.openingLine !== undefined ||
+      data.processingState !== undefined ||
       data.pipelineStatus !== undefined ||
       data.enrichmentData !== undefined ||
       data.status !== undefined,
@@ -886,7 +916,8 @@ export const AcqCompanyResponseSchema = z.object({
   category: z.string().nullable(),
   categoryPain: z.string().nullable(),
   segment: z.string().nullable(),
-  pipelineStatus: z.record(z.string(), z.unknown()).nullable(),
+  processingState: CompanyProcessingStateSchema.nullable(),
+  pipelineStatus: z.unknown().nullable().optional(),
   enrichmentData: z.record(z.string(), z.unknown()).nullable(),
   source: z.string().nullable(),
   batchId: z.string().nullable(),
@@ -936,7 +967,8 @@ export const AcqContactResponseSchema = z.object({
   openingLine: z.string().nullable(),
   source: z.string().nullable(),
   sourceId: z.string().nullable(),
-  pipelineStatus: z.record(z.string(), z.unknown()).nullable(),
+  processingState: ContactProcessingStateSchema.nullable(),
+  pipelineStatus: z.unknown().nullable().optional(),
   enrichmentData: z.record(z.string(), z.unknown()).nullable(),
   attioPersonId: z.string().nullable(),
   batchId: z.string().nullable(),
@@ -1063,6 +1095,7 @@ export const AcqListCompanyResponseSchema = z.object({
 // ---------------------------------------------------------------------------
 
 export const AcqCompanySchemas = {
+  CompanyProcessingState: CompanyProcessingStateSchema,
   CompanyIdParams: CompanyIdParamsSchema,
   ListCompaniesQuery: ListCompaniesQuerySchema,
   CreateCompanyRequest: CreateCompanyRequestSchema,
@@ -1073,6 +1106,7 @@ export const AcqCompanySchemas = {
 }
 
 export const AcqContactSchemas = {
+  ContactProcessingState: ContactProcessingStateSchema,
   ContactIdParams: ContactIdParamsSchema,
   ListContactsQuery: ListContactsQuerySchema,
   CreateContactRequest: CreateContactRequestSchema,
@@ -1116,7 +1150,10 @@ export const AcqListSchemas = {
   BuildPlanSnapshot: BuildPlanSnapshotSchema,
   BuildPlanSnapshotStep: BuildPlanSnapshotStepSchema,
   AcqListMetadata: AcqListMetadataSchema,
+  LeadGenStageKey: LeadGenStageKeySchema,
+  LeadGenCapabilityKey: LeadGenCapabilityKeySchema,
   ProcessingStageStatus: ProcessingStageStatusSchema,
+  ProcessingState: ProcessingStateSchema,
   ListStageCounts: ListStageCountsSchema,
   ListTelemetry: ListTelemetrySchema,
 
@@ -1195,6 +1232,7 @@ export type PipelineConfig = z.infer<typeof PipelineConfigSchema>
 export type BuildPlanSnapshotStep = z.infer<typeof BuildPlanSnapshotStepSchema>
 export type BuildPlanSnapshot = z.infer<typeof BuildPlanSnapshotSchema>
 export type AcqListMetadata = z.infer<typeof AcqListMetadataSchema>
+export type LeadGenCapabilityKey = z.infer<typeof LeadGenCapabilityKeySchema>
 export type ProcessingStageStatus = z.infer<typeof ProcessingStageStatusSchema>
 export type ListStageCountsInput = z.infer<typeof ListStageCountsSchema>['stageCounts']
 export type ListTelemetryInput = z.infer<typeof ListTelemetrySchema>