RagBenchCapstone10 / crudmilvus.py
Saiteja Solleti
milvas connection
2779cfc
raw
history blame
482 Bytes
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"))