Update app.py

app.py

@@ -23,6 +23,14 @@ from langchain.chains import LLMChain
 from langchain_core.prompts import PromptTemplate
 from langchain.memory.buffer import ConversationBufferMemory
 
+from pydantic import BaseModel, Field
+from langchain_core.runnables import (
+    RunnableLambda,
+    ConfigurableFieldSpec,
+    RunnablePassthrough,
+)
+from langchain_core.runnables.history import RunnableWithMessageHistory
+
 # ---------------------------------------------------for backend looks, example file:----------------------------------
 #with open('/home/user/.local/lib/python3.10/site-packages/socketio/async_server.py', 'r') as file:
     #content = file.read()
@@ -131,8 +139,8 @@ def init_session():
     #cl.user_session.set("memory", memory)
     # cl.user_session.set("connection_attempts", 0)
     # cl.user_session.set("connection_attempts", connection_attempts + 1)
-
-    llm_chain = RunnableWithHistory(
+  
+    llm_chain = RunnableWithMessageHistory(
         daysoff_assistant_prompt | llm
     ).configure_history(
         memory=conversation_memory,