import { Embeddings, EmbeddingsParams } from 'langchain/embeddings/base';
import { LLama as LLM } from '../index.js';
import '../llm/type.js';

declare class LLamaEmbeddings extends Embeddings {
    llm: LLM;
    constructor(params: EmbeddingsParams, llm: LLM);
    embedDocuments(documents: string[]): Promise<number[][]>;
    embedQuery(document: string): Promise<number[]>;
}

export { LLamaEmbeddings };