ESGToolKit / config.py
GirishaBuilds01's picture
Update config.py
37156ae verified
raw
history blame contribute delete
586 Bytes
import os
from qdrant_client import QdrantClient
from qdrant_client.http.models import VectorParams, Distance
COLLECTION = "esg_vectors"
VECTOR_SIZE = 512
os.makedirs("qdrant_db", exist_ok=True)
client = QdrantClient(path="qdrant_db")
def init():
collections = [c.name for c in client.get_collections().collections]
if COLLECTION not in collections:
client.create_collection(
collection_name=COLLECTION,
vectors_config=VectorParams(
size=VECTOR_SIZE,
distance=Distance.COSINE
)
)
init()