aiq.tool.nvidia_rag#

Attributes#

logger

Classes#

NVIDIARAGToolConfig

Tool used to search the NVIDIA Developer database for documents across a variety of NVIDIA asset types.

Functions#

nvidia_rag_tool(config, builder)

Module Contents#

logger#
class NVIDIARAGToolConfig(/, **data: Any)#

Bases: aiq.data_models.function.FunctionBaseConfig

Tool used to search the NVIDIA Developer database for documents across a variety of NVIDIA asset types.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

base_url: str = None#
timeout: int = None#
document_separator: str = None#
document_prompt: str = None#
top_k: int = None#
collection_name: str = None#
async nvidia_rag_tool(
config: NVIDIARAGToolConfig,
builder: aiq.builder.builder.Builder,
)#