nemo_microservices.types.retriever_pipeline_data

Module Contents

Classes

RetrieverPipelineData

Data

IndexEmbeddingModel
QueryEmbeddingModel
RerankerModel

API

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: 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: nemo_microservices.types.retriever_pipeline_data.RerankerModel | None

None

The reranker model.

top_k: int | None

None

The top k results to be used.