`nemo_microservices.types.retriever_pipeline_data`#

Module Contents#

nemo_microservices.types.retriever_pipeline_data.IndexEmbeddingModel: typing_extensions.TypeAlias#: None

nemo_microservices.types.retriever_pipeline_data.QueryEmbeddingModel: typing_extensions.TypeAlias#: None

nemo_microservices.types.retriever_pipeline_data.RerankerModel: typing_extensions.TypeAlias#: None

class nemo_microservices.types.retriever_pipeline_data.RetrieverPipelineData(/, **data: typing.Any)#

Bases: nemo_microservices._models.BaseModel

index_embedding_model: nemo_microservices.types.retriever_pipeline_data.IndexEmbeddingModel#

None

The index embedding model.

query_embedding_model: nemo_microservices.types.retriever_pipeline_data.QueryEmbeddingModel#

None

The query embedding model.

reranker_model: Optional[nemo_microservices.types.retriever_pipeline_data.RerankerModel]#

None

The reranker model.

top_k: Optional[int]#

None

The top k results to be used.