vecstore-js/dist/vecstore.js

A pluggable, browser-native vector database using IndexedDB with support for HNSW and local embeddings.

import { IDBStorageAdapter } from './adapters/idb.js'; // fallback if no adapter provided
import { CosineSearchAlgorithm } from './search/cosine.js';
export class VecStore {
    constructor(options) {
        this.embedder = options.embedder;
        this.store = options.store || new IDBStorageAdapter(options.dbName || 'vecstore');
        this.search = options.search || new CosineSearchAlgorithm();
        this.storeContent = options.storeContent ?? true; // Default to true for backward compatibility
    }
    async addDocument(id, content, metadata) {
        // For now, assume content is a string for embedding
        // Future: embedders can handle different content types
        const vector = await this.embedder.embed(content);
        const doc = {
            id,
            vector,
            ...(this.storeContent && { content }), // Conditionally include content
            metadata
        };
        await this.store.put(doc);
        // If search algorithm supports indexing, add to index immediately
        if (this.isIndexedSearch(this.search)) {
            await this.search.addDocument(doc);
        }
    }
    async query(queryContent, topK = 5) {
        // For now, assume queryContent is a string for embedding
        const queryVec = await this.embedder.embed(queryContent);
        // Use fast path if available (indexed search)
        if (this.isIndexedSearch(this.search)) {
            return this.search.searchIndex(queryVec, topK);
        }
        // Fallback to slow path (load all documents)
        const allDocs = await this.store.getAll();
        return this.search.search(queryVec, allDocs, topK);
    }
    // Type guard to check if search algorithm supports indexing
    isIndexedSearch(search) {
        return 'addDocument' in search && 'searchIndex' in search && 'initialize' in search;
    }
    // Initialize the search algorithm if it supports indexing
    async initialize() {
        if (this.isIndexedSearch(this.search)) {
            await this.search.initialize();
            // Load existing documents into the index
            const existingDocs = await this.store.getAll();
            for (const doc of existingDocs) {
                await this.search.addDocument(doc);
            }
        }
    }
}