Skip to content

Datastore chunk retriever

Defines base class for datastore-based chunk retrievers.

This module provides the DatastoreChunkRetriever class, which serves as a foundation for implementing retrieval systems that use BaseDataStore and return list[Chunk].

DatastoreChunkRetriever(data_store)

Bases: BaseRetriever[list[Chunk]]

Base class for datastore-based retrievers that return list[Chunk].

This intermediate class provides common functionality for retrievers that: 1. Use a BaseDataStore for retrieval 2. Return list[Chunk] as results 3. Support threshold filtering and top_k limiting

Subclasses should implement _retrieve_single to call the appropriate datastore capability (vector, fulltext, etc.).

Initialize with a datastore.

Parameters:

Name Type Description Default
data_store BaseDataStore

The datastore to retrieve from.

required

Raises:

Type Description
TypeError

If data_store is not an instance of BaseDataStore.