Update auditqa/process_chunks.py

auditqa/process_chunks.py

@@ -102,9 +102,23 @@ def load_chunks():
             qdrant_collections[file] = Qdrant.from_documents(
                 value,
                 embeddings,
-                location=":memory:", 
+                path="/data/local_qdrant",
                 collection_name=file,
             )
     print(qdrant_collections)
     print("vector embeddings done")
+    return qdrant_collections
+
+def get_local_qdrant(): 
+    config = getconfig("./model_params.cfg")
+    qdrant_collections = {}
+    embeddings = HuggingFaceEmbeddings(
+        model_kwargs = {'device': device},
+        encode_kwargs = {'normalize_embeddings': True},
+        model_name=config.get('retriever','MODEL'))
+    #list_ = ['Consolidated','District','Ministry','allreports']
+    #for val in list_:
+    client = QdrantClient(path="/data/local_qdrant") 
+    print(client.get_collections())
+    qdrant_collections['allreports'] = Qdrant(client=client, collection_name='allreports', embeddings=embeddings, )
     return qdrant_collections