NvidiaTextEmbedder Parameters

Check the init parameters you can specify for NvidiaTextEmbeder.

YAML Init Parameters

These are the parameters you can specify in pipeline YAML:

ParameterTypePossible valuesDescription
api_keySecretDefault: Secret.from_env_var("NVIDIA_API_KEY")The Nvidia API key.
Required.
modelStringDefault: NV-Embed-QA,The name of the model to use.
Required.
api_urlStringDefault: "https://ai.api.nvidia.com/v1/retrieval/nvidia"Custom API URL for NVIDIA NIM. The URL follows the format: https://host:port.
Required.
prefixStringDefault: ""(empty string)A string to add at the beginning of each text. Can be used to prepend the text with an instruction, as required by some embedding models, such as E5 and bge.
Required.
suffixStringDefault: ""(empty string)A string to add at the end of each text.
Required.
truncateUnion of EmbeddingTruncateMode and stringSTART
END
Default: None
Specifies how inputs longer than the maximum token length should be truncated. Possible values:

- START: The input is truncated from the start.
- END: The input is truncated from the end.
- None: The behavior depends on the model. See model documentation for information.
Optional.