langchain docstore

agent.py

@@ -17,6 +17,7 @@ from smolagents import DuckDuckGoSearchTool
 from smolagents import Tool
 from langchain.vectorstores import FAISS
 import faiss
+from langchain_community.docstore.in_memory import InMemoryDocstore
 
 # Load environment variables
 load_dotenv()
@@ -68,7 +69,8 @@ sys_msg = SystemMessage(content=system_prompt)
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
 embedding_dim = 768  # for 'all-mpnet-base-v2'
 empty_index = faiss.IndexFlatL2(embedding_dim)
-vector_store = FAISS(embedding_function=embeddings, index=empty_index, docstore={}, index_to_docstore_id={})
+docstore = InMemoryDocstore({})
+vector_store = FAISS(embedding_function=embeddings, index=empty_index, docstore=docstore, index_to_docstore_id={})
 
 
 retriever_tool = create_retriever_tool(

requirements.txt

@@ -19,3 +19,4 @@ python-dotenv
 smolagents
 faiss-cpu
 gradio[oauth]
+langchain-community