changes

src/main.py

@@ -16,7 +16,7 @@ from langchain.llms import OpenAI
 from langchain.memory import ConversationBufferMemory
 from langchain.vectorstores import Chroma
 from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.prompts import PromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate, AIMessagePromptTemplate, ChatPromptTemplate
+from langchain.prompts import PromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate, AIMessagePromptTemplate, ChatPromptTemplate, MessagesPlaceholder
 import datetime
 import shutil
 
@@ -87,8 +87,6 @@ def retreival(vector_store, k):
 
     CONTEXT: {context}
     =======
-    CHAT HISTORY: {chat_history}
-    =======
     FINAL ANSWER:"""
 
     human_template = """{question}"""
@@ -99,6 +97,7 @@ def retreival(vector_store, k):
     # Create the chat prompt templates
     messages = [
     SystemMessagePromptTemplate.from_template(system_template),
+    MessagesPlaceholder(variable_name="chat_history"),
     HumanMessagePromptTemplate.from_template(human_template),
     # AIMessagePromptTemplate.from_template(ai_template)
     ]