`nemo_microservices.types.retriever_pipeline_data_param`#

Module Contents#

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

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

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

class nemo_microservices.types.retriever_pipeline_data_param.RetrieverPipelineDataParam#

Bases: typing_extensions.TypedDict

index_embedding_model: typing_extensions.Required[nemo_microservices.types.retriever_pipeline_data_param.IndexEmbeddingModel]#

None

The index embedding model.

query_embedding_model: typing_extensions.Required[nemo_microservices.types.retriever_pipeline_data_param.QueryEmbeddingModel]#

None

The query embedding model.

reranker_model: nemo_microservices.types.retriever_pipeline_data_param.RerankerModel#

None

The reranker model.

top_k: int#

None

The top k results to be used.