nat.retriever.nemo_retriever.register

Classes

NemoRetrieverConfig

Configuration for a Retriever which pulls data from a Nemo Retriever service.

Functions

nemo_retriever(retriever_config, builder)
nemo_retriever_client(config, builder)

Module Contents

class NemoRetrieverConfig

Bases: nat.data_models.retriever.RetrieverBaseConfig

Configuration for a Retriever which pulls data from a Nemo Retriever service.

uri: pydantic.HttpUrl = None

collection_name: str | None = None

top_k: int | None = None

output_fields: list[str] | None = None

timeout: int = None

nvidia_api_key: nat.data_models.common.OptionalSecretStr = None

async nemo_retriever(

retriever_config: NemoRetrieverConfig,

builder: nat.builder.builder.Builder,

)

async nemo_retriever_client(

config: NemoRetrieverConfig,

builder: nat.builder.builder.Builder,

)