From 4e440404ff085769013844e8c61758c074d70afc Mon Sep 17 00:00:00 2001 From: Charles Packer Date: Tue, 31 Oct 2023 11:43:38 -0700 Subject: [PATCH] patch bug (#221) --- memgpt/main.py | 2 +- tests/test_load_archival.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/memgpt/main.py b/memgpt/main.py index 15197ce1..ecbc0ed0 100644 --- a/memgpt/main.py +++ b/memgpt/main.py @@ -229,7 +229,7 @@ async def main( memgpt_persona = persona if memgpt_persona is None: memgpt_persona = ( - personas.GPT35_DEFAULT if "gpt-3.5" in model else personas.DEFAULT_PRESET, + personas.GPT35_DEFAULT if "gpt-3.5" in model else personas.DEFAULT, None, # represents the personas dir in pymemgpt package ) else: diff --git a/tests/test_load_archival.py b/tests/test_load_archival.py index ba89989d..787bbee9 100644 --- a/tests/test_load_archival.py +++ b/tests/test_load_archival.py @@ -100,7 +100,7 @@ def test_load_database(): # create agents with defaults agent_config = AgentConfig( - persona=personas.DEFAULT_PRESET, + persona=personas.DEFAULT, human=humans.DEFAULT, model=DEFAULT_MEMGPT_MODEL, data_source="tmp_hf_dataset",