`nemo_microservices.types.retriever_pipeline_data_param1`#

Module Contents#

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

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

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

class nemo_microservices.types.retriever_pipeline_data_param1.RetrieverPipelineDataParam#

Bases: typing_extensions.TypedDict

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

None

The index embedding model.

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

None

The query embedding model.

reranker_model: nemo_microservices.types.retriever_pipeline_data_param1.RerankerModel#

None

The reranker model.

top_k: int#

None

The top k results to be used.