Fixed more deprecation error

config.py

@@ -1,5 +1,5 @@
 import os
-from langchain_community.chat_models import ChatOpenAI
+from langchain_openai import ChatOpenAI
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 
 

workflow.py

@@ -1,30 +1,29 @@
-from langchain.chains import ConversationChain
+#from langchain_openai import ChatOpenAI
 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_workflow():
-    """Set up the chatbot workflow with memory and prompt template."""
-    
+    """Set up the chatbot workflow using the updated RunnableWithMessageHistory."""
+
     # Define the prompt template for regular conversation
     prompt_template = PromptTemplate(
-        input_variables=["question"],
-        template="You are a helpful assistant. Answer the question: {question}"
+        input_variables=["input"],  # Match the memory variable name
+        template="You are a helpful assistant. Answer the question: {input}"
     )
-    
+
     # Create a memory object to remember previous conversations
-    memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
+    memory = ConversationBufferMemory(memory_key="input", return_messages=True)
 
-    chat_model = get_chat_model()  # Fetch the configured model from config.py
+    # Fetch the pre-configured model from config.py
+    chat_model = get_chat_model()
 
-    
-    # Create the conversation chain
-    conversation_chain = ConversationChain(
-        llm=chat_model,
+    # Create the updated RunnableWithMessageHistory
+    conversation_chain = RunnableWithMessageHistory(
         memory=memory,
-        verbose=True,
+        runnable=chat_model,
         prompt=prompt_template
     )
-    
+
     return conversation_chain