Skip to content

Commit c7cdc3d

Browse files
author
cinjospeh
committed
implement function get_last_message in DefaultGptsMessageMemeory
1 parent a1a6bb3 commit c7cdc3d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

dbgpt/agent/core/memory/gpts/default_gpts_memory.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,4 +146,7 @@ def get_by_conv_id(self, conv_id: str) -> List[GptsMessage]:
146146

147147
def get_last_message(self, conv_id: str) -> Optional[GptsMessage]:
148148
"""Get the last message in the conversation."""
149-
return None
149+
messages: List[GptsMessage] = self.get_by_conv_id(conv_id)
150+
if messages is None or len(messages) == 0:
151+
return None
152+
return messages[-1]

0 commit comments

Comments
 (0)