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

fix: Clean up and simplify docker entrypoint (#1235) (#1259)

Browse Source 
Co-authored-by: Ethan Knox <norton120@gmail.com>
This commit is contained in:
Sarah Wooders
2024-04-18 22:39:11 -07:00
committed by GitHub
parent d7d1bd1516
commit f3b5f5e7db
28 changed files with 1077 additions and 1029 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_server.py
View File

@@ -7,6 +7,7 @@ from dotenv import load_dotenv
from tests.config import TestMGPTConfig
utils.DEBUG = True
from memgpt.settings import settings
from memgpt.credentials import MemGPTCredentials
from memgpt.server.server import SyncServer
from memgpt.data_types import EmbeddingConfig, LLMConfig
@@ -19,9 +20,7 @@ def server():
wipe_config()
wipe_memgpt_home()
# Use os.getenv with a fallback to os.environ.get
db_url = os.getenv("MEMGPT_PGURI") or os.environ.get("MEMGPT_PGURI")
assert db_url, "Missing MEMGPT_PGURI"
db_url = settings.pg_db
if os.getenv("OPENAI_API_KEY"):
config = TestMGPTConfig(
@@ -180,7 +179,7 @@ def test_user_message(server, user_id, agent_id):
server.user_message(user_id=user_id, agent_id=agent_id, message="Hello?")
@pytest.mark.order5
@pytest.mark.order(5)
def test_get_recall_memory(server, user_id, agent_id):
# test recall memory cursor pagination
cursor1, messages_1 = server.get_agent_recall_cursor(user_id=user_id, agent_id=agent_id, reverse=True, limit=2)
@@ -207,7 +206,7 @@ def test_get_recall_memory(server, user_id, agent_id):
messages_1 = server.get_agent_messages(user_id=user_id, agent_id=agent_id, start=0, count=1)
assert len(messages_1) == 1
messages_2 = server.get_agent_messages(user_id=user_id, agent_id=agent_id, start=1, count=1000)
messages_3 = server.get_agent_messages(user_id=user_id, agent_id=agent_id, start=1, count=5)
messages_3 = server.get_agent_messages(user_id=user_id, agent_id=agent_id, start=1, count=2)
# not sure exactly how many messages there should be
assert len(messages_2) > len(messages_3)
# test safe empty return
@@ -215,7 +214,7 @@ def test_get_recall_memory(server, user_id, agent_id):
assert len(messages_none) == 0
@pytest.mark.order6
@pytest.mark.order(6)
def test_get_archival_memory(server, user_id, agent_id):
# test archival memory cursor pagination
cursor1, passages_1 = server.get_agent_archival_cursor(user_id=user_id, agent_id=agent_id, reverse=False, limit=2, order_by="text")