@vectorstores/upstash 0.1.3 → 0.1.4

package/dist/index.cjs

@@ -89,21 +89,31 @@ var env = require('@vectorstores/env');
         return results.every((result)=>result === "OK") ? "OK" : "NOT-OK";
     }
     /**
-   * Deletes a single record from the database by id.
-   * NOTE: Uses the collection property controlled by setCollection/getCollection.
-   * @param refDocId Unique identifier for the record to delete.
+   * Deletes all nodes from the database that belong to the given document.
+   * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
    * @returns Promise that resolves if the delete query did not throw an error.
    */ async delete(refDocId) {
-        await this.db.namespace(this.namespace).delete(refDocId);
+        // Query to find all node IDs with this ref_doc_id
+        const results = await this.db.query({
+            vector: new Array(1536).fill(0),
+            topK: 10000,
+            includeMetadata: true,
+            filter: `ref_doc_id = "${refDocId}"`
+        }, {
+            namespace: this.namespace
+        });
+        // Delete all matching nodes by their IDs
+        const idsToDelete = results.map((r)=>String(r.id));
+        if (idsToDelete.length > 0) {
+            await this.db.namespace(this.namespace).delete(idsToDelete);
+        }
     }
     /**
-   * Deletes a single record from the database by id.
-   * NOTE: Uses the collection property controlled by setCollection/getCollection.
-   * @param refDocId Unique identifier for the record to delete.
-   * @param deleteKwargs Required by VectorStore interface.  Currently ignored.
+   * Deletes multiple records from the database by their IDs.
+   * @param ids Array of node IDs to delete.
    * @returns Promise that resolves if the delete query did not throw an error.
-   */ async deleteMany(refDocId) {
-        await this.db.namespace(this.namespace).delete(refDocId);
+   */ async deleteMany(ids) {
+        await this.db.namespace(this.namespace).delete(ids);
     }
     /**
    * Query the vector store for the closest matching data to the query embeddings
@@ -176,6 +186,17 @@ var env = require('@vectorstores/env');
             metadata: core.nodeToMetadata(node)
         };
     }
+    async exists(refDocId) {
+        const results = await this.db.query({
+            vector: new Array(1536).fill(0),
+            topK: 1,
+            includeMetadata: false,
+            filter: `ref_doc_id = "${refDocId}"`
+        }, {
+            namespace: this.namespace
+        });
+        return results.length > 0;
+    }
 }
 
 exports.UpstashVectorStore = UpstashVectorStore;

package/dist/index.d.cts

@@ -47,20 +47,17 @@ declare class UpstashVectorStore extends BaseVectorStore {
     addPlainText(text: TextNode<Metadata>[]): Promise<string[]>;
     private upsertInBatches;
     /**
-     * Deletes a single record from the database by id.
-     * NOTE: Uses the collection property controlled by setCollection/getCollection.
-     * @param refDocId Unique identifier for the record to delete.
+     * Deletes all nodes from the database that belong to the given document.
+     * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
     delete(refDocId: string): Promise<void>;
     /**
-     * Deletes a single record from the database by id.
-     * NOTE: Uses the collection property controlled by setCollection/getCollection.
-     * @param refDocId Unique identifier for the record to delete.
-     * @param deleteKwargs Required by VectorStore interface.  Currently ignored.
+     * Deletes multiple records from the database by their IDs.
+     * @param ids Array of node IDs to delete.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
-    deleteMany(refDocId: string[]): Promise<void>;
+    deleteMany(ids: string[]): Promise<void>;
     /**
      * Query the vector store for the closest matching data to the query embeddings
      * @param query The VectorStoreQuery to be used
@@ -80,6 +77,7 @@ declare class UpstashVectorStore extends BaseVectorStore {
         data: string;
         metadata: Metadata;
     };
+    exists(refDocId: string): Promise<boolean>;
 }
 
 export { UpstashVectorStore };

package/dist/index.d.ts

@@ -47,20 +47,17 @@ declare class UpstashVectorStore extends BaseVectorStore {
     addPlainText(text: TextNode<Metadata>[]): Promise<string[]>;
     private upsertInBatches;
     /**
-     * Deletes a single record from the database by id.
-     * NOTE: Uses the collection property controlled by setCollection/getCollection.
-     * @param refDocId Unique identifier for the record to delete.
+     * Deletes all nodes from the database that belong to the given document.
+     * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
     delete(refDocId: string): Promise<void>;
     /**
-     * Deletes a single record from the database by id.
-     * NOTE: Uses the collection property controlled by setCollection/getCollection.
-     * @param refDocId Unique identifier for the record to delete.
-     * @param deleteKwargs Required by VectorStore interface.  Currently ignored.
+     * Deletes multiple records from the database by their IDs.
+     * @param ids Array of node IDs to delete.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
-    deleteMany(refDocId: string[]): Promise<void>;
+    deleteMany(ids: string[]): Promise<void>;
     /**
      * Query the vector store for the closest matching data to the query embeddings
      * @param query The VectorStoreQuery to be used
@@ -80,6 +77,7 @@ declare class UpstashVectorStore extends BaseVectorStore {
         data: string;
         metadata: Metadata;
     };
+    exists(refDocId: string): Promise<boolean>;
 }
 
 export { UpstashVectorStore };

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

@@ -47,20 +47,17 @@ declare class UpstashVectorStore extends BaseVectorStore {
     addPlainText(text: TextNode<Metadata>[]): Promise<string[]>;
     private upsertInBatches;
     /**
-     * Deletes a single record from the database by id.
-     * NOTE: Uses the collection property controlled by setCollection/getCollection.
-     * @param refDocId Unique identifier for the record to delete.
+     * Deletes all nodes from the database that belong to the given document.
+     * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
     delete(refDocId: string): Promise<void>;
     /**
-     * Deletes a single record from the database by id.
-     * NOTE: Uses the collection property controlled by setCollection/getCollection.
-     * @param refDocId Unique identifier for the record to delete.
-     * @param deleteKwargs Required by VectorStore interface.  Currently ignored.
+     * Deletes multiple records from the database by their IDs.
+     * @param ids Array of node IDs to delete.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
-    deleteMany(refDocId: string[]): Promise<void>;
+    deleteMany(ids: string[]): Promise<void>;
     /**
      * Query the vector store for the closest matching data to the query embeddings
      * @param query The VectorStoreQuery to be used
@@ -80,6 +77,7 @@ declare class UpstashVectorStore extends BaseVectorStore {
         data: string;
         metadata: Metadata;
     };
+    exists(refDocId: string): Promise<boolean>;
 }
 
 export { UpstashVectorStore };

package/dist/index.edge-light.js

@@ -87,21 +87,31 @@ import { getEnv } from '@vectorstores/env';
         return results.every((result)=>result === "OK") ? "OK" : "NOT-OK";
     }
     /**
-   * Deletes a single record from the database by id.
-   * NOTE: Uses the collection property controlled by setCollection/getCollection.
-   * @param refDocId Unique identifier for the record to delete.
+   * Deletes all nodes from the database that belong to the given document.
+   * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
    * @returns Promise that resolves if the delete query did not throw an error.
    */ async delete(refDocId) {
-        await this.db.namespace(this.namespace).delete(refDocId);
+        // Query to find all node IDs with this ref_doc_id
+        const results = await this.db.query({
+            vector: new Array(1536).fill(0),
+            topK: 10000,
+            includeMetadata: true,
+            filter: `ref_doc_id = "${refDocId}"`
+        }, {
+            namespace: this.namespace
+        });
+        // Delete all matching nodes by their IDs
+        const idsToDelete = results.map((r)=>String(r.id));
+        if (idsToDelete.length > 0) {
+            await this.db.namespace(this.namespace).delete(idsToDelete);
+        }
     }
     /**
-   * Deletes a single record from the database by id.
-   * NOTE: Uses the collection property controlled by setCollection/getCollection.
-   * @param refDocId Unique identifier for the record to delete.
-   * @param deleteKwargs Required by VectorStore interface.  Currently ignored.
+   * Deletes multiple records from the database by their IDs.
+   * @param ids Array of node IDs to delete.
    * @returns Promise that resolves if the delete query did not throw an error.
-   */ async deleteMany(refDocId) {
-        await this.db.namespace(this.namespace).delete(refDocId);
+   */ async deleteMany(ids) {
+        await this.db.namespace(this.namespace).delete(ids);
     }
     /**
    * Query the vector store for the closest matching data to the query embeddings
@@ -174,6 +184,17 @@ import { getEnv } from '@vectorstores/env';
             metadata: nodeToMetadata(node)
         };
     }
+    async exists(refDocId) {
+        const results = await this.db.query({
+            vector: new Array(1536).fill(0),
+            topK: 1,
+            includeMetadata: false,
+            filter: `ref_doc_id = "${refDocId}"`
+        }, {
+            namespace: this.namespace
+        });
+        return results.length > 0;
+    }
 }
 
 export { UpstashVectorStore };

package/dist/index.js

@@ -87,21 +87,31 @@ import { getEnv } from '@vectorstores/env';
         return results.every((result)=>result === "OK") ? "OK" : "NOT-OK";
     }
     /**
-   * Deletes a single record from the database by id.
-   * NOTE: Uses the collection property controlled by setCollection/getCollection.
-   * @param refDocId Unique identifier for the record to delete.
+   * Deletes all nodes from the database that belong to the given document.
+   * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
    * @returns Promise that resolves if the delete query did not throw an error.
    */ async delete(refDocId) {
-        await this.db.namespace(this.namespace).delete(refDocId);
+        // Query to find all node IDs with this ref_doc_id
+        const results = await this.db.query({
+            vector: new Array(1536).fill(0),
+            topK: 10000,
+            includeMetadata: true,
+            filter: `ref_doc_id = "${refDocId}"`
+        }, {
+            namespace: this.namespace
+        });
+        // Delete all matching nodes by their IDs
+        const idsToDelete = results.map((r)=>String(r.id));
+        if (idsToDelete.length > 0) {
+            await this.db.namespace(this.namespace).delete(idsToDelete);
+        }
     }
     /**
-   * Deletes a single record from the database by id.
-   * NOTE: Uses the collection property controlled by setCollection/getCollection.
-   * @param refDocId Unique identifier for the record to delete.
-   * @param deleteKwargs Required by VectorStore interface.  Currently ignored.
+   * Deletes multiple records from the database by their IDs.
+   * @param ids Array of node IDs to delete.
    * @returns Promise that resolves if the delete query did not throw an error.
-   */ async deleteMany(refDocId) {
-        await this.db.namespace(this.namespace).delete(refDocId);
+   */ async deleteMany(ids) {
+        await this.db.namespace(this.namespace).delete(ids);
     }
     /**
    * Query the vector store for the closest matching data to the query embeddings
@@ -174,6 +184,17 @@ import { getEnv } from '@vectorstores/env';
             metadata: nodeToMetadata(node)
         };
     }
+    async exists(refDocId) {
+        const results = await this.db.query({
+            vector: new Array(1536).fill(0),
+            topK: 1,
+            includeMetadata: false,
+            filter: `ref_doc_id = "${refDocId}"`
+        }, {
+            namespace: this.namespace
+        });
+        return results.length > 0;
+    }
 }
 
 export { UpstashVectorStore };

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@vectorstores/upstash",
   "description": "Upstash Storage for vectorstores",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",
@@ -34,11 +34,11 @@
     "directory": "packages/providers/storage/upstash"
   },
   "devDependencies": {
-    "@vectorstores/core": "0.1.3",
+    "@vectorstores/core": "0.1.4",
     "@vectorstores/env": "0.1.0"
   },
   "peerDependencies": {
-    "@vectorstores/core": "0.1.3",
+    "@vectorstores/core": "0.1.4",
     "@vectorstores/env": "0.1.0"
   },
   "dependencies": {