base

Base classes for retrieval.

Classes:

• RetrievalBase –

  Base Retrieval class.

• StringRet –

  String Retrieval class.

RetrievalBase

RetrievalBase(
    source: Any,
    splitter: TextSplitterBase = LangChainTextSplitter(
        'RecursiveCharacterTextSplitter'
    ),
    api_key: str = '',
    cache: Optional[Cache] = None,
    logs: dict[str, Any] = DEFAULT_LOGS,
)

Bases: RagoBase

Base Retrieval class.

Methods:

• get –

  Get the data from the source.

Source code in src/rago/retrieval/base.py

def __init__(
    self,
    source: Any,
    splitter: TextSplitterBase = LangChainTextSplitter(
        'RecursiveCharacterTextSplitter'
    ),
    api_key: str = '',
    cache: Optional[Cache] = None,
    logs: dict[str, Any] = DEFAULT_LOGS,
) -> None:
    """Initialize the Retrieval class."""
    if logs is DEFAULT_LOGS:
        logs = {}
    super().__init__(api_key=api_key, cache=cache, logs=logs)
    self.source = source
    self.splitter = splitter

    self._validate()
    self._setup()

get abstractmethod

get(query: str = '') -> Iterable[str]

Get the data from the source.

Source code in src/rago/retrieval/base.py

@abstractmethod
def get(self, query: str = '') -> Iterable[str]:
    """Get the data from the source."""
    return []

StringRet

StringRet(
    source: Any,
    splitter: TextSplitterBase = LangChainTextSplitter(
        'RecursiveCharacterTextSplitter'
    ),
    api_key: str = '',
    cache: Optional[Cache] = None,
    logs: dict[str, Any] = DEFAULT_LOGS,
)

Bases: RetrievalBase

String Retrieval class.

This is a very generic class that assumes that the input (source) is already a list of strings.

Methods:

• get –

  Get the data from the sources.

Source code in src/rago/retrieval/base.py

def __init__(
    self,
    source: Any,
    splitter: TextSplitterBase = LangChainTextSplitter(
        'RecursiveCharacterTextSplitter'
    ),
    api_key: str = '',
    cache: Optional[Cache] = None,
    logs: dict[str, Any] = DEFAULT_LOGS,
) -> None:
    """Initialize the Retrieval class."""
    if logs is DEFAULT_LOGS:
        logs = {}
    super().__init__(api_key=api_key, cache=cache, logs=logs)
    self.source = source
    self.splitter = splitter

    self._validate()
    self._setup()

get

get(query: str = '') -> Iterable[str]

Get the data from the sources.

Source code in src/rago/retrieval/base.py

def get(self, query: str = '') -> Iterable[str]:
    """Get the data from the sources."""
    return cast(list[str], self.source)