From 8b86fc8f8e43b28225257b61189009d1f4a0025f Mon Sep 17 00:00:00 2001 From: Vivian Fang Date: Mon, 30 Oct 2023 02:10:26 -0700 Subject: [PATCH] Fix persona selection for local LLM, bump version --- memgpt/main.py | 12 ++++++------ pyproject.toml | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/memgpt/main.py b/memgpt/main.py index b3ee4a67..5ae09e6f 100644 --- a/memgpt/main.py +++ b/memgpt/main.py @@ -209,26 +209,26 @@ async def main( if memgpt_persona is None: memgpt_persona = ( personas.GPT35_DEFAULT if "gpt-3.5" in model else personas.DEFAULT, - Config.personas_dir, + None, # represents the personas dir in pymemgpt package ) else: try: personas.get_persona_text(memgpt_persona, Config.custom_personas_dir) memgpt_persona = (memgpt_persona, Config.custom_personas_dir) except FileNotFoundError: - personas.get_persona_text(memgpt_persona, Config.personas_dir) - memgpt_persona = (memgpt_persona, Config.personas_dir) + personas.get_persona_text(memgpt_persona) + memgpt_persona = (memgpt_persona, None) human_persona = human if human_persona is None: - human_persona = (humans.DEFAULT, Config.humans_dir) + human_persona = (humans.DEFAULT, None) else: try: humans.get_human_text(human_persona, Config.custom_humans_dir) human_persona = (human_persona, Config.custom_humans_dir) except FileNotFoundError: - humans.get_human_text(human_persona, Config.humans_dir) - human_persona = (human_persona, Config.humans_dir) + humans.get_human_text(human_persona) + human_persona = (human_persona, None) print(persona, model, memgpt_persona) if archival_storage_files: diff --git a/pyproject.toml b/pyproject.toml index ba1419f1..413811a7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "pymemgpt" -version = "0.1.10" +version = "0.1.11" packages = [ {include = "memgpt"} ]