faiss

Module for faiss database.

Classes:

• FaissDB –

  Faiss Database.

FaissDB

Bases: DBBase

Faiss Database.

Methods:

• embed –

  Embed the documents into the database.

• search –

  Search an encoded query into vector database.

embed

embed(documents: Any) -> None

Embed the documents into the database.

Source code in src/rago/augmented/db/faiss.py

def embed(self, documents: Any) -> None:
    """Embed the documents into the database."""
    self.index = faiss.IndexFlatL2(documents.shape[1])
    self.index.add(documents)

search

search(
    query_encoded: Any, top_k: int = 2
) -> tuple[Iterable[float], Iterable[int]]

Search an encoded query into vector database.

Source code in src/rago/augmented/db/faiss.py

def search(
    self, query_encoded: Any, top_k: int = 2
) -> tuple[Iterable[float], Iterable[int]]:
    """Search an encoded query into vector database."""
    distances, indices = self.index.search(query_encoded, top_k)
    return distances, indices[0]