diff --git a/memgpt/utils.py b/memgpt/utils.py index 55c85864..52b044ab 100644 --- a/memgpt/utils.py +++ b/memgpt/utils.py @@ -15,7 +15,7 @@ import sqlite3 import fitz from tqdm import tqdm from memgpt.openai_tools import async_get_embedding_with_backoff -from memgpt.configs import memgpt_dir +from memgpt.config import memgpt_dir def count_tokens(s: str, model: str = "gpt-4") -> int: encoding = tiktoken.encoding_for_model(model) diff --git a/pyproject.toml b/pyproject.toml index d316929f..36224e11 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,9 @@ [tool.poetry] -name = "memgpt" +name = "pymemgpt" version = "0.1.0" +packages = [ + {include = "memgpt"} +] description = "Teaching LLMs memory management for unbounded context" authors = [ "Charles Packer ",