jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update add_messages and add_messagetables functions to return Message objects

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-25 20:20:12 -03:00
commit d1d5eb6e39
3 changed files with 15 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_messages_endpoints.py
View file

@ -30,11 +30,9 @@ def created_messages(session):
MessageCreate(text="Test message 3", sender="User", sender_name="User", session_id="session_id2"),
]
messagetables = [MessageTable.model_validate(message, from_attributes=True) for message in messages]
messagetables = add_messagetables(messagetables, session)
messages_read = [
MessageRead.model_validate(messagetable, from_attributes=True) for messagetable in messagetables
]
return messages_read
message_list = add_messagetables(messagetables, session)
return message_list
def test_delete_messages(client: TestClient, created_messages, logged_in_headers):