@vectorstores/mongodb 0.1.2 → 0.1.4

package/dist/index.cjs

@@ -1,12 +1,8 @@
 Object.defineProperty(exports, '__esModule', { value: true });
 
 var core = require('@vectorstores/core');
-var mongodb = require('mongodb');
 var env = require('@vectorstores/env');
-
-var version = "0.1.2";
-var pkg = {
-	version: version};
+var mongodb = require('mongodb');
 
 const DEFAULT_DB_NAME = "KVStoreDB";
 const DEFAULT_COLLECTION_NAME = "KVStoreCollection";
@@ -75,61 +71,9 @@ class MongoKVStore extends core.BaseKVStore {
     }
 }
 
-const DEFAULT_DATABASE = "DocumentStoreDB";
-const DEFAULT_COLLECTION = "DocumentStoreCollection";
-class MongoDocumentStore extends core.KVDocumentStore {
-    constructor({ mongoKVStore, namespace }){
-        super(mongoKVStore, namespace);
-    }
-    /**
-   * Static method for creating an instance using a MongoClient.
-   * @returns Instance of MongoDBDocumentStore
-   * @param mongoClient - MongoClient instance
-   * @param dbName - Database name
-   * @param collectionName - Collection name
-   * @example
-   * ```ts
-   * const mongoClient = new MongoClient("mongodb://localhost:27017");
-   * const documentStore = MongoDBDocumentStore.fromMongoClient(mongoClient, "my_db", "my_collection");
-   * ```
-   */ static fromMongoClient(mongoClient, dbName = DEFAULT_DATABASE, collectionName = DEFAULT_COLLECTION) {
-        const mongoKVStore = new MongoKVStore({
-            mongoClient,
-            dbName
-        });
-        mongoClient.appendMetadata({
-            name: "VECTORSTORES_MONGODB_DOC_STORE",
-            version: pkg.version
-        });
-        return new MongoDocumentStore({
-            mongoKVStore,
-            namespace: `${dbName}.${collectionName}`
-        });
-    }
-    /**
-   * Static method for creating an instance using a connection string.
-   * @returns Instance of MongoDBDocumentStore
-   * @param connectionString - MongoDB connection string
-   * @param dbName - Database name
-   * @param collectionName - Collection name
-   * @example
-   * ```ts
-   * const documentStore = MongoDBDocumentStore.fromConnectionString("mongodb://localhost:27017", "my_db", "my_collection");
-   * ```
-   */ static fromConnectionString(connectionString, dbName = DEFAULT_DATABASE, collectionName = DEFAULT_COLLECTION) {
-        const mongoClient = new mongodb.MongoClient(connectionString, {
-            appName: "VECTORSTORES_JS"
-        });
-        const mongoKVStore = new MongoKVStore({
-            mongoClient,
-            dbName
-        });
-        return new MongoDocumentStore({
-            mongoKVStore,
-            namespace: `${dbName}.${collectionName}`
-        });
-    }
-}
+var version = "0.1.4";
+var pkg = {
+	version: version};
 
 // define your Atlas Search index. See detail https://www.mongodb.com/docs/atlas/atlas-search/field-types/knn-vector/
 const DEFAULT_EMBEDDING_DEFINITION = {
@@ -335,6 +279,15 @@ function toMongoDBFilter(filters) {
         };
         return result;
     }
+    async exists(refDocId) {
+        const collection = await this.ensureCollection();
+        const count = await collection.countDocuments({
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
+        }, {
+            limit: 1
+        });
+        return count > 0;
+    }
 }
 
 /**
@@ -396,6 +349,5 @@ function toMongoDBFilter(filters) {
 }
 
 exports.MongoDBAtlasVectorSearch = MongoDBAtlasVectorSearch;
-exports.MongoDocumentStore = MongoDocumentStore;
 exports.MongoKVStore = MongoKVStore;
 exports.SimpleMongoReader = SimpleMongoReader;

package/dist/index.d.cts

@@ -1,4 +1,4 @@
-import { BaseKVStore, StoredValue, KVDocumentStore, BaseVectorStore, VectorStoreQuery, BaseEmbedding, VectorStoreBaseParams, BaseNode, VectorStoreQueryResult, BaseReader, Document } from '@vectorstores/core';
+import { BaseKVStore, StoredValue, BaseVectorStore, VectorStoreQuery, BaseEmbedding, VectorStoreBaseParams, BaseNode, VectorStoreQueryResult, BaseReader, Document } from '@vectorstores/core';
 import { MongoClient, BulkWriteOptions, Collection, Document as Document$1, Filter } from 'mongodb';
 
 interface MongoKVStoreConfig {
@@ -17,39 +17,6 @@ declare class MongoKVStore extends BaseKVStore {
     delete(key: string, collectionName?: string): Promise<boolean>;
 }
 
-interface MongoDBDocumentStoreConfig {
-    mongoKVStore: MongoKVStore;
-    namespace?: string;
-}
-declare class MongoDocumentStore extends KVDocumentStore {
-    constructor({ mongoKVStore, namespace }: MongoDBDocumentStoreConfig);
-    /**
-     * Static method for creating an instance using a MongoClient.
-     * @returns Instance of MongoDBDocumentStore
-     * @param mongoClient - MongoClient instance
-     * @param dbName - Database name
-     * @param collectionName - Collection name
-     * @example
-     * ```ts
-     * const mongoClient = new MongoClient("mongodb://localhost:27017");
-     * const documentStore = MongoDBDocumentStore.fromMongoClient(mongoClient, "my_db", "my_collection");
-     * ```
-     */
-    static fromMongoClient(mongoClient: MongoClient, dbName?: string, collectionName?: string): MongoDocumentStore;
-    /**
-     * Static method for creating an instance using a connection string.
-     * @returns Instance of MongoDBDocumentStore
-     * @param connectionString - MongoDB connection string
-     * @param dbName - Database name
-     * @param collectionName - Collection name
-     * @example
-     * ```ts
-     * const documentStore = MongoDBDocumentStore.fromConnectionString("mongodb://localhost:27017", "my_db", "my_collection");
-     * ```
-     */
-    static fromConnectionString(connectionString: string, dbName?: string, collectionName?: string): MongoDocumentStore;
-}
-
 /**
  * Vector store that uses MongoDB Atlas for storage and vector search.
  * This store uses the $vectorSearch aggregation stage to perform vector similarity search.
@@ -144,6 +111,7 @@ declare class MongoDBAtlasVectorSearch extends BaseVectorStore {
      * @returns List of nodes and their similarities
      */
     query(query: VectorStoreQuery, options?: object): Promise<VectorStoreQueryResult>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /**
@@ -173,4 +141,4 @@ declare class SimpleMongoReader implements BaseReader<Document> {
     loadData<TSchema extends Document$1 = Document$1>(dbName: string, collectionName: string, fieldNames?: string[], separator?: string, filterQuery?: Filter<TSchema>, maxDocs?: number, metadataNames?: string[]): Promise<Document[]>;
 }
 
-export { MongoDBAtlasVectorSearch, MongoDocumentStore, MongoKVStore, SimpleMongoReader };
+export { MongoDBAtlasVectorSearch, MongoKVStore, SimpleMongoReader };

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { BaseKVStore, StoredValue, KVDocumentStore, BaseVectorStore, VectorStoreQuery, BaseEmbedding, VectorStoreBaseParams, BaseNode, VectorStoreQueryResult, BaseReader, Document } from '@vectorstores/core';
+import { BaseKVStore, StoredValue, BaseVectorStore, VectorStoreQuery, BaseEmbedding, VectorStoreBaseParams, BaseNode, VectorStoreQueryResult, BaseReader, Document } from '@vectorstores/core';
 import { MongoClient, BulkWriteOptions, Collection, Document as Document$1, Filter } from 'mongodb';
 
 interface MongoKVStoreConfig {
@@ -17,39 +17,6 @@ declare class MongoKVStore extends BaseKVStore {
     delete(key: string, collectionName?: string): Promise<boolean>;
 }
 
-interface MongoDBDocumentStoreConfig {
-    mongoKVStore: MongoKVStore;
-    namespace?: string;
-}
-declare class MongoDocumentStore extends KVDocumentStore {
-    constructor({ mongoKVStore, namespace }: MongoDBDocumentStoreConfig);
-    /**
-     * Static method for creating an instance using a MongoClient.
-     * @returns Instance of MongoDBDocumentStore
-     * @param mongoClient - MongoClient instance
-     * @param dbName - Database name
-     * @param collectionName - Collection name
-     * @example
-     * ```ts
-     * const mongoClient = new MongoClient("mongodb://localhost:27017");
-     * const documentStore = MongoDBDocumentStore.fromMongoClient(mongoClient, "my_db", "my_collection");
-     * ```
-     */
-    static fromMongoClient(mongoClient: MongoClient, dbName?: string, collectionName?: string): MongoDocumentStore;
-    /**
-     * Static method for creating an instance using a connection string.
-     * @returns Instance of MongoDBDocumentStore
-     * @param connectionString - MongoDB connection string
-     * @param dbName - Database name
-     * @param collectionName - Collection name
-     * @example
-     * ```ts
-     * const documentStore = MongoDBDocumentStore.fromConnectionString("mongodb://localhost:27017", "my_db", "my_collection");
-     * ```
-     */
-    static fromConnectionString(connectionString: string, dbName?: string, collectionName?: string): MongoDocumentStore;
-}
-
 /**
  * Vector store that uses MongoDB Atlas for storage and vector search.
  * This store uses the $vectorSearch aggregation stage to perform vector similarity search.
@@ -144,6 +111,7 @@ declare class MongoDBAtlasVectorSearch extends BaseVectorStore {
      * @returns List of nodes and their similarities
      */
     query(query: VectorStoreQuery, options?: object): Promise<VectorStoreQueryResult>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /**
@@ -173,4 +141,4 @@ declare class SimpleMongoReader implements BaseReader<Document> {
     loadData<TSchema extends Document$1 = Document$1>(dbName: string, collectionName: string, fieldNames?: string[], separator?: string, filterQuery?: Filter<TSchema>, maxDocs?: number, metadataNames?: string[]): Promise<Document[]>;
 }
 
-export { MongoDBAtlasVectorSearch, MongoDocumentStore, MongoKVStore, SimpleMongoReader };
+export { MongoDBAtlasVectorSearch, MongoKVStore, SimpleMongoReader };

package/dist/index.edge-light.d.ts

@@ -1,4 +1,4 @@
-import { BaseKVStore, StoredValue, KVDocumentStore, BaseVectorStore, VectorStoreQuery, BaseEmbedding, VectorStoreBaseParams, BaseNode, VectorStoreQueryResult, BaseReader, Document } from '@vectorstores/core';
+import { BaseKVStore, StoredValue, BaseVectorStore, VectorStoreQuery, BaseEmbedding, VectorStoreBaseParams, BaseNode, VectorStoreQueryResult, BaseReader, Document } from '@vectorstores/core';
 import { MongoClient, BulkWriteOptions, Collection, Document as Document$1, Filter } from 'mongodb';
 
 interface MongoKVStoreConfig {
@@ -17,39 +17,6 @@ declare class MongoKVStore extends BaseKVStore {
     delete(key: string, collectionName?: string): Promise<boolean>;
 }
 
-interface MongoDBDocumentStoreConfig {
-    mongoKVStore: MongoKVStore;
-    namespace?: string;
-}
-declare class MongoDocumentStore extends KVDocumentStore {
-    constructor({ mongoKVStore, namespace }: MongoDBDocumentStoreConfig);
-    /**
-     * Static method for creating an instance using a MongoClient.
-     * @returns Instance of MongoDBDocumentStore
-     * @param mongoClient - MongoClient instance
-     * @param dbName - Database name
-     * @param collectionName - Collection name
-     * @example
-     * ```ts
-     * const mongoClient = new MongoClient("mongodb://localhost:27017");
-     * const documentStore = MongoDBDocumentStore.fromMongoClient(mongoClient, "my_db", "my_collection");
-     * ```
-     */
-    static fromMongoClient(mongoClient: MongoClient, dbName?: string, collectionName?: string): MongoDocumentStore;
-    /**
-     * Static method for creating an instance using a connection string.
-     * @returns Instance of MongoDBDocumentStore
-     * @param connectionString - MongoDB connection string
-     * @param dbName - Database name
-     * @param collectionName - Collection name
-     * @example
-     * ```ts
-     * const documentStore = MongoDBDocumentStore.fromConnectionString("mongodb://localhost:27017", "my_db", "my_collection");
-     * ```
-     */
-    static fromConnectionString(connectionString: string, dbName?: string, collectionName?: string): MongoDocumentStore;
-}
-
 /**
  * Vector store that uses MongoDB Atlas for storage and vector search.
  * This store uses the $vectorSearch aggregation stage to perform vector similarity search.
@@ -144,6 +111,7 @@ declare class MongoDBAtlasVectorSearch extends BaseVectorStore {
      * @returns List of nodes and their similarities
      */
     query(query: VectorStoreQuery, options?: object): Promise<VectorStoreQueryResult>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /**
@@ -173,4 +141,4 @@ declare class SimpleMongoReader implements BaseReader<Document> {
     loadData<TSchema extends Document$1 = Document$1>(dbName: string, collectionName: string, fieldNames?: string[], separator?: string, filterQuery?: Filter<TSchema>, maxDocs?: number, metadataNames?: string[]): Promise<Document[]>;
 }
 
-export { MongoDBAtlasVectorSearch, MongoDocumentStore, MongoKVStore, SimpleMongoReader };
+export { MongoDBAtlasVectorSearch, MongoKVStore, SimpleMongoReader };

package/dist/index.edge-light.js

@@ -1,10 +1,6 @@
-import { BaseKVStore, KVDocumentStore, BaseVectorStore, nodeToMetadata, MetadataMode, metadataDictToNode, FilterCondition, Document } from '@vectorstores/core';
-import { MongoClient } from 'mongodb';
+import { BaseKVStore, BaseVectorStore, nodeToMetadata, MetadataMode, metadataDictToNode, FilterCondition, Document } from '@vectorstores/core';
 import { getEnv } from '@vectorstores/env';
-
-var version = "0.1.2";
-var pkg = {
-	version: version};
+import { MongoClient } from 'mongodb';
 
 const DEFAULT_DB_NAME = "KVStoreDB";
 const DEFAULT_COLLECTION_NAME = "KVStoreCollection";
@@ -73,61 +69,9 @@ class MongoKVStore extends BaseKVStore {
     }
 }
 
-const DEFAULT_DATABASE = "DocumentStoreDB";
-const DEFAULT_COLLECTION = "DocumentStoreCollection";
-class MongoDocumentStore extends KVDocumentStore {
-    constructor({ mongoKVStore, namespace }){
-        super(mongoKVStore, namespace);
-    }
-    /**
-   * Static method for creating an instance using a MongoClient.
-   * @returns Instance of MongoDBDocumentStore
-   * @param mongoClient - MongoClient instance
-   * @param dbName - Database name
-   * @param collectionName - Collection name
-   * @example
-   * ```ts
-   * const mongoClient = new MongoClient("mongodb://localhost:27017");
-   * const documentStore = MongoDBDocumentStore.fromMongoClient(mongoClient, "my_db", "my_collection");
-   * ```
-   */ static fromMongoClient(mongoClient, dbName = DEFAULT_DATABASE, collectionName = DEFAULT_COLLECTION) {
-        const mongoKVStore = new MongoKVStore({
-            mongoClient,
-            dbName
-        });
-        mongoClient.appendMetadata({
-            name: "VECTORSTORES_MONGODB_DOC_STORE",
-            version: pkg.version
-        });
-        return new MongoDocumentStore({
-            mongoKVStore,
-            namespace: `${dbName}.${collectionName}`
-        });
-    }
-    /**
-   * Static method for creating an instance using a connection string.
-   * @returns Instance of MongoDBDocumentStore
-   * @param connectionString - MongoDB connection string
-   * @param dbName - Database name
-   * @param collectionName - Collection name
-   * @example
-   * ```ts
-   * const documentStore = MongoDBDocumentStore.fromConnectionString("mongodb://localhost:27017", "my_db", "my_collection");
-   * ```
-   */ static fromConnectionString(connectionString, dbName = DEFAULT_DATABASE, collectionName = DEFAULT_COLLECTION) {
-        const mongoClient = new MongoClient(connectionString, {
-            appName: "VECTORSTORES_JS"
-        });
-        const mongoKVStore = new MongoKVStore({
-            mongoClient,
-            dbName
-        });
-        return new MongoDocumentStore({
-            mongoKVStore,
-            namespace: `${dbName}.${collectionName}`
-        });
-    }
-}
+var version = "0.1.4";
+var pkg = {
+	version: version};
 
 // define your Atlas Search index. See detail https://www.mongodb.com/docs/atlas/atlas-search/field-types/knn-vector/
 const DEFAULT_EMBEDDING_DEFINITION = {
@@ -333,6 +277,15 @@ function toMongoDBFilter(filters) {
         };
         return result;
     }
+    async exists(refDocId) {
+        const collection = await this.ensureCollection();
+        const count = await collection.countDocuments({
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
+        }, {
+            limit: 1
+        });
+        return count > 0;
+    }
 }
 
 /**
@@ -393,4 +346,4 @@ function toMongoDBFilter(filters) {
     }
 }
 
-export { MongoDBAtlasVectorSearch, MongoDocumentStore, MongoKVStore, SimpleMongoReader };
+export { MongoDBAtlasVectorSearch, MongoKVStore, SimpleMongoReader };

package/dist/index.js

@@ -1,10 +1,6 @@
-import { BaseKVStore, KVDocumentStore, BaseVectorStore, nodeToMetadata, MetadataMode, metadataDictToNode, FilterCondition, Document } from '@vectorstores/core';
-import { MongoClient } from 'mongodb';
+import { BaseKVStore, BaseVectorStore, nodeToMetadata, MetadataMode, metadataDictToNode, FilterCondition, Document } from '@vectorstores/core';
 import { getEnv } from '@vectorstores/env';
-
-var version = "0.1.2";
-var pkg = {
-	version: version};
+import { MongoClient } from 'mongodb';
 
 const DEFAULT_DB_NAME = "KVStoreDB";
 const DEFAULT_COLLECTION_NAME = "KVStoreCollection";
@@ -73,61 +69,9 @@ class MongoKVStore extends BaseKVStore {
     }
 }
 
-const DEFAULT_DATABASE = "DocumentStoreDB";
-const DEFAULT_COLLECTION = "DocumentStoreCollection";
-class MongoDocumentStore extends KVDocumentStore {
-    constructor({ mongoKVStore, namespace }){
-        super(mongoKVStore, namespace);
-    }
-    /**
-   * Static method for creating an instance using a MongoClient.
-   * @returns Instance of MongoDBDocumentStore
-   * @param mongoClient - MongoClient instance
-   * @param dbName - Database name
-   * @param collectionName - Collection name
-   * @example
-   * ```ts
-   * const mongoClient = new MongoClient("mongodb://localhost:27017");
-   * const documentStore = MongoDBDocumentStore.fromMongoClient(mongoClient, "my_db", "my_collection");
-   * ```
-   */ static fromMongoClient(mongoClient, dbName = DEFAULT_DATABASE, collectionName = DEFAULT_COLLECTION) {
-        const mongoKVStore = new MongoKVStore({
-            mongoClient,
-            dbName
-        });
-        mongoClient.appendMetadata({
-            name: "VECTORSTORES_MONGODB_DOC_STORE",
-            version: pkg.version
-        });
-        return new MongoDocumentStore({
-            mongoKVStore,
-            namespace: `${dbName}.${collectionName}`
-        });
-    }
-    /**
-   * Static method for creating an instance using a connection string.
-   * @returns Instance of MongoDBDocumentStore
-   * @param connectionString - MongoDB connection string
-   * @param dbName - Database name
-   * @param collectionName - Collection name
-   * @example
-   * ```ts
-   * const documentStore = MongoDBDocumentStore.fromConnectionString("mongodb://localhost:27017", "my_db", "my_collection");
-   * ```
-   */ static fromConnectionString(connectionString, dbName = DEFAULT_DATABASE, collectionName = DEFAULT_COLLECTION) {
-        const mongoClient = new MongoClient(connectionString, {
-            appName: "VECTORSTORES_JS"
-        });
-        const mongoKVStore = new MongoKVStore({
-            mongoClient,
-            dbName
-        });
-        return new MongoDocumentStore({
-            mongoKVStore,
-            namespace: `${dbName}.${collectionName}`
-        });
-    }
-}
+var version = "0.1.4";
+var pkg = {
+	version: version};
 
 // define your Atlas Search index. See detail https://www.mongodb.com/docs/atlas/atlas-search/field-types/knn-vector/
 const DEFAULT_EMBEDDING_DEFINITION = {
@@ -333,6 +277,15 @@ function toMongoDBFilter(filters) {
         };
         return result;
     }
+    async exists(refDocId) {
+        const collection = await this.ensureCollection();
+        const count = await collection.countDocuments({
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
+        }, {
+            limit: 1
+        });
+        return count > 0;
+    }
 }
 
 /**
@@ -393,4 +346,4 @@ function toMongoDBFilter(filters) {
     }
 }
 
-export { MongoDBAtlasVectorSearch, MongoDocumentStore, MongoKVStore, SimpleMongoReader };
+export { MongoDBAtlasVectorSearch, MongoKVStore, SimpleMongoReader };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@vectorstores/mongodb",
   "description": "MongoDB Storage for vectorstores",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -36,11 +36,11 @@
   "devDependencies": {
     "vitest": "2.1.0",
     "mongodb-memory-server": "^10.1.4",
-    "@vectorstores/core": "0.1.2",
+    "@vectorstores/core": "0.1.4",
     "@vectorstores/env": "0.1.0"
   },
   "peerDependencies": {
-    "@vectorstores/core": "0.1.2",
+    "@vectorstores/core": "0.1.4",
     "@vectorstores/env": "0.1.0"
   },
   "dependencies": {