@microfox/ai-worker 1.0.2 → 1.0.3

package/dist/chunk-BJPQY2NJ.mjs.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/client.ts"],"sourcesContent":["/**\n * Client for dispatching background worker jobs.\n *\n * In production, dispatching happens via the workers HTTP API:\n *   POST /workers/trigger  -> enqueues message to SQS on the workers service side\n *\n * This avoids requiring AWS credentials in your Next.js app.\n */\n\nimport type { ZodType, z } from 'zod';\nimport type { WorkerQueueConfig, WorkerQueueContext } from './queue.js';\n\nexport interface WorkerQueueRegistry {\n  getQueueById(queueId: string): WorkerQueueConfig | undefined;\n  invokeMapInput?: (\n    queueId: string,\n    stepIndex: number,\n    prevOutput: unknown,\n    initialInput: unknown\n  ) => Promise<unknown> | unknown;\n}\n\nexport interface DispatchOptions {\n  /**\n   * Optional webhook callback URL to notify when the job finishes.\n   * Only called when provided. Default: no webhook (use job store / MongoDB only).\n   */\n  webhookUrl?: string;\n  /**\n   * Controls how dispatch executes.\n   * - \"auto\" (default): local inline execution in development unless WORKERS_LOCAL_MODE=false.\n   * - \"local\": force inline execution (no SQS).\n   * - \"remote\": force SQS/Lambda dispatch even in development.\n   */\n  mode?: 'auto' | 'local' | 'remote';\n  jobId?: string;\n  metadata?: Record<string, any>;\n  /**\n   * In-memory queue registry for dispatchQueue. Required when using dispatchQueue.\n   * Pass a registry that imports from your .queue.ts definitions (works on Vercel/serverless).\n   */\n  registry?: WorkerQueueRegistry;\n  /**\n   * Optional callback to create a queue job record before dispatching.\n   * Called with queueJobId (= first worker's jobId), queueId, and firstStep.\n   */\n  onCreateQueueJob?: (params: {\n    queueJobId: string;\n    queueId: string;\n    firstStep: { workerId: string; workerJobId: string };\n    metadata?: Record<string, unknown>;\n  }) => Promise<void>;\n}\n\nexport interface DispatchResult {\n  messageId: string;\n  status: 'queued';\n  jobId: string;\n}\n\nexport interface DispatchQueueResult extends DispatchResult {\n  queueId: string;\n}\n\nexport interface SerializedContext {\n  requestId?: string;\n  userId?: string;\n  traceId?: string;\n  [key: string]: any;\n}\n\n/**\n * Derives the full /workers/trigger URL from env.\n * Exported for use by local dispatchWorker (worker-to-worker in dev).\n *\n * Preferred env vars:\n * - WORKER_BASE_URL: base URL of the workers service (e.g. https://.../prod)\n * - NEXT_PUBLIC_WORKER_BASE_URL: same, but exposed to the browser\n *\n * Legacy env vars (still supported for backwards compatibility):\n * - WORKERS_TRIGGER_API_URL / NEXT_PUBLIC_WORKERS_TRIGGER_API_URL\n * - WORKERS_CONFIG_API_URL / NEXT_PUBLIC_WORKERS_CONFIG_API_URL\n */\nexport function getWorkersTriggerUrl(): string {\n  const raw =\n    process.env.WORKER_BASE_URL ||\n    process.env.NEXT_PUBLIC_WORKER_BASE_URL ||\n    process.env.WORKERS_TRIGGER_API_URL ||\n    process.env.NEXT_PUBLIC_WORKERS_TRIGGER_API_URL ||\n    process.env.WORKERS_CONFIG_API_URL ||\n    process.env.NEXT_PUBLIC_WORKERS_CONFIG_API_URL;\n\n  if (!raw) {\n    throw new Error(\n      'WORKER_BASE_URL (preferred) or NEXT_PUBLIC_WORKER_BASE_URL is required for background workers'\n    );\n  }\n\n  const url = new URL(raw);\n  url.search = '';\n  url.hash = '';\n\n  const path = url.pathname || '';\n\n  // If the user pointed at a specific endpoint, normalize back to the service root.\n  url.pathname = path.replace(/\\/?workers\\/(trigger|config)\\/?$/, '');\n\n  const basePath = url.pathname.replace(/\\/+$/, '');\n  url.pathname = `${basePath}/workers/trigger`.replace(/\\/+$/, '');\n\n  return url.toString();\n}\n\n/**\n * Serializes context data for transmission to Lambda.\n * Only serializes safe, JSON-compatible properties.\n */\nfunction serializeContext(ctx: any): SerializedContext {\n  const serialized: SerializedContext = {};\n\n  if (ctx.requestId) {\n    serialized.requestId = ctx.requestId;\n  }\n\n  // Extract any additional serializable metadata\n  if (ctx.metadata && typeof ctx.metadata === 'object') {\n    Object.assign(serialized, ctx.metadata);\n  }\n\n  // Allow custom context serialization via a helper property\n  if (ctx._serializeContext && typeof ctx._serializeContext === 'function') {\n    const custom = ctx._serializeContext();\n    Object.assign(serialized, custom);\n  }\n\n  return serialized;\n}\n\n\n/**\n * Dispatches a background worker job to SQS.\n *\n * @param workerId - The ID of the worker to dispatch\n * @param input - The input data for the worker (will be validated against inputSchema)\n * @param inputSchema - Zod schema for input validation\n * @param options - Dispatch options including webhook URL\n * @param ctx - Optional context object (only serializable parts will be sent)\n * @returns Promise resolving to dispatch result with messageId and jobId\n */\nexport async function dispatch<INPUT_SCHEMA extends ZodType<any>>(\n  workerId: string,\n  input: z.input<INPUT_SCHEMA>,\n  inputSchema: INPUT_SCHEMA,\n  options: DispatchOptions,\n  ctx?: any\n): Promise<DispatchResult> {\n  // Validate input against schema\n  const validatedInput = inputSchema.parse(input);\n\n  // Generate job ID if not provided\n  const jobId =\n    options.jobId || `job-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;\n\n  // Resolve /workers/trigger endpoint URL\n  const triggerUrl = getWorkersTriggerUrl();\n\n  // Serialize context (only safe, JSON-compatible parts)\n  const serializedContext = ctx ? serializeContext(ctx) : {};\n\n  // Job updates use MongoDB only; never pass jobStoreUrl/origin URL.\n  const messageBody = {\n    workerId,\n    jobId,\n    input: validatedInput,\n    context: serializedContext,\n    webhookUrl: options.webhookUrl,\n    metadata: options.metadata || {},\n    timestamp: new Date().toISOString(),\n  };\n\n  const headers: Record<string, string> = {\n    'Content-Type': 'application/json',\n  };\n  const triggerKey = process.env.WORKERS_TRIGGER_API_KEY;\n  if (triggerKey) {\n    headers['x-workers-trigger-key'] = triggerKey;\n  }\n\n  const response = await fetch(triggerUrl, {\n    method: 'POST',\n    headers,\n    body: JSON.stringify({\n      workerId,\n      body: messageBody,\n    }),\n  });\n\n  if (!response.ok) {\n    const text = await response.text().catch(() => '');\n    throw new Error(\n      `Failed to trigger worker \"${workerId}\": ${response.status} ${response.statusText}${text ? ` - ${text}` : ''}`\n    );\n  }\n\n  const data = (await response.json().catch(() => ({}))) as any;\n  const messageId = data?.messageId ? String(data.messageId) : `trigger-${jobId}`;\n\n  return {\n    messageId,\n    status: 'queued',\n    jobId,\n  };\n}\n\n/**\n * Local development mode: runs the handler immediately in the same process.\n * This bypasses SQS and Lambda for faster iteration during development.\n *\n * @param handler - The worker handler function\n * @param input - The input data\n * @param ctx - The context object\n * @returns The handler result\n */\nexport async function dispatchLocal<INPUT, OUTPUT>(\n  handler: (params: { input: INPUT; ctx: any }) => Promise<OUTPUT>,\n  input: INPUT,\n  ctx?: any\n): Promise<OUTPUT> {\n  return handler({ input, ctx: ctx || {} });\n}\n\n/**\n * Dispatches a queue by ID, using a generated registry of .queue.ts\n * definitions to determine the first worker and initial input mapping.\n *\n * This API intentionally mirrors the ergonomics of dispatching a single\n * worker, but under the hood it embeds queue context into the job input\n * so queue-aware wrappers can chain subsequent steps.\n */\nexport async function dispatchQueue<InitialInput = any>(\n  queueId: string,\n  initialInput: InitialInput,\n  options: DispatchOptions = {},\n  ctx?: any\n): Promise<DispatchQueueResult> {\n  const registry = options.registry;\n  if (!registry?.getQueueById) {\n    throw new Error(\n      'dispatchQueue requires options.registry with getQueueById. ' +\n        'Use getQueueRegistry() from your workflows registry (e.g. app/api/workflows/registry/workers) and pass { registry: await getQueueRegistry() }.'\n    );\n  }\n  const { getQueueById, invokeMapInput } = registry;\n  const queue = getQueueById(queueId);\n\n  if (!queue) {\n    throw new Error(`Worker queue \"${queueId}\" not found in registry`);\n  }\n\n  if (!queue.steps || queue.steps.length === 0) {\n    throw new Error(`Worker queue \"${queueId}\" has no steps defined`);\n  }\n\n  const stepIndex = 0;\n  const firstStep = queue.steps[stepIndex];\n  const firstWorkerId = firstStep.workerId;\n\n  if (!firstWorkerId) {\n    throw new Error(\n      `Worker queue \"${queueId}\" has an invalid first step (missing workerId)`\n    );\n  }\n\n  // Compute the first step's input:\n  // - If a mapping function is configured and the registry exposes invokeMapInput,\n  //   use it (prevOutput is undefined for the first step).\n  // - Otherwise, default to the initial input.\n  let firstInput: unknown = initialInput;\n  if (firstStep.mapInputFromPrev && typeof invokeMapInput === 'function') {\n    firstInput = await invokeMapInput(\n      queueId,\n      stepIndex,\n      undefined,\n      initialInput\n    );\n  }\n\n  const jobId =\n    options.jobId ||\n    `job-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;\n\n  const queueContext: WorkerQueueContext<InitialInput> = {\n    id: queueId,\n    stepIndex,\n    initialInput,\n    queueJobId: jobId,\n  };\n\n  // Create queue job record if callback provided (for progress tracking)\n  if (options.onCreateQueueJob) {\n    try {\n      await options.onCreateQueueJob({\n        queueJobId: jobId,\n        queueId,\n        firstStep: { workerId: firstWorkerId, workerJobId: jobId },\n        metadata: options.metadata as Record<string, unknown> | undefined,\n      });\n    } catch (err: any) {\n      console.warn('[dispatchQueue] onCreateQueueJob failed:', err?.message ?? err);\n    }\n  }\n\n  // Embed queue context into the worker input under a reserved key.\n  const normalizedFirstInput =\n    firstInput !== null && typeof firstInput === 'object'\n      ? (firstInput as Record<string, any>)\n      : { value: firstInput };\n\n  const inputWithQueue = {\n    ...normalizedFirstInput,\n    __workerQueue: queueContext,\n  };\n\n  const metadataWithQueue = {\n    ...(options.metadata || {}),\n    __workerQueue: queueContext,\n  };\n\n  const triggerUrl = getWorkersTriggerUrl();\n  const serializedContext = ctx ? serializeContext(ctx) : {};\n\n  const messageBody = {\n    workerId: firstWorkerId,\n    jobId,\n    input: inputWithQueue,\n    context: serializedContext,\n    webhookUrl: options.webhookUrl,\n    metadata: metadataWithQueue,\n    timestamp: new Date().toISOString(),\n  };\n\n  const headers: Record<string, string> = {\n    'Content-Type': 'application/json',\n  };\n  const triggerKey = process.env.WORKERS_TRIGGER_API_KEY;\n  if (triggerKey) {\n    headers['x-workers-trigger-key'] = triggerKey;\n  }\n\n  const response = await fetch(triggerUrl, {\n    method: 'POST',\n    headers,\n    body: JSON.stringify({\n      workerId: firstWorkerId,\n      body: messageBody,\n    }),\n  });\n\n  if (!response.ok) {\n    const text = await response.text().catch(() => '');\n    throw new Error(\n      `Failed to trigger queue \"${queueId}\" (worker \"${firstWorkerId}\"): ` +\n        `${response.status} ${response.statusText}${text ? ` - ${text}` : ''}`\n    );\n  }\n\n  const data = (await response.json().catch(() => ({}))) as any;\n  const messageId = data?.messageId ? String(data.messageId) : `trigger-${jobId}`;\n\n  return {\n    queueId,\n    messageId,\n    status: 'queued',\n    jobId,\n  };\n}\n\n"],"mappings":";AAmFO,SAAS,uBAA+B;AAC7C,QAAM,MACJ,QAAQ,IAAI,mBACZ,QAAQ,IAAI,+BACZ,QAAQ,IAAI,2BACZ,QAAQ,IAAI,uCACZ,QAAQ,IAAI,0BACZ,QAAQ,IAAI;AAEd,MAAI,CAAC,KAAK;AACR,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,QAAM,MAAM,IAAI,IAAI,GAAG;AACvB,MAAI,SAAS;AACb,MAAI,OAAO;AAEX,QAAM,OAAO,IAAI,YAAY;AAG7B,MAAI,WAAW,KAAK,QAAQ,oCAAoC,EAAE;AAElE,QAAM,WAAW,IAAI,SAAS,QAAQ,QAAQ,EAAE;AAChD,MAAI,WAAW,GAAG,QAAQ,mBAAmB,QAAQ,QAAQ,EAAE;AAE/D,SAAO,IAAI,SAAS;AACtB;AAMA,SAAS,iBAAiB,KAA6B;AACrD,QAAM,aAAgC,CAAC;AAEvC,MAAI,IAAI,WAAW;AACjB,eAAW,YAAY,IAAI;AAAA,EAC7B;AAGA,MAAI,IAAI,YAAY,OAAO,IAAI,aAAa,UAAU;AACpD,WAAO,OAAO,YAAY,IAAI,QAAQ;AAAA,EACxC;AAGA,MAAI,IAAI,qBAAqB,OAAO,IAAI,sBAAsB,YAAY;AACxE,UAAM,SAAS,IAAI,kBAAkB;AACrC,WAAO,OAAO,YAAY,MAAM;AAAA,EAClC;AAEA,SAAO;AACT;AAaA,eAAsB,SACpB,UACA,OACA,aACA,SACA,KACyB;AAEzB,QAAM,iBAAiB,YAAY,MAAM,KAAK;AAG9C,QAAM,QACJ,QAAQ,SAAS,OAAO,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,OAAO,GAAG,CAAC,CAAC;AAG/E,QAAM,aAAa,qBAAqB;AAGxC,QAAM,oBAAoB,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAGzD,QAAM,cAAc;AAAA,IAClB;AAAA,IACA;AAAA,IACA,OAAO;AAAA,IACP,SAAS;AAAA,IACT,YAAY,QAAQ;AAAA,IACpB,UAAU,QAAQ,YAAY,CAAC;AAAA,IAC/B,YAAW,oBAAI,KAAK,GAAE,YAAY;AAAA,EACpC;AAEA,QAAM,UAAkC;AAAA,IACtC,gBAAgB;AAAA,EAClB;AACA,QAAM,aAAa,QAAQ,IAAI;AAC/B,MAAI,YAAY;AACd,YAAQ,uBAAuB,IAAI;AAAA,EACrC;AAEA,QAAM,WAAW,MAAM,MAAM,YAAY;AAAA,IACvC,QAAQ;AAAA,IACR;AAAA,IACA,MAAM,KAAK,UAAU;AAAA,MACnB;AAAA,MACA,MAAM;AAAA,IACR,CAAC;AAAA,EACH,CAAC;AAED,MAAI,CAAC,SAAS,IAAI;AAChB,UAAM,OAAO,MAAM,SAAS,KAAK,EAAE,MAAM,MAAM,EAAE;AACjD,UAAM,IAAI;AAAA,MACR,6BAA6B,QAAQ,MAAM,SAAS,MAAM,IAAI,SAAS,UAAU,GAAG,OAAO,MAAM,IAAI,KAAK,EAAE;AAAA,IAC9G;AAAA,EACF;AAEA,QAAM,OAAQ,MAAM,SAAS,KAAK,EAAE,MAAM,OAAO,CAAC,EAAE;AACpD,QAAM,YAAY,MAAM,YAAY,OAAO,KAAK,SAAS,IAAI,WAAW,KAAK;AAE7E,SAAO;AAAA,IACL;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,EACF;AACF;AAWA,eAAsB,cACpB,SACA,OACA,KACiB;AACjB,SAAO,QAAQ,EAAE,OAAO,KAAK,OAAO,CAAC,EAAE,CAAC;AAC1C;AAUA,eAAsB,cACpB,SACA,cACA,UAA2B,CAAC,GAC5B,KAC8B;AAC9B,QAAM,WAAW,QAAQ;AACzB,MAAI,CAAC,UAAU,cAAc;AAC3B,UAAM,IAAI;AAAA,MACR;AAAA,IAEF;AAAA,EACF;AACA,QAAM,EAAE,cAAc,eAAe,IAAI;AACzC,QAAM,QAAQ,aAAa,OAAO;AAElC,MAAI,CAAC,OAAO;AACV,UAAM,IAAI,MAAM,iBAAiB,OAAO,yBAAyB;AAAA,EACnE;AAEA,MAAI,CAAC,MAAM,SAAS,MAAM,MAAM,WAAW,GAAG;AAC5C,UAAM,IAAI,MAAM,iBAAiB,OAAO,wBAAwB;AAAA,EAClE;AAEA,QAAM,YAAY;AAClB,QAAM,YAAY,MAAM,MAAM,SAAS;AACvC,QAAM,gBAAgB,UAAU;AAEhC,MAAI,CAAC,eAAe;AAClB,UAAM,IAAI;AAAA,MACR,iBAAiB,OAAO;AAAA,IAC1B;AAAA,EACF;AAMA,MAAI,aAAsB;AAC1B,MAAI,UAAU,oBAAoB,OAAO,mBAAmB,YAAY;AACtE,iBAAa,MAAM;AAAA,MACjB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,QAAM,QACJ,QAAQ,SACR,OAAO,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,OAAO,GAAG,CAAC,CAAC;AAE9D,QAAM,eAAiD;AAAA,IACrD,IAAI;AAAA,IACJ;AAAA,IACA;AAAA,IACA,YAAY;AAAA,EACd;AAGA,MAAI,QAAQ,kBAAkB;AAC5B,QAAI;AACF,YAAM,QAAQ,iBAAiB;AAAA,QAC7B,YAAY;AAAA,QACZ;AAAA,QACA,WAAW,EAAE,UAAU,eAAe,aAAa,MAAM;AAAA,QACzD,UAAU,QAAQ;AAAA,MACpB,CAAC;AAAA,IACH,SAAS,KAAU;AACjB,cAAQ,KAAK,4CAA4C,KAAK,WAAW,GAAG;AAAA,IAC9E;AAAA,EACF;AAGA,QAAM,uBACJ,eAAe,QAAQ,OAAO,eAAe,WACxC,aACD,EAAE,OAAO,WAAW;AAE1B,QAAM,iBAAiB;AAAA,IACrB,GAAG;AAAA,IACH,eAAe;AAAA,EACjB;AAEA,QAAM,oBAAoB;AAAA,IACxB,GAAI,QAAQ,YAAY,CAAC;AAAA,IACzB,eAAe;AAAA,EACjB;AAEA,QAAM,aAAa,qBAAqB;AACxC,QAAM,oBAAoB,MAAM,iBAAiB,GAAG,IAAI,CAAC;AAEzD,QAAM,cAAc;AAAA,IAClB,UAAU;AAAA,IACV;AAAA,IACA,OAAO;AAAA,IACP,SAAS;AAAA,IACT,YAAY,QAAQ;AAAA,IACpB,UAAU;AAAA,IACV,YAAW,oBAAI,KAAK,GAAE,YAAY;AAAA,EACpC;AAEA,QAAM,UAAkC;AAAA,IACtC,gBAAgB;AAAA,EAClB;AACA,QAAM,aAAa,QAAQ,IAAI;AAC/B,MAAI,YAAY;AACd,YAAQ,uBAAuB,IAAI;AAAA,EACrC;AAEA,QAAM,WAAW,MAAM,MAAM,YAAY;AAAA,IACvC,QAAQ;AAAA,IACR;AAAA,IACA,MAAM,KAAK,UAAU;AAAA,MACnB,UAAU;AAAA,MACV,MAAM;AAAA,IACR,CAAC;AAAA,EACH,CAAC;AAED,MAAI,CAAC,SAAS,IAAI;AAChB,UAAM,OAAO,MAAM,SAAS,KAAK,EAAE,MAAM,MAAM,EAAE;AACjD,UAAM,IAAI;AAAA,MACR,4BAA4B,OAAO,cAAc,aAAa,OACzD,SAAS,MAAM,IAAI,SAAS,UAAU,GAAG,OAAO,MAAM,IAAI,KAAK,EAAE;AAAA,IACxE;AAAA,EACF;AAEA,QAAM,OAAQ,MAAM,SAAS,KAAK,EAAE,MAAM,OAAO,CAAC,EAAE;AACpD,QAAM,YAAY,MAAM,YAAY,OAAO,KAAK,SAAS,IAAI,WAAW,KAAK;AAE7E,SAAO;AAAA,IACL;AAAA,IACA;AAAA,IACA,QAAQ;AAAA,IACR;AAAA,EACF;AACF;","names":[]}