nat.tool.document_search

Attributes

logger

Classes

MilvusDocumentSearchToolConfig

This tool retrieves relevant documents for a given user query. The input query is mapped to the most appropriate

Functions

document_search(config, builder)

Module Contents

logger

class MilvusDocumentSearchToolConfig

Bases: nat.data_models.function.FunctionBaseConfig

This tool retrieves relevant documents for a given user query. The input query is mapped to the most appropriate Milvus collection database. This will return relevant documents from the selected collection.

base_url: str = None

top_k: int = None

timeout: int = None

llm_name: nat.data_models.component_ref.LLMRef = None

collection_names: list = None

collection_descriptions: list = None

async document_search(

config: MilvusDocumentSearchToolConfig,

builder: nat.builder.builder.Builder,

)