@roadiehq/rag-ai-backend-embeddings-openai/dist/RoadieOpenAiAugmenter.cjs.js

The OpenAI backend module for the @roadiehq/rag-ai plugin.

'use strict';

var openai = require('@langchain/openai');
var ragAiBackendRetrievalAugmenter = require('@roadiehq/rag-ai-backend-retrieval-augmenter');

class RoadieOpenAiAugmenter extends ragAiBackendRetrievalAugmenter.DefaultVectorAugmentationIndexer {
  constructor(config) {
    const embeddings = new openai.OpenAIEmbeddings({
      configuration: {
        baseURL: config.config.openAiBaseUrl
      },
      openAIApiKey: config.config.openAiApiKey,
      // In Node.js defaults to process.env.OPENAI_API_KEY
      batchSize: config.config.batchSize,
      // Default value if omitted is 512. Max is 2048
      modelName: config.config.modelName ? config.config.modelName : "text-embedding-3-small",
      dimensions: config.config.embeddingsDimensions
    });
    super({ ...config, embeddings });
  }
}

exports.RoadieOpenAiAugmenter = RoadieOpenAiAugmenter;
//# sourceMappingURL=RoadieOpenAiAugmenter.cjs.js.map