Create my_memory_logic.py

my_memory_logic.py

@@ -0,0 +1,21 @@
+# my_memory_logic.py
+from langchain.memory import ConversationBufferMemory
+from langchain.chat_models import ChatOpenAI
+from langchain.chains import LLMChain
+from langchain.prompts.chat import ChatPromptTemplate, SystemMessagePromptTemplate, MessagesPlaceholder, HumanMessagePromptTemplate
+
+memory = ConversationBufferMemory(return_messages=True)
+
+restatement_system_prompt = (
+    "Given a chat history and the latest user question "
+    "which might reference context in the chat history, ..."
+)
+
+restatement_prompt = ChatPromptTemplate.from_messages([
+    SystemMessagePromptTemplate.from_template(restatement_system_prompt),
+    MessagesPlaceholder(variable_name="chat_history"),
+    HumanMessagePromptTemplate.from_template("{input}")
+])
+
+restatement_llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.0)
+restatement_chain = LLMChain(llm=restatement_llm, prompt=restatement_prompt)