UNPKG

embeddings-js

Version:

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

22 lines (21 loc) 609 B
import { OpenAIEmbeddings } from '@langchain/openai'; export class AdaEmbeddings { constructor() { Object.defineProperty(this, "model", { enumerable: true, configurable: true, writable: true, value: void 0 }); this.model = new OpenAIEmbeddings({ modelName: 'text-embedding-ada-002', maxConcurrency: 3, maxRetries: 5 }); } getDimensions() { return 1536; } embedDocuments(texts) { return this.model.embedDocuments(texts); } embedQuery(text) { return this.model.embedQuery(text); } }