import { Embeddings, EmbeddingsParams } from "@langchain/core/embeddings";
//#region src/embeddings.d.ts
interface TogetherAIEmbeddingsParams extends EmbeddingsParams {
  /**
   * The API key to use for the Together AI API.
   * @default {process.env.TOGETHER_AI_API_KEY}
   */
  apiKey?: string;
  /**
   * Model name to use.
   * Alias for `model`.
   * @default {"togethercomputer/m2-bert-80M-8k-retrieval"}
   */
  modelName?: string;
  /**
   * Model name to use.
   * @default {"togethercomputer/m2-bert-80M-8k-retrieval"}
   */
  model?: string;
  /**
   * Timeout to use when making requests to Together AI.
   */
  timeout?: number;
  /**
   * The maximum number of documents to embed in a single logical batch.
   * @default {512}
   */
  batchSize?: number;
  /**
   * Whether to strip new lines from the input text.
   * @default {false}
   */
  stripNewLines?: boolean;
}
/**
 * Class for generating embeddings using the Together AI embeddings API.
 */
declare class TogetherAIEmbeddings extends Embeddings implements TogetherAIEmbeddingsParams {
  lc_serializable: boolean;
  lc_namespace: string[];
  modelName: string;
  model: string;
  apiKey: string;
  batchSize: number;
  stripNewLines: boolean;
  timeout?: number;
  private embeddingsAPIUrl;
  get lc_secrets(): {
    [key: string]: string;
  } | undefined;
  constructor(fields?: Partial<TogetherAIEmbeddingsParams>);
  private constructHeaders;
  private constructBody;
  embedDocuments(texts: string[]): Promise<number[][]>;
  embedQuery(text: string): Promise<number[]>;
  private embeddingWithRetry;
}
//#endregion
export { TogetherAIEmbeddings, TogetherAIEmbeddingsParams };
//# sourceMappingURL=embeddings.d.cts.map