From 78534f4852ae2a3d0543acd90170ab2ebf8248cb Mon Sep 17 00:00:00 2001 From: cpacker Date: Mon, 15 Jan 2024 14:07:57 -0800 Subject: [PATCH] fix: patched another test that used dummy user_id's to instead use the default user_id --- tests/test_base_functions.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/test_base_functions.py b/tests/test_base_functions.py index 0c872cd2..1b8f1bea 100644 --- a/tests/test_base_functions.py +++ b/tests/test_base_functions.py @@ -1,7 +1,10 @@ +import os +import uuid + from memgpt import MemGPT +from memgpt.config import MemGPTConfig from memgpt import constants import memgpt.functions.function_sets.base as base_functions -import os from .utils import wipe_config @@ -28,7 +31,9 @@ def create_test_agent(): ) global agent_obj - agent_obj = client.server._get_or_load_agent(user_id="NULL", agent_id=agent_state.id) + config = MemGPTConfig.load() + user_id = uuid.UUID(config.anon_clientid) + agent_obj = client.server._get_or_load_agent(user_id=user_id, agent_id=agent_state.id) def test_archival():