Exposes Rockset's vector store/search functionality

Hierarchy

  • VectorStore
    • RocksetStore

Constructors

Properties

FilterType: string
client: MainApi
collectionName: string
embeddingKey: string
embeddings: EmbeddingsInterface
filter?: string
textKey: string
workspaceName: string

Accessors

Methods

  • Embeds and adds Documents to the store.

    Parameters

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

      The documents to store

    Returns Promise<undefined | string[]>

    The _id's of the documents added

  • Adds vectors to the store given their corresponding Documents

    Parameters

    • vectors: number[][]

      The vectors to store

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

      The Documents they represent

    Returns Promise<undefined | string[]>

    The _id's of the added documents

  • Parameters

    • OptionalkOrFields: number | Partial<VectorStoreRetrieverInput<RocksetStore>>
    • Optionalfilter: string
    • Optionalcallbacks: Callbacks
    • Optionaltags: string[]
    • Optionalmetadata: Record<string, unknown>
    • Optionalverbose: boolean

    Returns VectorStoreRetriever<RocksetStore>

  • Deletes the collection this RocksetStore uses

    Parameters

    • OptionalwaitUntilDeletion: boolean

      Whether to sleep until the collection is ready to be queried

    Returns Promise<void>

  • 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<string>
    • _callbacks: undefined | Callbacks

    Returns Promise<DocumentInterface<Record<string, any>>[]>

    • List of documents selected by maximal marginal relevance.
  • Parameters

    • query: string
    • Optionalk: number
    • Optionalfilter: string
    • Optional_callbacks: Callbacks

    Returns Promise<DocumentInterface<Record<string, any>>[]>

  • Gets the most relevant documents to a query along with their similarity score. The returned documents are ordered by similarity (most similar at the first index)

    Parameters

    • query: number[]

      The embedded query to search the store by

    • k: number

      The number of documents to retreive

    • Optionalfilter: string

      The SQL WHERE clause to filter by

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

  • Parameters

    • query: string
    • Optionalk: number
    • Optionalfilter: string
    • Optional_callbacks: Callbacks

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

  • Returns Serialized

  • Constructs, adds docs to, and returns a RocksetStore object

    Parameters

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

      The Documents to store

    • embeddings: EmbeddingsInterface

      The object used to embed queries and page content

    • dbConfig: RocksetLibArgs

      The options to be passed into the RocksetStore constructor

    Returns Promise<RocksetStore>

  • Constructs and returns a RocksetStore object given texts to store.

    Parameters

    • texts: string[]

      The texts to store

    • metadatas: object | object[]

      The metadatas that correspond to

    • embeddings: EmbeddingsInterface

      The object used to embed queries and page content

    • dbConfig: RocksetLibArgs

      The options to be passed into the RocksetStore constructor

    Returns Promise<RocksetStore>