Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat: return MemGPTMessage for Python Client (#1738)

Browse Source
This commit is contained in:
Sarah Wooders
2024-09-10 21:28:26 -07:00
committed by GitHub
parent 9fadd22fb8
commit c8fcbff39e
6 changed files with 49 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_client.py
View File

@@ -127,7 +127,7 @@ def test_agent_interactions(client: Union[LocalClient, RESTClient], agent: Agent
message = "Hello, agent!"
print("Sending message", message)
response = client.user_message(agent_id=agent.id, message=message)
response = client.user_message(agent_id=agent.id, message=message, include_full_message=True)
print("Response", response)
assert isinstance(response.usage, MemGPTUsageStatistics)
assert response.usage.step_count == 1
@@ -401,11 +401,7 @@ def test_message_update(client: Union[LocalClient, RESTClient], agent: AgentStat
"""Test that we can update the details of a message"""
# create a message
message_response = client.send_message(
agent_id=agent.id,
message="Test message",
role="user",
)
message_response = client.send_message(agent_id=agent.id, message="Test message", role="user", include_full_message=True)
print("Messages=", message_response)
assert isinstance(message_response, MemGPTResponse)
assert isinstance(message_response.messages[-1], Message)