abacusai.api_class.document_retriever
=====================================

.. py:module:: abacusai.api_class.document_retriever


Attributes
----------

.. autoapisummary::

   abacusai.api_class.document_retriever.DocumentRetrieverConfig


Classes
-------

.. autoapisummary::

   abacusai.api_class.document_retriever.VectorStoreConfig


Module Contents
---------------

.. py:class:: VectorStoreConfig

   Bases: :py:obj:`abacusai.api_class.abstract.ApiClass`


   Config for indexing options of a document retriever. Default values of optional arguments are heuristically selected by the Abacus.AI platform based on the underlying data.

   :param chunk_size: The size of text chunks in the vector store.
   :type chunk_size: int
   :param chunk_overlap_fraction: The fraction of overlap between chunks.
   :type chunk_overlap_fraction: float
   :param text_encoder: Encoder used to index texts from the documents.
   :type text_encoder: VectorStoreTextEncoder
   :param chunk_size_factors: Chunking data with multiple sizes. The specified list of factors are used to calculate more sizes, in addition to `chunk_size`.
   :type chunk_size_factors: list
   :param score_multiplier_column: If provided, will use the values in this metadata column to modify the relevance score of returned chunks for all queries.
   :type score_multiplier_column: str
   :param prune_vectors: Transform vectors using SVD so that the average component of vectors in the corpus are removed.
   :type prune_vectors: bool


   .. py:attribute:: chunk_size
      :type:  int


   .. py:attribute:: chunk_overlap_fraction
      :type:  float


   .. py:attribute:: text_encoder
      :type:  abacusai.api_class.enums.VectorStoreTextEncoder


   .. py:attribute:: chunk_size_factors
      :type:  list


   .. py:attribute:: score_multiplier_column
      :type:  str


   .. py:attribute:: prune_vectors
      :type:  bool


.. py:data:: DocumentRetrieverConfig

