Class that provides an interface to a Postgres vector database. It extends the VectorStore base class and implements methods for adding documents and vectors, performing similarity searches, and ensuring the existence of a table in the database.

Hierarchy

Properties

FilterType: Metadata
chunkSize: number = 500
contentColumnName: string
embeddings: Embeddings
idColumnName: string
metadataColumnName: string
pool: Pool
tableName: string
vectorColumnName: string
client?: PoolClient
filter?: Metadata

Methods

  • Method to add documents to the vector store. It converts the documents into vectors, and adds them to the store.

    Parameters

    • documents: Document<Record<string, any>>[]

      Array of Document instances.

    Returns Promise<void>

    Promise that resolves when the documents have been added.

  • Method to add vectors to the vector store. It converts the vectors into rows and inserts them into the database.

    Parameters

    • vectors: number[][]

      Array of vectors.

    • documents: Document<Record<string, any>>[]

      Array of Document instances.

    Returns Promise<void>

    Promise that resolves when the vectors have been added.

  • Closes all the clients in the pool and terminates the pool.

    Returns Promise<void>

    Promise that resolves when all clients are closed and the pool is terminated.

  • Method to ensure the existence of the table in the database. It creates the table if it does not already exist.

    Returns Promise<void>

    Promise that resolves when the table has been ensured.

  • Method to perform a similarity search in the vector store. It returns the k most similar documents to the query vector, along with their similarity scores.

    Parameters

    • query: number[]

      Query vector.

    • k: number

      Number of most similar documents to return.

    • Optional filter: Metadata

      Optional filter to apply to the search.

    Returns Promise<[Document<Record<string, any>>, number][]>

    Promise that resolves with an array of tuples, each containing a Document and its similarity score.

  • Static method to create a new PGVectorStore instance from an array of texts and their metadata. It converts the texts into Document instances and adds them to the store.

    Parameters

    • texts: string[]

      Array of texts.

    • metadatas: object | object[]

      Array of metadata objects or a single metadata object.

    • embeddings: Embeddings

      Embeddings instance.

    • dbConfig: PGVectorStoreArgs

      PGVectorStoreArgs instance.

    Returns Promise<PGVectorStore>

    Promise that resolves with a new instance of PGVectorStore.

Generated using TypeDoc