RetrieverPipelineDataParam#
- class nemo_microservices.types.RetrieverPipelineDataParam
Bases:
TypedDict- index_embedding_model: Required[str | ModelEvParam]
The index embedding model.
- query_embedding_model: Required[str | ModelEvParam]
The query embedding model.
- reranker_model: str | ModelEvParam
The reranker model.
- top_k: int
The top k results to be used.