@vectorstores/azure 0.1.3 → 0.1.5

package/dist/index.cjs

@@ -2,8 +2,8 @@ Object.defineProperty(exports, '__esModule', { value: true });
 
 var core = require('@vectorstores/core');
 var storage_cjs = require('./storage.cjs');
-var searchDocuments = require('@azure/search-documents');
 var identity = require('@azure/identity');
+var searchDocuments = require('@azure/search-documents');
 var env = require('@vectorstores/env');
 var mongodb = require('mongodb');
 var cosmos = require('@azure/cosmos');
@@ -365,11 +365,10 @@ const vectorStore = new AzureAISearchVectorStore({
 const documents = await new SimpleDirectoryReader().loadData(
   "data/paul_graham/",
 );
-const storageContext = await storageContextFromDefaults({ vectorStore });
 
-// Create index from documents with the specified storage context
+// Create index from documents with the specified vector store
 const index = await VectorStoreIndex.fromDocuments(documents, {
-  storageContext,
+  vectorStore,
   docStoreStrategy: DocStoreStrategy.UPSERTS,
 });
 
@@ -789,7 +788,7 @@ console.log({ response });
         }
     }
     #buildCredentials(options) {
-        let { credential: credential, key, endpoint, indexName } = options;
+        let { credential, key, endpoint, indexName } = options;
         // validate and use credential
         if (credential) {
             // if credential are provided, ensure they are an instance of valid credential instances
@@ -1121,9 +1120,19 @@ console.log({ response });
         // Execute the search and return the result
         return await azureQueryResultSearch.search();
     }
+    async exists(refDocId) {
+        const results = await this._searchClient.search("*", {
+            filter: `ref_doc_id eq '${refDocId}'`,
+            top: 1
+        });
+        for await (const _result of results.results){
+            return true;
+        }
+        return false;
+    }
 }
 
-var version = "0.1.3";
+var version = "0.1.5";
 var pkg = {
 	version: version};
 
@@ -1206,13 +1215,14 @@ var pkg = {
         return Object.values(insertResult.insertedIds).map((id)=>String(id));
     }
     /**
-   * Removes specified documents from the AzureCosmosDBMongoDBVectorStore.
-   * @param params Parameters for the delete operation.
+   * Deletes all nodes from the collection that belong to the given document.
+   * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
+   * @param deleteOptions Additional delete options.
    * @returns A promise that resolves when the documents have been removed.
-   */ async delete(id, deleteOptions) {
+   */ async delete(refDocId, deleteOptions) {
         const collection = await this.ensureCollection();
         await collection.deleteMany({
-            id: id
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
         }, deleteOptions);
     }
     async query(query, options) {
@@ -1350,6 +1360,15 @@ var pkg = {
             await collection.dropIndex(indexName);
         }
     }
+    async exists(refDocId) {
+        const collection = await this.ensureCollection();
+        const count = await collection.countDocuments({
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
+        }, {
+            limit: 1
+        });
+        return count > 0;
+    }
 }
 
 const USER_AGENT_SUFFIX = "vectorstores-cdbnosql-vectorstore-javascript";
@@ -1532,14 +1551,28 @@ class AzureCosmosDBNoSqlVectorStore extends core.BaseVectorStore {
         return ids;
     }
     /**
-   * Delete a document from the CosmosDB container.
+   * Delete all nodes from the CosmosDB container that belong to the given document.
    *
-   * @param refDocId - The id of the document to delete
-   * @param deleteOptions - Any options to pass to the container.item.delete function
+   * @param refDocId - Reference document ID - all nodes with this ref_doc_id will be deleted.
+   * @param deleteOptions - Any options to pass to the delete operations.
    * @returns Promise that resolves if the delete query did not throw an error.
    */ async delete(refDocId, deleteOptions) {
         await this.initialize();
-        await this.container.item(refDocId).delete(deleteOptions);
+        // Query all items with matching ref_doc_id
+        const querySpec = {
+            query: `SELECT c.id FROM c WHERE c.metadata.ref_doc_id = @refDocId`,
+            parameters: [
+                {
+                    name: "@refDocId",
+                    value: refDocId
+                }
+            ]
+        };
+        const { resources: items } = await this.container.items.query(querySpec).fetchAll();
+        // Delete each matching item
+        for (const item of items){
+            await this.container.item(item.id, item.id).delete(deleteOptions);
+        }
     }
     /**
    * Performs a vector similarity search query in the CosmosDB container.
@@ -1633,6 +1666,19 @@ class AzureCosmosDBNoSqlVectorStore extends core.BaseVectorStore {
         });
         this.container = container;
     }
+    async exists(refDocId) {
+        await this.initialize();
+        const { resources } = await this.container.items.query({
+            query: `SELECT VALUE COUNT(1) FROM c WHERE c.metadata.ref_doc_id = @refDocId`,
+            parameters: [
+                {
+                    name: "@refDocId",
+                    value: refDocId
+                }
+            ]
+        }).fetchAll();
+        return (resources[0] ?? 0) > 0;
+    }
 }
 
 exports.AzureAISearchVectorStore = AzureAISearchVectorStore;

package/dist/index.d.cts

@@ -1,8 +1,8 @@
 import { SqlQuerySpec, CosmosClient, DatabaseRequest, ContainerRequest, VectorEmbeddingPolicy, IndexingPolicy } from '@azure/cosmos';
 import { BaseReader, Document, BaseNode, BaseVectorStore, VectorStoreBaseParams, MetadataFilters, VectorStoreQuery, VectorStoreQueryResult } from '@vectorstores/core';
 export * from './storage.cjs';
-import { AzureKeyCredential, SearchIndexClient, SearchClient, LexicalAnalyzerName, KnownVectorSearchCompressionKind, KnownVectorSearchAlgorithmKind, VectorizedQuery } from '@azure/search-documents';
 import { DefaultAzureCredential, ManagedIdentityCredential, TokenCredential } from '@azure/identity';
+import { AzureKeyCredential, SearchIndexClient, SearchClient, LexicalAnalyzerName, KnownVectorSearchCompressionKind, KnownVectorSearchAlgorithmKind, VectorizedQuery } from '@azure/search-documents';
 import { MongoClient, Collection } from 'mongodb';
 
 type SimpleCosmosDBReaderLoaderConfig = {
@@ -247,11 +247,10 @@ const vectorStore = new AzureAISearchVectorStore({
 const documents = await new SimpleDirectoryReader().loadData(
   "data/paul_graham/",
 );
-const storageContext = await storageContextFromDefaults({ vectorStore });
 
-// Create index from documents with the specified storage context
+// Create index from documents with the specified vector store
 const index = await VectorStoreIndex.fromDocuments(documents, {
-  storageContext,
+  vectorStore,
   docStoreStrategy: DocStoreStrategy.UPSERTS,
 });
 
@@ -303,6 +302,7 @@ declare class AzureAISearchVectorStore<T extends R> extends BaseVectorStore {
     query(query: VectorStoreQuery & {
         queryStr: string;
     }): Promise<VectorStoreQueryResult>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /** Azure Cosmos DB for MongoDB vCore Similarity type. */
@@ -388,11 +388,12 @@ declare class AzureCosmosDBMongoDBVectorStore extends BaseVectorStore {
     ensureCollection(): Promise<Collection>;
     add(nodes: BaseNode[]): Promise<string[]>;
     /**
-     * Removes specified documents from the AzureCosmosDBMongoDBVectorStore.
-     * @param params Parameters for the delete operation.
+     * Deletes all nodes from the collection that belong to the given document.
+     * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
+     * @param deleteOptions Additional delete options.
      * @returns A promise that resolves when the documents have been removed.
      */
-    delete(id: string, deleteOptions?: object): Promise<void>;
+    delete(refDocId: string, deleteOptions?: object): Promise<void>;
     query(query: VectorStoreQuery, options: AzureCosmosDBMongoDBQueryOptions): Promise<VectorStoreQueryResult>;
     /**
      * Creates an index on the collection with the specified index name during
@@ -435,6 +436,7 @@ declare class AzureCosmosDBMongoDBVectorStore extends BaseVectorStore {
      * @returns A promise that resolves when the index has been deleted.
      */
     deleteIndex(indexName: string): Promise<void>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /** Azure Cosmos DB for NoSQL database creation options. */
@@ -534,10 +536,10 @@ declare class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
      */
     add(nodes: BaseNode[]): Promise<string[]>;
     /**
-     * Delete a document from the CosmosDB container.
+     * Delete all nodes from the CosmosDB container that belong to the given document.
      *
-     * @param refDocId - The id of the document to delete
-     * @param deleteOptions - Any options to pass to the container.item.delete function
+     * @param refDocId - Reference document ID - all nodes with this ref_doc_id will be deleted.
+     * @param deleteOptions - Any options to pass to the delete operations.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
     delete(refDocId: string, deleteOptions?: object): Promise<void>;
@@ -552,6 +554,7 @@ declare class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
      * Initialize the CosmosDB container.
      */
     private init;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 declare class AzureQueryResultSearchBase<T extends R> {

package/dist/index.d.ts

@@ -1,8 +1,8 @@
 import { SqlQuerySpec, CosmosClient, DatabaseRequest, ContainerRequest, VectorEmbeddingPolicy, IndexingPolicy } from '@azure/cosmos';
 import { BaseReader, Document, BaseNode, BaseVectorStore, VectorStoreBaseParams, MetadataFilters, VectorStoreQuery, VectorStoreQueryResult } from '@vectorstores/core';
 export * from './storage.js';
-import { AzureKeyCredential, SearchIndexClient, SearchClient, LexicalAnalyzerName, KnownVectorSearchCompressionKind, KnownVectorSearchAlgorithmKind, VectorizedQuery } from '@azure/search-documents';
 import { DefaultAzureCredential, ManagedIdentityCredential, TokenCredential } from '@azure/identity';
+import { AzureKeyCredential, SearchIndexClient, SearchClient, LexicalAnalyzerName, KnownVectorSearchCompressionKind, KnownVectorSearchAlgorithmKind, VectorizedQuery } from '@azure/search-documents';
 import { MongoClient, Collection } from 'mongodb';
 
 type SimpleCosmosDBReaderLoaderConfig = {
@@ -247,11 +247,10 @@ const vectorStore = new AzureAISearchVectorStore({
 const documents = await new SimpleDirectoryReader().loadData(
   "data/paul_graham/",
 );
-const storageContext = await storageContextFromDefaults({ vectorStore });
 
-// Create index from documents with the specified storage context
+// Create index from documents with the specified vector store
 const index = await VectorStoreIndex.fromDocuments(documents, {
-  storageContext,
+  vectorStore,
   docStoreStrategy: DocStoreStrategy.UPSERTS,
 });
 
@@ -303,6 +302,7 @@ declare class AzureAISearchVectorStore<T extends R> extends BaseVectorStore {
     query(query: VectorStoreQuery & {
         queryStr: string;
     }): Promise<VectorStoreQueryResult>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /** Azure Cosmos DB for MongoDB vCore Similarity type. */
@@ -388,11 +388,12 @@ declare class AzureCosmosDBMongoDBVectorStore extends BaseVectorStore {
     ensureCollection(): Promise<Collection>;
     add(nodes: BaseNode[]): Promise<string[]>;
     /**
-     * Removes specified documents from the AzureCosmosDBMongoDBVectorStore.
-     * @param params Parameters for the delete operation.
+     * Deletes all nodes from the collection that belong to the given document.
+     * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
+     * @param deleteOptions Additional delete options.
      * @returns A promise that resolves when the documents have been removed.
      */
-    delete(id: string, deleteOptions?: object): Promise<void>;
+    delete(refDocId: string, deleteOptions?: object): Promise<void>;
     query(query: VectorStoreQuery, options: AzureCosmosDBMongoDBQueryOptions): Promise<VectorStoreQueryResult>;
     /**
      * Creates an index on the collection with the specified index name during
@@ -435,6 +436,7 @@ declare class AzureCosmosDBMongoDBVectorStore extends BaseVectorStore {
      * @returns A promise that resolves when the index has been deleted.
      */
     deleteIndex(indexName: string): Promise<void>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /** Azure Cosmos DB for NoSQL database creation options. */
@@ -534,10 +536,10 @@ declare class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
      */
     add(nodes: BaseNode[]): Promise<string[]>;
     /**
-     * Delete a document from the CosmosDB container.
+     * Delete all nodes from the CosmosDB container that belong to the given document.
      *
-     * @param refDocId - The id of the document to delete
-     * @param deleteOptions - Any options to pass to the container.item.delete function
+     * @param refDocId - Reference document ID - all nodes with this ref_doc_id will be deleted.
+     * @param deleteOptions - Any options to pass to the delete operations.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
     delete(refDocId: string, deleteOptions?: object): Promise<void>;
@@ -552,6 +554,7 @@ declare class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
      * Initialize the CosmosDB container.
      */
     private init;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 declare class AzureQueryResultSearchBase<T extends R> {

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

@@ -1,8 +1,8 @@
 import { SqlQuerySpec, CosmosClient, DatabaseRequest, ContainerRequest, VectorEmbeddingPolicy, IndexingPolicy } from '@azure/cosmos';
 import { BaseReader, Document, BaseNode, BaseVectorStore, VectorStoreBaseParams, MetadataFilters, VectorStoreQuery, VectorStoreQueryResult } from '@vectorstores/core';
 export * from './storage.js';
-import { AzureKeyCredential, SearchIndexClient, SearchClient, LexicalAnalyzerName, KnownVectorSearchCompressionKind, KnownVectorSearchAlgorithmKind, VectorizedQuery } from '@azure/search-documents';
 import { DefaultAzureCredential, ManagedIdentityCredential, TokenCredential } from '@azure/identity';
+import { AzureKeyCredential, SearchIndexClient, SearchClient, LexicalAnalyzerName, KnownVectorSearchCompressionKind, KnownVectorSearchAlgorithmKind, VectorizedQuery } from '@azure/search-documents';
 import { MongoClient, Collection } from 'mongodb';
 
 type SimpleCosmosDBReaderLoaderConfig = {
@@ -247,11 +247,10 @@ const vectorStore = new AzureAISearchVectorStore({
 const documents = await new SimpleDirectoryReader().loadData(
   "data/paul_graham/",
 );
-const storageContext = await storageContextFromDefaults({ vectorStore });
 
-// Create index from documents with the specified storage context
+// Create index from documents with the specified vector store
 const index = await VectorStoreIndex.fromDocuments(documents, {
-  storageContext,
+  vectorStore,
   docStoreStrategy: DocStoreStrategy.UPSERTS,
 });
 
@@ -303,6 +302,7 @@ declare class AzureAISearchVectorStore<T extends R> extends BaseVectorStore {
     query(query: VectorStoreQuery & {
         queryStr: string;
     }): Promise<VectorStoreQueryResult>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /** Azure Cosmos DB for MongoDB vCore Similarity type. */
@@ -388,11 +388,12 @@ declare class AzureCosmosDBMongoDBVectorStore extends BaseVectorStore {
     ensureCollection(): Promise<Collection>;
     add(nodes: BaseNode[]): Promise<string[]>;
     /**
-     * Removes specified documents from the AzureCosmosDBMongoDBVectorStore.
-     * @param params Parameters for the delete operation.
+     * Deletes all nodes from the collection that belong to the given document.
+     * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
+     * @param deleteOptions Additional delete options.
      * @returns A promise that resolves when the documents have been removed.
      */
-    delete(id: string, deleteOptions?: object): Promise<void>;
+    delete(refDocId: string, deleteOptions?: object): Promise<void>;
     query(query: VectorStoreQuery, options: AzureCosmosDBMongoDBQueryOptions): Promise<VectorStoreQueryResult>;
     /**
      * Creates an index on the collection with the specified index name during
@@ -435,6 +436,7 @@ declare class AzureCosmosDBMongoDBVectorStore extends BaseVectorStore {
      * @returns A promise that resolves when the index has been deleted.
      */
     deleteIndex(indexName: string): Promise<void>;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 /** Azure Cosmos DB for NoSQL database creation options. */
@@ -534,10 +536,10 @@ declare class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
      */
     add(nodes: BaseNode[]): Promise<string[]>;
     /**
-     * Delete a document from the CosmosDB container.
+     * Delete all nodes from the CosmosDB container that belong to the given document.
      *
-     * @param refDocId - The id of the document to delete
-     * @param deleteOptions - Any options to pass to the container.item.delete function
+     * @param refDocId - Reference document ID - all nodes with this ref_doc_id will be deleted.
+     * @param deleteOptions - Any options to pass to the delete operations.
      * @returns Promise that resolves if the delete query did not throw an error.
      */
     delete(refDocId: string, deleteOptions?: object): Promise<void>;
@@ -552,6 +554,7 @@ declare class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
      * Initialize the CosmosDB container.
      */
     private init;
+    exists(refDocId: string): Promise<boolean>;
 }
 
 declare class AzureQueryResultSearchBase<T extends R> {

package/dist/index.edge-light.js

@@ -1,7 +1,7 @@
 import { Document, metadataDictToNode, BaseVectorStore, MetadataMode, nodeToMetadata, FilterOperator, FilterCondition, VectorStoreQueryMode } from '@vectorstores/core';
 export * from './storage.edge-light.js';
-import { KnownVectorSearchAlgorithmKind, KnownAnalyzerNames, KnownSearchFieldDataType, KnownVectorSearchAlgorithmMetric, KnownVectorSearchCompressionKind, SearchClient, SearchIndexClient, AzureKeyCredential, IndexDocumentsBatch } from '@azure/search-documents';
 import { DefaultAzureCredential, ManagedIdentityCredential } from '@azure/identity';
+import { KnownVectorSearchAlgorithmKind, KnownAnalyzerNames, KnownSearchFieldDataType, KnownVectorSearchAlgorithmMetric, KnownVectorSearchCompressionKind, SearchClient, SearchIndexClient, AzureKeyCredential, IndexDocumentsBatch } from '@azure/search-documents';
 import { consoleLogger, getEnv } from '@vectorstores/env';
 import { MongoClient } from 'mongodb';
 import { VectorEmbeddingDistanceFunction, VectorEmbeddingDataType, VectorIndexType, CosmosClient } from '@azure/cosmos';
@@ -363,11 +363,10 @@ const vectorStore = new AzureAISearchVectorStore({
 const documents = await new SimpleDirectoryReader().loadData(
   "data/paul_graham/",
 );
-const storageContext = await storageContextFromDefaults({ vectorStore });
 
-// Create index from documents with the specified storage context
+// Create index from documents with the specified vector store
 const index = await VectorStoreIndex.fromDocuments(documents, {
-  storageContext,
+  vectorStore,
   docStoreStrategy: DocStoreStrategy.UPSERTS,
 });
 
@@ -787,7 +786,7 @@ console.log({ response });
         }
     }
     #buildCredentials(options) {
-        let { credential: credential, key, endpoint, indexName } = options;
+        let { credential, key, endpoint, indexName } = options;
         // validate and use credential
         if (credential) {
             // if credential are provided, ensure they are an instance of valid credential instances
@@ -1119,9 +1118,19 @@ console.log({ response });
         // Execute the search and return the result
         return await azureQueryResultSearch.search();
     }
+    async exists(refDocId) {
+        const results = await this._searchClient.search("*", {
+            filter: `ref_doc_id eq '${refDocId}'`,
+            top: 1
+        });
+        for await (const _result of results.results){
+            return true;
+        }
+        return false;
+    }
 }
 
-var version = "0.1.3";
+var version = "0.1.5";
 var pkg = {
 	version: version};
 
@@ -1204,13 +1213,14 @@ var pkg = {
         return Object.values(insertResult.insertedIds).map((id)=>String(id));
     }
     /**
-   * Removes specified documents from the AzureCosmosDBMongoDBVectorStore.
-   * @param params Parameters for the delete operation.
+   * Deletes all nodes from the collection that belong to the given document.
+   * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
+   * @param deleteOptions Additional delete options.
    * @returns A promise that resolves when the documents have been removed.
-   */ async delete(id, deleteOptions) {
+   */ async delete(refDocId, deleteOptions) {
         const collection = await this.ensureCollection();
         await collection.deleteMany({
-            id: id
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
         }, deleteOptions);
     }
     async query(query, options) {
@@ -1348,6 +1358,15 @@ var pkg = {
             await collection.dropIndex(indexName);
         }
     }
+    async exists(refDocId) {
+        const collection = await this.ensureCollection();
+        const count = await collection.countDocuments({
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
+        }, {
+            limit: 1
+        });
+        return count > 0;
+    }
 }
 
 const USER_AGENT_SUFFIX = "vectorstores-cdbnosql-vectorstore-javascript";
@@ -1530,14 +1549,28 @@ class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
         return ids;
     }
     /**
-   * Delete a document from the CosmosDB container.
+   * Delete all nodes from the CosmosDB container that belong to the given document.
    *
-   * @param refDocId - The id of the document to delete
-   * @param deleteOptions - Any options to pass to the container.item.delete function
+   * @param refDocId - Reference document ID - all nodes with this ref_doc_id will be deleted.
+   * @param deleteOptions - Any options to pass to the delete operations.
    * @returns Promise that resolves if the delete query did not throw an error.
    */ async delete(refDocId, deleteOptions) {
         await this.initialize();
-        await this.container.item(refDocId).delete(deleteOptions);
+        // Query all items with matching ref_doc_id
+        const querySpec = {
+            query: `SELECT c.id FROM c WHERE c.metadata.ref_doc_id = @refDocId`,
+            parameters: [
+                {
+                    name: "@refDocId",
+                    value: refDocId
+                }
+            ]
+        };
+        const { resources: items } = await this.container.items.query(querySpec).fetchAll();
+        // Delete each matching item
+        for (const item of items){
+            await this.container.item(item.id, item.id).delete(deleteOptions);
+        }
     }
     /**
    * Performs a vector similarity search query in the CosmosDB container.
@@ -1631,6 +1664,19 @@ class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
         });
         this.container = container;
     }
+    async exists(refDocId) {
+        await this.initialize();
+        const { resources } = await this.container.items.query({
+            query: `SELECT VALUE COUNT(1) FROM c WHERE c.metadata.ref_doc_id = @refDocId`,
+            parameters: [
+                {
+                    name: "@refDocId",
+                    value: refDocId
+                }
+            ]
+        }).fetchAll();
+        return (resources[0] ?? 0) > 0;
+    }
 }
 
 export { AzureAISearchVectorStore, AzureAISearchVectorStoreConfig, AzureCosmosDBMongoDBSimilarityType, AzureCosmosDBMongoDBVectorStore, AzureCosmosDBNoSqlVectorStore, AzureQueryResultSearchBase, AzureQueryResultSearchDefault, AzureQueryResultSearchHybrid, AzureQueryResultSearchSemanticHybrid, AzureQueryResultSearchSparse, IndexManagement, MetadataIndexFieldType, SimpleCosmosDBReader };

package/dist/index.js

@@ -1,7 +1,7 @@
 import { Document, metadataDictToNode, BaseVectorStore, MetadataMode, nodeToMetadata, FilterOperator, FilterCondition, VectorStoreQueryMode } from '@vectorstores/core';
 export * from './storage.js';
-import { KnownVectorSearchAlgorithmKind, KnownAnalyzerNames, KnownSearchFieldDataType, KnownVectorSearchAlgorithmMetric, KnownVectorSearchCompressionKind, SearchClient, SearchIndexClient, AzureKeyCredential, IndexDocumentsBatch } from '@azure/search-documents';
 import { DefaultAzureCredential, ManagedIdentityCredential } from '@azure/identity';
+import { KnownVectorSearchAlgorithmKind, KnownAnalyzerNames, KnownSearchFieldDataType, KnownVectorSearchAlgorithmMetric, KnownVectorSearchCompressionKind, SearchClient, SearchIndexClient, AzureKeyCredential, IndexDocumentsBatch } from '@azure/search-documents';
 import { consoleLogger, getEnv } from '@vectorstores/env';
 import { MongoClient } from 'mongodb';
 import { VectorEmbeddingDistanceFunction, VectorEmbeddingDataType, VectorIndexType, CosmosClient } from '@azure/cosmos';
@@ -363,11 +363,10 @@ const vectorStore = new AzureAISearchVectorStore({
 const documents = await new SimpleDirectoryReader().loadData(
   "data/paul_graham/",
 );
-const storageContext = await storageContextFromDefaults({ vectorStore });
 
-// Create index from documents with the specified storage context
+// Create index from documents with the specified vector store
 const index = await VectorStoreIndex.fromDocuments(documents, {
-  storageContext,
+  vectorStore,
   docStoreStrategy: DocStoreStrategy.UPSERTS,
 });
 
@@ -787,7 +786,7 @@ console.log({ response });
         }
     }
     #buildCredentials(options) {
-        let { credential: credential, key, endpoint, indexName } = options;
+        let { credential, key, endpoint, indexName } = options;
         // validate and use credential
         if (credential) {
             // if credential are provided, ensure they are an instance of valid credential instances
@@ -1119,9 +1118,19 @@ console.log({ response });
         // Execute the search and return the result
         return await azureQueryResultSearch.search();
     }
+    async exists(refDocId) {
+        const results = await this._searchClient.search("*", {
+            filter: `ref_doc_id eq '${refDocId}'`,
+            top: 1
+        });
+        for await (const _result of results.results){
+            return true;
+        }
+        return false;
+    }
 }
 
-var version = "0.1.3";
+var version = "0.1.5";
 var pkg = {
 	version: version};
 
@@ -1204,13 +1213,14 @@ var pkg = {
         return Object.values(insertResult.insertedIds).map((id)=>String(id));
     }
     /**
-   * Removes specified documents from the AzureCosmosDBMongoDBVectorStore.
-   * @param params Parameters for the delete operation.
+   * Deletes all nodes from the collection that belong to the given document.
+   * @param refDocId Reference document ID - all nodes with this ref_doc_id will be deleted.
+   * @param deleteOptions Additional delete options.
    * @returns A promise that resolves when the documents have been removed.
-   */ async delete(id, deleteOptions) {
+   */ async delete(refDocId, deleteOptions) {
         const collection = await this.ensureCollection();
         await collection.deleteMany({
-            id: id
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
         }, deleteOptions);
     }
     async query(query, options) {
@@ -1348,6 +1358,15 @@ var pkg = {
             await collection.dropIndex(indexName);
         }
     }
+    async exists(refDocId) {
+        const collection = await this.ensureCollection();
+        const count = await collection.countDocuments({
+            [`${this.metadataKey}.ref_doc_id`]: refDocId
+        }, {
+            limit: 1
+        });
+        return count > 0;
+    }
 }
 
 const USER_AGENT_SUFFIX = "vectorstores-cdbnosql-vectorstore-javascript";
@@ -1530,14 +1549,28 @@ class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
         return ids;
     }
     /**
-   * Delete a document from the CosmosDB container.
+   * Delete all nodes from the CosmosDB container that belong to the given document.
    *
-   * @param refDocId - The id of the document to delete
-   * @param deleteOptions - Any options to pass to the container.item.delete function
+   * @param refDocId - Reference document ID - all nodes with this ref_doc_id will be deleted.
+   * @param deleteOptions - Any options to pass to the delete operations.
    * @returns Promise that resolves if the delete query did not throw an error.
    */ async delete(refDocId, deleteOptions) {
         await this.initialize();
-        await this.container.item(refDocId).delete(deleteOptions);
+        // Query all items with matching ref_doc_id
+        const querySpec = {
+            query: `SELECT c.id FROM c WHERE c.metadata.ref_doc_id = @refDocId`,
+            parameters: [
+                {
+                    name: "@refDocId",
+                    value: refDocId
+                }
+            ]
+        };
+        const { resources: items } = await this.container.items.query(querySpec).fetchAll();
+        // Delete each matching item
+        for (const item of items){
+            await this.container.item(item.id, item.id).delete(deleteOptions);
+        }
     }
     /**
    * Performs a vector similarity search query in the CosmosDB container.
@@ -1631,6 +1664,19 @@ class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
         });
         this.container = container;
     }
+    async exists(refDocId) {
+        await this.initialize();
+        const { resources } = await this.container.items.query({
+            query: `SELECT VALUE COUNT(1) FROM c WHERE c.metadata.ref_doc_id = @refDocId`,
+            parameters: [
+                {
+                    name: "@refDocId",
+                    value: refDocId
+                }
+            ]
+        }).fetchAll();
+        return (resources[0] ?? 0) > 0;
+    }
 }
 
 export { AzureAISearchVectorStore, AzureAISearchVectorStoreConfig, AzureCosmosDBMongoDBSimilarityType, AzureCosmosDBMongoDBVectorStore, AzureCosmosDBNoSqlVectorStore, AzureQueryResultSearchBase, AzureQueryResultSearchDefault, AzureQueryResultSearchHybrid, AzureQueryResultSearchSemanticHybrid, AzureQueryResultSearchSparse, IndexManagement, MetadataIndexFieldType, SimpleCosmosDBReader };