From 0f6737735d44cb35cc7c1377a88dabad0982e659 Mon Sep 17 00:00:00 2001 From: Matt Florence Date: Thu, 29 Jun 2023 21:10:28 -0400 Subject: [PATCH] Order messages in PostgresChatMessageHistory (#6830) Fixes issue: https://github.com/hwchase17/langchain/issues/6829 This guarantees message history is in the correct order. --------- Co-authored-by: Dev 2049 --- langchain/memory/chat_message_histories/postgres.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/langchain/memory/chat_message_histories/postgres.py b/langchain/memory/chat_message_histories/postgres.py index 1b3ee7200bf..3d06346d265 100644 --- a/langchain/memory/chat_message_histories/postgres.py +++ b/langchain/memory/chat_message_histories/postgres.py @@ -49,7 +49,9 @@ class PostgresChatMessageHistory(BaseChatMessageHistory): @property def messages(self) -> List[BaseMessage]: # type: ignore """Retrieve the messages from PostgreSQL""" - query = f"SELECT message FROM {self.table_name} WHERE session_id = %s;" + query = ( + f"SELECT message FROM {self.table_name} WHERE session_id = %s ORDER BY id;" + ) self.cursor.execute(query, (self.session_id,)) items = [record["message"] for record in self.cursor.fetchall()] messages = messages_from_dict(items)