Update app/vectorstore.py
Browse files- app/vectorstore.py +10 -1
app/vectorstore.py
CHANGED
|
@@ -172,9 +172,18 @@ 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 |
# 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")
|
|
|
|
| 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")
|