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

Bases: aiq.data_models.retriever.RetrieverBaseConfig

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

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.

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

async milvus_retriever(

retriever_config: MilvusRetrieverConfig,

builder: aiq.builder.builder.Builder,

)

async milvus_retriever_client(

config: MilvusRetrieverConfig,

builder: aiq.builder.builder.Builder,

)