Class that is a wrapper around MongoDB Atlas Vector Search. It is used to store embeddings in MongoDB documents, create a vector search index, and perform K-Nearest Neighbors (KNN) search with an approximate nearest neighbor algorithm.

Hierarchy (view full)

Constructors

Properties

FilterType: MongoDBAtlasFilter

Methods

  • Method to add documents to the MongoDB collection. It first converts the documents to vectors using the embeddings and then calls the addVectors method.

    Parameters

    • documents: Document[]

      Documents to be added.

    • Optional options: {
          ids?: string[];
      }
      • Optional ids?: string[]

    Returns Promise<any[]>

    Promise that resolves when the documents have been added.

  • Method to add vectors and their corresponding documents to the MongoDB collection.

    Parameters

    • vectors: number[][]

      Vectors to be added.

    • documents: Document[]

      Corresponding documents to be added.

    • Optional options: {
          ids?: string[];
      }
      • Optional ids?: string[]

    Returns Promise<any[]>

    Promise that resolves when the vectors and documents have been added.

  • Delete documents from the collection

    Parameters

    • params: {
          ids: any[];
      }
      • ids: any[]

    Returns Promise<void>

    • A promise that resolves when all documents deleted
  • Return documents selected using the maximal marginal relevance. Maximal marginal relevance optimizes for similarity to the query AND diversity among selected documents.

    Parameters

    • query: string

      Text to look up documents similar to.

    • options: MaxMarginalRelevanceSearchOptions<this["FilterType"]>

    Returns Promise<Document[]>

    • List of documents selected by maximal marginal relevance.
  • Method that performs a similarity search on the vectors stored in the MongoDB collection. It returns a list of documents and their corresponding similarity scores.

    Parameters

    • query: number[]

      Query vector for the similarity search.

    • k: number

      Number of nearest neighbors to return.

    • Optional filter: MongoDBAtlasFilter

      Optional filter to be applied.

    Returns Promise<[Document, number][]>

    Promise that resolves to a list of documents and their corresponding similarity scores.

  • Static method to fix the precision of the array that ensures that every number in this array is always float when casted to other types. This is needed since MongoDB Atlas Vector Search does not cast integer inside vector search to float automatically. This method shall introduce a hint of error but should be safe to use since introduced error is very small, only applies to integer numbers returned by embeddings, and most embeddings shall not have precision as high as 15 decimal places.

    Parameters

    • array: number[]

      Array of number to be fixed.

    Returns number[]

  • Static method to create an instance of MongoDBAtlasVectorSearch from a list of documents. It first converts the documents to vectors and then adds them to the MongoDB collection.

    Parameters

    • docs: Document[]

      List of documents to be converted to vectors.

    • embeddings: EmbeddingsInterface

      Embeddings to be used for conversion.

    • dbConfig: MongoDBAtlasVectorSearchLibArgs & {
          ids?: string[];
      }

      Database configuration for MongoDB Atlas.

    Returns Promise<MongoDBAtlasVectorSearch>

    Promise that resolves to a new instance of MongoDBAtlasVectorSearch.

  • Static method to create an instance of MongoDBAtlasVectorSearch from a list of texts. It first converts the texts to vectors and then adds them to the MongoDB collection.

    Parameters

    • texts: string[]

      List of texts to be converted to vectors.

    • metadatas: object | object[]

      Metadata for the texts.

    • embeddings: EmbeddingsInterface

      Embeddings to be used for conversion.

    • dbConfig: MongoDBAtlasVectorSearchLibArgs & {
          ids?: string[];
      }

      Database configuration for MongoDB Atlas.

    Returns Promise<MongoDBAtlasVectorSearch>

    Promise that resolves to a new instance of MongoDBAtlasVectorSearch.

Generated using TypeDoc