import { Co as withRemoteHttpResponse, Qo as uploadBatchJsonlFile, fs as postJsonWithRetry, os as EmbeddingBatchExecutionParams } from "../../types-CM3a6Vd6.js";
import { t as EmbeddingBatchStatus } from "../../memory-core-host-engine-embeddings-mqN4TUVX.js";
import { n as VoyageEmbeddingClient } from "../../embedding-provider-j3YkE03U.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;
  uploadBatchJsonlFile: typeof uploadBatchJsonlFile;
  withRemoteHttpResponse: typeof withRemoteHttpResponse;
};
declare function fetchVoyageBatchStatus(params: {
  client: VoyageEmbeddingClient;
  batchId: string;
  deps: VoyageBatchDeps;
  maxResponseBytes?: number;
}): Promise<VoyageBatchStatus>;
declare function readVoyageBatchError(params: {
  client: VoyageEmbeddingClient;
  errorFileId: string;
  deps: VoyageBatchDeps;
  maxResponseBytes?: number;
}): Promise<string | undefined>;
declare function runVoyageEmbeddingBatches(params: {
  client: VoyageEmbeddingClient;
  agentId: string;
  requests: VoyageBatchRequest[];
  deps?: Partial<VoyageBatchDeps>;
} & EmbeddingBatchExecutionParams): Promise<Map<string, number[]>>;
declare const testing: {
  readonly fetchVoyageBatchStatus: typeof fetchVoyageBatchStatus;
  readonly readVoyageBatchError: typeof readVoyageBatchError;
  readonly VOYAGE_BATCH_RESPONSE_MAX_BYTES: number;
};
//#endregion
export { runVoyageEmbeddingBatches, testing };