github/MemGPT
7
0
Fork 0
mirror of https://github.com/cpacker/MemGPT.git synced 2025-06-03 04:30:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: hotfix for PR #1693 (#1697)

Browse Source
This commit is contained in:
Charles Packer 2024-08-29 16:14:02 -07:00 committed by GitHub
parent 69c8540992
commit 7e4ca0601a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_client.py
View File

@ -8,7 +8,7 @@ from dotenv import load_dotenv
from memgpt import Admin, create_client from memgpt import Admin, create_client
from memgpt.constants import DEFAULT_PRESET from memgpt.constants import DEFAULT_PRESET
from memgpt.schemas.memgpt_message import InternalMonologue from memgpt.schemas.message import Message
from memgpt.schemas.usage import MemGPTUsageStatistics from memgpt.schemas.usage import MemGPTUsageStatistics
# from tests.utils import create_config # from tests.utils import create_config
@ -127,7 +127,7 @@ def test_agent_interactions(client, agent):
assert response.usage.step_count == 1 assert response.usage.step_count == 1
assert response.usage.total_tokens > 0 assert response.usage.total_tokens > 0
assert response.usage.completion_tokens > 0 assert response.usage.completion_tokens > 0
assert isinstance(response.messages[0], InternalMonologue) assert isinstance(response.messages[0], Message)
print(response.messages) print(response.messages)
# TODO: add streaming tests # TODO: add streaming tests