Retrieval API¶

Class responsible for retrieving data from given vector store.

Parameters:

• vector_store (BaseVectorStore) –

  Vector store / vector index to retrieve data from.

• method (Literal['simple'], default: 'simple' ) –

  Method describing how data should be retrieved.

• number_of_chunks (int) –

  Number of chunks to retrieve.

• search_mode (Literal['vector', 'hybrid'], default: "vector" ) –

  Search mode passed to the vector store: "vector" or "hybrid".

• ranker_strategy (str | None, default: None ) –

  Ranking strategy for hybrid search: "rrf", "weighted", or "normalized".

• ranker_k (int | None, default: None ) –

  Parameter k for the ranking function.

• ranker_alpha (float | None, default: None ) –

  Alpha parameter for weighted ranking strategy.

38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

def __init__(
    self,
    vector_store: BaseVectorStore,
    number_of_chunks: int,
    method: Literal["simple"] = "simple",
    search_mode: Literal["vector", "hybrid"] = "vector",
    ranker_strategy: str | None = None,
    ranker_k: int | None = None,
    ranker_alpha: float | None = None,
):
    self._vector_store = vector_store
    self.method = method
    self.number_of_chunks = number_of_chunks
    self.search_mode = search_mode
    self.ranker_strategy = ranker_strategy
    self.ranker_k = ranker_k
    self.ranker_alpha = ranker_alpha

retrieve(query: str, **kwargs) -> list[AI4RAGChunk]

56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

def retrieve(self, query: str, **kwargs) -> list[AI4RAGChunk]:
    """Retrieve relevant chunks from vector store.

    Parameters
    ----------
    query : str
        Question for which chunks should be retrieved.

    Returns
    -------
    list[AI4RAGChunk]
        Chunks with their metadata corresponding to the query.
    """
    _number_of_chunks = kwargs.get("number_of_chunks", self.number_of_chunks)

    return self._vector_store.search(
        query,
        k=_number_of_chunks,
        search_mode=self.search_mode,
        ranker_strategy=self.ranker_strategy,
        ranker_k=self.ranker_k,
        ranker_alpha=self.ranker_alpha,
    )