RetrieverPipelineData

class nemo_microservices.types.RetrieverPipelineData(*args: Any, **kwargs: Any)

Bases: BaseModel

index_embedding_model: str | ModelEv

The index embedding model.

query_embedding_model: str | ModelEv

The query embedding model.

reranker_model: str | ModelEv | None = None

The reranker model.

top_k: int | None = None

The top k results to be used.