nat.retriever.milvus.register

Classes

MilvusRetrieverConfig

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

Functions

milvus_retriever(retriever_config, builder)
milvus_retriever_client(config, builder)

Module Contents

class MilvusRetrieverConfig

Bases: nat.data_models.retriever.RetrieverBaseConfig

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

uri: pydantic.HttpUrl = None

connection_args: dict = None

embedding_model: str = None

collection_name: str | None = None

content_field: str = None

top_k: int | None = None

output_fields: list[str] | None = None

search_params: dict = None

vector_field: str = None

description: str | None = None

use_async_client: bool = None

async milvus_retriever(

retriever_config: MilvusRetrieverConfig,

builder: nat.builder.builder.Builder,

)

async milvus_retriever_client(

config: MilvusRetrieverConfig,

builder: nat.builder.builder.Builder,

)