embeddings-js/dist/embeddings/cohere-embeddings.js

A NodeJS RAG framework to easily work with LLMs and custom datasets

import { CohereEmbeddings as LangChainCohereEmbeddings } from '@langchain/cohere';
export class CohereEmbeddings {
    constructor() {
        Object.defineProperty(this, "model", {
            enumerable: true,
            configurable: true,
            writable: true,
            value: void 0
        });
        this.model = new LangChainCohereEmbeddings({
            maxConcurrency: 3,
            maxRetries: 5,
        });
    }
    getDimensions() {
        return 4096;
    }
    embedDocuments(texts) {
        return this.model.embedDocuments(texts);
    }
    embedQuery(text) {
        return this.model.embedQuery(text);
    }
}