github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 17:08:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Order messages in PostgresChatMessageHistory (#6830)

Browse Source 
Fixes issue: https://github.com/hwchase17/langchain/issues/6829

This guarantees message history is in the correct order. 

---------

Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
This commit is contained in:
Matt Florence 2023-06-29 21:10:28 -04:00 committed by GitHub
parent e9950392dd
commit 0f6737735d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/memory/chat_message_histories/postgres.py
View File

@ -49,7 +49,9 @@ class PostgresChatMessageHistory(BaseChatMessageHistory):
@property @property
def messages(self) -> List[BaseMessage]: # type: ignore def messages(self) -> List[BaseMessage]: # type: ignore
"""Retrieve the messages from PostgreSQL""" """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,)) self.cursor.execute(query, (self.session_id,))
items = [record["message"] for record in self.cursor.fetchall()] items = [record["message"] for record in self.cursor.fetchall()]
messages = messages_from_dict(items) messages = messages_from_dict(items)