Update agent.py

agent.py

@@ -342,24 +342,6 @@ for name in enabled_tool_names:
     tools.append(tool_map[name])
 
     
-# -----------------------------
-# Create FAISS Vector Store
-# -----------------------------
-import faiss
-class MyVectorStore:
-    def __init__(self, index: faiss.Index):
-        self.index = index
-
-    def save_local(self, path: str):
-        # Save the FAISS index to the specified file
-        faiss.write_index(self.index, path)
-        print(f"Index saved to {path}") 
-    
-    @classmethod
-    def load_local(cls, path: str):
-        # Load the FAISS index from the specified file
-        index = faiss.read_index(path)
-        return cls(index)
 
 # -----------------------------
 # Prepare Documents
@@ -431,8 +413,7 @@ class MyVectorStore:
         index = faiss.read_index(path)
         return cls(index)
 
-# Usage Example
-if __name__ == "__main__":
+
     # Process JSON data
     with open("questions.json", "r", encoding="utf-8") as f:
         json_data = json.load(f)
@@ -465,7 +446,7 @@ if __name__ == "__main__":
 # Create LangChain Retriever Tool
 # -----------------------------
 
-retriever = FAISS.load_local("faiss_index.index", embedding_model).as_retriever()
+retriever = FAISS. loaded_store("/home/wendy/Downloads/faiss_index.index", embedding_model).as_retriever()
 
 question_retriever_tool = create_retriever_tool(
     retriever=retriever,