@roadiehq/rag-ai-storage-pgvector
Version:
30 lines (26 loc) • 895 B
JavaScript
;
var migrations = require('../database/migrations.cjs.js');
var RoadiePgVectorStore = require('./RoadiePgVectorStore.cjs.js');
async function createRoadiePgVectorStore({
logger,
database,
config
}) {
logger.info("Starting Roadie PgVectorStore");
const dbClient = await database.getClient();
await migrations.applyDatabaseMigrations(dbClient);
const pgVectorConfig = config.getOptionalConfig("ai.storage.pgVector");
const options = {};
if (pgVectorConfig) {
options.amount = pgVectorConfig.getOptionalNumber("amount");
options.chunkSize = pgVectorConfig.getOptionalNumber("chunkSize");
}
return RoadiePgVectorStore.RoadiePgVectorStore.initialize({
logger,
db: dbClient,
chunkSize: options?.chunkSize,
amount: options?.amount
});
}
exports.createRoadiePgVectorStore = createRoadiePgVectorStore;
//# sourceMappingURL=index.cjs.js.map