Update Index.py

Index.py

@@ -96,7 +96,7 @@ def _upsert_docs(doc, embeddings, vector_doc_path: str, db_present: bool):
     print(vector_doc_path)
     if db_present:
         print(1)
-        max_index = _max_index_id(f"{vector_doc_path}/documents")
+        max_index = _max_index_id(f"{vector_doc_path}")
         print(max_index)
         embeddings.upsert(_stream(doc, 500, max_index["max_index"]))
         print("Embeddings done!!")
@@ -106,7 +106,7 @@ def _upsert_docs(doc, embeddings, vector_doc_path: str, db_present: bool):
         print(2)
         embeddings.index(_stream(doc, 500, 0))
         embeddings.save(vector_doc_path)
-        max_index = _max_index_id(f"{vector_doc_path}/documents")
+        max_index = _max_index_id(f"{vector_doc_path}")
         print(max_index)
     # check
     # max_index = _max_index_id(f"{vector_doc_path}/documents")
@@ -157,7 +157,7 @@ def get_domain_file_path(domain: str, file_path: str):
         max_index = _upsert_docs(
             doc=doc,
             embeddings=embeddings,
-            vector_doc_path=f"{os.getcwd()}/index/{domain}/documents",
+            vector_doc_path=f"{os.getcwd()}/index/{domain}",
             db_present=bool_value,
         )
         # print("-------")
@@ -167,7 +167,7 @@ def get_domain_file_path(domain: str, file_path: str):
         max_index = _upsert_docs(
             doc=doc,
             embeddings=embeddings,
-            vector_doc_path=f"{os.getcwd()}/index/{domain}/documents",
+            vector_doc_path=f"{os.getcwd()}/index/{domain}",
             db_present=bool_value,
         )
     # print("Final - output : ", max_index)