Update my_memory_logic.py

my_memory_logic.py

@@ -4,8 +4,7 @@ import os
 
 # Import your actual RAG chain (or pipeline) from pipeline.py
 # We'll assume `rag_chain` is exposed by pipeline.py
-from pipeline import run_with_chain  
-
+from pipeline import run_with_chain_context  
 # We'll import the session-based classes from langchain_core
 # If they're in different modules, adjust accordingly.
 from langchain_core.chat_history import BaseChatMessageHistory
@@ -32,7 +31,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(
-    run_with_chain,               # the main chain from pipeline.py
+    run_with_chain_context,               # 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