import os
from pymilvus import connections, FieldSchema, CollectionSchema, DataType, Collection
from sentence_transformers import SentenceTransformer

milvus_token = os.getenv("MILVUS_TOKEN")

COLLECTION_NAME = "final_ragbench_document_embeddings"
MILVUS_CLOUD_URI = "https://in03-7b4da1b7b588a88.serverless.gcp-us-west1.cloud.zilliz.com"
connections.connect("default", uri=MILVUS_CLOUD_URI, token=milvus_token)

# Verify connection
print(connections.get_connection_addr("default"))