GirishaBuilds01 commited on
Commit
a4c2dfd
·
verified ·
1 Parent(s): adb434d

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +26 -0
config.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from qdrant_client import QdrantClient
3
+ from qdrant_client.http.models import VectorParams, Distance
4
+
5
+ VECTOR_SIZE = 384
6
+ COLLECTION = "esg_vectors"
7
+
8
+ os.makedirs("qdrant_db", exist_ok=True)
9
+
10
+ client = QdrantClient(path="qdrant_db")
11
+
12
+ def init_collection():
13
+
14
+ collections = [c.name for c in client.get_collections().collections]
15
+
16
+ if COLLECTION not in collections:
17
+
18
+ client.create_collection(
19
+ collection_name=COLLECTION,
20
+ vectors_config=VectorParams(
21
+ size=VECTOR_SIZE,
22
+ distance=Distance.COSINE
23
+ )
24
+ )
25
+
26
+ init_collection()