UNPKG

@roadiehq/rag-ai-backend-embeddings-openai

Version:

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

37 lines (33 loc) 1.11 kB
'use strict'; var RoadieOpenAiAugmenter = require('./RoadieOpenAiAugmenter.cjs.js'); async function initializeOpenAiEmbeddings({ logger, tokenManager, auth, vectorStore, catalogApi, discovery, config }) { logger.info("Initializing Roadie OpenAI Embeddings"); const openAiConfig = config.get("ai.embeddings.openai"); const embeddingsOptions = config.getOptionalConfig("ai.embeddings"); const augmentationOptions = {}; if (embeddingsOptions) { augmentationOptions.chunkSize = embeddingsOptions.getOptionalNumber("chunkSize"); augmentationOptions.chunkOverlap = embeddingsOptions.getOptionalNumber("chunkOverlap"); augmentationOptions.concurrencyLimit = embeddingsOptions.getOptionalNumber("concurrencyLimit"); } return new RoadieOpenAiAugmenter.RoadieOpenAiAugmenter({ vectorStore, catalogApi, discovery, augmentationOptions, logger: logger.child({ label: "roadie-openai-embeddings" }), tokenManager, auth, config: openAiConfig }); } exports.initializeOpenAiEmbeddings = initializeOpenAiEmbeddings; //# sourceMappingURL=index.cjs.js.map