@xiaohui-wang/mcpadvisor/build/services/common/vector/VectorDB.js

MCP Advisor & Installation - Find the right MCP server for your needs

import { MemoryVectorDB } from './XenovaVectorDB.js';
import logger from '../../../utils/logger.js';
export class VectorDB {
    db;
    _collectionId;
    constructor() {
        this._collectionId = `nacos_mcp_router-collection-${process.pid}`;
        this.db = new MemoryVectorDB({ numDimensions: 384, clearOnStart: true });
        logger.info(`VectorDB initialized with collection ID: ${this._collectionId}`);
    }
    async start() {
        // MemoryVectorDB initialization is done in constructor
        logger.debug('VectorDB start called');
        return Promise.resolve();
    }
    async isReady() {
        // MemoryVectorDB is always ready
        return Promise.resolve(true);
    }
    async getCollectionCount() {
        return this.db.getCount();
    }
    updateData(ids, documents, metadatas) {
        if (!documents)
            return;
        documents.forEach((doc, i) => {
            this.db.add(doc, { id: ids[i], ...(metadatas ? metadatas[i] : {}) });
        });
        this.db.save();
        logger.debug(`Updated vector database with ${documents.length} documents`);
    }
    async query(query, count) {
        logger.debug(`Querying vector database: ${query.substring(0, 50)}...`);
        const results = await this.db.search(query, count);
        const response = {
            ids: [results.map((r) => r.metadata.id)],
            documents: [results.map((r) => r.text)],
            metadatas: [results.map((r) => r.metadata)],
            distances: [results.map((r) => r.distance)],
            included: []
        };
        logger.debug(`Found ${results.length} results for query`);
        return response;
    }
    async get(ids) {
        const all = this.db['metadatas'] || [];
        const found = all.filter((m) => ids.includes(m.id));
        const response = {
            ids,
            documents: found.map((m) => m.text),
            metadatas: found,
            included: []
        };
        logger.debug(`Retrieved ${found.length} documents by ID`);
        return response;
    }
    async clear() {
        this.db.clear();
        logger.info('Vector database cleared');
    }
}
export default VectorDB;