base

Base classes for database.

Classes:

• DBBase –

  Base class for vector database.

DBBase

Base class for vector database.

Methods:

• embed –

  Embed the documents into the database.

• search –

  Search a query from documents.

embed abstractmethod

embed(documents: Any) -> None

Embed the documents into the database.

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

@abstractmethod
def embed(self, documents: Any) -> None:
    """Embed the documents into the database."""
    ...

search abstractmethod

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

Search a query from documents.

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

@abstractmethod
def search(
    self, query_encoded: Any, top_k: int = 2
) -> tuple[Iterable[float], Union[Iterable[str], Iterable[int]]]:
    """Search a query from documents."""
    ...