ppsingh commited on
Commit
29b43fb
1 Parent(s): 376f540

Update auditqa/doc_process.py

Browse files
Files changed (1) hide show
  1. auditqa/doc_process.py +2 -3
auditqa/doc_process.py CHANGED
@@ -86,15 +86,14 @@ def process_pdf():
86
  qdrant_collections[file] = Qdrant.from_documents(
87
  value,
88
  embeddings,
89
- path=f"./data/{file}",
90
  collection_name=file,
91
  )
92
 
93
  print("done")
94
  return qdrant_collections
95
 
96
- def get_local_qdrant(name):
97
- client = QdrantClient(path=f"./data/{name}")
98
  print(client.get_collections())
99
  embeddings = HuggingFaceEmbeddings(
100
  model_kwargs = {'device': device},
 
86
  qdrant_collections[file] = Qdrant.from_documents(
87
  value,
88
  embeddings,
89
+ path=f"./data/local_qdrant",
90
  collection_name=file,
91
  )
92
 
93
  print("done")
94
  return qdrant_collections
95
 
96
+ def get_local_qdrant(client, name):
 
97
  print(client.get_collections())
98
  embeddings = HuggingFaceEmbeddings(
99
  model_kwargs = {'device': device},