Update variables.py

variables.py

@@ -6,6 +6,7 @@ import pathlib
 
 from langchain.embeddings import HuggingFaceEmbeddings,HuggingFaceInstructEmbeddings
 from langchain.vectorstores import FAISS
+from langchain.memory import ConversationBufferMemory
 from langchain.chat_models.openai import ChatOpenAI
 from langchain.callbacks.base import CallbackManager
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
@@ -97,6 +98,7 @@ def embed_tweets(query,_prompt,_docsearch):
     chain_type_kwargs = {"prompt": _prompt}
     chain = VectorDBQA.from_chain_type(
     ChatOpenAI(temperature=0), 
+    memory = ConversationBufferMemory(memory_key="chat_history", input_key="question"),
     chain_type="stuff", 
     vectorstore=_docsearch,
     chain_type_kwargs=chain_type_kwargs,