EDITED SAVE FUNCTION
Browse files- app/vectorstore.py +1 -10
app/vectorstore.py
CHANGED
|
@@ -172,18 +172,9 @@ class vectorstore:
|
|
| 172 |
def save_vectorstore_with_timestamp_and_without(self, vectorstore, embedder_model=None):
|
| 173 |
vector_db_folder = os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")), "VectorDB")
|
| 174 |
os.makedirs(vector_db_folder, exist_ok=True)
|
| 175 |
-
|
| 176 |
-
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
| 177 |
-
folder_name = f"vectorstore_{timestamp}"
|
| 178 |
-
folder_path = os.path.join(vector_db_folder, folder_name)
|
| 179 |
-
|
| 180 |
-
# Save the LangChain FAISS vectorstore
|
| 181 |
-
vectorstore.save_local(folder_path)
|
| 182 |
-
|
| 183 |
# Save metadata with embedder model name
|
| 184 |
metadata = {"embedder_model": embedder_model or "all-MiniLM-L6-v2"}
|
| 185 |
-
with open(os.path.join(folder_path, "metadata.json"), "w") as f:
|
| 186 |
-
json.dump(metadata, f)
|
| 187 |
|
| 188 |
# Save to main folder
|
| 189 |
main_folder = os.path.join(vector_db_folder, "vectorstore_mainV2")
|
|
|
|
| 172 |
def save_vectorstore_with_timestamp_and_without(self, vectorstore, embedder_model=None):
|
| 173 |
vector_db_folder = os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")), "VectorDB")
|
| 174 |
os.makedirs(vector_db_folder, exist_ok=True)
|
| 175 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 176 |
# Save metadata with embedder model name
|
| 177 |
metadata = {"embedder_model": embedder_model or "all-MiniLM-L6-v2"}
|
|
|
|
|
|
|
| 178 |
|
| 179 |
# Save to main folder
|
| 180 |
main_folder = os.path.join(vector_db_folder, "vectorstore_mainV2")
|