Saltar al contenido principal

ParallelRetriever

Retriever
DashAI.back.models.RAG.retrievers.composite.ParallelRetriever

Queries multiple retrievers in parallel and merges their results.

Results are deduplicated by (document_id, document_position).

Parameters​

enable_retrieve_then_expand : boolean, default=False
Widen each retrieved chunk with its neighbouring chunks before the chunks are returned as the retriever's context.
expand_window : integer, default=3
Total number of chunks in the expanded window, the retrieved one included. Odd values are centred exactly on the retrieved chunk; even values keep the extra chunk after it. Windows never leave the document, and overlapping windows merge into one concatenated chunk.
children : array, default=[]
List of child retrievers queried in parallel.
merge_strategy : string, default=round_robin
'round_robin': alternates results from each retriever. 'interleave': merges preserving internal order.

Methods​

retrieve(self, query, **kwargs) -> Tuple[List[DashAI.back.models.RAG.documents.chunk.Chunk], Dict[str, Any]]

Defined on ParallelRetriever

Retrieve chunks from all children in parallel and merge.

score_chunks(self, chunk_ids: List[int], query: str) -> List[Tuple[int, float]]

Defined on ParallelRetriever

Aggregate chunk scores across all children.

add(self, child: DashAI.back.models.RAG.retrievers.base_retriever.BaseRetriever) -> None

Defined on CompositeRetriever

Add a child retriever.

close(self) -> None

Defined on CompositeRetriever

Release every child's resources and this composite's own.

Returns

None
Nothing.

expand_retrieved_chunks(self, chunks: List[DashAI.back.models.RAG.documents.chunk.Chunk], window_size: int) -> List[DashAI.back.models.RAG.documents.chunk.Chunk]

Defined on BaseRetriever

Expand chunks into centred windows and merge overlapping windows.

get_children(self) -> List[DashAI.back.models.RAG.retrievers.base_retriever.BaseRetriever]

Defined on CompositeRetriever

Return a copy of the children list.

get_credential(self, name: str)

Defined on ConfigObject

Resolve a registered credential component by name.

Parameters

name : str
Credential component class name (e.g. "HuggingFaceCredential").

Returns

BaseCredential
An instance of the requested credential component.

get_id(self) -> int | None

Defined on BaseRetriever

Return the database ID of this retriever, or None.

get_metadata(cls) -> Dict[str, Any]

Defined on BaseRetriever

Return the UI metadata shared by every retriever.

Returns

Dict[str, Any]
Dictionary with the icon shown for the retriever in the frontend and, when declared, its representation template and the category the retriever pickers group it under.

get_representation(cls) -> DashAI.back.core.utils.MultilingualString | None

Defined on BaseRetriever

Return the retriever's representation template, if any.

Returns

MultilingualString | None
The template with {param} / {nested.param} placeholders, or None when the frontend should fall back to the component description.

get_schema(cls) -> dict

Defined on ConfigObject

Generates the component related Json Schema.

Returns

dict
Dictionary representing the Json Schema of the component.

get_schema_summary(self) -> DashAI.back.models.RAG.retrievers.retrieval_contracts.SchemaSummary

Defined on BaseRetriever

Return a recursive, JSON-serializable summary of this retriever.

Returns

SchemaSummary
{"class": ..., "params": ...} plus children when present.

get_schema_summary_parameters(cls) -> Dict[str, str]

Defined on BaseRetriever

Return a copy of the summary parameter mapping for the frontend.

init_model(self) -> None

Defined on BaseRetriever

Called by the factory after inject_infra().

inject_infra(self, env_RAG_path: str | os.PathLike, chunks: Dict[int, Dict[int, DashAI.back.models.RAG.documents.chunk.Chunk]], persistence: Any, documents: Optional[Dict[int, DashAI.back.models.RAG.documents.source_document.SourceDocument]] = None) -> None

Defined on BaseRetriever

Inject runtime infrastructure after schema validation.

load(self, filename: str = '') -> None

Defined on BaseRetriever

Restore the retriever's state from disk.

merge_expanded_chunks(self, chunks: List[DashAI.back.models.RAG.documents.chunk.Chunk]) -> List[DashAI.back.models.RAG.documents.chunk.Chunk]

Defined on BaseRetriever

Merge overlapping expanded units without widening plain chunks.

remove(self, child: DashAI.back.models.RAG.retrievers.base_retriever.BaseRetriever) -> None

Defined on CompositeRetriever

Remove a child retriever.

save(self, filename: str = '') -> None

Defined on BaseRetriever

Persist the retriever's state to disk.

set_id(self, id: int) -> None

Defined on BaseRetriever

Assign a database ID to this retriever.

set_trace_params(self, params: Dict[str, Any]) -> None

Defined on BaseRetriever

Store a JSON-serializable snapshot of this retriever's configuration.

Parameters

params : Dict[str, Any]
Raw construction parameters captured by the factory before any nested component reference is consumed. A deep copy is stored so later mutations of the original mapping cannot affect trace rendering.

validate_and_transform(self, raw_data: dict) -> dict

Defined on ConfigObject

It takes the data given by the user to initialize the model and returns it with all the objects that the model needs to work.

Parameters

raw_data : dict
A dictionary with the data provided by the user to initialize the model.

Returns

dict
A validated dictionary with the necessary objects.