ShadowTEM commited on
Commit
7418d29
·
verified ·
1 Parent(s): ea4caa0

Update app/vectorstore.py

Browse files
Files changed (1) hide show
  1. 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")