Update my_memory_logic.py

my_memory_logic.py

@@ -1,5 +1,3 @@
-# my_memory_logic.py
-
 import os
 
 # Import the PipelineRunnable from pipeline.py
@@ -34,8 +32,16 @@ conversational_rag_chain = RunnableWithMessageHistory(
 # 3) Convenience function for session-based memory
 ###############################################################################
 def run_with_session_memory(user_query: str, session_id: str) -> str:
+    if not user_query:
+        raise ValueError("User query is missing. Ensure the 'input' key is provided.")
+    
+    # Prepare input dictionary with the correct structure
+    input_data = {"input": user_query}
+    
+    # Ensure the session ID is passed correctly in the configuration
     response = conversational_rag_chain.invoke(
-        {"input": user_query},
-        config={"configurable": {"session_id": session_id}}
+        input_data,  # Pass the dictionary containing the 'input' key
+        config={"configurable": {"session_id": session_id}}  # Pass session ID in the config
     )
-    return response["answer"]
+    
+    return response.get("answer", "No answer returned from the chain.")