Document Retrievers package

The package provides methods for creating and getting a RAG retreiver for a particular document store.

Submodules

sax.core.synthesis.rag.base_retrieval module

class sax.core.synthesis.rag.base_retrieval.BaseRetriever

Bases: object

Base class for all document retriever wrappers

abstract get_retriever()

Get the embedded retriever

sax.core.synthesis.rag.documents_retrieval module

class sax.core.synthesis.rag.documents_retrieval.DocumentRetrieverLLM(modelClass: BaseLLM, docLocation, filterFiles=None, chunk_size=None, chunk_overlap=None, retrieved_results=None, dbPath=None)

Bases: BaseRetriever

A document retrieval imeplementation based on Chroma vector store

get_retriever()

Return the embedded retriever stored in this wrapper

Module contents