Update custom_llm.py

custom_llm.py

@@ -49,7 +49,8 @@ def custom_chain_with_history(llm, memory):
     
     def prompt_memory(_):
       t = ""
-      for x in memory.chat_memory.messages:
+      # for x in memory.chat_memory.messages:
+      for x in memory.messages:
         t += f"<|assistant|>\n<s>{x.content}</s>\n" if type(x) is AIMessage else f"<|user|>\n{x.content}\n"
       return "" if len(t) == 0 else t