fix server tests

letta-ai · Jun 30, 2024 · fe24dad · fe24dad
1 parent 5ba118f
commit fe24dad
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_server.py b/tests/test_server.py
@@ -10,6 +10,7 @@
 utils.DEBUG = True
 from memgpt.config import MemGPTConfig
 from memgpt.credentials import MemGPTCredentials
+from memgpt.memory import ChatMemory
 from memgpt.server.server import SyncServer
 from memgpt.settings import settings
 
@@ -69,9 +70,7 @@ def user_id(server):
 def agent_id(server, user_id):
     # create agent
     agent_state = server.create_agent(
-        user_id=user_id,
-        name="test_agent",
-        tools=BASE_TOOLS,
+        user_id=user_id, name="test_agent", tools=BASE_TOOLS, memory=ChatMemory(human="I am Chad", persona="I love testing")
     )
     print(f"Created agent\n{agent_state}")
     yield agent_state.id