Spaces:
Running
on
T4
Running
on
T4
Update auditqa/doc_process.py
Browse files- 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 |
-
|
| 89 |
-
|
| 90 |
-
|
| 91 |
-
|
| 92 |
-
|
| 93 |
-
|
| 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 |
|