Function pollForCompletion

server/src/utils/batch.ts:114–158

* Poll a batch until processing_status is 'ended' or timeout.

(
  client: Anthropic,
  batchId: string,
  options: { pollIntervalMs: number; timeoutMs: number; operationName: string },
)

 * Poll a batch until processing_status is 'ended' or timeout.
 */
async function pollForCompletion(
  client: Anthropic,
  batchId: string,
  options: { pollIntervalMs: number; timeoutMs: number; operationName: string },
): Promise<MessageBatch> {
  const deadline = Date.now() + options.timeoutMs;

  while (true) {
    const batch = await client.messages.batches.retrieve(batchId);

    if (batch.processing_status === 'ended') {
      logger.info(
        {
          batchId,
          operation: options.operationName,
          counts: batch.request_counts,
        },
        'Batch processing ended',
      );
      return batch;
    }

    if (Date.now() >= deadline) {
      break;
    }

    const remaining = batch.request_counts.processing;
    const succeeded = batch.request_counts.succeeded;
    logger.debug(
      { batchId, processing: remaining, succeeded, operation: options.operationName },
      'Batch still processing',
    );

    await sleep(Math.min(options.pollIntervalMs, deadline - Date.now()));
  }

  // Timeout — cancel and throw
  logger.warn({ batchId, operation: options.operationName }, 'Batch timed out, canceling');
  try {
    await client.messages.batches.cancel(batchId);
  } catch (cancelErr) {
    logger.warn({ batchId, error: cancelErr }, 'Failed to cancel timed-out batch');
  }
  throw new Error(`Batch ${batchId} timed out after ${options.timeoutMs}ms`);
}

/**
 * Stream JSONL results from a completed batch and return as a Map.