@hazeljs/rag 0.2.0-alpha.1

package/dist/vector-stores/chroma.store.js

@@ -0,0 +1,242 @@
+"use strict";
+/**
+ * ChromaDB Vector Store
+ * Lightweight, embedded vector database
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChromaVectorStore = void 0;
+class ChromaVectorStore {
+    constructor(embeddingProvider, config) {
+        this.collection = null;
+        this.initialized = false;
+        this.embeddingProvider = embeddingProvider;
+        this.collectionName = config.collectionName;
+        // Initialize ChromaDB client
+        // eslint-disable-next-line @typescript-eslint/no-require-imports
+        const { ChromaClient } = require('chromadb');
+        const clientConfig = {};
+        if (config.url) {
+            clientConfig.path = config.url;
+        }
+        if (config.auth) {
+            clientConfig.auth = config.auth;
+        }
+        this.client = new ChromaClient(clientConfig);
+    }
+    async initialize() {
+        try {
+            // Get or create collection
+            this.collection = await this.client.getOrCreateCollection({
+                name: this.collectionName,
+                metadata: { 'hnsw:space': 'cosine' }, // Use cosine similarity
+            });
+            this.initialized = true;
+        }
+        catch (error) {
+            throw new Error(`Failed to initialize ChromaDB collection: ${error}`);
+        }
+    }
+    async addDocuments(documents) {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        const ids = [];
+        const embeddings = [];
+        const metadatas = [];
+        const documentsTexts = [];
+        // Generate embeddings for all documents
+        const texts = documents.map((doc) => doc.content);
+        const generatedEmbeddings = await this.embeddingProvider.embedBatch(texts);
+        for (let i = 0; i < documents.length; i++) {
+            const doc = documents[i];
+            const id = doc.id || this.generateId();
+            ids.push(id);
+            embeddings.push(generatedEmbeddings[i]);
+            documentsTexts.push(doc.content);
+            metadatas.push(doc.metadata || {});
+        }
+        // Add to ChromaDB
+        await this.collection.add({
+            ids,
+            embeddings,
+            documents: documentsTexts,
+            metadatas,
+        });
+        return ids;
+    }
+    async search(query, options) {
+        // Generate embedding for query
+        const queryEmbedding = await this.embeddingProvider.embed(query);
+        return this.searchByVector(queryEmbedding, options);
+    }
+    async searchByVector(embedding, options) {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        const topK = options?.topK || 5;
+        const minScore = options?.minScore;
+        const filter = options?.filter;
+        try {
+            // Build query
+            const queryParams = {
+                queryEmbeddings: [embedding],
+                nResults: topK,
+            };
+            // Add metadata filter if provided
+            if (filter) {
+                queryParams.where = filter;
+            }
+            const response = await this.collection.query(queryParams);
+            // Transform results
+            const results = [];
+            if (response.ids && response.ids[0]) {
+                for (let i = 0; i < response.ids[0].length; i++) {
+                    // ChromaDB returns distances, convert to similarity scores
+                    // For cosine distance: similarity = 1 - distance
+                    const distance = response.distances?.[0]?.[i] || 0;
+                    const score = 1 - distance;
+                    // Apply minimum score filter
+                    if (minScore && score < minScore) {
+                        continue;
+                    }
+                    results.push({
+                        id: response.ids[0][i],
+                        content: response.documents?.[0]?.[i] || '',
+                        metadata: response.metadatas?.[0]?.[i] || {},
+                        score,
+                    });
+                }
+            }
+            return results;
+        }
+        catch (error) {
+            throw new Error(`ChromaDB search failed: ${error}`);
+        }
+    }
+    async deleteDocuments(ids) {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        try {
+            await this.collection.delete({
+                ids,
+            });
+        }
+        catch (error) {
+            throw new Error(`Failed to delete documents: ${error}`);
+        }
+    }
+    async updateDocument(id, document) {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        try {
+            // Get existing document
+            const existing = await this.getDocument(id);
+            if (!existing) {
+                throw new Error(`Document with id ${id} not found`);
+            }
+            // Merge updates
+            const updated = { ...existing, ...document };
+            // Generate new embedding if content changed
+            let embedding = existing.embedding;
+            if (document.content && document.content !== existing.content) {
+                embedding = await this.embeddingProvider.embed(document.content);
+            }
+            // Update in ChromaDB (delete and re-add)
+            await this.collection.delete({ ids: [id] });
+            await this.collection.add({
+                ids: [id],
+                embeddings: [embedding],
+                documents: [updated.content],
+                metadatas: [updated.metadata || {}],
+            });
+        }
+        catch (error) {
+            throw new Error(`Failed to update document: ${error}`);
+        }
+    }
+    async getDocument(id) {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        try {
+            const response = await this.collection.get({
+                ids: [id],
+                include: ['embeddings', 'documents', 'metadatas'],
+            });
+            if (!response.ids || response.ids.length === 0) {
+                return null;
+            }
+            return {
+                id: response.ids[0],
+                content: response.documents?.[0] || '',
+                metadata: response.metadatas?.[0] || {},
+                embedding: response.embeddings?.[0],
+            };
+        }
+        catch {
+            return null;
+        }
+    }
+    async clear() {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        try {
+            // Delete the collection
+            await this.client.deleteCollection({ name: this.collectionName });
+            // Recreate it
+            await this.initialize();
+        }
+        catch (error) {
+            throw new Error(`Failed to clear vector store: ${error}`);
+        }
+    }
+    /**
+     * Get collection statistics
+     */
+    async getStats() {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        try {
+            const count = await this.collection.count();
+            return { count };
+        }
+        catch (error) {
+            throw new Error(`Failed to get stats: ${error}`);
+        }
+    }
+    /**
+     * Peek at first N documents
+     */
+    async peek(limit = 10) {
+        if (!this.collection) {
+            throw new Error('Collection not initialized');
+        }
+        try {
+            const response = await this.collection.peek({ limit });
+            const documents = [];
+            if (response.ids) {
+                for (let i = 0; i < response.ids.length; i++) {
+                    documents.push({
+                        id: response.ids[i],
+                        content: response.documents?.[i] || '',
+                        metadata: response.metadatas?.[i] || {},
+                        embedding: response.embeddings?.[i],
+                    });
+                }
+            }
+            return documents;
+        }
+        catch (error) {
+            throw new Error(`Failed to peek documents: ${error}`);
+        }
+    }
+    generateId() {
+        return `chroma_${Date.now()}_${Math.random().toString(36).substring(2, 15)}`;
+    }
+}
+exports.ChromaVectorStore = ChromaVectorStore;
+//# sourceMappingURL=chroma.store.js.map

package/dist/vector-stores/chroma.store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chroma.store.js","sourceRoot":"","sources":["../../src/vector-stores/chroma.store.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAoBH,MAAa,iBAAiB;IAO5B,YAAY,iBAAoC,EAAE,MAAoB;QAL9D,eAAU,GAAsB,IAAI,CAAC;QAGrC,gBAAW,GAAY,KAAK,CAAC;QAGnC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;QAE5C,6BAA6B;QAC7B,iEAAiE;QACjE,MAAM,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QAE7C,MAAM,YAAY,GAA4B,EAAE,CAAC;QACjD,IAAI,MAAM,CAAC,GAAG,EAAE,CAAC;YACf,YAAY,CAAC,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC;QACjC,CAAC;QACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,YAAY,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QAClC,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,YAAY,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC;YACH,2BAA2B;YAC3B,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;gBACxD,IAAI,EAAE,IAAI,CAAC,cAAc;gBACzB,QAAQ,EAAE,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,wBAAwB;aAC/D,CAAC,CAAC;YAEH,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,6CAA6C,KAAK,EAAE,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAqB;QACtC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,GAAG,GAAa,EAAE,CAAC;QACzB,MAAM,UAAU,GAAe,EAAE,CAAC;QAClC,MAAM,SAAS,GAA8B,EAAE,CAAC;QAChD,MAAM,cAAc,GAAa,EAAE,CAAC;QAEpC,wCAAwC;QACxC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAE3E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAEvC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACb,UAAU,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC;YACxC,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACjC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;QACrC,CAAC;QAED,kBAAkB;QAClB,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YACxB,GAAG;YACH,UAAU;YACV,SAAS,EAAE,cAAc;YACzB,SAAS;SACV,CAAC,CAAC;QAEH,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAsB;QAChD,+BAA+B;QAC/B,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAEjE,OAAO,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,SAAmB,EAAE,OAAsB;QAC9D,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,CAAC;QACnC,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAE/B,IAAI,CAAC;YACH,cAAc;YACd,MAAM,WAAW,GAIb;gBACF,eAAe,EAAE,CAAC,SAAS,CAAC;gBAC5B,QAAQ,EAAE,IAAI;aACf,CAAC;YAEF,kCAAkC;YAClC,IAAI,MAAM,EAAE,CAAC;gBACX,WAAW,CAAC,KAAK,GAAG,MAAM,CAAC;YAC7B,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAE1D,oBAAoB;YACpB,MAAM,OAAO,GAAmB,EAAE,CAAC;YAEnC,IAAI,QAAQ,CAAC,GAAG,IAAI,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC;gBACpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAChD,2DAA2D;oBAC3D,iDAAiD;oBACjD,MAAM,QAAQ,GAAG,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;oBACnD,MAAM,KAAK,GAAG,CAAC,GAAG,QAAQ,CAAC;oBAE3B,6BAA6B;oBAC7B,IAAI,QAAQ,IAAI,KAAK,GAAG,QAAQ,EAAE,CAAC;wBACjC,SAAS;oBACX,CAAC;oBAED,OAAO,CAAC,IAAI,CAAC;wBACX,EAAE,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;wBACtB,OAAO,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;wBAC3C,QAAQ,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;wBAC5C,KAAK;qBACN,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,2BAA2B,KAAK,EAAE,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,GAAa;QACjC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;gBAC3B,GAAG;aACJ,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,+BAA+B,KAAK,EAAE,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,QAA2B;QAC1D,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC;YACH,wBAAwB;YACxB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;YACtD,CAAC;YAED,gBAAgB;YAChB,MAAM,OAAO,GAAG,EAAE,GAAG,QAAQ,EAAE,GAAG,QAAQ,EAAE,CAAC;YAE7C,4CAA4C;YAC5C,IAAI,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;YACnC,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,KAAK,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC9D,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACnE,CAAC;YAED,yCAAyC;YACzC,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBACxB,GAAG,EAAE,CAAC,EAAE,CAAC;gBACT,UAAU,EAAE,CAAC,SAAU,CAAC;gBACxB,SAAS,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;gBAC5B,SAAS,EAAE,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;aACpC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU;QAC1B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBACzC,GAAG,EAAE,CAAC,EAAE,CAAC;gBACT,OAAO,EAAE,CAAC,YAAY,EAAE,WAAW,EAAE,WAAW,CAAC;aAClD,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,QAAQ,CAAC,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC/C,OAAO,IAAI,CAAC;YACd,CAAC;YAED,OAAO;gBACL,EAAE,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;gBACnB,OAAO,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;gBACtC,QAAQ,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;gBACvC,SAAS,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;aACpC,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC;YACH,wBAAwB;YACxB,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;YAElE,cAAc;YACd,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YAC5C,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,KAAK,EAAE,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE;QAC3B,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YACvD,MAAM,SAAS,GAAe,EAAE,CAAC;YAEjC,IAAI,QAAQ,CAAC,GAAG,EAAE,CAAC;gBACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC7C,SAAS,CAAC,IAAI,CAAC;wBACb,EAAE,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;wBACnB,OAAO,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;wBACtC,QAAQ,EAAE,QAAQ,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE;wBACvC,SAAS,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;qBACpC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,EAAE,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,OAAO,UAAU,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;IAC/E,CAAC;CACF;AArRD,8CAqRC"}

package/dist/vector-stores/memory-vector-store.d.ts

@@ -0,0 +1,20 @@
+/**
+ * In-Memory Vector Store
+ * Useful for development and testing
+ */
+import { VectorStore, Document, SearchResult, QueryOptions, EmbeddingProvider } from '../types';
+export declare class MemoryVectorStore implements VectorStore {
+    private documents;
+    private embeddingProvider;
+    constructor(embeddingProvider: EmbeddingProvider);
+    initialize(): Promise<void>;
+    addDocuments(documents: Document[]): Promise<string[]>;
+    search(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    searchByVector(embedding: number[], options?: QueryOptions): Promise<SearchResult[]>;
+    deleteDocuments(ids: string[]): Promise<void>;
+    updateDocument(id: string, document: Partial<Document>): Promise<void>;
+    getDocument(id: string): Promise<Document | null>;
+    clear(): Promise<void>;
+    private generateId;
+}
+//# sourceMappingURL=memory-vector-store.d.ts.map

package/dist/vector-stores/memory-vector-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-vector-store.d.ts","sourceRoot":"","sources":["../../src/vector-stores/memory-vector-store.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAGhG,qBAAa,iBAAkB,YAAW,WAAW;IACnD,OAAO,CAAC,SAAS,CAAoC;IACrD,OAAO,CAAC,iBAAiB,CAAoB;gBAEjC,iBAAiB,EAAE,iBAAiB;IAI1C,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAwBtD,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAK1E,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAwCxF,eAAe,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAM7C,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgBtE,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;IAIjD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B,OAAO,CAAC,UAAU;CAGnB"}

package/dist/vector-stores/memory-vector-store.js

@@ -0,0 +1,94 @@
+"use strict";
+/**
+ * In-Memory Vector Store
+ * Useful for development and testing
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MemoryVectorStore = void 0;
+const similarity_1 = require("../utils/similarity");
+class MemoryVectorStore {
+    constructor(embeddingProvider) {
+        this.documents = new Map();
+        this.embeddingProvider = embeddingProvider;
+    }
+    async initialize() {
+        // No initialization needed for memory store
+    }
+    async addDocuments(documents) {
+        const ids = [];
+        // Generate embeddings for documents
+        const texts = documents.map((doc) => doc.content);
+        const embeddings = await this.embeddingProvider.embedBatch(texts);
+        for (let i = 0; i < documents.length; i++) {
+            const doc = documents[i];
+            const id = doc.id || this.generateId();
+            const embedding = embeddings[i];
+            this.documents.set(id, {
+                ...doc,
+                id,
+                embedding,
+            });
+            ids.push(id);
+        }
+        return ids;
+    }
+    async search(query, options = {}) {
+        const queryEmbedding = await this.embeddingProvider.embed(query);
+        return this.searchByVector(queryEmbedding, options);
+    }
+    async searchByVector(embedding, options = {}) {
+        const { topK = 5, filter = {}, includeMetadata = true, includeEmbedding = false, minScore = 0, } = options;
+        const results = [];
+        for (const [id, doc] of this.documents.entries()) {
+            if (!doc.embedding)
+                continue;
+            // Apply metadata filters
+            if (Object.keys(filter).length > 0) {
+                const matchesFilter = Object.entries(filter).every(([key, value]) => doc.metadata?.[key] === value);
+                if (!matchesFilter)
+                    continue;
+            }
+            // Calculate similarity
+            const score = (0, similarity_1.cosineSimilarity)(embedding, doc.embedding);
+            if (score >= minScore) {
+                results.push({
+                    id,
+                    content: doc.content,
+                    metadata: includeMetadata ? doc.metadata : undefined,
+                    embedding: includeEmbedding ? doc.embedding : undefined,
+                    score,
+                });
+            }
+        }
+        // Sort by score descending and return top K
+        return results.sort((a, b) => b.score - a.score).slice(0, topK);
+    }
+    async deleteDocuments(ids) {
+        for (const id of ids) {
+            this.documents.delete(id);
+        }
+    }
+    async updateDocument(id, document) {
+        const existing = this.documents.get(id);
+        if (!existing) {
+            throw new Error(`Document with id ${id} not found`);
+        }
+        const updated = { ...existing, ...document };
+        // Re-generate embedding if content changed
+        if (document.content && document.content !== existing.content) {
+            updated.embedding = await this.embeddingProvider.embed(document.content);
+        }
+        this.documents.set(id, updated);
+    }
+    async getDocument(id) {
+        return this.documents.get(id) || null;
+    }
+    async clear() {
+        this.documents.clear();
+    }
+    generateId() {
+        return `doc_${Date.now()}_${Math.random().toString(36).substring(2, 9)}`;
+    }
+}
+exports.MemoryVectorStore = MemoryVectorStore;
+//# sourceMappingURL=memory-vector-store.js.map

package/dist/vector-stores/memory-vector-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-vector-store.js","sourceRoot":"","sources":["../../src/vector-stores/memory-vector-store.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAGH,oDAAuD;AAEvD,MAAa,iBAAiB;IAI5B,YAAY,iBAAoC;QAHxC,cAAS,GAA0B,IAAI,GAAG,EAAE,CAAC;QAInD,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;IAC7C,CAAC;IAED,KAAK,CAAC,UAAU;QACd,4CAA4C;IAC9C,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAqB;QACtC,MAAM,GAAG,GAAa,EAAE,CAAC;QAEzB,oCAAoC;QACpC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAElE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,MAAM,GAAG,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;YACzB,MAAM,EAAE,GAAG,GAAG,CAAC,EAAE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACvC,MAAM,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YAEhC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE;gBACrB,GAAG,GAAG;gBACN,EAAE;gBACF,SAAS;aACV,CAAC,CAAC;YAEH,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,UAAwB,EAAE;QACpD,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACjE,OAAO,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,SAAmB,EAAE,UAAwB,EAAE;QAClE,MAAM,EACJ,IAAI,GAAG,CAAC,EACR,MAAM,GAAG,EAAE,EACX,eAAe,GAAG,IAAI,EACtB,gBAAgB,GAAG,KAAK,EACxB,QAAQ,GAAG,CAAC,GACb,GAAG,OAAO,CAAC;QAEZ,MAAM,OAAO,GAAmB,EAAE,CAAC;QAEnC,KAAK,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;YACjD,IAAI,CAAC,GAAG,CAAC,SAAS;gBAAE,SAAS;YAE7B,yBAAyB;YACzB,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnC,MAAM,aAAa,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAK,CAChD,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,KAAK,KAAK,CAChD,CAAC;gBACF,IAAI,CAAC,aAAa;oBAAE,SAAS;YAC/B,CAAC;YAED,uBAAuB;YACvB,MAAM,KAAK,GAAG,IAAA,6BAAgB,EAAC,SAAS,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;YAEzD,IAAI,KAAK,IAAI,QAAQ,EAAE,CAAC;gBACtB,OAAO,CAAC,IAAI,CAAC;oBACX,EAAE;oBACF,OAAO,EAAE,GAAG,CAAC,OAAO;oBACpB,QAAQ,EAAE,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;oBACpD,SAAS,EAAE,gBAAgB,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;oBACvD,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,4CAA4C;QAC5C,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,GAAa;QACjC,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;YACrB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,QAA2B;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,GAAG,QAAQ,EAAE,GAAG,QAAQ,EAAE,CAAC;QAE7C,2CAA2C;QAC3C,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,KAAK,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC9D,OAAO,CAAC,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,OAAmB,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU;QAC1B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAEO,UAAU;QAChB,OAAO,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAC3E,CAAC;CACF;AAlHD,8CAkHC"}

package/dist/vector-stores/pinecone.store.d.ts

@@ -0,0 +1,34 @@
+/**
+ * Pinecone Vector Store
+ * Production-ready vector database integration
+ */
+import { VectorStore, Document, SearchResult, QueryOptions } from '../types';
+import { EmbeddingProvider } from '../types';
+export interface PineconeConfig {
+    apiKey: string;
+    environment: string;
+    indexName: string;
+    namespace?: string;
+    textKey?: string;
+    metadataKey?: string;
+}
+export declare class PineconeVectorStore implements VectorStore {
+    private client;
+    private index;
+    private embeddingProvider;
+    private namespace;
+    private textKey;
+    private metadataKey;
+    private initialized;
+    constructor(embeddingProvider: EmbeddingProvider, config: PineconeConfig);
+    initialize(): Promise<void>;
+    addDocuments(documents: Document[]): Promise<string[]>;
+    search(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    searchByVector(embedding: number[], options?: QueryOptions): Promise<SearchResult[]>;
+    deleteDocuments(ids: string[]): Promise<void>;
+    updateDocument(id: string, document: Partial<Document>): Promise<void>;
+    getDocument(id: string): Promise<Document | null>;
+    clear(): Promise<void>;
+    private generateId;
+}
+//# sourceMappingURL=pinecone.store.d.ts.map

package/dist/vector-stores/pinecone.store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pinecone.store.d.ts","sourceRoot":"","sources":["../../src/vector-stores/pinecone.store.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAQ7C,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,qBAAa,mBAAoB,YAAW,WAAW;IACrD,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,iBAAiB,CAAoB;IAC7C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,WAAW,CAAkB;gBAEzB,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,EAAE,cAAc;IAiBlE,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAW3B,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAgCtD,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAOtE,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAgCpF,eAAe,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAI7C,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IA6BtE,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;IAoBjD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAI5B,OAAO,CAAC,UAAU;CAGnB"}

package/dist/vector-stores/pinecone.store.js

@@ -0,0 +1,146 @@
+"use strict";
+/**
+ * Pinecone Vector Store
+ * Production-ready vector database integration
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PineconeVectorStore = void 0;
+class PineconeVectorStore {
+    constructor(embeddingProvider, config) {
+        this.initialized = false;
+        this.embeddingProvider = embeddingProvider;
+        this.namespace = config.namespace || '';
+        this.textKey = config.textKey || 'text';
+        this.metadataKey = config.metadataKey || 'metadata';
+        // Initialize Pinecone client
+        // eslint-disable-next-line @typescript-eslint/no-require-imports
+        const { Pinecone } = require('@pinecone-database/pinecone');
+        this.client = new Pinecone({
+            apiKey: config.apiKey,
+            environment: config.environment,
+        });
+        this.index = this.client.index(config.indexName);
+    }
+    async initialize() {
+        // Pinecone doesn't require explicit initialization
+        // but we can verify the index exists
+        try {
+            await this.index.describeIndexStats();
+            this.initialized = true;
+        }
+        catch (error) {
+            throw new Error(`Failed to initialize Pinecone index: ${error}`);
+        }
+    }
+    async addDocuments(documents) {
+        const ids = [];
+        // Generate embeddings for all documents
+        const texts = documents.map((doc) => doc.content);
+        const embeddings = await this.embeddingProvider.embedBatch(texts);
+        // Prepare vectors for upsert
+        const vectors = documents.map((doc, idx) => {
+            const id = this.generateId();
+            ids.push(id);
+            return {
+                id,
+                values: embeddings[idx],
+                metadata: {
+                    [this.textKey]: doc.content,
+                    [this.metadataKey]: doc.metadata || {},
+                },
+            };
+        });
+        // Upsert to Pinecone in batches
+        const batchSize = 100;
+        for (let i = 0; i < vectors.length; i += batchSize) {
+            const batch = vectors.slice(i, i + batchSize);
+            await this.index.namespace(this.namespace).upsert(batch);
+        }
+        return ids;
+    }
+    async search(query, options) {
+        // Generate embedding for query
+        const queryEmbedding = await this.embeddingProvider.embed(query);
+        return this.searchByVector(queryEmbedding, options);
+    }
+    async searchByVector(embedding, options) {
+        const topK = options?.topK || 5;
+        const minScore = options?.minScore;
+        const filter = options?.filter;
+        // Query Pinecone
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const queryOptions = {
+            vector: embedding,
+            topK,
+            includeMetadata: true,
+        };
+        if (filter && Object.keys(filter).length > 0) {
+            queryOptions.filter = filter;
+        }
+        const queryResponse = await this.index.namespace(this.namespace).query(queryOptions);
+        // Transform results
+        const results = queryResponse.matches
+            .filter((match) => !minScore || (match.score ?? 0) >= minScore)
+            .map((match) => ({
+            id: match.id,
+            content: match.metadata[this.textKey],
+            metadata: match.metadata[this.metadataKey],
+            score: match.score,
+        }));
+        return results;
+    }
+    async deleteDocuments(ids) {
+        await this.index.namespace(this.namespace).deleteMany(ids);
+    }
+    async updateDocument(id, document) {
+        // Get existing document
+        const existing = await this.getDocument(id);
+        if (!existing) {
+            throw new Error(`Document with id ${id} not found`);
+        }
+        // Merge updates
+        const updated = { ...existing, ...document };
+        // Generate new embedding if content changed
+        let embedding = existing.embedding;
+        if (document.content && document.content !== existing.content) {
+            embedding = await this.embeddingProvider.embed(document.content);
+        }
+        // Upsert updated document
+        await this.index.namespace(this.namespace).upsert([
+            {
+                id,
+                values: embedding,
+                metadata: {
+                    [this.textKey]: updated.content,
+                    [this.metadataKey]: updated.metadata || {},
+                },
+            },
+        ]);
+    }
+    async getDocument(id) {
+        try {
+            const response = await this.index.namespace(this.namespace).fetch([id]);
+            const vector = response.vectors[id];
+            if (!vector) {
+                return null;
+            }
+            return {
+                id,
+                content: vector.metadata[this.textKey],
+                metadata: vector.metadata[this.metadataKey],
+                embedding: vector.values,
+            };
+        }
+        catch {
+            return null;
+        }
+    }
+    async clear() {
+        await this.index.namespace(this.namespace).deleteAll();
+    }
+    generateId() {
+        return `doc_${Date.now()}_${Math.random().toString(36).substring(2, 15)}`;
+    }
+}
+exports.PineconeVectorStore = PineconeVectorStore;
+//# sourceMappingURL=pinecone.store.js.map

package/dist/vector-stores/pinecone.store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pinecone.store.js","sourceRoot":"","sources":["../../src/vector-stores/pinecone.store.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAoBH,MAAa,mBAAmB;IAS9B,YAAY,iBAAoC,EAAE,MAAsB;QAFhE,gBAAW,GAAY,KAAK,CAAC;QAGnC,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC;QACxC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,IAAI,UAAU,CAAC;QAEpD,6BAA6B;QAC7B,iEAAiE;QACjE,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,6BAA6B,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,GAAG,IAAI,QAAQ,CAAC;YACzB,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,mDAAmD;QACnD,qCAAqC;QACrC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,CAAC;YACtC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,wCAAwC,KAAK,EAAE,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,SAAqB;QACtC,MAAM,GAAG,GAAa,EAAE,CAAC;QAEzB,wCAAwC;QACxC,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAClD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAElE,6BAA6B;QAC7B,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACzC,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YAC7B,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAEb,OAAO;gBACL,EAAE;gBACF,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC;gBACvB,QAAQ,EAAE;oBACR,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,OAAO;oBAC3B,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,GAAG,CAAC,QAAQ,IAAI,EAAE;iBACvC;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,gCAAgC;QAChC,MAAM,SAAS,GAAG,GAAG,CAAC;QACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAC9C,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,OAAsB;QAChD,+BAA+B;QAC/B,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAEjE,OAAO,IAAI,CAAC,cAAc,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,SAAmB,EAAE,OAAsB;QAC9D,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,CAAC;QACnC,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAE/B,iBAAiB;QACjB,8DAA8D;QAC9D,MAAM,YAAY,GAAQ;YACxB,MAAM,EAAE,SAAS;YACjB,IAAI;YACJ,eAAe,EAAE,IAAI;SACtB,CAAC;QAEF,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC7C,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC;QAC/B,CAAC;QAED,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAErF,oBAAoB;QACpB,MAAM,OAAO,GAAmB,aAAa,CAAC,OAAO;aAClD,MAAM,CAAC,CAAC,KAAyB,EAAE,EAAE,CAAC,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,QAAQ,CAAC;aAClF,GAAG,CAAC,CAAC,KAAuE,EAAE,EAAE,CAAC,CAAC;YACjF,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;YACrC,QAAQ,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC;YAC1C,KAAK,EAAE,KAAK,CAAC,KAAK;SACnB,CAAC,CAAC,CAAC;QAEN,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,GAAa;QACjC,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;IAC7D,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,EAAU,EAAE,QAA2B;QAC1D,wBAAwB;QACxB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;QACtD,CAAC;QAED,gBAAgB;QAChB,MAAM,OAAO,GAAG,EAAE,GAAG,QAAQ,EAAE,GAAG,QAAQ,EAAE,CAAC;QAE7C,4CAA4C;QAC5C,IAAI,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC;QACnC,IAAI,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,KAAK,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC9D,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACnE,CAAC;QAED,0BAA0B;QAC1B,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;YAChD;gBACE,EAAE;gBACF,MAAM,EAAE,SAAU;gBAClB,QAAQ,EAAE;oBACR,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,OAAO;oBAC/B,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,OAAO,CAAC,QAAQ,IAAI,EAAE;iBAC3C;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,EAAU;QAC1B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YACxE,MAAM,MAAM,GAAG,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAEpC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO,IAAI,CAAC;YACd,CAAC;YAED,OAAO;gBACL,EAAE;gBACF,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;gBACtC,QAAQ,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC;gBAC3C,SAAS,EAAE,MAAM,CAAC,MAAM;aACzB,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,SAAS,EAAE,CAAC;IACzD,CAAC;IAEO,UAAU;QAChB,OAAO,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;IAC5E,CAAC;CACF;AAxKD,kDAwKC"}

package/dist/vector-stores/qdrant.store.d.ts

@@ -0,0 +1,33 @@
+/**
+ * Qdrant Vector Store
+ * High-performance Rust-based vector database integration
+ */
+import { VectorStore, Document, SearchResult, QueryOptions } from '../types';
+import { EmbeddingProvider } from '../types';
+export interface QdrantConfig {
+    url: string;
+    apiKey?: string;
+    collectionName: string;
+    vectorSize?: number;
+    /** Skip client-server version check (avoids warning when Qdrant is unreachable). Default: true */
+    checkCompatibility?: boolean;
+}
+export declare class QdrantVectorStore implements VectorStore {
+    private client;
+    private embeddingProvider;
+    private collectionName;
+    private vectorSize;
+    private initialized;
+    constructor(embeddingProvider: EmbeddingProvider, config: QdrantConfig);
+    initialize(): Promise<void>;
+    addDocuments(documents: Document[]): Promise<string[]>;
+    search(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    searchByVector(embedding: number[], options?: QueryOptions): Promise<SearchResult[]>;
+    deleteDocuments(ids: string[]): Promise<void>;
+    updateDocument(id: string, document: Partial<Document>): Promise<void>;
+    getDocument(id: string): Promise<Document | null>;
+    clear(): Promise<void>;
+    private buildFilter;
+    private generateId;
+}
+//# sourceMappingURL=qdrant.store.d.ts.map

package/dist/vector-stores/qdrant.store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"qdrant.store.d.ts","sourceRoot":"","sources":["../../src/vector-stores/qdrant.store.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAM7C,MAAM,WAAW,YAAY;IAC3B,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,kGAAkG;IAClG,kBAAkB,CAAC,EAAE,OAAO,CAAC;CAC9B;AAED,qBAAa,iBAAkB,YAAW,WAAW;IACnD,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,iBAAiB,CAAoB;IAC7C,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,WAAW,CAAkB;gBAEzB,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,EAAE,YAAY;IAehE,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAwB3B,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA+BtD,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAOtE,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAmCpF,eAAe,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAO7C,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgCtE,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;IAwBjD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAM5B,OAAO,CAAC,WAAW;IAgBnB,OAAO,CAAC,UAAU;CAGnB"}