aiq.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(/, **data: Any)

Bases: aiq.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.

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

top_k: int = None

timeout: int = None

llm_name: aiq.data_models.component_ref.LLMRef = None

collection_names: list = None

collection_descriptions: list = None

async document_search(

config: MilvusDocumentSearchToolConfig,

builder: aiq.builder.builder.Builder,

)