Updated get_workflow funtion with get_session_history funtion

workflow.py

@@ -1,28 +1,34 @@
-#from langchain_openai import ChatOpenAI
+from langchain.schema import BaseMemory
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 from langchain_core.runnables.history import RunnableWithMessageHistory
 from config import get_chat_model
 
+def get_session_history(memory: BaseMemory):
+    """Retrieve the session history from memory."""
+    return memory.chat_memory.messages if hasattr(memory, "chat_memory") else []
+
 def get_workflow():
-    """Set up the chatbot workflow using the updated RunnableWithMessageHistory."""
+    """Set up the chatbot workflow with memory and prompt template."""
 
-    # Define the prompt template for regular conversation
+    # Define the prompt template for conversation
     prompt_template = PromptTemplate(
-        input_variables=["input"],  # Match the memory variable name
-        template="You are a helpful assistant. Answer the question: {input}"
+        input_variables=["input"],
+        template="You are a helpful assistant. Answer the input: {input}"
     )
 
-    # Create a memory object to remember previous conversations
+    # Create memory for conversation
     memory = ConversationBufferMemory(memory_key="input", return_messages=True)
 
-    # Fetch the pre-configured model from config.py
+    # Fetch the chat model
     chat_model = get_chat_model()
 
-    # Create the updated RunnableWithMessageHistory
+    # Use RunnableWithMessageHistory for session memory
     conversation_chain = RunnableWithMessageHistory(
+        llm=chat_model,
+        get_session_history=lambda: get_session_history(memory),
         memory=memory,
-        runnable=chat_model,
+        verbose=True,
         prompt=prompt_template
     )