UPDATE FUNCTION CHAIN IN MEMORY.PY

my_memory_logic.py

@@ -32,7 +32,7 @@ def get_session_history(session_id: str) -> BaseChatMessageHistory:
 # This wraps your `rag_chain` so it automatically reads/writes
 # conversation history from get_session_history for each session.
 conversational_rag_chain = RunnableWithMessageHistory(
-    rag_chain,               # the main chain from pipeline.py
+    run_with_chain,               # the main chain from pipeline.py
     get_session_history,     # fetches or creates ChatMessageHistory by session_id
     input_messages_key="input",       # key in the dict for user's new query
     history_messages_key="chat_history",  # key for existing chat logs