Update agent.py

agent.py

@@ -353,7 +353,7 @@ class MyVectorStore:
 
     def save_local(self, path: str):
         # Save the FAISS index to the specified file
-        faiss.write_index(self.index, "/home/wendy/Downloads")
+        faiss.write_index(self.index, path)
         print(f"Index saved to {path}") 
     
     @classmethod
@@ -397,10 +397,17 @@ vector_store = FAISS.from_documents(docs, embedding_model)
 
 # Save the FAISS index
 vector_store = MyVectorStore(index)
-vector_store.save_local("/home/wt/Downloads/faiss_index.index")
 
-# Load the FAISS index later
-loaded_vector_store = MyVectorStore.load_local("faiss_index.index")
+# Assuming you already have a FAISS index
+index = faiss.IndexFlatL2(768)  # Or whatever dimensionality your embeddings have
+vector_store = MyVectorStore(index)
+
+# Save
+vector_store.save_local("/home/wendy/Downloads/faiss_index.index")
+
+# Load
+loaded_store = MyVectorStore.load_local("/home/wendy/Downloads/faiss_index.index")
+