Update agent.py

agent.py

@@ -221,11 +221,12 @@ for task in tasks:
 # Initialize HuggingFace Embedding model
 embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
 
-# Create FAISS VectorStore from documents
-vector_store = FAISS.from_documents(docs, embedding_model)
-#vector_store = FAISS.load_local("faiss_index", embedding_model)
 
-# Save the FAISS index locally
+from langchain_community.document_loaders import WikipediaLoader
+
+wiki_docs = WikipediaLoader(query="Mercedes Sosa", load_max_docs=3).load()
+all_docs = docs + wiki_docs  # `docs` is your existing list
+vector_store = FAISS.from_documents(all_docs, embedding_model)
 vector_store.save_local("faiss_index")