import { j as ProviderBatchOutputLine } from "../../memory-core-host-engine-embeddings-r8ihp70f.js";
import { n as OpenAiEmbeddingClient } from "../../embedding-provider-a1uT4q_-.js";

//#region extensions/openai/embedding-batch.d.ts
type EmbeddingBatchExecutionParams = {
  wait: boolean;
  pollIntervalMs: number;
  timeoutMs: number;
  concurrency: number;
  debug?: (message: string, data?: Record<string, unknown>) => void;
};
type OpenAiBatchRequest = {
  custom_id: string;
  method: "POST";
  url: "/v1/embeddings";
  body: {
    model: string;
    input: string;
  };
};
type OpenAiBatchOutputLine = ProviderBatchOutputLine;
declare const OPENAI_BATCH_ENDPOINT = "/v1/embeddings";
declare function parseOpenAiBatchOutput(text: string): OpenAiBatchOutputLine[];
declare function runOpenAiEmbeddingBatches(params: {
  openAi: OpenAiEmbeddingClient;
  agentId: string;
  requests: OpenAiBatchRequest[];
  maxJsonlBytes?: number;
} & EmbeddingBatchExecutionParams): Promise<Map<string, number[]>>;
//#endregion
export { OPENAI_BATCH_ENDPOINT, parseOpenAiBatchOutput, runOpenAiEmbeddingBatches };