ppsingh commited on
Commit
aaf6c15
1 Parent(s): 331d7cb

Update auditqa/doc_process.py

Browse files
Files changed (1) hide show
  1. auditqa/doc_process.py +8 -9
auditqa/doc_process.py CHANGED
@@ -85,13 +85,14 @@ def process_pdf():
85
 
86
 
87
  for file,value in all_documents.items():
88
- print("emebddings for:",file)
89
- qdrant_collections[file] = Qdrant.from_documents(
90
- value,
91
- embeddings,
92
- location=":memory:",
93
- collection_name=file,
94
- )
 
95
  print(qdrant_collections)
96
  print("vector embeddings done")
97
  return qdrant_collections
@@ -108,6 +109,4 @@ def get_local_qdrant():
108
  print(client.get_collections())
109
  qdrant_collections[val] = Qdrant(client=client, collection_name=val, embeddings=embeddings, )
110
  return qdrant_collections
111
-
112
-
113
 
 
85
 
86
 
87
  for file,value in all_documents.items():
88
+ if file == "allreports"
89
+ print("emebddings for:",file)
90
+ qdrant_collections[file] = Qdrant.from_documents(
91
+ value,
92
+ embeddings,
93
+ location=":memory:",
94
+ collection_name=file,
95
+ )
96
  print(qdrant_collections)
97
  print("vector embeddings done")
98
  return qdrant_collections
 
109
  print(client.get_collections())
110
  qdrant_collections[val] = Qdrant(client=client, collection_name=val, embeddings=embeddings, )
111
  return qdrant_collections
 
 
112