rag_server/applications/utils/milvus/collection.py

from pymilvus import connections, CollectionSchema, Collection
from applications.utils.milvus.field import fields
from applications.config import MILVUS_CONFIG


connections.connect("default", **MILVUS_CONFIG)

schema = CollectionSchema(
    fields, description="Chunk multi-vector embeddings with metadata"
)
milvus_collection = Collection(name="chunk_multi_embeddings", schema=schema)


print("Connecting to Milvus Server...successfully")


__all__ = ["milvus_collection"]