Refactor memory.py to ensure messages have a timestamp before adding them to the database
This commit is contained in:
parent
9ad10b33f2
commit
080af42802
1 changed files with 2 additions and 1 deletions
|
|
@ -74,7 +74,8 @@ def add_messages(messages: Message | list[Message], flow_id: Optional[str] = Non
|
|||
|
||||
messages_models: list[MessageModel] = []
|
||||
for msg in messages:
|
||||
msg.timestamp = monitor_service.get_timestamp()
|
||||
if not msg.timestamp:
|
||||
msg.timestamp = monitor_service.get_timestamp()
|
||||
messages_models.append(MessageModel.from_message(msg, flow_id=flow_id))
|
||||
|
||||
for message_model in messages_models:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue