import { Jt as uploadBatchJsonlFile, ln as postJsonWithRetry, rn as EmbeddingBatchExecutionParams, sn as EmbeddingBatchStatus, yt as withRemoteHttpResponse } from "../../types-CaqpvD9U.js";
import { n as VoyageEmbeddingClient } from "../../embedding-provider-CKGXc9m-.js";

//#region extensions/voyage/embedding-batch.d.ts
/**
 * Voyage Batch API Input Line format.
 * See: https://docs.voyageai.com/docs/batch-inference
 */
type VoyageBatchRequest = {
  custom_id: string;
  body: {
    input: string | string[];
  };
};
type VoyageBatchStatus = EmbeddingBatchStatus;
type VoyageBatchDeps = {
  now: () => number;
  sleep: (ms: number) => Promise<void>;
  postJsonWithRetry: typeof postJsonWithRetry<VoyageBatchStatus>;
  uploadBatchJsonlFile: typeof uploadBatchJsonlFile;
  withRemoteHttpResponse: typeof withRemoteHttpResponse;
};
declare function runVoyageEmbeddingBatches(params: {
  client: VoyageEmbeddingClient;
  agentId: string;
  requests: VoyageBatchRequest[];
  deps?: Partial<VoyageBatchDeps>;
} & EmbeddingBatchExecutionParams): Promise<Map<string, number[]>>;
//#endregion
export { runVoyageEmbeddingBatches };