latent.rag.protocols

Extension-point protocols for RAG components.

Classes

EmbeddingProvider

EmbeddingProvider()

Protocol for text embedding providers.

QueryExpander

QueryExpander()

Protocol for query expansion (e.g. Hebrew synonyms).

Retriever

Retriever()

Protocol for vector/keyword retrieval backends.

Methods

EmbeddingProvider.embed_documents

embed_documents(texts: list[str]) -> list[list[float]]

EmbeddingProvider.embed_query

embed_query(text: str) -> list[float]

QueryExpander.expand

expand(query: str) -> list[str]

QueryExpander.expand_tokens

expand_tokens(query: str) -> list[str]

Retriever.add_texts

add_texts(texts: list[str], metadatas: list[dict] | None = None) -> None

Retriever.delete_collection

delete_collection() -> None

Retriever.search

search(query: str, k: int = 5) -> list[RetrievedChunk]

Retriever.search_multi

search_multi(queries: list[str], k: int = 5) -> list[RetrievedChunk]

Retriever.search_with_threshold

search_with_threshold(query: str, k: int = 5, threshold: float = 0.0) -> list[RetrievedChunk]