Update custom_llm.py

custom_llm.py

@@ -45,7 +45,7 @@ def custom_chain_with_history(llm, memory):
       print(len(docs))
       return "\n".join([f"{i+1}. {d.page_content}" for i,d in enumerate(docs)])
     
-    return {"chat_history":lambda x:prompt_memory(x['memory'])} | prompt | llm
+    return {"chat_history":lambda x:prompt_memory(x['memory']), "question":lambda x:x['question']} | prompt | llm
 
 
 class CustomLLM(LLM):